mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-03-24 22:52:12 -03:00
feat(debug): replace websocket with ComfyUI UI system for metadata display
- Update DebugMetadata node to return metadata via ComfyUI's UI system instead of websocket - Add new JsonDisplayWidget Vue component for displaying metadata in the UI - Remove dependency on PromptServer and websocket communication - Improve error handling with proper UI feedback - Maintain backward compatibility with existing metadata collection system
This commit is contained in:
159
vue-widgets/src/components/JsonDisplayWidget.vue
Normal file
159
vue-widgets/src/components/JsonDisplayWidget.vue
Normal file
@@ -0,0 +1,159 @@
|
||||
<template>
|
||||
<div class="json-display-widget">
|
||||
<div class="json-content" ref="contentRef">
|
||||
<pre v-if="hasMetadata" v-html="highlightedJson"></pre>
|
||||
<div v-else class="placeholder">No metadata available</div>
|
||||
</div>
|
||||
</div>
|
||||
</template>
|
||||
|
||||
<script setup lang="ts">
|
||||
import { ref, computed, onMounted } from 'vue'
|
||||
|
||||
interface JsonDisplayWidget {
|
||||
serializeValue?: () => Promise<unknown>
|
||||
value?: unknown
|
||||
onSetValue?: (v: unknown) => void
|
||||
}
|
||||
|
||||
const props = defineProps<{
|
||||
widget: JsonDisplayWidget
|
||||
node: { id: number; onExecuted?: (output: Record<string, unknown>) => void }
|
||||
}>()
|
||||
|
||||
const metadata = ref<Record<string, unknown> | null>(null)
|
||||
|
||||
const hasMetadata = computed(() =>
|
||||
metadata.value !== null && Object.keys(metadata.value).length > 0
|
||||
)
|
||||
|
||||
const highlightedJson = computed(() => {
|
||||
if (!metadata.value) return ''
|
||||
const jsonStr = JSON.stringify(metadata.value, null, 2)
|
||||
return syntaxHighlight(jsonStr)
|
||||
})
|
||||
|
||||
// Color scheme matching original json_display_widget.js
|
||||
const colors = {
|
||||
key: '#6ad6f5', // Light blue for keys
|
||||
string: '#98c379', // Soft green for strings
|
||||
number: '#e5c07b', // Amber for numbers
|
||||
boolean: '#c678dd', // Purple for booleans
|
||||
null: '#7f848e' // Gray for null
|
||||
}
|
||||
|
||||
function syntaxHighlight(json: string): string {
|
||||
// Escape HTML entities
|
||||
json = json.replace(/&/g, '&').replace(/</g, '<').replace(/>/g, '>')
|
||||
|
||||
// Apply syntax highlighting with regex
|
||||
return json.replace(
|
||||
/("(\\u[a-zA-Z0-9]{4}|\\[^u]|[^\\"])*"(\s*:)?|\b(true|false|null)\b|-?\d+(?:\.\d*)?(?:[eE][+\-]?\d+)?)/g,
|
||||
(match) => {
|
||||
let color = colors.number
|
||||
|
||||
if (/^"/.test(match)) {
|
||||
if (/:$/.test(match)) {
|
||||
// Key
|
||||
color = colors.key
|
||||
match = match.replace(/:$/, '')
|
||||
return `<span style="color:${color};">${match}</span>:`
|
||||
} else {
|
||||
// String value
|
||||
color = colors.string
|
||||
}
|
||||
} else if (/true|false/.test(match)) {
|
||||
color = colors.boolean
|
||||
} else if (/null/.test(match)) {
|
||||
color = colors.null
|
||||
}
|
||||
|
||||
return `<span style="color:${color};">${match}</span>`
|
||||
}
|
||||
)
|
||||
}
|
||||
|
||||
onMounted(() => {
|
||||
// Display-only widget - return null on serialization to avoid saving large metadata
|
||||
props.widget.serializeValue = async () => null
|
||||
|
||||
// Handle external value updates (e.g., loading workflow, paste)
|
||||
props.widget.onSetValue = (v) => {
|
||||
if (v && typeof v === 'object') {
|
||||
metadata.value = v as Record<string, unknown>
|
||||
}
|
||||
}
|
||||
|
||||
// Restore from saved value if exists (for workflow loading)
|
||||
if (props.widget.value && typeof props.widget.value === 'object') {
|
||||
metadata.value = props.widget.value as Record<string, unknown>
|
||||
}
|
||||
|
||||
// Override onExecuted to handle backend UI updates
|
||||
// Following the pattern from LoraRandomizerWidget.vue
|
||||
const originalOnExecuted = (props.node as any).onExecuted?.bind(props.node)
|
||||
|
||||
;(props.node as any).onExecuted = function(output: any) {
|
||||
// Update metadata from backend ui return
|
||||
if (output?.metadata !== undefined) {
|
||||
let metadataValue = output.metadata
|
||||
|
||||
// ComfyUI wraps ui values in arrays, unwrap if needed
|
||||
if (Array.isArray(metadataValue)) {
|
||||
metadataValue = metadataValue[0]
|
||||
}
|
||||
|
||||
// If it's a string (JSON), parse it
|
||||
if (typeof metadataValue === 'string') {
|
||||
try {
|
||||
metadataValue = JSON.parse(metadataValue)
|
||||
} catch (e) {
|
||||
console.error('[JsonDisplayWidget] Failed to parse JSON:', e)
|
||||
}
|
||||
}
|
||||
|
||||
metadata.value = metadataValue
|
||||
}
|
||||
|
||||
// Call original onExecuted if it exists
|
||||
if (originalOnExecuted) {
|
||||
return originalOnExecuted(output)
|
||||
}
|
||||
}
|
||||
})
|
||||
</script>
|
||||
|
||||
<style scoped>
|
||||
.json-display-widget {
|
||||
padding: 8px;
|
||||
background: rgba(40, 44, 52, 0.6);
|
||||
border-radius: 6px;
|
||||
height: 100%;
|
||||
display: flex;
|
||||
flex-direction: column;
|
||||
overflow: hidden;
|
||||
box-sizing: border-box;
|
||||
}
|
||||
|
||||
.json-content {
|
||||
flex: 1;
|
||||
overflow: auto;
|
||||
font-family: monospace;
|
||||
font-size: 12px;
|
||||
line-height: 1.5;
|
||||
white-space: pre-wrap;
|
||||
color: rgba(226, 232, 240, 0.9);
|
||||
}
|
||||
|
||||
.json-content pre {
|
||||
margin: 0;
|
||||
padding: 0;
|
||||
}
|
||||
|
||||
.placeholder {
|
||||
font-style: italic;
|
||||
color: rgba(226, 232, 240, 0.6);
|
||||
text-align: center;
|
||||
padding: 20px 0;
|
||||
}
|
||||
</style>
|
||||
@@ -2,6 +2,7 @@ import { createApp, type App as VueApp } from 'vue'
|
||||
import PrimeVue from 'primevue/config'
|
||||
import LoraPoolWidget from '@/components/LoraPoolWidget.vue'
|
||||
import LoraRandomizerWidget from '@/components/LoraRandomizerWidget.vue'
|
||||
import JsonDisplayWidget from '@/components/JsonDisplayWidget.vue'
|
||||
import type { LoraPoolConfig, LegacyLoraPoolConfig, RandomizerConfig } from './composables/types'
|
||||
|
||||
const LORA_POOL_WIDGET_MIN_WIDTH = 500
|
||||
@@ -9,6 +10,8 @@ const LORA_POOL_WIDGET_MIN_HEIGHT = 400
|
||||
const LORA_RANDOMIZER_WIDGET_MIN_WIDTH = 500
|
||||
const LORA_RANDOMIZER_WIDGET_MIN_HEIGHT = 510
|
||||
const LORA_RANDOMIZER_WIDGET_MAX_HEIGHT = LORA_RANDOMIZER_WIDGET_MIN_HEIGHT
|
||||
const JSON_DISPLAY_WIDGET_MIN_WIDTH = 300
|
||||
const JSON_DISPLAY_WIDGET_MIN_HEIGHT = 200
|
||||
|
||||
// @ts-ignore - ComfyUI external module
|
||||
import { app } from '../../../scripts/app.js'
|
||||
@@ -207,6 +210,72 @@ function createLoraRandomizerWidget(node) {
|
||||
return { widget }
|
||||
}
|
||||
|
||||
// @ts-ignore
|
||||
function createJsonDisplayWidget(node) {
|
||||
const container = document.createElement('div')
|
||||
container.id = `json-display-widget-${node.id}`
|
||||
container.style.width = '100%'
|
||||
container.style.height = '100%'
|
||||
container.style.display = 'flex'
|
||||
container.style.flexDirection = 'column'
|
||||
container.style.overflow = 'hidden'
|
||||
|
||||
forwardMiddleMouseToCanvas(container)
|
||||
|
||||
let internalValue: Record<string, unknown> | undefined
|
||||
|
||||
const widget = node.addDOMWidget(
|
||||
'metadata',
|
||||
'JSON_DISPLAY',
|
||||
container,
|
||||
{
|
||||
getValue() {
|
||||
return internalValue
|
||||
},
|
||||
setValue(v: Record<string, unknown>) {
|
||||
internalValue = v
|
||||
if (typeof widget.onSetValue === 'function') {
|
||||
widget.onSetValue(v)
|
||||
}
|
||||
},
|
||||
serialize: false, // Display-only widget - don't save metadata in workflows
|
||||
getMinHeight() {
|
||||
return JSON_DISPLAY_WIDGET_MIN_HEIGHT
|
||||
}
|
||||
}
|
||||
)
|
||||
|
||||
const vueApp = createApp(JsonDisplayWidget, {
|
||||
widget,
|
||||
node
|
||||
})
|
||||
|
||||
vueApp.use(PrimeVue, {
|
||||
unstyled: true,
|
||||
ripple: false
|
||||
})
|
||||
|
||||
vueApp.mount(container)
|
||||
vueApps.set(node.id + 20000, vueApp) // Offset to avoid collision with other widgets
|
||||
|
||||
widget.computeLayoutSize = () => {
|
||||
const minWidth = JSON_DISPLAY_WIDGET_MIN_WIDTH
|
||||
const minHeight = JSON_DISPLAY_WIDGET_MIN_HEIGHT
|
||||
|
||||
return { minHeight, minWidth }
|
||||
}
|
||||
|
||||
widget.onRemove = () => {
|
||||
const vueApp = vueApps.get(node.id + 20000)
|
||||
if (vueApp) {
|
||||
vueApp.unmount()
|
||||
vueApps.delete(node.id + 20000)
|
||||
}
|
||||
}
|
||||
|
||||
return { widget }
|
||||
}
|
||||
|
||||
app.registerExtension({
|
||||
name: 'LoraManager.VueWidgets',
|
||||
|
||||
@@ -238,5 +307,20 @@ app.registerExtension({
|
||||
return addLorasWidgetCache(node, 'loras', { isRandomizerNode }, callback)
|
||||
}
|
||||
}
|
||||
},
|
||||
|
||||
// Add display-only widget to Debug Metadata node
|
||||
// @ts-ignore
|
||||
async beforeRegisterNodeDef(nodeType, nodeData) {
|
||||
if (nodeData.name === 'Debug Metadata (LoraManager)') {
|
||||
const onNodeCreated = nodeType.prototype.onNodeCreated
|
||||
|
||||
nodeType.prototype.onNodeCreated = function () {
|
||||
onNodeCreated?.apply(this, [])
|
||||
|
||||
// Add the JSON display widget
|
||||
createJsonDisplayWidget(this)
|
||||
}
|
||||
}
|
||||
}
|
||||
})
|
||||
|
||||
Reference in New Issue
Block a user