ima2-gen 1.1.8 → 1.1.9

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (230) hide show
  1. package/README.md +56 -27
  2. package/bin/commands/annotate.js +137 -0
  3. package/bin/commands/annotate.ts +118 -0
  4. package/bin/commands/cancel.js +37 -33
  5. package/bin/commands/cancel.ts +45 -0
  6. package/bin/commands/canvas-versions.js +91 -0
  7. package/bin/commands/canvas-versions.ts +80 -0
  8. package/bin/commands/cardnews.js +293 -0
  9. package/bin/commands/cardnews.ts +248 -0
  10. package/bin/commands/comfy.js +63 -0
  11. package/bin/commands/comfy.ts +54 -0
  12. package/bin/commands/config.js +270 -0
  13. package/bin/commands/config.ts +265 -0
  14. package/bin/commands/edit.js +97 -72
  15. package/bin/commands/edit.ts +116 -0
  16. package/bin/commands/gen.js +140 -118
  17. package/bin/commands/gen.ts +176 -0
  18. package/bin/commands/history.js +164 -0
  19. package/bin/commands/history.ts +145 -0
  20. package/bin/commands/ls.js +60 -42
  21. package/bin/commands/ls.ts +60 -0
  22. package/bin/commands/metadata.js +45 -0
  23. package/bin/commands/metadata.ts +36 -0
  24. package/bin/commands/multimode.js +159 -0
  25. package/bin/commands/multimode.ts +146 -0
  26. package/bin/commands/node.js +176 -0
  27. package/bin/commands/node.ts +157 -0
  28. package/bin/commands/observability.js +201 -0
  29. package/bin/commands/observability.ts +176 -0
  30. package/bin/commands/ping.js +26 -20
  31. package/bin/commands/ping.ts +29 -0
  32. package/bin/commands/prompt.js +506 -0
  33. package/bin/commands/prompt.ts +421 -0
  34. package/bin/commands/ps.js +78 -71
  35. package/bin/commands/ps.ts +78 -0
  36. package/bin/commands/session.js +308 -0
  37. package/bin/commands/session.ts +265 -0
  38. package/bin/commands/show.js +75 -40
  39. package/bin/commands/show.ts +69 -0
  40. package/bin/ima2.js +324 -310
  41. package/bin/ima2.ts +444 -0
  42. package/bin/lib/args.js +75 -66
  43. package/bin/lib/args.ts +73 -0
  44. package/bin/lib/browser-id.js +15 -0
  45. package/bin/lib/browser-id.ts +16 -0
  46. package/bin/lib/client.js +91 -83
  47. package/bin/lib/client.ts +109 -0
  48. package/bin/lib/error-hints.js +14 -17
  49. package/bin/lib/error-hints.ts +23 -0
  50. package/bin/lib/files.js +26 -28
  51. package/bin/lib/files.ts +39 -0
  52. package/bin/lib/output.js +44 -42
  53. package/bin/lib/output.ts +58 -0
  54. package/bin/lib/platform.js +60 -56
  55. package/bin/lib/platform.ts +97 -0
  56. package/bin/lib/sse.js +73 -0
  57. package/bin/lib/sse.ts +73 -0
  58. package/bin/lib/star-prompt.js +69 -76
  59. package/bin/lib/star-prompt.ts +97 -0
  60. package/bin/lib/storage-doctor.js +34 -35
  61. package/bin/lib/storage-doctor.ts +38 -0
  62. package/config.js +147 -243
  63. package/config.ts +331 -0
  64. package/docs/API.md +48 -8
  65. package/docs/CLI.md +190 -0
  66. package/docs/FAQ.ko.md +5 -5
  67. package/docs/FAQ.md +5 -5
  68. package/docs/README.ja.md +71 -25
  69. package/docs/README.ko.md +61 -24
  70. package/docs/README.zh-CN.md +73 -27
  71. package/lib/assetLifecycle.js +130 -130
  72. package/lib/assetLifecycle.ts +142 -0
  73. package/lib/canvasVersionStore.js +135 -153
  74. package/lib/canvasVersionStore.ts +181 -0
  75. package/lib/cardNewsGenerator.js +127 -142
  76. package/lib/cardNewsGenerator.ts +162 -0
  77. package/lib/cardNewsJobStore.js +78 -84
  78. package/lib/cardNewsJobStore.ts +107 -0
  79. package/lib/cardNewsManifestStore.js +88 -93
  80. package/lib/cardNewsManifestStore.ts +112 -0
  81. package/lib/cardNewsPlanner.js +157 -152
  82. package/lib/cardNewsPlanner.ts +180 -0
  83. package/lib/cardNewsPlannerClient.js +101 -98
  84. package/lib/cardNewsPlannerClient.ts +114 -0
  85. package/lib/cardNewsPlannerPrompt.js +56 -56
  86. package/lib/cardNewsPlannerPrompt.ts +60 -0
  87. package/lib/cardNewsPlannerSchema.js +231 -223
  88. package/lib/cardNewsPlannerSchema.ts +259 -0
  89. package/lib/cardNewsRoleTemplateStore.js +39 -41
  90. package/lib/cardNewsRoleTemplateStore.ts +47 -0
  91. package/lib/cardNewsTemplateStore.js +171 -175
  92. package/lib/cardNewsTemplateStore.ts +210 -0
  93. package/lib/codexDetect.js +44 -47
  94. package/lib/codexDetect.ts +69 -0
  95. package/lib/comfyBridge.js +164 -184
  96. package/lib/comfyBridge.ts +214 -0
  97. package/lib/db.js +41 -51
  98. package/lib/db.ts +166 -0
  99. package/lib/errorClassify.js +62 -78
  100. package/lib/errorClassify.ts +100 -0
  101. package/lib/generationErrors.js +140 -103
  102. package/lib/generationErrors.ts +125 -0
  103. package/lib/historyList.js +149 -147
  104. package/lib/historyList.ts +164 -0
  105. package/lib/imageMetadata.js +86 -89
  106. package/lib/imageMetadata.ts +111 -0
  107. package/lib/imageMetadataStore.js +46 -51
  108. package/lib/imageMetadataStore.ts +67 -0
  109. package/lib/imageModels.js +38 -45
  110. package/lib/imageModels.ts +52 -0
  111. package/lib/inflight.js +131 -150
  112. package/lib/inflight.ts +204 -0
  113. package/lib/localImportStore.js +87 -93
  114. package/lib/localImportStore.ts +111 -0
  115. package/lib/logger.js +105 -112
  116. package/lib/logger.ts +150 -0
  117. package/lib/nodeStore.js +65 -64
  118. package/lib/nodeStore.ts +81 -0
  119. package/lib/oauthLauncher.js +61 -59
  120. package/lib/oauthLauncher.ts +64 -0
  121. package/lib/oauthNormalize.js +15 -19
  122. package/lib/oauthNormalize.ts +30 -0
  123. package/lib/oauthProxy.js +834 -832
  124. package/lib/oauthProxy.ts +995 -0
  125. package/lib/openDirectory.js +41 -40
  126. package/lib/openDirectory.ts +45 -0
  127. package/lib/pngInfo.js +18 -20
  128. package/lib/pngInfo.ts +26 -0
  129. package/lib/promptImport/curatedSources.js +125 -129
  130. package/lib/promptImport/curatedSources.ts +139 -0
  131. package/lib/promptImport/discoveryRegistry.js +185 -203
  132. package/lib/promptImport/discoveryRegistry.ts +236 -0
  133. package/lib/promptImport/errors.js +10 -10
  134. package/lib/promptImport/errors.ts +18 -0
  135. package/lib/promptImport/githubDiscovery.js +209 -219
  136. package/lib/promptImport/githubDiscovery.ts +248 -0
  137. package/lib/promptImport/githubFolder.js +253 -259
  138. package/lib/promptImport/githubFolder.ts +308 -0
  139. package/lib/promptImport/githubSource.js +189 -200
  140. package/lib/promptImport/githubSource.ts +239 -0
  141. package/lib/promptImport/gptImageHints.js +49 -56
  142. package/lib/promptImport/gptImageHints.ts +68 -0
  143. package/lib/promptImport/parsePromptCandidates.js +108 -123
  144. package/lib/promptImport/parsePromptCandidates.ts +153 -0
  145. package/lib/promptImport/promptIndex.js +190 -208
  146. package/lib/promptImport/promptIndex.ts +248 -0
  147. package/lib/promptImport/rankPromptCandidates.js +46 -43
  148. package/lib/promptImport/rankPromptCandidates.ts +49 -0
  149. package/lib/providerOptions.js +31 -0
  150. package/lib/providerOptions.ts +41 -0
  151. package/lib/referenceImageCompress.js +51 -62
  152. package/lib/referenceImageCompress.ts +75 -0
  153. package/lib/refs.js +93 -81
  154. package/lib/refs.ts +117 -0
  155. package/lib/requestLogger.js +32 -38
  156. package/lib/requestLogger.ts +48 -0
  157. package/lib/responsesImageAdapter.js +351 -0
  158. package/lib/responsesImageAdapter.ts +352 -0
  159. package/lib/runtimePorts.js +71 -73
  160. package/lib/runtimePorts.ts +93 -0
  161. package/lib/sessionStore.js +179 -230
  162. package/lib/sessionStore.ts +272 -0
  163. package/lib/storageMigration.js +247 -245
  164. package/lib/storageMigration.ts +284 -0
  165. package/lib/styleSheet.js +86 -90
  166. package/lib/styleSheet.ts +128 -0
  167. package/lib/systemTrash.js +18 -0
  168. package/lib/systemTrash.ts +20 -0
  169. package/package.json +26 -10
  170. package/routes/annotations.js +76 -79
  171. package/routes/annotations.ts +95 -0
  172. package/routes/canvasVersions.js +50 -54
  173. package/routes/canvasVersions.ts +64 -0
  174. package/routes/cardNews.js +158 -171
  175. package/routes/cardNews.ts +183 -0
  176. package/routes/comfy.js +23 -31
  177. package/routes/comfy.ts +39 -0
  178. package/routes/edit.js +183 -214
  179. package/routes/edit.ts +230 -0
  180. package/routes/generate.js +269 -291
  181. package/routes/generate.ts +309 -0
  182. package/routes/health.js +102 -107
  183. package/routes/health.ts +114 -0
  184. package/routes/history.js +136 -144
  185. package/routes/history.ts +153 -0
  186. package/routes/imageImport.js +27 -27
  187. package/routes/imageImport.ts +33 -0
  188. package/routes/index.js +17 -17
  189. package/routes/index.ts +35 -0
  190. package/routes/metadata.js +60 -64
  191. package/routes/metadata.ts +71 -0
  192. package/routes/multimode.js +228 -263
  193. package/routes/multimode.ts +280 -0
  194. package/routes/nodes.js +378 -424
  195. package/routes/nodes.ts +455 -0
  196. package/routes/promptImport.js +284 -324
  197. package/routes/promptImport.ts +354 -0
  198. package/routes/prompts.js +333 -360
  199. package/routes/prompts.ts +379 -0
  200. package/routes/sessions.js +277 -285
  201. package/routes/sessions.ts +292 -0
  202. package/routes/storage.js +29 -31
  203. package/routes/storage.ts +39 -0
  204. package/server.js +189 -196
  205. package/server.ts +235 -0
  206. package/ui/dist/.vite/manifest.json +101 -0
  207. package/ui/dist/assets/CardNewsWorkspace-BJOCey7Z.js +2 -0
  208. package/ui/dist/assets/NodeCanvas-BZV40eAE.css +1 -0
  209. package/ui/dist/assets/NodeCanvas-C3dzYNsk.js +7 -0
  210. package/ui/dist/assets/PromptImportDialog-Dqu1VpUh.js +2 -0
  211. package/ui/dist/assets/PromptImportDiscoverySection-Dg8T9X0L.js +1 -0
  212. package/ui/dist/assets/PromptImportFolderSection-DBaqsFO4.js +1 -0
  213. package/ui/dist/assets/PromptLibraryPanel-p5QqR97M.js +2 -0
  214. package/ui/dist/assets/SettingsWorkspace-B5bSAZ6u.js +1 -0
  215. package/ui/dist/assets/index-C9cXwiWE.js +25 -0
  216. package/ui/dist/assets/index-CGMIkZXn.css +1 -0
  217. package/ui/dist/assets/index-Cvld7dUZ.js +1 -0
  218. package/ui/dist/index.html +2 -2
  219. package/assets/phase-a-bg-cleanup-test.png +0 -0
  220. package/assets/screenshot.png +0 -0
  221. package/assets/screenshots/classic-generate-light.png +0 -0
  222. package/assets/screenshots/node-graph-branching.png +0 -0
  223. package/assets/screenshots/settings-oauth-generation.png +0 -0
  224. package/assets/screenshots/settings-workspace.png +0 -0
  225. package/assets/screenshots/style-sheet-editor.png +0 -0
  226. package/integrations/comfyui/ima2_gen_bridge/__pycache__/__init__.cpython-313.pyc +0 -0
  227. package/integrations/comfyui/ima2_gen_bridge/__pycache__/nodes.cpython-313.pyc +0 -0
  228. package/ui/dist/assets/index-BDffwmLs.css +0 -1
  229. package/ui/dist/assets/index-D0fdHLkJ.js +0 -31
  230. package/ui/dist/assets/index-D0fdHLkJ.js.map +0 -1
@@ -1,239 +1,228 @@
1
1
  import { createHash } from "node:crypto";
2
2
  import { promptImportError } from "./errors.js";
3
-
4
3
  const ALLOWED_HOSTS = new Set(["github.com", "raw.githubusercontent.com"]);
5
4
  const SUPPORTED_EXTENSIONS = new Set(["md", "markdown", "txt"]);
6
5
  const OWNER_REPO_RE = /^[A-Za-z0-9_.-]+$/;
7
-
8
6
  function safeDecode(value) {
9
- try {
10
- return decodeURIComponent(value);
11
- } catch {
12
- throw promptImportError("INVALID_GITHUB_SOURCE", "Invalid encoded GitHub path");
13
- }
7
+ try {
8
+ return decodeURIComponent(value);
9
+ }
10
+ catch {
11
+ throw promptImportError("INVALID_GITHUB_SOURCE", "Invalid encoded GitHub path");
12
+ }
14
13
  }
15
-
16
14
  function assertCleanPath(path) {
17
- const lower = path.toLowerCase();
18
- if (path.includes("\0") || lower.includes("%00")) {
19
- throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub path contains a null byte");
20
- }
21
- if (/%2f|%5c/i.test(path)) {
22
- throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub path contains an encoded slash");
23
- }
24
- const decoded = safeDecode(path);
25
- if (decoded.includes("\\") || decoded.split("/").includes("..")) {
26
- throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub path traversal is not allowed");
27
- }
28
- return decoded.replace(/^\/+/, "");
15
+ const lower = path.toLowerCase();
16
+ if (path.includes("\0") || lower.includes("%00")) {
17
+ throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub path contains a null byte");
18
+ }
19
+ if (/%2f|%5c/i.test(path)) {
20
+ throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub path contains an encoded slash");
21
+ }
22
+ const decoded = safeDecode(path);
23
+ if (decoded.includes("\\") || decoded.split("/").includes("..")) {
24
+ throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub path traversal is not allowed");
25
+ }
26
+ return decoded.replace(/^\/+/, "");
29
27
  }
30
-
31
28
  function extensionForPath(path) {
32
- const match = /\.([A-Za-z0-9]+)$/.exec(path);
33
- return match?.[1]?.toLowerCase() ?? "";
29
+ const match = /\.([A-Za-z0-9]+)$/.exec(path);
30
+ return match?.[1]?.toLowerCase() ?? "";
34
31
  }
35
-
36
32
  function assertSupportedFilePath(path) {
37
- const ext = extensionForPath(path);
38
- if (!ext) {
39
- throw promptImportError("FOLDER_IMPORT_DEFERRED", "Folder import is planned for a later version", 422);
40
- }
41
- if (!SUPPORTED_EXTENSIONS.has(ext)) {
42
- throw promptImportError("UNSUPPORTED_EXTENSION", "Only .md, .markdown, and .txt files are supported");
43
- }
44
- return ext;
33
+ const ext = extensionForPath(path);
34
+ if (!ext) {
35
+ throw promptImportError("FOLDER_IMPORT_DEFERRED", "Folder import is planned for a later version", 422);
36
+ }
37
+ if (!SUPPORTED_EXTENSIONS.has(ext)) {
38
+ throw promptImportError("UNSUPPORTED_EXTENSION", "Only .md, .markdown, and .txt files are supported");
39
+ }
40
+ return ext;
45
41
  }
46
-
47
42
  function assertOwnerRepo(owner, repo) {
48
- if (!OWNER_REPO_RE.test(owner || "") || !OWNER_REPO_RE.test(repo || "")) {
49
- throw promptImportError("INVALID_GITHUB_SOURCE", "Invalid GitHub owner or repository");
50
- }
43
+ if (!OWNER_REPO_RE.test(owner || "") || !OWNER_REPO_RE.test(repo || "")) {
44
+ throw promptImportError("INVALID_GITHUB_SOURCE", "Invalid GitHub owner or repository");
45
+ }
51
46
  }
52
-
53
47
  function normalizeUrlInput(input) {
54
- let url;
55
- try {
56
- url = new URL(input);
57
- } catch {
58
- return null;
59
- }
60
- if (!["http:", "https:"].includes(url.protocol)) {
61
- throw promptImportError("INVALID_GITHUB_SOURCE", "Only http(s) GitHub URLs are supported");
62
- }
63
- if (!ALLOWED_HOSTS.has(url.hostname)) {
64
- throw promptImportError("INVALID_GITHUB_SOURCE", "Only GitHub file URLs are supported");
65
- }
66
-
67
- const parts = url.pathname.split("/").filter(Boolean);
68
- if (url.hostname === "github.com") {
69
- const [owner, repo, marker, ref, ...pathParts] = parts;
70
- assertOwnerRepo(owner, repo);
71
- if (marker !== "blob") {
72
- throw promptImportError("FOLDER_IMPORT_DEFERRED", "Only GitHub file URLs are supported in PR1", 422);
48
+ let url;
49
+ try {
50
+ url = new URL(input);
73
51
  }
52
+ catch {
53
+ return null;
54
+ }
55
+ if (!["http:", "https:"].includes(url.protocol)) {
56
+ throw promptImportError("INVALID_GITHUB_SOURCE", "Only http(s) GitHub URLs are supported");
57
+ }
58
+ if (!ALLOWED_HOSTS.has(url.hostname)) {
59
+ throw promptImportError("INVALID_GITHUB_SOURCE", "Only GitHub file URLs are supported");
60
+ }
61
+ const parts = url.pathname.split("/").filter(Boolean);
62
+ if (url.hostname === "github.com") {
63
+ const [owner, repo, marker, ref, ...pathParts] = parts;
64
+ assertOwnerRepo(owner, repo);
65
+ if (marker !== "blob") {
66
+ throw promptImportError("FOLDER_IMPORT_DEFERRED", "Only GitHub file URLs are supported in PR1", 422);
67
+ }
68
+ if (!ref || pathParts.length === 0) {
69
+ throw promptImportError("FOLDER_IMPORT_DEFERRED", "Folder import is planned for a later version", 422);
70
+ }
71
+ const path = assertCleanPath(pathParts.join("/"));
72
+ const ext = assertSupportedFilePath(path);
73
+ return {
74
+ kind: "github",
75
+ owner,
76
+ repo,
77
+ ref: safeDecode(ref),
78
+ path,
79
+ extension: ext,
80
+ htmlUrl: url.toString(),
81
+ rawUrl: `https://raw.githubusercontent.com/${owner}/${repo}/${encodeURIComponent(safeDecode(ref))}/${path}`,
82
+ tags: ["github", `repo:${owner}/${repo}`, `ref:${safeDecode(ref)}`, `file:${path.split("/").pop()}`, `ext:${ext}`],
83
+ };
84
+ }
85
+ const [owner, repo, ref, ...pathParts] = parts;
86
+ assertOwnerRepo(owner, repo);
74
87
  if (!ref || pathParts.length === 0) {
75
- throw promptImportError("FOLDER_IMPORT_DEFERRED", "Folder import is planned for a later version", 422);
88
+ throw promptImportError("FOLDER_IMPORT_DEFERRED", "Folder import is planned for a later version", 422);
76
89
  }
77
90
  const path = assertCleanPath(pathParts.join("/"));
78
91
  const ext = assertSupportedFilePath(path);
79
92
  return {
80
- kind: "github",
81
- owner,
82
- repo,
83
- ref: safeDecode(ref),
84
- path,
85
- extension: ext,
86
- htmlUrl: url.toString(),
87
- rawUrl: `https://raw.githubusercontent.com/${owner}/${repo}/${encodeURIComponent(safeDecode(ref))}/${path}`,
88
- tags: ["github", `repo:${owner}/${repo}`, `ref:${safeDecode(ref)}`, `file:${path.split("/").pop()}`, `ext:${ext}`],
93
+ kind: "github",
94
+ owner,
95
+ repo,
96
+ ref: safeDecode(ref),
97
+ path,
98
+ extension: ext,
99
+ htmlUrl: `https://github.com/${owner}/${repo}/blob/${safeDecode(ref)}/${path}`,
100
+ rawUrl: url.toString(),
101
+ tags: ["github", `repo:${owner}/${repo}`, `ref:${safeDecode(ref)}`, `file:${path.split("/").pop()}`, `ext:${ext}`],
89
102
  };
90
- }
91
-
92
- const [owner, repo, ref, ...pathParts] = parts;
93
- assertOwnerRepo(owner, repo);
94
- if (!ref || pathParts.length === 0) {
95
- throw promptImportError("FOLDER_IMPORT_DEFERRED", "Folder import is planned for a later version", 422);
96
- }
97
- const path = assertCleanPath(pathParts.join("/"));
98
- const ext = assertSupportedFilePath(path);
99
- return {
100
- kind: "github",
101
- owner,
102
- repo,
103
- ref: safeDecode(ref),
104
- path,
105
- extension: ext,
106
- htmlUrl: `https://github.com/${owner}/${repo}/blob/${safeDecode(ref)}/${path}`,
107
- rawUrl: url.toString(),
108
- tags: ["github", `repo:${owner}/${repo}`, `ref:${safeDecode(ref)}`, `file:${path.split("/").pop()}`, `ext:${ext}`],
109
- };
110
103
  }
111
-
112
104
  function normalizeShorthand(input) {
113
- const match = /^([A-Za-z0-9_.-]+)\/([A-Za-z0-9_.-]+)(?:@([^:]+))?:(.+)$/.exec(input);
114
- if (!match) {
115
- throw promptImportError("INVALID_GITHUB_SOURCE", "Enter a GitHub file URL or owner/repo:path");
116
- }
117
- const [, owner, repo, rawRef, rawPath] = match;
118
- assertOwnerRepo(owner, repo);
119
- const ref = rawRef ? safeDecode(rawRef.trim()) : "main";
120
- if (ref.includes("/")) {
121
- throw promptImportError("AMBIGUOUS_GITHUB_REF", "Branches with slashes need GitHub API folder support planned for PR3");
122
- }
123
- const path = assertCleanPath(rawPath.trim());
124
- const ext = assertSupportedFilePath(path);
125
- return {
126
- kind: "github",
127
- owner,
128
- repo,
129
- ref,
130
- path,
131
- extension: ext,
132
- htmlUrl: `https://github.com/${owner}/${repo}/blob/${ref}/${path}`,
133
- rawUrl: `https://raw.githubusercontent.com/${owner}/${repo}/${encodeURIComponent(ref)}/${path}`,
134
- tags: ["github", `repo:${owner}/${repo}`, `ref:${ref}`, `file:${path.split("/").pop()}`, `ext:${ext}`],
135
- };
105
+ const match = /^([A-Za-z0-9_.-]+)\/([A-Za-z0-9_.-]+)(?:@([^:]+))?:(.+)$/.exec(input);
106
+ if (!match) {
107
+ throw promptImportError("INVALID_GITHUB_SOURCE", "Enter a GitHub file URL or owner/repo:path");
108
+ }
109
+ const [, owner, repo, rawRef, rawPath] = match;
110
+ assertOwnerRepo(owner, repo);
111
+ const ref = rawRef ? safeDecode(rawRef.trim()) : "main";
112
+ if (ref.includes("/")) {
113
+ throw promptImportError("AMBIGUOUS_GITHUB_REF", "Branches with slashes need GitHub API folder support planned for PR3");
114
+ }
115
+ const path = assertCleanPath(rawPath.trim());
116
+ const ext = assertSupportedFilePath(path);
117
+ return {
118
+ kind: "github",
119
+ owner,
120
+ repo,
121
+ ref,
122
+ path,
123
+ extension: ext,
124
+ htmlUrl: `https://github.com/${owner}/${repo}/blob/${ref}/${path}`,
125
+ rawUrl: `https://raw.githubusercontent.com/${owner}/${repo}/${encodeURIComponent(ref)}/${path}`,
126
+ tags: ["github", `repo:${owner}/${repo}`, `ref:${ref}`, `file:${path.split("/").pop()}`, `ext:${ext}`],
127
+ };
136
128
  }
137
-
138
129
  export function buildGitHubRawFileSource({ owner, repo, ref = "main", path }) {
139
- assertOwnerRepo(owner, repo);
140
- const cleanRef = safeDecode(String(ref || "main").trim());
141
- if (!cleanRef || cleanRef.includes("/")) {
142
- throw promptImportError("AMBIGUOUS_GITHUB_REF", "Branches with slashes need GitHub API folder support planned for PR3");
143
- }
144
- const cleanPath = assertCleanPath(String(path || "").trim());
145
- const ext = assertSupportedFilePath(cleanPath);
146
- return {
147
- kind: "github",
148
- owner,
149
- repo,
150
- ref: cleanRef,
151
- path: cleanPath,
152
- extension: ext,
153
- htmlUrl: `https://github.com/${owner}/${repo}/blob/${cleanRef}/${cleanPath}`,
154
- rawUrl: `https://raw.githubusercontent.com/${owner}/${repo}/${encodeURIComponent(cleanRef)}/${cleanPath}`,
155
- tags: ["github", `repo:${owner}/${repo}`, `ref:${cleanRef}`, `file:${cleanPath.split("/").pop()}`, `ext:${ext}`],
156
- };
130
+ assertOwnerRepo(owner, repo);
131
+ const cleanRef = safeDecode(String(ref || "main").trim());
132
+ if (!cleanRef || cleanRef.includes("/")) {
133
+ throw promptImportError("AMBIGUOUS_GITHUB_REF", "Branches with slashes need GitHub API folder support planned for PR3");
134
+ }
135
+ const cleanPath = assertCleanPath(String(path || "").trim());
136
+ const ext = assertSupportedFilePath(cleanPath);
137
+ return {
138
+ kind: "github",
139
+ owner,
140
+ repo,
141
+ ref: cleanRef,
142
+ path: cleanPath,
143
+ extension: ext,
144
+ htmlUrl: `https://github.com/${owner}/${repo}/blob/${cleanRef}/${cleanPath}`,
145
+ rawUrl: `https://raw.githubusercontent.com/${owner}/${repo}/${encodeURIComponent(cleanRef)}/${cleanPath}`,
146
+ tags: ["github", `repo:${owner}/${repo}`, `ref:${cleanRef}`, `file:${cleanPath.split("/").pop()}`, `ext:${ext}`],
147
+ };
157
148
  }
158
-
159
149
  function validateFinalFetchUrl(rawUrl) {
160
- let url;
161
- try {
162
- url = new URL(rawUrl);
163
- } catch {
164
- throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub fetch returned an invalid final URL");
165
- }
166
- if (!["http:", "https:"].includes(url.protocol)) {
167
- throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub fetch returned an unsupported protocol");
168
- }
169
- if (!ALLOWED_HOSTS.has(url.hostname)) {
170
- throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub fetch redirected to an unsupported host");
171
- }
172
- const parts = url.pathname.split("/").filter(Boolean);
173
- if (url.hostname === "github.com") {
174
- const marker = parts[2];
175
- if (parts.length < 5 || marker !== "blob") {
176
- throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub fetch redirected to a non-file page");
177
- }
178
- const finalPath = assertCleanPath(parts.slice(4).join("/"));
150
+ let url;
151
+ try {
152
+ url = new URL(rawUrl);
153
+ }
154
+ catch {
155
+ throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub fetch returned an invalid final URL");
156
+ }
157
+ if (!["http:", "https:"].includes(url.protocol)) {
158
+ throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub fetch returned an unsupported protocol");
159
+ }
160
+ if (!ALLOWED_HOSTS.has(url.hostname)) {
161
+ throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub fetch redirected to an unsupported host");
162
+ }
163
+ const parts = url.pathname.split("/").filter(Boolean);
164
+ if (url.hostname === "github.com") {
165
+ const marker = parts[2];
166
+ if (parts.length < 5 || marker !== "blob") {
167
+ throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub fetch redirected to a non-file page");
168
+ }
169
+ const finalPath = assertCleanPath(parts.slice(4).join("/"));
170
+ assertSupportedFilePath(finalPath);
171
+ return;
172
+ }
173
+ if (parts.length < 4) {
174
+ throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub fetch returned a non-file path");
175
+ }
176
+ const finalPath = assertCleanPath(parts.slice(3).join("/"));
179
177
  assertSupportedFilePath(finalPath);
180
- return;
181
- }
182
- if (parts.length < 4) {
183
- throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub fetch returned a non-file path");
184
- }
185
- const finalPath = assertCleanPath(parts.slice(3).join("/"));
186
- assertSupportedFilePath(finalPath);
187
178
  }
188
-
189
179
  export function normalizeGitHubSource(input) {
190
- const trimmed = typeof input === "string" ? input.trim() : "";
191
- if (!trimmed) {
192
- throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub source is required");
193
- }
194
- return normalizeUrlInput(trimmed) ?? normalizeShorthand(trimmed);
180
+ const trimmed = typeof input === "string" ? input.trim() : "";
181
+ if (!trimmed) {
182
+ throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub source is required");
183
+ }
184
+ return normalizeUrlInput(trimmed) ?? normalizeShorthand(trimmed);
195
185
  }
196
-
197
186
  export async function fetchGitHubSource(source, limits) {
198
- const controller = new AbortController();
199
- const timer = setTimeout(() => controller.abort(), limits.fetchTimeoutMs);
200
- try {
201
- const response = await fetch(source.rawUrl, { signal: controller.signal });
202
- validateFinalFetchUrl(response.url);
203
- if (!response.ok) {
204
- throw promptImportError("INVALID_GITHUB_SOURCE", `GitHub file fetch failed with ${response.status}`, 422);
205
- }
206
- const contentLength = Number(response.headers.get("content-length") || 0);
207
- if (contentLength > limits.maxFileBytesForPreview) {
208
- throw promptImportError("REMOTE_FILE_TOO_LARGE", "Remote file is too large", 413);
209
- }
210
- const buffer = await response.arrayBuffer();
211
- if (buffer.byteLength > limits.maxFileBytesForPreview) {
212
- throw promptImportError("REMOTE_FILE_TOO_LARGE", "Remote file is too large", 413);
213
- }
214
- const text = new TextDecoder("utf-8", { fatal: false }).decode(buffer);
215
- return {
216
- text,
217
- finalUrl: response.url,
218
- etag: response.headers.get("etag"),
219
- sizeBytes: buffer.byteLength,
220
- contentHash: createHash("sha256").update(Buffer.from(buffer)).digest("hex"),
221
- };
222
- } catch (error) {
223
- if (error?.name === "AbortError") {
224
- throw promptImportError("REMOTE_FETCH_TIMEOUT", "GitHub fetch timed out", 504);
225
- }
226
- throw error;
227
- } finally {
228
- clearTimeout(timer);
229
- }
187
+ const controller = new AbortController();
188
+ const timer = setTimeout(() => controller.abort(), limits.fetchTimeoutMs);
189
+ try {
190
+ const response = await fetch(source.rawUrl, { signal: controller.signal });
191
+ validateFinalFetchUrl(response.url);
192
+ if (!response.ok) {
193
+ throw promptImportError("INVALID_GITHUB_SOURCE", `GitHub file fetch failed with ${response.status}`, 422);
194
+ }
195
+ const contentLength = Number(response.headers.get("content-length") || 0);
196
+ if (contentLength > limits.maxFileBytesForPreview) {
197
+ throw promptImportError("REMOTE_FILE_TOO_LARGE", "Remote file is too large", 413);
198
+ }
199
+ const buffer = await response.arrayBuffer();
200
+ if (buffer.byteLength > limits.maxFileBytesForPreview) {
201
+ throw promptImportError("REMOTE_FILE_TOO_LARGE", "Remote file is too large", 413);
202
+ }
203
+ const text = new TextDecoder("utf-8", { fatal: false }).decode(buffer);
204
+ return {
205
+ text,
206
+ finalUrl: response.url,
207
+ etag: response.headers.get("etag"),
208
+ sizeBytes: buffer.byteLength,
209
+ contentHash: createHash("sha256").update(Buffer.from(buffer)).digest("hex"),
210
+ };
211
+ }
212
+ catch (error) {
213
+ if (error?.name === "AbortError") {
214
+ throw promptImportError("REMOTE_FETCH_TIMEOUT", "GitHub fetch timed out", 504);
215
+ }
216
+ throw error;
217
+ }
218
+ finally {
219
+ clearTimeout(timer);
220
+ }
230
221
  }
231
-
232
222
  export async function fetchGitHubSourceText(source, limits) {
233
- const result = await fetchGitHubSource(source, limits);
234
- return result.text;
223
+ const result = await fetchGitHubSource(source, limits);
224
+ return result.text;
235
225
  }
236
-
237
226
  export function isSupportedPromptFileName(filename) {
238
- return SUPPORTED_EXTENSIONS.has(extensionForPath(filename || ""));
227
+ return SUPPORTED_EXTENSIONS.has(extensionForPath(filename || ""));
239
228
  }
@@ -0,0 +1,239 @@
1
+ import { createHash } from "node:crypto";
2
+ import { promptImportError } from "./errors.js";
3
+
4
+ const ALLOWED_HOSTS = new Set(["github.com", "raw.githubusercontent.com"]);
5
+ const SUPPORTED_EXTENSIONS = new Set(["md", "markdown", "txt"]);
6
+ const OWNER_REPO_RE = /^[A-Za-z0-9_.-]+$/;
7
+
8
+ function safeDecode(value) {
9
+ try {
10
+ return decodeURIComponent(value);
11
+ } catch {
12
+ throw promptImportError("INVALID_GITHUB_SOURCE", "Invalid encoded GitHub path");
13
+ }
14
+ }
15
+
16
+ function assertCleanPath(path) {
17
+ const lower = path.toLowerCase();
18
+ if (path.includes("\0") || lower.includes("%00")) {
19
+ throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub path contains a null byte");
20
+ }
21
+ if (/%2f|%5c/i.test(path)) {
22
+ throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub path contains an encoded slash");
23
+ }
24
+ const decoded = safeDecode(path);
25
+ if (decoded.includes("\\") || decoded.split("/").includes("..")) {
26
+ throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub path traversal is not allowed");
27
+ }
28
+ return decoded.replace(/^\/+/, "");
29
+ }
30
+
31
+ function extensionForPath(path) {
32
+ const match = /\.([A-Za-z0-9]+)$/.exec(path);
33
+ return match?.[1]?.toLowerCase() ?? "";
34
+ }
35
+
36
+ function assertSupportedFilePath(path) {
37
+ const ext = extensionForPath(path);
38
+ if (!ext) {
39
+ throw promptImportError("FOLDER_IMPORT_DEFERRED", "Folder import is planned for a later version", 422);
40
+ }
41
+ if (!SUPPORTED_EXTENSIONS.has(ext)) {
42
+ throw promptImportError("UNSUPPORTED_EXTENSION", "Only .md, .markdown, and .txt files are supported");
43
+ }
44
+ return ext;
45
+ }
46
+
47
+ function assertOwnerRepo(owner, repo) {
48
+ if (!OWNER_REPO_RE.test(owner || "") || !OWNER_REPO_RE.test(repo || "")) {
49
+ throw promptImportError("INVALID_GITHUB_SOURCE", "Invalid GitHub owner or repository");
50
+ }
51
+ }
52
+
53
+ function normalizeUrlInput(input) {
54
+ let url;
55
+ try {
56
+ url = new URL(input);
57
+ } catch {
58
+ return null;
59
+ }
60
+ if (!["http:", "https:"].includes(url.protocol)) {
61
+ throw promptImportError("INVALID_GITHUB_SOURCE", "Only http(s) GitHub URLs are supported");
62
+ }
63
+ if (!ALLOWED_HOSTS.has(url.hostname)) {
64
+ throw promptImportError("INVALID_GITHUB_SOURCE", "Only GitHub file URLs are supported");
65
+ }
66
+
67
+ const parts = url.pathname.split("/").filter(Boolean);
68
+ if (url.hostname === "github.com") {
69
+ const [owner, repo, marker, ref, ...pathParts] = parts;
70
+ assertOwnerRepo(owner, repo);
71
+ if (marker !== "blob") {
72
+ throw promptImportError("FOLDER_IMPORT_DEFERRED", "Only GitHub file URLs are supported in PR1", 422);
73
+ }
74
+ if (!ref || pathParts.length === 0) {
75
+ throw promptImportError("FOLDER_IMPORT_DEFERRED", "Folder import is planned for a later version", 422);
76
+ }
77
+ const path = assertCleanPath(pathParts.join("/"));
78
+ const ext = assertSupportedFilePath(path);
79
+ return {
80
+ kind: "github",
81
+ owner,
82
+ repo,
83
+ ref: safeDecode(ref),
84
+ path,
85
+ extension: ext,
86
+ htmlUrl: url.toString(),
87
+ rawUrl: `https://raw.githubusercontent.com/${owner}/${repo}/${encodeURIComponent(safeDecode(ref))}/${path}`,
88
+ tags: ["github", `repo:${owner}/${repo}`, `ref:${safeDecode(ref)}`, `file:${path.split("/").pop()}`, `ext:${ext}`],
89
+ };
90
+ }
91
+
92
+ const [owner, repo, ref, ...pathParts] = parts;
93
+ assertOwnerRepo(owner, repo);
94
+ if (!ref || pathParts.length === 0) {
95
+ throw promptImportError("FOLDER_IMPORT_DEFERRED", "Folder import is planned for a later version", 422);
96
+ }
97
+ const path = assertCleanPath(pathParts.join("/"));
98
+ const ext = assertSupportedFilePath(path);
99
+ return {
100
+ kind: "github",
101
+ owner,
102
+ repo,
103
+ ref: safeDecode(ref),
104
+ path,
105
+ extension: ext,
106
+ htmlUrl: `https://github.com/${owner}/${repo}/blob/${safeDecode(ref)}/${path}`,
107
+ rawUrl: url.toString(),
108
+ tags: ["github", `repo:${owner}/${repo}`, `ref:${safeDecode(ref)}`, `file:${path.split("/").pop()}`, `ext:${ext}`],
109
+ };
110
+ }
111
+
112
+ function normalizeShorthand(input) {
113
+ const match = /^([A-Za-z0-9_.-]+)\/([A-Za-z0-9_.-]+)(?:@([^:]+))?:(.+)$/.exec(input);
114
+ if (!match) {
115
+ throw promptImportError("INVALID_GITHUB_SOURCE", "Enter a GitHub file URL or owner/repo:path");
116
+ }
117
+ const [, owner, repo, rawRef, rawPath] = match;
118
+ assertOwnerRepo(owner, repo);
119
+ const ref = rawRef ? safeDecode(rawRef.trim()) : "main";
120
+ if (ref.includes("/")) {
121
+ throw promptImportError("AMBIGUOUS_GITHUB_REF", "Branches with slashes need GitHub API folder support planned for PR3");
122
+ }
123
+ const path = assertCleanPath(rawPath.trim());
124
+ const ext = assertSupportedFilePath(path);
125
+ return {
126
+ kind: "github",
127
+ owner,
128
+ repo,
129
+ ref,
130
+ path,
131
+ extension: ext,
132
+ htmlUrl: `https://github.com/${owner}/${repo}/blob/${ref}/${path}`,
133
+ rawUrl: `https://raw.githubusercontent.com/${owner}/${repo}/${encodeURIComponent(ref)}/${path}`,
134
+ tags: ["github", `repo:${owner}/${repo}`, `ref:${ref}`, `file:${path.split("/").pop()}`, `ext:${ext}`],
135
+ };
136
+ }
137
+
138
+ export function buildGitHubRawFileSource({ owner, repo, ref = "main", path }) {
139
+ assertOwnerRepo(owner, repo);
140
+ const cleanRef = safeDecode(String(ref || "main").trim());
141
+ if (!cleanRef || cleanRef.includes("/")) {
142
+ throw promptImportError("AMBIGUOUS_GITHUB_REF", "Branches with slashes need GitHub API folder support planned for PR3");
143
+ }
144
+ const cleanPath = assertCleanPath(String(path || "").trim());
145
+ const ext = assertSupportedFilePath(cleanPath);
146
+ return {
147
+ kind: "github",
148
+ owner,
149
+ repo,
150
+ ref: cleanRef,
151
+ path: cleanPath,
152
+ extension: ext,
153
+ htmlUrl: `https://github.com/${owner}/${repo}/blob/${cleanRef}/${cleanPath}`,
154
+ rawUrl: `https://raw.githubusercontent.com/${owner}/${repo}/${encodeURIComponent(cleanRef)}/${cleanPath}`,
155
+ tags: ["github", `repo:${owner}/${repo}`, `ref:${cleanRef}`, `file:${cleanPath.split("/").pop()}`, `ext:${ext}`],
156
+ };
157
+ }
158
+
159
+ function validateFinalFetchUrl(rawUrl) {
160
+ let url;
161
+ try {
162
+ url = new URL(rawUrl);
163
+ } catch {
164
+ throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub fetch returned an invalid final URL");
165
+ }
166
+ if (!["http:", "https:"].includes(url.protocol)) {
167
+ throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub fetch returned an unsupported protocol");
168
+ }
169
+ if (!ALLOWED_HOSTS.has(url.hostname)) {
170
+ throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub fetch redirected to an unsupported host");
171
+ }
172
+ const parts = url.pathname.split("/").filter(Boolean);
173
+ if (url.hostname === "github.com") {
174
+ const marker = parts[2];
175
+ if (parts.length < 5 || marker !== "blob") {
176
+ throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub fetch redirected to a non-file page");
177
+ }
178
+ const finalPath = assertCleanPath(parts.slice(4).join("/"));
179
+ assertSupportedFilePath(finalPath);
180
+ return;
181
+ }
182
+ if (parts.length < 4) {
183
+ throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub fetch returned a non-file path");
184
+ }
185
+ const finalPath = assertCleanPath(parts.slice(3).join("/"));
186
+ assertSupportedFilePath(finalPath);
187
+ }
188
+
189
+ export function normalizeGitHubSource(input) {
190
+ const trimmed = typeof input === "string" ? input.trim() : "";
191
+ if (!trimmed) {
192
+ throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub source is required");
193
+ }
194
+ return normalizeUrlInput(trimmed) ?? normalizeShorthand(trimmed);
195
+ }
196
+
197
+ export async function fetchGitHubSource(source, limits) {
198
+ const controller = new AbortController();
199
+ const timer = setTimeout(() => controller.abort(), limits.fetchTimeoutMs);
200
+ try {
201
+ const response = await fetch(source.rawUrl, { signal: controller.signal });
202
+ validateFinalFetchUrl(response.url);
203
+ if (!response.ok) {
204
+ throw promptImportError("INVALID_GITHUB_SOURCE", `GitHub file fetch failed with ${response.status}`, 422);
205
+ }
206
+ const contentLength = Number(response.headers.get("content-length") || 0);
207
+ if (contentLength > limits.maxFileBytesForPreview) {
208
+ throw promptImportError("REMOTE_FILE_TOO_LARGE", "Remote file is too large", 413);
209
+ }
210
+ const buffer = await response.arrayBuffer();
211
+ if (buffer.byteLength > limits.maxFileBytesForPreview) {
212
+ throw promptImportError("REMOTE_FILE_TOO_LARGE", "Remote file is too large", 413);
213
+ }
214
+ const text = new TextDecoder("utf-8", { fatal: false }).decode(buffer);
215
+ return {
216
+ text,
217
+ finalUrl: response.url,
218
+ etag: response.headers.get("etag"),
219
+ sizeBytes: buffer.byteLength,
220
+ contentHash: createHash("sha256").update(Buffer.from(buffer)).digest("hex"),
221
+ };
222
+ } catch (error) {
223
+ if (error?.name === "AbortError") {
224
+ throw promptImportError("REMOTE_FETCH_TIMEOUT", "GitHub fetch timed out", 504);
225
+ }
226
+ throw error;
227
+ } finally {
228
+ clearTimeout(timer);
229
+ }
230
+ }
231
+
232
+ export async function fetchGitHubSourceText(source, limits) {
233
+ const result = await fetchGitHubSource(source, limits);
234
+ return result.text;
235
+ }
236
+
237
+ export function isSupportedPromptFileName(filename) {
238
+ return SUPPORTED_EXTENSIONS.has(extensionForPath(filename || ""));
239
+ }