Graph component

src/lib/components/ComfyApp.svelte

@@ -20,6 +20,7 @@
  import ComfyProperties from "./ComfyProperties.svelte";
  import queueState from "$lib/stores/queueState";
  import ComfyUnlockUIButton from "./ComfyUnlockUIButton.svelte";
+	import ComfyGraphView from "./ComfyGraphView.svelte";
 
  export let app: ComfyApp = undefined;
  let imageViewer: ImageViewer;
@@ -59,6 +60,7 @@
      $layoutState.currentSelection = []
 
  let graphSize = 0;
+ let graphTransitioning = false;
 
  function toggleGraph() {
      if (graphSize == 0) {
@@ -121,10 +123,6 @@
      }
  }
 
- function doRecenter(): void {
-     app.lCanvas.recenter();
- }
-
  $: if ($uiState.uiUnlocked && !hasShownUIHelpToast) {
      hasShownUIHelpToast = true;
      toast.push("Right-click to open context menu.")
@@ -145,14 +143,21 @@
  }
 
  $: if (containerElem) {
-     let wrappers = containerElem.querySelectorAll<HTMLDivElement>(".pane-wrapper")
-     for (const wrapper of wrappers) {
-         const paneNode = wrapper.parentNode as HTMLElement; // get the node inside the <Pane/>
-         paneNode.ontransitionend = () => {
+     const canvas = containerElem.querySelector<HTMLDivElement>("#graph-canvas")
+     if (canvas) {
+         console.warn("reg!");
+         const paneNode = canvas.closest(".splitpanes__pane")
+         if (paneNode) {
+             (paneNode as HTMLElement).ontransitionstart = () => {
+                 graphTransitioning = true
+             }
+             (paneNode as HTMLElement).ontransitionend = () => {
+                 graphTransitioning = false
                  app.resizeCanvas()
              }
          }
      }
+ }
 
  onMount(async () => {
      await app.setup();
@@ -190,9 +195,7 @@
                         <ComfyUIPane bind:this={uiPane} {app} />
                     </Pane>
                     <Pane bind:size={graphSize}>
-                        <div class="canvas-wrapper pane-wrapper">
-                            <canvas id="graph-canvas" />
-                        </div>
+                        <ComfyGraphView {app} transitioning={graphTransitioning} />
                     </Pane>
                 </Splitpanes>
             </Pane>
@@ -226,9 +229,6 @@
             <Button variant="secondary" on:click={doLoadDefault}>
                 Load Default
             </Button>
-            <Button variant="secondary" on:click={doRecenter}>
-                Recenter
-            </Button>
             <Button variant="secondary" on:click={doRefreshCombos}>
                 🔄
             </Button>
@@ -273,15 +273,6 @@
      height: 100vh;
  }
 
- #comfy-ui {
- }
-
- #comfy-graph {
- }
-
- #graph-canvas {
- }
-
  #bottombar {
      padding-top: 0.5em;
      display: flex;
@@ -302,12 +293,6 @@
      }
  }
 
- .canvas-wrapper {
-     width: 100%;
-     height: 100%;
-     background-color: #333;
- }
-
  .sidebar-wrapper {
      width: 100%;
      height: 100%;

src/lib/components/ComfyApp.ts

@@ -106,6 +106,8 @@ export default class ComfyApp {
         LiteGraph.release_link_on_empty_shows_menu = true;
         LiteGraph.alt_drag_do_clone_nodes = true;
 
+        (window as any).LiteGraph = LiteGraph;
+
         // await this.#invokeExtensionsAsync("init");
         await this.registerNodes();
 

src/lib/components/ComfyGraphView.svelte

@@ -0,0 +1,76 @@
+<script lang="ts">
+ import { Button } from "@gradio/button";
+ import type ComfyApp from "./ComfyApp";
+
+ export let app: ComfyApp;
+ export let transitioning: boolean = false;
+
+ function doRecenter(): void {
+     app?.lCanvas?.recenter();
+ }
+</script>
+
+<div class="wrapper">
+    <div class="canvas-wrapper pane-wrapper">
+        <canvas id="graph-canvas" />
+    </div>
+    <div class="bar">
+        {#if !transitioning}
+            <span class="left">
+                <button on:click={doRecenter}>Recenter</button>
+            </span>
+        {/if}
+    </div>
+</div>
+
+<style lang="scss">
+ $bar-height: 3em;
+
+ .wrapper {
+     width: 100%;
+     height: 100%;
+ }
+
+ .canvas-wrapper {
+     width: 100%;
+     height: calc(100% - $bar-height);
+     background-color: #333;
+ }
+
+ .bar {
+     padding: 0.2em 0.5em;
+     display: flex;
+     align-items: center;
+     width: 100%;
+     height: $bar-height;
+     background-color: #3A3A3A;
+     border: 2px solid #2A2A2A;
+     gap: var(--layout-gap);
+     margin-top: auto;
+     overflow-x: auto;
+
+     > .left {
+         flex-shrink: 0;
+     }
+
+     > .right {
+         margin-left: auto
+     }
+
+     button {
+         color: #EEE;
+         border: 1px solid #888;
+         padding: 2px 5px;
+         background-color: #666;
+
+         &:hover {
+             background-color: #999;
+             border-color: #AAA;
+         }
+         &:active {
+             background-color: #555;
+             border-color: #777;
+         }
+     }
+ }
+</style>

src/lib/components/ComfyProperties.svelte

@@ -65,7 +65,7 @@
  }
 
  function validNodeProperty(spec: AttributesSpec, node: LGraphNode | null): boolean {
-     if (node == null)
+     if (node == null || spec.location !== "nodeProps")
          return false;
 
      if (spec.canShow && !spec.canShow(node))
@@ -77,8 +77,21 @@
      return spec.name in node.properties
  }
 
+ function validNodeVar(spec: AttributesSpec, node: LGraphNode | null): boolean {
+     if (node == null || spec.location !== "nodeVars")
+         return false;
+
+     if (spec.canShow && !spec.canShow(node))
+         return false;
+
+     if (spec.validNodeTypes) {
+         return spec.validNodeTypes.indexOf(node.type) !== -1;
+     }
+     return spec.name in node
+ }
+
  function validWidgetAttribute(spec: AttributesSpec, widget: IDragItem | null): boolean {
-     if (widget == null)
+     if (widget == null || spec.location !== "widget")
          return false;
      if (spec.canShow)
          return spec.canShow(widget);
@@ -86,6 +99,13 @@
      return spec.name in widget.attrs
  }
 
+ function validWorkflowAttribute(spec: AttributesSpec): boolean {
+     if (spec.location !== "workflow")
+         return false;
+
+     return spec.name in $layoutState.attrs
+ }
+
  function getAttribute(target: IDragItem, spec: AttributesSpec): any {
      let value = target.attrs[spec.name]
      if (value == null)
@@ -200,7 +220,7 @@
                     </span>
                 </div>
                 {#each category.specs as spec(spec.id)}
-                    {#if spec.location === "widget" && validWidgetAttribute(spec, target)}
+                    {#if validWidgetAttribute(spec, target)}
                         <div class="props-entry">
                             {#if spec.type === "string"}
                                 <TextBox
@@ -237,7 +257,7 @@
                             {/if}
                         </div>
                     {:else if node}
-                        {#if spec.location === "nodeProps" && validNodeProperty(spec, node)}
+                        {#if validNodeProperty(spec, node)}
                             <div class="props-entry">
                                 {#if spec.type === "string"}
                                     <TextBox
@@ -273,7 +293,7 @@
                                                     />
                                 {/if}
                             </div>
-                        {:else if spec.location === "nodeVars" && spec.name in node}
+                        {:else if validNodeVar(spec, node)}
                             <div class="props-entry">
                                 {#if spec.type === "string"}
                                     <TextBox
@@ -310,7 +330,7 @@
                                 {/if}
                             </div>
                         {/if}
-                    {:else if spec.location === "workflow" && spec.name in $layoutState.attrs}
+                    {:else if !node && !target && validWorkflowAttribute(spec)}
                         <div class="props-entry">
                             {#if spec.type === "string"}
                                 <TextBox

src/lib/components/ComfyUnlockUIButton.svelte

@@ -9,7 +9,7 @@
  }
 </script>
 
-<div class="wrapper button lg" class:toggled>
+<div class="unlock-button" class:toggled>
     <Button on:click={toggle} variant={toggled ? "primary" : "secondary"}>
         {#if toggled}
             <LockOpen2 />
@@ -20,7 +20,7 @@
 </div>
 
 <style lang="scss">
- .wrapper {
+ .unlock-button {
      display: inline-flex;
      width: var(--size-12);
      height: var(--size-12);

src/scss/ux.scss

@@ -856,3 +856,23 @@ input[type=range]::-ms-fill-upper {
         color: var(--ae-input-color) !important;
     }
 }
+
+.unlock-button {
+    > .lg {
+        border-color: var(--ae-subpanel-border-color) !important;
+    }
+    &:hover svg {
+        color: var(--ae-main-bg-color) !important;
+    }
+    svg {
+        color: var(--ae-input-color) !important;
+    }
+    &.toggled {
+        svg {
+            color: var(--ae-icon-color) !important;
+        }
+        &:hover svg {
+            color: var(--ae-main-bg-color) !important;
+        }
+    }
+}