Fix node library bookmark preview

Comfy-Org · Aug 25, 2024 · 50c70bc · 50c70bc
1 parent 7a7188a
commit 50c70bc
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 5 deletions.
diff --git a/browser_tests/menu.spec.ts b/browser_tests/menu.spec.ts
@@ -105,6 +105,12 @@ test.describe('Menu', () => {
       )
       // Verify the bookmark node with the same name is added to the tree.
       expect(await tab.getNode('KSampler (Advanced)').count()).toBe(2)
+
+      // Hover on the bookmark node to display the preview
+      await comfyPage.page.hover('.node-tree-leaf.bookmark')
+      expect(await comfyPage.page.isVisible('.node-lib-node-preview')).toBe(
+        true
+      )
     })
 
     test('Ignores unrecognized node', async ({ comfyPage }) => {

diff --git a/src/components/sidebar/tabs/NodeLibrarySidebarTab.vue b/src/components/sidebar/tabs/NodeLibrarySidebarTab.vue
@@ -24,10 +24,7 @@
         :pt="{
           nodeLabel: 'node-lib-tree-node-label',
           nodeContent: ({ props }) => ({
-            onClick: (e: MouseEvent) => onNodeContentClick(e, props.node)
-          }),
-          nodeChildren: ({ props }) => ({
-            'data-comfy-node-name': props.node?.data?.name,
+            onClick: (e: MouseEvent) => onNodeContentClick(e, props.node),
             onMouseenter: (event: MouseEvent) =>
               handleNodeHover(event, props.node?.data?.name),
             onMouseleave: () => {

diff --git a/src/components/sidebar/tabs/nodeLibrary/NodeTreeLeaf.vue b/src/components/sidebar/tabs/nodeLibrary/NodeTreeLeaf.vue
@@ -1,5 +1,5 @@
 <template>
-  <div class="node-tree-leaf" ref="container">
+  <div :class="['node-tree-leaf', { bookmark: isBookmarked }]" ref="container">
     <div class="node-content">
       <Tag
         v-if="node.experimental"