From 3fa03465781fdc37df85948aa56dbcf18bde98db Mon Sep 17 00:00:00 2001
From: Palazzini Luca <luca.palazzini3@studio.unibo.it>
Date: Tue, 17 Sep 2024 14:40:15 +0200
Subject: [PATCH] Fix node copy-paste issue

---
 web/visual_area_prompt.js   |  8 +++++---
 web/widgets/graph_widget.js | 10 ++++++++--
 2 files changed, 13 insertions(+), 5 deletions(-)

diff --git a/web/visual_area_prompt.js b/web/visual_area_prompt.js
index 389cf49..dc858a4 100644
--- a/web/visual_area_prompt.js
+++ b/web/visual_area_prompt.js
@@ -36,7 +36,7 @@ function updateWidgetValues(node) {
 
 app.registerExtension({
    name: 'fuwuffy.' + _ID,
-   async beforeRegisterNodeDef(nodeType, nodeData, _app) {
+   async beforeRegisterNodeDef(nodeType, nodeData, app) {
       // Discard other nodes
       if (nodeData.name !== _ID) {
          return;
@@ -59,7 +59,7 @@ app.registerExtension({
             }, { min: 0, max: 4096, step: 640, precision: 0 });
          });
          // Add the canvas
-         addAreaGraphWidget(this);
+         addAreaGraphWidget(app, this, "area_conditioning_canvas");
          // Add area selection control
          addNumberInput(this, "area_id", 0, (value, _, node) => {
             node.index = value;
@@ -116,7 +116,9 @@ app.registerExtension({
             }
             // Remove unlinked dynamic nodes
             if (slot.link === null) {
-               this.removeInput(idx);
+               if (this.graph) {
+                  this.removeInput(idx);
+               }
                continue;
             }
             idx += 1;
diff --git a/web/widgets/graph_widget.js b/web/widgets/graph_widget.js
index ccc3320..bd0f958 100644
--- a/web/widgets/graph_widget.js
+++ b/web/widgets/graph_widget.js
@@ -81,15 +81,16 @@ function computeCanvasSize(node, size) {
    node.canvasHeight = remainingHeight;
 }
 
-export function addAreaGraphWidget(node) {
+export function addAreaGraphWidget(app, node, name) {
    const widget = {
       type: WIDGET_CANVAS_TYPE,
-      name: "AreaConditioningCanvas",
+      name: name,
       draw: function(ctx, node, widgetWidth, widgetY) {
          if (!node.canvasHeight) {
             computeCanvasSize(node, node.size);
          }
          // Canvas variables
+         const visible = app.canvas.ds.scale > 0.5;
          const transform = ctx.getTransform();
          const widgetHeight = node.canvasHeight;
          const imageWidth = node.properties["image_width"] || 512;
@@ -106,6 +107,7 @@ export function addAreaGraphWidget(node) {
             position: "absolute",
             zIndex: 1,
             pointerEvents: "none",
+            display: visible ? "block" : "none"
          });
          // Calculate canvas draw dimensions
          const backgroundWidth = imageWidth * scale;
@@ -141,6 +143,9 @@ export function addAreaGraphWidget(node) {
          // Draw the canvas's background and border
          drawRect(widgetX - CANVAS_BORDER, widgetYOffset - CANVAS_BORDER, backgroundWidth + CANVAS_BORDER * 2, backgroundHeight + CANVAS_BORDER * 2, borderColor);
          drawRect(widgetX, widgetYOffset, backgroundWidth, backgroundHeight, backgroundColor);
+         if (!visible) {
+            return;
+         }
          // Draw a grid
          for (const value of CANVAS_GRID_VALUES) {
             const [x1, y1, w1, h1] = getDrawArea([value, 0.0, 0.002, 1.0]);
@@ -181,6 +186,7 @@ export function addAreaGraphWidget(node) {
    }
    widget.canvas = document.createElement("canvas");
    widget.canvas.className = "fuwuffy-area-canvas";
+   widget.canvas.style.display = "none";
    widget.parent = node;
    document.body.appendChild(widget.canvas);
    node.addCustomWidget(widget);