| import {app} from "../../scripts/app.js"; |
| import {sceneDataFieldPresentation} from "./h3_scene_data_core.mjs?v=0.6.2"; |
|
|
| const NODE_NAME = "MiniMaxH3SceneDataExtract"; |
|
|
| function fieldWidget(node) { |
| return node.widgets?.find((widget) => widget.name === "field") ?? null; |
| } |
|
|
| function refreshOutput(node) { |
| const output = node.outputs?.[0]; |
| if (!output) return; |
| const presentation = sceneDataFieldPresentation(fieldWidget(node)?.value); |
| output.name = presentation.name; |
| output.label = presentation.name; |
| output.type = presentation.type; |
| node.setDirtyCanvas?.(true, true); |
| } |
|
|
| function install(node) { |
| const widget = fieldWidget(node); |
| if (!widget || widget._h3SceneDataWrapped) { |
| refreshOutput(node); |
| return; |
| } |
| widget._h3SceneDataWrapped = true; |
| const original = widget.callback; |
| widget.callback = function () { |
| const result = original?.apply(this, arguments); |
| refreshOutput(node); |
| return result; |
| }; |
| refreshOutput(node); |
| } |
|
|
| app.registerExtension({ |
| name: "minimax.h3.sceneDataExtract", |
| async beforeRegisterNodeDef(nodeType, nodeData) { |
| if (nodeData.name !== NODE_NAME) return; |
|
|
| const created = nodeType.prototype.onNodeCreated; |
| nodeType.prototype.onNodeCreated = function () { |
| const result = created?.apply(this, arguments); |
| install(this); |
| return result; |
| }; |
|
|
| const configured = nodeType.prototype.onConfigure; |
| nodeType.prototype.onConfigure = function () { |
| const result = configured?.apply(this, arguments); |
| install(this); |
| return result; |
| }; |
| }, |
| }); |
|
|
