@enslo/sd-metadata 1.4.0 → 1.4.2

package/README.ja.md

@@ -153,7 +153,7 @@ if (result.status === 'success') {
 > 本番環境では `@latest` の代わりに特定のバージョンを指定してください：
 >
 > ```text
-> https://cdn.jsdelivr.net/npm/@enslo/sd-metadata@1.4.0/dist/index.js
+> https://cdn.jsdelivr.net/npm/@enslo/sd-metadata@1.4.1/dist/index.js
 > ```
 
 ### 応用例

package/README.md

@@ -153,7 +153,7 @@ if (result.status === 'success') {
 > For production use, pin to a specific version instead of `@latest`:
 >
 > ```text
-> https://cdn.jsdelivr.net/npm/@enslo/sd-metadata@1.4.0/dist/index.js
+> https://cdn.jsdelivr.net/npm/@enslo/sd-metadata@1.4.2/dist/index.js
 > ```
 
 ### Advanced Examples

package/dist/index.js

@@ -314,7 +314,7 @@ function detectSoftware(entries) {
   return null;
 }
 function detectUniqueKeywords(entryRecord) {
-  if (entryRecord.Software === "NovelAI") {
+  if (entryRecord.Software?.startsWith("NovelAI")) {
     return "novelai";
   }
   if ("invokeai_metadata" in entryRecord) {
@@ -345,6 +345,12 @@ function detectFromCommentJson(comment) {
     if ("invokeai_metadata" in parsed) {
       return "invokeai";
     }
+    if ("generation_data" in parsed) {
+      return "tensorart";
+    }
+    if ("smproj" in parsed) {
+      return "stability-matrix";
+    }
     if ("prompt" in parsed && "workflow" in parsed) {
       const workflow = parsed.workflow;
       const prompt = parsed.prompt;
@@ -574,7 +580,7 @@ function parseFooocus(entries) {
 // src/parsers/hf-space.ts
 function parseHfSpace(entries) {
   const entryRecord = buildEntryRecord(entries);
-  const parametersText = entryRecord.parameters;
+  const parametersText = entryRecord.parameters ?? entryRecord.Comment;
   if (!parametersText) {
     return Result.error({ type: "unsupportedFormat" });
   }
@@ -671,7 +677,7 @@ function parseInvokeAI(entries) {
 // src/parsers/novelai.ts
 function parseNovelAI(entries) {
   const entryRecord = buildEntryRecord(entries);
-  if (entryRecord.Software !== "NovelAI") {
+  if (!entryRecord.Software?.startsWith("NovelAI")) {
     return Result.error({ type: "unsupportedFormat" });
   }
   const commentText = entryRecord.Comment;
@@ -727,7 +733,7 @@ function parseNovelAI(entries) {
 // src/parsers/ruined-fooocus.ts
 function parseRuinedFooocus(entries) {
   const entryRecord = buildEntryRecord(entries);
-  const jsonText = entryRecord.parameters;
+  const jsonText = entryRecord.parameters ?? entryRecord.Comment;
   if (!jsonText || !jsonText.startsWith("{")) {
     return Result.error({ type: "unsupportedFormat" });
   }
@@ -775,7 +781,20 @@ function parseStabilityMatrix(entries) {
     ...comfyResult.value,
     software: "stability-matrix"
   };
-  const jsonText = entryRecord["parameters-json"];
+  let jsonText = entryRecord["parameters-json"];
+  if (!jsonText && entryRecord.Comment?.startsWith("{")) {
+    const commentParsed = parseJson(
+      entryRecord.Comment
+    );
+    if (commentParsed.ok) {
+      const commentData = commentParsed.value;
+      if (typeof commentData["parameters-json"] === "string") {
+        jsonText = commentData["parameters-json"];
+      } else if (typeof commentData["parameters-json"] === "object") {
+        jsonText = JSON.stringify(commentData["parameters-json"]);
+      }
+    }
+  }
   if (jsonText) {
     const parsed = parseJson(jsonText);
     if (parsed.ok) {
@@ -874,7 +893,26 @@ function parseSwarmUI(entries) {
 // src/parsers/tensorart.ts
 function parseTensorArt(entries) {
   const entryRecord = buildEntryRecord(entries);
-  const dataText = entryRecord.generation_data;
+  let dataText = entryRecord.generation_data;
+  let promptChunk = entryRecord.prompt;
+  if (!dataText && entryRecord.Comment?.startsWith("{")) {
+    const commentParsed = parseJson(
+      entryRecord.Comment
+    );
+    if (commentParsed.ok) {
+      const commentData = commentParsed.value;
+      if (typeof commentData.generation_data === "string") {
+        dataText = commentData.generation_data;
+      } else if (typeof commentData.generation_data === "object") {
+        dataText = JSON.stringify(commentData.generation_data);
+      }
+      if (typeof commentData.prompt === "string") {
+        promptChunk = commentData.prompt;
+      } else if (typeof commentData.prompt === "object") {
+        promptChunk = JSON.stringify(commentData.prompt);
+      }
+    }
+  }
   if (!dataText) {
     return Result.error({ type: "unsupportedFormat" });
   }
@@ -889,7 +927,6 @@ function parseTensorArt(entries) {
   const data = parsed.value;
   const width = data.width ?? 0;
   const height = data.height ?? 0;
-  const promptChunk = entryRecord.prompt;
   if (!promptChunk) {
     return Result.error({ type: "unsupportedFormat" });
   }
@@ -1580,7 +1617,7 @@ function tryExpandNovelAIWebpFormat(text) {
     return null;
   }
   const outer = outerParsed.value;
-  if (typeof outer !== "object" || outer === null || outer.Software !== "NovelAI" || typeof outer.Comment !== "string") {
+  if (typeof outer !== "object" || outer === null || typeof outer.Software === "string" && !outer.Software.startsWith("NovelAI") || typeof outer.Comment !== "string") {
     return null;
   }
   const entries = [{ keyword: "Software", text: "NovelAI" }];
@@ -1758,35 +1795,18 @@ var stringify = (value) => {
 };
 
 // src/converters/chunk-encoding.ts
-var CHUNK_ENCODING_STRATEGIES = {
-  // Dynamic selection tools
-  a1111: "dynamic",
-  forge: "dynamic",
-  "forge-neo": "dynamic",
-  "sd-webui": "dynamic",
-  invokeai: "dynamic",
-  novelai: "dynamic",
-  "sd-next": "dynamic",
-  easydiffusion: "dynamic",
-  // Unicode escape tools (spec-compliant)
-  comfyui: "text-unicode-escape",
-  swarmui: "text-unicode-escape",
-  fooocus: "text-unicode-escape",
-  "ruined-fooocus": "text-unicode-escape",
-  "hf-space": "text-unicode-escape",
-  // Raw UTF-8 tools (non-compliant but compatible)
-  "stability-matrix": "text-utf8-raw",
-  tensorart: "text-utf8-raw"
-};
-function getEncodingStrategy(tool) {
-  return CHUNK_ENCODING_STRATEGIES[tool] ?? "text-unicode-escape";
-}
 function escapeUnicode(text) {
   return text.replace(/[\u0100-\uffff]/g, (char) => {
     const code = char.charCodeAt(0).toString(16).padStart(4, "0");
     return `\\u${code}`;
   });
 }
+function unescapeUnicode(text) {
+  return text.replace(
+    /\\u([0-9a-fA-F]{4})/g,
+    (_, hex) => String.fromCharCode(Number.parseInt(hex, 16))
+  );
+}
 function hasNonLatin1(text) {
   return /[^\x00-\xFF]/.test(text);
 }
@@ -1825,15 +1845,11 @@ function convertA1111SegmentsToPng(segments) {
   if (!userComment) {
     return [];
   }
-  return createEncodedChunk(
-    "parameters",
-    userComment.data,
-    getEncodingStrategy("a1111")
-  );
+  return createEncodedChunk("parameters", userComment.data, "dynamic");
 }
 
-// src/converters/comfyui.ts
-function convertComfyUIPngToSegments(chunks) {
+// src/converters/base-json.ts
+function convertKvPngToSegments(chunks) {
   const data = {};
   for (const chunk of chunks) {
     const parsed = parseJson(chunk.text);
@@ -1850,6 +1866,29 @@ function convertComfyUIPngToSegments(chunks) {
     }
   ];
 }
+function convertKvSegmentsToPng(segments, encodingStrategy) {
+  const userComment = findSegment(segments, "exifUserComment");
+  if (!userComment) {
+    return [];
+  }
+  const parsed = parseJson(userComment.data);
+  if (!parsed.ok) {
+    return [];
+  }
+  return Object.entries(parsed.value).flatMap(([keyword, value]) => {
+    const text = stringify(value);
+    return createEncodedChunk(
+      keyword,
+      text !== void 0 ? unescapeUnicode(text) : void 0,
+      encodingStrategy
+    );
+  });
+}
+
+// src/converters/comfyui.ts
+function convertComfyUIPngToSegments(chunks) {
+  return convertKvPngToSegments(chunks);
+}
 var tryParseExtendedFormat = (segments) => {
   const imageDescription = findSegment(segments, "exifImageDescription");
   const make = findSegment(segments, "exifMake");
@@ -1857,34 +1896,17 @@ var tryParseExtendedFormat = (segments) => {
     return null;
   }
   return [
-    ...createEncodedChunk("prompt", make?.data, getEncodingStrategy("comfyui")),
+    ...createEncodedChunk("prompt", make?.data, "text-unicode-escape"),
     ...createEncodedChunk(
       "workflow",
       imageDescription?.data,
-      getEncodingStrategy("comfyui")
+      "text-unicode-escape"
     )
   ];
 };
 var tryParseSaveImagePlusFormat = (segments) => {
-  const userComment = findSegment(segments, "exifUserComment");
-  if (!userComment) {
-    return null;
-  }
-  const parsed = parseJson(userComment.data);
-  if (!parsed.ok) {
-    return createEncodedChunk(
-      "prompt",
-      userComment.data,
-      getEncodingStrategy("comfyui")
-    );
-  }
-  return Object.entries(parsed.value).flatMap(
-    ([keyword, value]) => createEncodedChunk(
-      keyword,
-      stringify(value),
-      getEncodingStrategy("comfyui")
-    )
-  );
+  const chunks = convertKvSegmentsToPng(segments, "text-utf8-raw");
+  return chunks.length > 0 ? chunks : null;
 };
 function convertComfyUISegmentsToPng(segments) {
   return tryParseExtendedFormat(segments) ?? tryParseSaveImagePlusFormat(segments) ?? [];
@@ -1911,93 +1933,36 @@ function convertEasyDiffusionSegmentsToPng(segments) {
   if (!parsed.ok) {
     return [];
   }
-  return Object.entries(parsed.value).flatMap(([keyword, value]) => {
-    const text = value != null ? typeof value === "string" ? value : String(value) : void 0;
-    if (!text) return [];
-    return createEncodedChunk(
-      keyword,
-      text,
-      getEncodingStrategy("easydiffusion")
-    );
-  });
+  return Object.entries(parsed.value).flatMap(
+    ([keyword, value]) => createEncodedChunk(keyword, stringify(value), "dynamic")
+  );
 }
 
 // src/converters/invokeai.ts
 function convertInvokeAIPngToSegments(chunks) {
-  const data = {};
-  for (const chunk of chunks) {
-    const parsed = parseJson(chunk.text);
-    if (parsed.ok) {
-      data[chunk.keyword] = parsed.value;
-    } else {
-      data[chunk.keyword] = chunk.text;
-    }
-  }
-  return [
-    {
-      source: { type: "exifUserComment" },
-      data: JSON.stringify(data)
-    }
-  ];
+  return convertKvPngToSegments(chunks);
 }
 function convertInvokeAISegmentsToPng(segments) {
-  const userComment = findSegment(segments, "exifUserComment");
-  if (!userComment) {
-    return [];
-  }
-  const parsed = parseJson(userComment.data);
-  if (!parsed.ok) {
-    return createEncodedChunk(
-      "invokeai_metadata",
-      userComment.data,
-      getEncodingStrategy("invokeai")
-    );
-  }
-  const metadataText = stringify(parsed.value.invokeai_metadata);
-  const graphText = stringify(parsed.value.invokeai_graph);
-  const chunks = [
-    ...createEncodedChunk(
-      "invokeai_metadata",
-      metadataText,
-      getEncodingStrategy("invokeai")
-    ),
-    ...createEncodedChunk(
-      "invokeai_graph",
-      graphText,
-      getEncodingStrategy("invokeai")
-    )
-  ];
-  if (chunks.length > 0) {
-    return chunks;
-  }
-  return createEncodedChunk(
-    "invokeai_metadata",
-    userComment.data,
-    getEncodingStrategy("invokeai")
-  );
+  return convertKvSegmentsToPng(segments, "dynamic");
 }
 
 // src/converters/novelai.ts
 var NOVELAI_TITLE = "NovelAI generated image";
 var NOVELAI_SOFTWARE = "NovelAI";
 function convertNovelaiPngToSegments(chunks) {
-  const comment = chunks.find((c) => c.keyword === "Comment");
-  if (!comment) {
-    return [];
-  }
   const description = chunks.find((c) => c.keyword === "Description");
+  const descriptionSegment = description && {
+    source: { type: "exifImageDescription" },
+    data: `\0\0\0\0${description.text}`
+  };
   const data = buildUserCommentJson(chunks);
-  const descriptionSegment = description ? [
-    {
-      source: { type: "exifImageDescription" },
-      data: `\0\0\0\0${description.text}`
-    }
-  ] : [];
   const userCommentSegment = {
     source: { type: "exifUserComment" },
     data: JSON.stringify(data)
   };
-  return [...descriptionSegment, userCommentSegment];
+  return [descriptionSegment, userCommentSegment].filter(
+    (segment) => segment !== void 0
+  );
 }
 function buildUserCommentJson(chunks) {
   return NOVELAI_KEY_ORDER.map((key) => {
@@ -2034,16 +1999,11 @@ function parseSegments(userCommentSeg, descriptionSeg) {
     descriptionSeg,
     stringify(jsonData.Description)
   );
-  const descriptionChunks = descriptionText ? createEncodedChunk(
-    "Description",
-    descriptionText,
-    getEncodingStrategy("novelai")
-  ) : [];
   return [
     // Title (required, use default if missing)
     createTextChunk("Title", stringify(jsonData.Title) ?? NOVELAI_TITLE),
     // Description (optional, prefer exifImageDescription over JSON)
-    ...descriptionChunks,
+    createEncodedChunk("Description", descriptionText, "dynamic"),
     // Software (required, use default if missing)
     createTextChunk(
       "Software",
@@ -2075,17 +2035,13 @@ function createPngToSegments(keyword) {
     return !chunk ? [] : [{ source: { type: "exifUserComment" }, data: chunk.text }];
   };
 }
-function createSegmentsToPng(keyword) {
+function createSegmentsToPng(keyword, encodingStrategy) {
   return (segments) => {
     const userComment = segments.find(
       (s) => s.source.type === "exifUserComment"
     );
     if (!userComment) return [];
-    return createEncodedChunk(
-      keyword,
-      userComment.data,
-      getEncodingStrategy(keyword)
-    );
+    return createEncodedChunk(keyword, userComment.data, encodingStrategy);
   };
 }
 
@@ -2096,11 +2052,10 @@ function convertSwarmUIPngToSegments(chunks) {
     return [];
   }
   const parsed = parseJson(parametersChunk.text);
-  const data = parsed.ok ? parsed.value : parametersChunk.text;
   const segments = [
     {
       source: { type: "exifUserComment" },
-      data: typeof data === "string" ? data : JSON.stringify(data)
+      data: parsed.ok ? JSON.stringify(parsed.value) : parametersChunk.text
     }
   ];
   const promptChunk = chunks.find((c) => c.keyword === "prompt");
@@ -2114,27 +2069,13 @@ function convertSwarmUIPngToSegments(chunks) {
 }
 function convertSwarmUISegmentsToPng(segments) {
   const userComment = findSegment(segments, "exifUserComment");
-  if (!userComment) {
-    return [];
-  }
-  const chunks = [];
   const make = findSegment(segments, "exifMake");
-  if (make) {
-    chunks.push(
-      ...createEncodedChunk(
-        "prompt",
-        make.data,
-        getEncodingStrategy("swarmui")
-      )
-    );
-  }
-  chunks.push(
-    ...createEncodedChunk(
-      "parameters",
-      userComment.data,
-      getEncodingStrategy("swarmui")
-    )
-  );
+  const chunks = [
+    // Restore node graph first if present (extended format)
+    createEncodedChunk("prompt", make?.data, "text-unicode-escape"),
+    // Add parameters chunk second (always present)
+    createEncodedChunk("parameters", userComment?.data, "text-unicode-escape")
+  ].flat();
   return chunks;
 }
 
@@ -2203,11 +2144,11 @@ var convertEasyDiffusion = createFormatConverter(
 );
 var convertFooocus = createFormatConverter(
   createPngToSegments("Comment"),
-  createSegmentsToPng("Comment")
+  createSegmentsToPng("Comment", "text-unicode-escape")
 );
 var convertRuinedFooocus = createFormatConverter(
   createPngToSegments("parameters"),
-  createSegmentsToPng("parameters")
+  createSegmentsToPng("parameters", "text-unicode-escape")
 );
 var convertSwarmUI = createFormatConverter(
   convertSwarmUIPngToSegments,
@@ -2219,7 +2160,7 @@ var convertInvokeAI = createFormatConverter(
 );
 var convertHfSpace = createFormatConverter(
   createPngToSegments("parameters"),
-  createSegmentsToPng("parameters")
+  createSegmentsToPng("parameters", "text-unicode-escape")
 );
 var softwareConverters = {
   // NovelAI
@@ -3002,8 +2943,7 @@ function writeAsWebUI(data, metadata) {
   return Result.ok(writeResult.value);
 }
 function createPngChunks(text) {
-  const strategy = getEncodingStrategy("a1111");
-  return createEncodedChunk("parameters", text, strategy);
+  return createEncodedChunk("parameters", text, "dynamic");
 }
 function createExifSegments(text) {
   return [