custom_nodes/ComfyUI-Crystools/web/extensions.ts

import { app, TLGraphNode, ComfyApp } from './comfy/index.js';
import type { ComfyNode } from './comfy/index.js';
import { displayContext } from './common.js';

const crystoolsExtensionsSerialized: Record<string, string> = {
  // 'External parameter from JSON file [Crystools]': 'Crystools.Utils.ExternalParameterFromJson',
  'Read JSON file [Crystools]': 'Crystools.Utils.ReadJsonFile',
  'JSON extractor [Crystools]': 'Crystools.Utils.JsonExtractor',
};

const crystoolsExtensions: Record<string, string> = {
  'Get resolution [Crystools]': 'Crystools.Image.GetResolution',
  'Preview from image [Crystools]': 'Crystools.Image.PreviewFromImage',
  'Preview from metadata [Crystools]': 'Crystools.Image.PreviewFromMetadata',
  'Metadata comparator [Crystools]': 'Crystools.Metadata.MetadataComparator',
  'Stats system [Crystools]': 'Crystools.Utils.StatsSystem',
  'Show any to JSON [Crystools]': 'Crystools.Debugger.ConsoleAnyToJson',
};

Object.keys(crystoolsExtensionsSerialized).forEach(prop => {
  // @ts-ignore
  crystoolsExtensions[prop] = crystoolsExtensionsSerialized[prop];
});

Object.keys(crystoolsExtensions).forEach(key => {
  app.registerExtension({
    name: crystoolsExtensions[key],
    beforeRegisterNodeDef(nodeType: ComfyNode, nodeData: TLGraphNode, appFromArg: ComfyApp) {
      if (nodeData.name === key) {
        if (nodeData.name in crystoolsExtensionsSerialized) {
          displayContext(nodeType, appFromArg, 0, true); // serialize_widgets = true
        } else {
          displayContext(nodeType, appFromArg, 0);
        }
      }
    },
  });
});