@tldraw/editor 4.2.0-next.3abbdad51d2e → 4.2.0-next.54bc357bbff2

package/dist-esm/lib/editor/Editor.mjs

@@ -859,26 +859,30 @@ class Editor extends (_a = EventEmitter, _getIsShapeHiddenCache_dec = [computed]
    * after the editor has already been initialized.
    *
    * @param Tool - The tool to set.
+   * @param parent - The parent state node to set the tool on.
    *
    * @public
    */
-  setTool(Tool) {
-    if (hasOwnProperty(this.root.children, Tool.id)) {
+  setTool(Tool, parent) {
+    parent ??= this.root;
+    if (hasOwnProperty(parent.children, Tool.id)) {
       throw Error(`Can't override tool with id "${Tool.id}"`);
     }
-    this.root.children[Tool.id] = new Tool(this, this.root);
+    parent.children[Tool.id] = new Tool(this, parent);
   }
   /**
    * Remove a tool. Useful if you need to remove a tool from the state chart on demand,
    * after the editor has already been initialized.
    *
    * @param Tool - The tool to delete.
+   * @param parent - The parent state node to remove the tool from.
    *
    * @public
    */
-  removeTool(Tool) {
-    if (hasOwnProperty(this.root.children, Tool.id)) {
-      delete this.root.children[Tool.id];
+  removeTool(Tool, parent) {
+    parent ??= this.root;
+    if (hasOwnProperty(parent.children, Tool.id)) {
+      delete parent.children[Tool.id];
     }
   }
   /**