worsoft-frontend-codegen-local-mcp 0.1.52 → 0.1.54

package/mcp_server.js

@@ -5,7 +5,7 @@ const fs = require('fs');
 const path = require('path');
 
 const SERVER_NAME = 'worsoft-codegen-local';
-const SERVER_VERSION = '0.1.52';
+const SERVER_VERSION = '0.1.54';
 const PROTOCOL_VERSION = '2024-11-05';
 const TOOL_NAME = 'worsoft_codegen_local_generate_frontend';
 const STYLE_CATALOG_PATH = path.join(__dirname, 'assets', 'style-catalog.json');
@@ -462,9 +462,9 @@ function toConstantCase(value) {
     .toUpperCase();
 }
 
-function parseDictRegistryEntries(fileContent) {
-  const blockMatch = String(fileContent || '').match(/export const DictRegistry\s*=\s*{([\s\S]*?)}\s*as const;/);
-  if (!blockMatch) return [];
+function parseDictRegistryEntries(fileContent) {
+  const blockMatch = String(fileContent || '').match(/export const DictRegistry\s*=\s*{([\s\S]*?)}\s*as const;/);
+  if (!blockMatch) return [];
 
   const entries = [];
   const entryRegex = /^\s*([A-Z0-9_]+)\s*:\s*['"]([^'"]+)['"]\s*,?\s*$/gm;
@@ -473,23 +473,78 @@ function parseDictRegistryEntries(fileContent) {
     entries.push({ key: match[1], value: match[2] });
     match = entryRegex.exec(blockMatch[1]);
   }
-  return entries;
-}
-
-function renderDictRegistryContent(entries) {
-  const lines = entries.map((entry) => `  ${entry.key}: '${entry.value}',`);
-  return [
-    'export const DictRegistry = {',
-    ...lines,
-    '} as const;',
-    '',
-    'export type DictRegistryKey = keyof typeof DictRegistry;',
-    'export type DictType = (typeof DictRegistry)[DictRegistryKey];',
-    '',
-  ].join('\n');
-}
-
-function isPlainObject(value) {
+  return entries;
+}
+
+function renderDictRegistryContent(entries) {
+  const lines = entries.map((entry) => `  ${entry.key}: '${entry.value}',`);
+  return [
+    'export const DictRegistry = {',
+    ...lines,
+    '} as const;',
+    '',
+    'export type DictRegistryKey = keyof typeof DictRegistry;',
+    'export type DictType = (typeof DictRegistry)[DictRegistryKey];',
+    '',
+    'export const DictSemanticValues = {',
+    '  billState: {',
+    "    editing: '0',",
+    "    processing: '1',",
+    "    paused: '2',",
+    "    terminated: '3',",
+    "    completed: '4',",
+    '  },',
+    '} as const;',
+    '',
+    'export const isBillStateEditing = (value: unknown): boolean =>',
+    "  String(value ?? '') === DictSemanticValues.billState.editing;",
+    '',
+  ].join('\n');
+}
+
+function patchDictRegistryContent(fileContent, entriesToAdd) {
+  const source = String(fileContent || '');
+  const blockMatch = source.match(/export const DictRegistry\s*=\s*{([\s\S]*?)}\s*as const;/);
+  if (!blockMatch) return null;
+
+  const block = blockMatch[0];
+  const insertAt = block.lastIndexOf('}');
+  if (insertAt < 0) return null;
+
+  const linesToAdd = entriesToAdd.map((entry) => `  ${entry.key}: '${entry.value}',`).join('\n');
+  if (!linesToAdd) return source;
+
+  const prefix = block.slice(0, insertAt).replace(/\s*$/, '');
+  const suffix = block.slice(insertAt);
+  const nextBlock = `${prefix}\n${linesToAdd}\n${suffix}`;
+
+  return source.replace(block, nextBlock);
+}
+
+const DICT_REGISTRY_SEMANTIC_HELPERS = [
+  '',
+  'export const DictSemanticValues = {',
+  '  billState: {',
+  "    editing: '0',",
+  "    processing: '1',",
+  "    paused: '2',",
+  "    terminated: '3',",
+  "    completed: '4',",
+  '  },',
+  '} as const;',
+  '',
+  'export const isBillStateEditing = (value: unknown): boolean =>',
+  "  String(value ?? '') === DictSemanticValues.billState.editing;",
+  '',
+].join('\n');
+
+function ensureDictRegistrySemanticHelpers(fileContent) {
+  const source = String(fileContent || '').replace(/\s*$/, '\n');
+  if (source.includes('export const isBillStateEditing')) return source;
+  return `${source}${DICT_REGISTRY_SEMANTIC_HELPERS}`;
+}
+
+function isPlainObject(value) {
   return Boolean(value) && typeof value === 'object' && !Array.isArray(value);
 }
 
@@ -720,33 +775,43 @@ function prepareZhCnLocaleFile(model, mergeExisting) {
   };
 }
 
-function prepareDictRegistry(frontendPath, dictTypes) {
-  const registryPath = path.join(frontendPath, 'src', 'enums', 'dict-registry.ts');
-  const exists = fs.existsSync(registryPath);
-  const existingEntries = exists ? parseDictRegistryEntries(readUtf8File(registryPath)) : [];
-  const entries = existingEntries.map((entry) => ({ ...entry }));
-  const keyByValue = new Map(entries.map((entry) => [entry.value, entry.key]));
-  const existingByKey = new Map(entries.map((entry) => [entry.key, entry.value]));
-  const usedKeys = new Set(entries.map((entry) => entry.key));
-  let changed = !exists;
-
-  for (const dictType of dictTypes) {
-    if (!dictType || keyByValue.has(dictType)) continue;
-    const key = buildUniqueDictRegistryKey(dictType, usedKeys, existingByKey);
-    entries.push({ key, value: dictType });
-    keyByValue.set(dictType, key);
-    existingByKey.set(key, dictType);
-    usedKeys.add(key);
-    changed = true;
-  }
-
-  return {
-    path: registryPath,
-    entries,
-    keyByValue,
-    needsWrite: changed,
-  };
-}
+function prepareDictRegistry(frontendPath, dictTypes) {
+  const registryPath = path.join(frontendPath, 'src', 'enums', 'dict-registry.ts');
+  const exists = fs.existsSync(registryPath);
+  const currentContent = exists ? readUtf8File(registryPath) : '';
+  const existingEntries = exists ? parseDictRegistryEntries(currentContent) : [];
+  const entries = existingEntries.map((entry) => ({ ...entry }));
+  const keyByValue = new Map(entries.map((entry) => [entry.value, entry.key]));
+  const existingByKey = new Map(entries.map((entry) => [entry.key, entry.value]));
+  const usedKeys = new Set(entries.map((entry) => entry.key));
+  const entriesToAdd = [];
+  let changed = !exists;
+
+  for (const dictType of dictTypes) {
+    if (!dictType || keyByValue.has(dictType)) continue;
+    const key = buildUniqueDictRegistryKey(dictType, usedKeys, existingByKey);
+    const entry = { key, value: dictType };
+    entries.push(entry);
+    entriesToAdd.push(entry);
+    keyByValue.set(dictType, key);
+    existingByKey.set(key, dictType);
+    usedKeys.add(key);
+    changed = true;
+  }
+
+  const patchedContent = exists ? patchDictRegistryContent(currentContent, entriesToAdd) : renderDictRegistryContent(entries);
+  const content = patchedContent ? ensureDictRegistrySemanticHelpers(patchedContent) : null;
+  const needsSemanticHelpers = exists && Boolean(content) && content !== currentContent;
+
+  return {
+    path: registryPath,
+    entries,
+    keyByValue,
+    content,
+    isCompatible: !exists || Boolean(content),
+    needsWrite: changed || needsSemanticHelpers,
+  };
+}
 
 function ensureCrudSchemaSupportFile(frontendPath) {
   const schemaPath = path.join(frontendPath, 'src', 'utils', 'crudSchema.ts');
@@ -817,10 +882,16 @@ function maybeWriteSharedSupport(sharedSupport, writeToDisk) {
     writeSupportFile(sharedSupport.crudSchema.path, sharedSupport.crudSchema.content);
   }
 
-  if (sharedSupport.dictRegistry.needsWrite) {
-    writeSupportFile(sharedSupport.dictRegistry.path, renderDictRegistryContent(sharedSupport.dictRegistry.entries));
-  }
-}
+  if (sharedSupport.dictRegistry.needsWrite) {
+    if (!sharedSupport.dictRegistry.isCompatible) {
+      throw new Error(
+        'Detected an existing src/enums/dict-registry.ts that MCP could not merge safely. ' +
+          'Please align the DictRegistry export shape manually before enabling writeSupportFiles.'
+      );
+    }
+    writeSupportFile(sharedSupport.dictRegistry.path, sharedSupport.dictRegistry.content);
+  }
+}
 
 function buildSupportNote(sharedSupport, localeZhSupport) {
   const notes = [];
@@ -831,14 +902,21 @@ function buildSupportNote(sharedSupport, localeZhSupport) {
     );
   }
 
-  if (sharedSupport.crudSchema.exists && !sharedSupport.crudSchema.isCompatible) {
-    notes.push(
-      'Detected an existing src/utils/crudSchema.ts that does not match the expected helper signature. ' +
-        'MCP preserved the existing file and did not overwrite it. Generated pages now depend on that file being manually aligned.'
-    );
-  }
-
-  if (localeZhSupport.exists && !localeZhSupport.isCompatible) {
+  if (sharedSupport.crudSchema.exists && !sharedSupport.crudSchema.isCompatible) {
+    notes.push(
+      'Detected an existing src/utils/crudSchema.ts that does not match the expected helper signature. ' +
+        'MCP preserved the existing file and did not overwrite it. Generated pages now depend on that file being manually aligned.'
+    );
+  }
+
+  if (sharedSupport.dictRegistry.exists && !sharedSupport.dictRegistry.isCompatible) {
+    notes.push(
+      'Detected an existing src/enums/dict-registry.ts that MCP could not merge safely. ' +
+        'MCP preserved the existing file and did not overwrite it.'
+    );
+  }
+
+  if (localeZhSupport.exists && !localeZhSupport.isCompatible) {
     notes.push(
       `Detected an existing ${path.relative(localeZhSupport.frontendPath, localeZhSupport.path).replace(/\\/g, '/')} that MCP could not parse. ` +
         'The file was preserved and not updated, so new Chinese i18n keys may need to be merged manually.'

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "worsoft-frontend-codegen-local-mcp",
-  "version": "0.1.52",
+  "version": "0.1.54",
   "description": "Worsoft frontend local-template code generation MCP server.",
   "license": "UNLICENSED",
   "author": "worsoft <sw@worsoft.vip>",