ima2-gen 1.1.7 → 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 (229) 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 -190
  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 +105 -0
  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 +135 -0
  130. package/lib/promptImport/curatedSources.ts +139 -0
  131. package/lib/promptImport/discoveryRegistry.js +218 -0
  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 +238 -0
  136. package/lib/promptImport/githubDiscovery.ts +248 -0
  137. package/lib/promptImport/githubFolder.js +302 -0
  138. package/lib/promptImport/githubFolder.ts +308 -0
  139. package/lib/promptImport/githubSource.js +194 -171
  140. package/lib/promptImport/githubSource.ts +239 -0
  141. package/lib/promptImport/gptImageHints.js +61 -0
  142. package/lib/promptImport/gptImageHints.ts +68 -0
  143. package/lib/promptImport/parsePromptCandidates.js +110 -112
  144. package/lib/promptImport/parsePromptCandidates.ts +153 -0
  145. package/lib/promptImport/promptIndex.js +230 -0
  146. package/lib/promptImport/promptIndex.ts +248 -0
  147. package/lib/promptImport/rankPromptCandidates.js +52 -0
  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 +33 -0
  187. package/routes/imageImport.ts +33 -0
  188. package/routes/index.js +18 -16
  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 +291 -152
  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 +6 -3
  219. package/assets/screenshot.png +0 -0
  220. package/assets/screenshots/classic-generate-light.png +0 -0
  221. package/assets/screenshots/node-graph-branching.png +0 -0
  222. package/assets/screenshots/settings-oauth-generation.png +0 -0
  223. package/assets/screenshots/settings-workspace.png +0 -0
  224. package/assets/screenshots/style-sheet-editor.png +0 -0
  225. package/integrations/comfyui/ima2_gen_bridge/__pycache__/__init__.cpython-313.pyc +0 -0
  226. package/integrations/comfyui/ima2_gen_bridge/__pycache__/nodes.cpython-313.pyc +0 -0
  227. package/ui/dist/assets/index-DARPdT4Q.css +0 -1
  228. package/ui/dist/assets/index-ht80GMq4.js +0 -31
  229. package/ui/dist/assets/index-ht80GMq4.js.map +0 -1
@@ -1,205 +1,228 @@
1
+ import { createHash } from "node:crypto";
1
2
  import { promptImportError } from "./errors.js";
2
-
3
3
  const ALLOWED_HOSTS = new Set(["github.com", "raw.githubusercontent.com"]);
4
4
  const SUPPORTED_EXTENSIONS = new Set(["md", "markdown", "txt"]);
5
5
  const OWNER_REPO_RE = /^[A-Za-z0-9_.-]+$/;
6
-
7
6
  function safeDecode(value) {
8
- try {
9
- return decodeURIComponent(value);
10
- } catch {
11
- throw promptImportError("INVALID_GITHUB_SOURCE", "Invalid encoded GitHub path");
12
- }
7
+ try {
8
+ return decodeURIComponent(value);
9
+ }
10
+ catch {
11
+ throw promptImportError("INVALID_GITHUB_SOURCE", "Invalid encoded GitHub path");
12
+ }
13
13
  }
14
-
15
14
  function assertCleanPath(path) {
16
- const lower = path.toLowerCase();
17
- if (path.includes("\0") || lower.includes("%00")) {
18
- throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub path contains a null byte");
19
- }
20
- if (/%2f|%5c/i.test(path)) {
21
- throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub path contains an encoded slash");
22
- }
23
- const decoded = safeDecode(path);
24
- if (decoded.includes("\\") || decoded.split("/").includes("..")) {
25
- throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub path traversal is not allowed");
26
- }
27
- 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(/^\/+/, "");
28
27
  }
29
-
30
28
  function extensionForPath(path) {
31
- const match = /\.([A-Za-z0-9]+)$/.exec(path);
32
- return match?.[1]?.toLowerCase() ?? "";
29
+ const match = /\.([A-Za-z0-9]+)$/.exec(path);
30
+ return match?.[1]?.toLowerCase() ?? "";
33
31
  }
34
-
35
32
  function assertSupportedFilePath(path) {
36
- const ext = extensionForPath(path);
37
- if (!ext) {
38
- throw promptImportError("FOLDER_IMPORT_DEFERRED", "Folder import is planned for a later version", 422);
39
- }
40
- if (!SUPPORTED_EXTENSIONS.has(ext)) {
41
- throw promptImportError("UNSUPPORTED_EXTENSION", "Only .md, .markdown, and .txt files are supported");
42
- }
43
- 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;
44
41
  }
45
-
46
42
  function assertOwnerRepo(owner, repo) {
47
- if (!OWNER_REPO_RE.test(owner || "") || !OWNER_REPO_RE.test(repo || "")) {
48
- throw promptImportError("INVALID_GITHUB_SOURCE", "Invalid GitHub owner or repository");
49
- }
43
+ if (!OWNER_REPO_RE.test(owner || "") || !OWNER_REPO_RE.test(repo || "")) {
44
+ throw promptImportError("INVALID_GITHUB_SOURCE", "Invalid GitHub owner or repository");
45
+ }
50
46
  }
51
-
52
47
  function normalizeUrlInput(input) {
53
- let url;
54
- try {
55
- url = new URL(input);
56
- } catch {
57
- return null;
58
- }
59
- if (!["http:", "https:"].includes(url.protocol)) {
60
- throw promptImportError("INVALID_GITHUB_SOURCE", "Only http(s) GitHub URLs are supported");
61
- }
62
- if (!ALLOWED_HOSTS.has(url.hostname)) {
63
- throw promptImportError("INVALID_GITHUB_SOURCE", "Only GitHub file URLs are supported");
64
- }
65
-
66
- const parts = url.pathname.split("/").filter(Boolean);
67
- if (url.hostname === "github.com") {
68
- const [owner, repo, marker, ref, ...pathParts] = parts;
69
- assertOwnerRepo(owner, repo);
70
- if (marker !== "blob") {
71
- throw promptImportError("FOLDER_IMPORT_DEFERRED", "Only GitHub file URLs are supported in PR1", 422);
48
+ let url;
49
+ try {
50
+ url = new URL(input);
72
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);
73
87
  if (!ref || pathParts.length === 0) {
74
- 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);
75
89
  }
76
90
  const path = assertCleanPath(pathParts.join("/"));
77
91
  const ext = assertSupportedFilePath(path);
78
92
  return {
79
- kind: "github",
80
- owner,
81
- repo,
82
- ref: safeDecode(ref),
83
- path,
84
- extension: ext,
85
- htmlUrl: url.toString(),
86
- rawUrl: `https://raw.githubusercontent.com/${owner}/${repo}/${encodeURIComponent(safeDecode(ref))}/${path}`,
87
- 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}`],
88
102
  };
89
- }
90
-
91
- const [owner, repo, ref, ...pathParts] = parts;
92
- assertOwnerRepo(owner, repo);
93
- if (!ref || pathParts.length === 0) {
94
- throw promptImportError("FOLDER_IMPORT_DEFERRED", "Folder import is planned for a later version", 422);
95
- }
96
- const path = assertCleanPath(pathParts.join("/"));
97
- const ext = assertSupportedFilePath(path);
98
- return {
99
- kind: "github",
100
- owner,
101
- repo,
102
- ref: safeDecode(ref),
103
- path,
104
- extension: ext,
105
- htmlUrl: `https://github.com/${owner}/${repo}/blob/${safeDecode(ref)}/${path}`,
106
- rawUrl: url.toString(),
107
- tags: ["github", `repo:${owner}/${repo}`, `ref:${safeDecode(ref)}`, `file:${path.split("/").pop()}`, `ext:${ext}`],
108
- };
109
103
  }
110
-
111
104
  function normalizeShorthand(input) {
112
- const match = /^([A-Za-z0-9_.-]+)\/([A-Za-z0-9_.-]+)(?:@([^:]+))?:(.+)$/.exec(input);
113
- if (!match) {
114
- throw promptImportError("INVALID_GITHUB_SOURCE", "Enter a GitHub file URL or owner/repo:path");
115
- }
116
- const [, owner, repo, rawRef, rawPath] = match;
117
- assertOwnerRepo(owner, repo);
118
- const ref = rawRef ? safeDecode(rawRef.trim()) : "main";
119
- if (ref.includes("/")) {
120
- throw promptImportError("AMBIGUOUS_GITHUB_REF", "Branches with slashes need GitHub API folder support planned for PR3");
121
- }
122
- const path = assertCleanPath(rawPath.trim());
123
- const ext = assertSupportedFilePath(path);
124
- return {
125
- kind: "github",
126
- owner,
127
- repo,
128
- ref,
129
- path,
130
- extension: ext,
131
- htmlUrl: `https://github.com/${owner}/${repo}/blob/${ref}/${path}`,
132
- rawUrl: `https://raw.githubusercontent.com/${owner}/${repo}/${encodeURIComponent(ref)}/${path}`,
133
- tags: ["github", `repo:${owner}/${repo}`, `ref:${ref}`, `file:${path.split("/").pop()}`, `ext:${ext}`],
134
- };
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
+ };
128
+ }
129
+ export function buildGitHubRawFileSource({ owner, repo, ref = "main", path }) {
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
+ };
135
148
  }
136
-
137
149
  function validateFinalFetchUrl(rawUrl) {
138
- let url;
139
- try {
140
- url = new URL(rawUrl);
141
- } catch {
142
- throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub fetch returned an invalid final URL");
143
- }
144
- if (!["http:", "https:"].includes(url.protocol)) {
145
- throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub fetch returned an unsupported protocol");
146
- }
147
- if (!ALLOWED_HOSTS.has(url.hostname)) {
148
- throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub fetch redirected to an unsupported host");
149
- }
150
- const parts = url.pathname.split("/").filter(Boolean);
151
- if (url.hostname === "github.com") {
152
- const marker = parts[2];
153
- if (parts.length < 5 || marker !== "blob") {
154
- throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub fetch redirected to a non-file page");
155
- }
156
- 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("/"));
157
177
  assertSupportedFilePath(finalPath);
158
- return;
159
- }
160
- if (parts.length < 4) {
161
- throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub fetch returned a non-file path");
162
- }
163
- const finalPath = assertCleanPath(parts.slice(3).join("/"));
164
- assertSupportedFilePath(finalPath);
165
178
  }
166
-
167
179
  export function normalizeGitHubSource(input) {
168
- const trimmed = typeof input === "string" ? input.trim() : "";
169
- if (!trimmed) {
170
- throw promptImportError("INVALID_GITHUB_SOURCE", "GitHub source is required");
171
- }
172
- 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);
185
+ }
186
+ export async function fetchGitHubSource(source, limits) {
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
+ }
173
221
  }
174
-
175
222
  export async function fetchGitHubSourceText(source, limits) {
176
- const controller = new AbortController();
177
- const timer = setTimeout(() => controller.abort(), limits.fetchTimeoutMs);
178
- try {
179
- const response = await fetch(source.rawUrl, { signal: controller.signal });
180
- validateFinalFetchUrl(response.url);
181
- if (!response.ok) {
182
- throw promptImportError("INVALID_GITHUB_SOURCE", `GitHub file fetch failed with ${response.status}`, 422);
183
- }
184
- const contentLength = Number(response.headers.get("content-length") || 0);
185
- if (contentLength > limits.maxFileBytesForPreview) {
186
- throw promptImportError("REMOTE_FILE_TOO_LARGE", "Remote file is too large", 413);
187
- }
188
- const buffer = await response.arrayBuffer();
189
- if (buffer.byteLength > limits.maxFileBytesForPreview) {
190
- throw promptImportError("REMOTE_FILE_TOO_LARGE", "Remote file is too large", 413);
191
- }
192
- return new TextDecoder("utf-8", { fatal: false }).decode(buffer);
193
- } catch (error) {
194
- if (error?.name === "AbortError") {
195
- throw promptImportError("REMOTE_FETCH_TIMEOUT", "GitHub fetch timed out", 504);
196
- }
197
- throw error;
198
- } finally {
199
- clearTimeout(timer);
200
- }
223
+ const result = await fetchGitHubSource(source, limits);
224
+ return result.text;
201
225
  }
202
-
203
226
  export function isSupportedPromptFileName(filename) {
204
- return SUPPORTED_EXTENSIONS.has(extensionForPath(filename || ""));
227
+ return SUPPORTED_EXTENSIONS.has(extensionForPath(filename || ""));
205
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
+ }