@modusensus/dsh-mneme 0.8.13 → 0.8.15

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 (271) hide show
  1. package/LICENSE +21 -21
  2. package/README.md +655 -626
  3. package/assets/icon.svg +3 -3
  4. package/bin/cli.mjs +689 -689
  5. package/bin/dsh-mneme-mcp.mjs +510 -510
  6. package/bin/dsh-mneme-serve.mjs +151 -0
  7. package/cordis.patch.yml +21 -21
  8. package/docs/CONFIGURATION.md +263 -0
  9. package/docs/CONTINUITY.md +124 -0
  10. package/docs/DAEMON.md +46 -0
  11. package/docs/ENTITIES.md +245 -0
  12. package/docs/LOCAL_MODEL.md +222 -0
  13. package/docs/SEMANTIC.md +256 -0
  14. package/docs/SLEEP.md +164 -0
  15. package/docs/STORAGE.md +89 -0
  16. package/lib/api-standalone.js +673 -572
  17. package/lib/api.js +1437 -1433
  18. package/lib/bootstrap.js +211 -211
  19. package/lib/client.js +5444 -5295
  20. package/lib/commands.js +95 -95
  21. package/lib/config.js +880 -836
  22. package/lib/content-hash.js +54 -54
  23. package/lib/continuity.js +220 -220
  24. package/lib/document-index.js +94 -94
  25. package/lib/document.js +303 -303
  26. package/lib/dream/clustering.js +118 -118
  27. package/lib/dream/decisions.js +674 -674
  28. package/lib/dream/narratives.js +44 -44
  29. package/lib/dream/sleep.js +819 -819
  30. package/lib/dream.js +1501 -1501
  31. package/lib/embedding.js +177 -177
  32. package/lib/entities/extractor.js +252 -252
  33. package/lib/graph/anchoring.js +179 -179
  34. package/lib/guide.js +74 -74
  35. package/lib/heat.js +144 -144
  36. package/lib/hot-memory.js +46 -46
  37. package/lib/index.js +535 -671
  38. package/lib/inject.js +464 -450
  39. package/lib/lang.js +248 -229
  40. package/lib/local-embedder.js +358 -358
  41. package/lib/maintenance.js +142 -142
  42. package/lib/mirror.js +330 -330
  43. package/lib/organize.js +450 -450
  44. package/lib/peak-hours.js +117 -117
  45. package/lib/placement.js +13 -0
  46. package/lib/pre-inject-gate.js +340 -0
  47. package/lib/quality-filter.js +126 -126
  48. package/lib/recall-stats.js +175 -175
  49. package/lib/reranker.js +269 -269
  50. package/lib/runtime/adopt.js +176 -176
  51. package/lib/runtime/closure.js +172 -172
  52. package/lib/runtime/download.js +247 -247
  53. package/lib/runtime/layout.js +265 -265
  54. package/lib/runtime/loader.js +252 -252
  55. package/lib/runtime/model-fetch.js +361 -361
  56. package/lib/runtime/provision.js +237 -237
  57. package/lib/runtime/tarball.js +130 -130
  58. package/lib/runtime/verify.js +223 -204
  59. package/lib/scope.js +146 -146
  60. package/lib/search/adaptive.js +60 -60
  61. package/lib/search/bm25.js +96 -96
  62. package/lib/semantic.js +229 -0
  63. package/lib/sensitive-scan.js +209 -209
  64. package/lib/serve.js +215 -0
  65. package/lib/service.js +2663 -2596
  66. package/lib/settings.js +517 -509
  67. package/lib/store.js +3007 -3007
  68. package/lib/summarize.js +891 -856
  69. package/lib/tools.js +743 -737
  70. package/lib/vector-index.js +110 -110
  71. package/lib/version-check.js +122 -122
  72. package/lib/write-admission.js +402 -402
  73. package/locale/en.json +6 -6
  74. package/locale/zh.json +6 -6
  75. package/package.json +8 -4
  76. package/runtime-manifest.json +526 -526
  77. package/scripts/benchmark-embed.js +201 -201
  78. package/scripts/benchmark-recall.js +249 -249
  79. package/scripts/benchmark-rerank.js +166 -166
  80. package/scripts/build-runtime-manifest.mjs +224 -224
  81. package/scripts/changelog-prep.mjs +36 -0
  82. package/scripts/check-sync.js +42 -42
  83. package/scripts/e2e-dsh.js +220 -220
  84. package/scripts/mneme-runtime.mjs +277 -277
  85. package/scripts/stress-dsh.js +255 -255
  86. package/scripts/sync-lib.js +47 -47
  87. package/scripts/sync-test-badge.mjs +69 -69
  88. package/scripts/test-count-sync.mjs +44 -44
  89. package/src/api-standalone.js +673 -572
  90. package/src/api.js +1437 -1433
  91. package/src/bootstrap.js +211 -211
  92. package/src/commands.js +95 -95
  93. package/src/config.js +880 -836
  94. package/src/content-hash.js +54 -54
  95. package/src/continuity.js +220 -220
  96. package/src/document-index.js +94 -94
  97. package/src/document.js +303 -303
  98. package/src/dream/clustering.js +118 -118
  99. package/src/dream/decisions.js +674 -674
  100. package/src/dream/narratives.js +44 -44
  101. package/src/dream/sleep.js +819 -819
  102. package/src/dream.js +1501 -1501
  103. package/src/embedding.js +177 -177
  104. package/src/entities/extractor.js +252 -252
  105. package/src/graph/anchoring.js +179 -179
  106. package/src/guide.js +74 -74
  107. package/src/heat.js +144 -144
  108. package/src/hot-memory.js +46 -46
  109. package/src/index.js +535 -671
  110. package/src/inject.js +464 -450
  111. package/src/lang.js +248 -229
  112. package/src/local-embedder.js +358 -358
  113. package/src/maintenance.js +142 -142
  114. package/src/mirror.js +330 -330
  115. package/src/organize.js +450 -450
  116. package/src/peak-hours.js +117 -117
  117. package/src/placement.js +13 -0
  118. package/src/pre-inject-gate.js +340 -0
  119. package/src/quality-filter.js +126 -126
  120. package/src/recall-stats.js +175 -175
  121. package/src/reranker.js +269 -269
  122. package/src/runtime/adopt.js +176 -176
  123. package/src/runtime/closure.js +172 -172
  124. package/src/runtime/download.js +247 -247
  125. package/src/runtime/layout.js +265 -265
  126. package/src/runtime/loader.js +252 -252
  127. package/src/runtime/model-fetch.js +361 -361
  128. package/src/runtime/provision.js +237 -237
  129. package/src/runtime/tarball.js +130 -130
  130. package/src/runtime/verify.js +223 -204
  131. package/src/scope.js +146 -146
  132. package/src/search/adaptive.js +60 -60
  133. package/src/search/bm25.js +96 -96
  134. package/src/semantic.js +229 -0
  135. package/src/sensitive-scan.js +209 -209
  136. package/src/serve.js +215 -0
  137. package/src/service.js +2663 -2596
  138. package/src/settings.js +517 -509
  139. package/src/store.js +3007 -3007
  140. package/src/summarize.js +891 -856
  141. package/src/tools.js +743 -737
  142. package/src/vector-index.js +110 -110
  143. package/src/version-check.js +122 -122
  144. package/src/write-admission.js +402 -402
  145. package/test/adaptive-inject-budget.test.js +85 -85
  146. package/test/api-routes-gaps.test.js +218 -218
  147. package/test/api.test.js +1702 -1688
  148. package/test/audit.test.js +448 -448
  149. package/test/baseline-persist.test.js +248 -248
  150. package/test/benchmark.test.js +94 -94
  151. package/test/bootstrap.test.js +220 -220
  152. package/test/changelog-prep.test.js +60 -0
  153. package/test/client.test.js +1230 -1169
  154. package/test/clustering.test.js +100 -100
  155. package/test/commands.test.js +106 -106
  156. package/test/config.test.js +66 -66
  157. package/test/conflict-freeze.test.js +300 -300
  158. package/test/content-hash.test.js +204 -204
  159. package/test/continuity.test.js +333 -333
  160. package/test/document-dir.test.js +233 -233
  161. package/test/document.test.js +671 -671
  162. package/test/dream-archive-guardrail.test.js +116 -116
  163. package/test/dream-candidate-hybrid.test.js +281 -281
  164. package/test/dream-coverage-degrade.test.js +86 -86
  165. package/test/dream-failure-backoff.test.js +173 -173
  166. package/test/dream-peak-hours.test.js +194 -194
  167. package/test/dream.test.js +1385 -1383
  168. package/test/embedding-ready.test.js +196 -196
  169. package/test/entities.test.js +627 -547
  170. package/test/entity-recall.test.js +179 -179
  171. package/test/epistemic.test.js +325 -325
  172. package/test/fnew-0112.test.js +311 -311
  173. package/test/fnew-03.test.js +422 -422
  174. package/test/graph-anchoring.test.js +184 -184
  175. package/test/graph-api.test.js +175 -175
  176. package/test/graph-inject-budget.test.js +103 -103
  177. package/test/graph-passive-confirm.test.js +91 -91
  178. package/test/graph-weight-evolution.test.js +104 -104
  179. package/test/heat.test.js +175 -175
  180. package/test/helpers/dream-mock.js +82 -82
  181. package/test/helpers/peer-worker.mjs +25 -25
  182. package/test/helpers/runtime-source.js +60 -60
  183. package/test/helpers/tar-builder.js +80 -80
  184. package/test/helpers/write-admission-samples.js +308 -308
  185. package/test/hot-memory.test.js +145 -145
  186. package/test/inject-content-cap.test.js +91 -91
  187. package/test/inject-first-round-bm25.test.js +89 -89
  188. package/test/inject-heat.test.js +77 -61
  189. package/test/inject-parent-gate.test.js +106 -106
  190. package/test/inject-pools.test.js +194 -194
  191. package/test/inject-recall.test.js +105 -105
  192. package/test/inject-rotation.test.js +161 -161
  193. package/test/inject.test.js +420 -420
  194. package/test/lang.test.js +87 -87
  195. package/test/lastrunat-persist.test.js +101 -101
  196. package/test/lib-smoke.test.js +96 -96
  197. package/test/llm-audit.test.js +598 -598
  198. package/test/llm-message-source.test.js +207 -207
  199. package/test/local-embedder.test.js +292 -292
  200. package/test/maintenance.test.js +391 -391
  201. package/test/mcp-stdio.test.js +385 -385
  202. package/test/mirror-dirty.test.js +424 -424
  203. package/test/mirror-edit-digest.test.js +187 -187
  204. package/test/mirror-generation.test.js +499 -499
  205. package/test/mirror-types.test.js +208 -208
  206. package/test/mirror.test.js +249 -249
  207. package/test/narrative-bars.test.js +162 -162
  208. package/test/organize.test.js +418 -418
  209. package/test/peer-blockers.test.js +193 -193
  210. package/test/placement.test.js +36 -0
  211. package/test/policy-epoch.test.js +259 -259
  212. package/test/pre-inject-gate.test.js +434 -0
  213. package/test/quality-filter.test.js +186 -186
  214. package/test/reasoning-effort.test.js +752 -752
  215. package/test/recall-anti-update.test.js +82 -82
  216. package/test/recall-evals.test.js +253 -253
  217. package/test/recall-layer.test.js +326 -326
  218. package/test/recall-stats.test.js +189 -189
  219. package/test/receipt-chain.test.js +451 -451
  220. package/test/reflection.test.js +226 -226
  221. package/test/reindex-backfill.test.js +62 -62
  222. package/test/reranker-single-logit.test.js +73 -73
  223. package/test/reranker.test.js +197 -197
  224. package/test/resident-status.test.js +83 -83
  225. package/test/runtime-adopt.test.js +206 -206
  226. package/test/runtime-cli.test.js +89 -89
  227. package/test/runtime-closure.test.js +193 -193
  228. package/test/runtime-download.test.js +302 -302
  229. package/test/runtime-layout.test.js +163 -163
  230. package/test/runtime-loader.test.js +255 -255
  231. package/test/runtime-manifest.test.js +159 -159
  232. package/test/runtime-model-fetch.test.js +348 -348
  233. package/test/runtime-provision.test.js +101 -101
  234. package/test/runtime-tarball.test.js +81 -81
  235. package/test/runtime-verify.test.js +319 -279
  236. package/test/scope-candidates.test.js +239 -239
  237. package/test/scope-explicit.test.js +452 -452
  238. package/test/scope-inject-soft.test.js +106 -106
  239. package/test/scope-retrieval.test.js +197 -197
  240. package/test/scope-storage.test.js +255 -255
  241. package/test/scope-strict.test.js +367 -343
  242. package/test/search-fusion.test.js +90 -90
  243. package/test/semantic.test.js +124 -124
  244. package/test/sensitive-scan.test.js +207 -207
  245. package/test/serve-bin.test.js +121 -0
  246. package/test/serve.test.js +196 -0
  247. package/test/service-search.test.js +174 -174
  248. package/test/service.test.js +405 -405
  249. package/test/settings.test.js +261 -261
  250. package/test/skip-invalid-restore.test.js +262 -262
  251. package/test/sleep-action-set.test.js +422 -422
  252. package/test/sleep-heat.test.js +126 -126
  253. package/test/sleep-scheduler-rearm.test.js +89 -89
  254. package/test/sleep.test.js +527 -527
  255. package/test/standalone-api.test.js +751 -544
  256. package/test/store-vector-cache.test.js +98 -98
  257. package/test/store.test.js +396 -396
  258. package/test/stress.test.js +209 -209
  259. package/test/summarize-occurred-at.test.js +127 -0
  260. package/test/summarize-reasoning-effort.test.js +189 -189
  261. package/test/summarize.test.js +1465 -1465
  262. package/test/supersede-state-tracking.test.js +394 -0
  263. package/test/test-count-sync.test.js +60 -60
  264. package/test/tools-dto-schema-matrix.test.js +325 -325
  265. package/test/tools-memory-get-schema.test.js +65 -65
  266. package/test/tools.test.js +428 -374
  267. package/test/updated-at-semantics.test.js +113 -113
  268. package/test/vector-index.test.js +221 -221
  269. package/test/version-check.test.js +152 -152
  270. package/test/webServer-optional.test.js +82 -82
  271. package/test/write-admission.test.js +668 -668
@@ -1,172 +1,172 @@
1
- // 依赖闭包遍历 + 平台过滤(issue #131 / PR-A)
2
- //
3
- // 用途:adopt(从宿主 node_modules 收编)与 export(导出给别的机器)要回答的是
4
- // 同一个问题——「一份能跑的运行时,到底需要哪些包,各自该落到哪个位置」。
5
- //
6
- // 两个刻意的设计:
7
- //
8
- // 1) 按源布局镜像,不压平。入口 transformers.node.mjs 内部是裸导入
9
- // (`import sharp from "sharp"`、`import * as ONNX_NODE from "onnxruntime-node"`),
10
- // 靠 Node 逐级向上查找解析。只要目标目录的嵌套结构与源一致,解析行为就一致;
11
- // 压平会把「同一个包的两个版本分别落在不同层」这类情形悄悄改成另一种结果。
12
- // 所以 rel 保留 `a/node_modules/b` 这种层级。
13
- //
14
- // 2) 跳过 peerDependencies。peer 由宿主提供(例如 @deepseek-ai/* 那些服务),
15
- // 插件自管的运行时不该把宿主的实现复制进来——复制了反而会和宿主的实例分叉。
16
- //
17
- // @module dsh-mneme/runtime/closure
18
- import { existsSync } from "node:fs";
19
- import { dirname, join, relative, sep } from "node:path";
20
- import { readJsonSafe } from "./layout.js";
21
-
22
- /** 闭包条目数上限:纯属防呆,命中即截断并如实上报,不静默丢包。 */
23
- export const DEFAULT_MAX_PACKAGES = 400;
24
-
25
- /**
26
- * 包是否匹配当前平台。npm 用 package.json 的 os / cpu / libc 字段声明平台,
27
- * 支持 `["win32"]` 正向匹配与 `["!darwin"]` 反向排除两种写法。
28
- * 字段缺省表示「不限平台」。
29
- *
30
- * libc(glibc / musl)与前两者不同:**只有确实知道本机 libc 时才据此过滤**。
31
- * 传 null(拿不到证据)就一律放行 —— 反过来「拿不到就当 musl」会把 glibc 变体滤掉,
32
- * 装出一份必然 import 不起来的运行时,而本仓库没有 Linux 机器能验证这条路径。
33
- * 少省十几兆,换「不会静默装错」。
34
- * @param {any} manifest - 包的 package.json(或清单条目)。
35
- * @param {string} platform - process.platform 取值。
36
- * @param {string} arch - process.arch 取值。
37
- * @param {string|null} [libc] - 本机 libc;null/缺省 = 不据此过滤。
38
- * @returns {boolean} 是否匹配。
39
- */
40
- export function matchesPlatform(manifest, platform, arch, libc = null) {
41
- const check = (values, actual) => {
42
- if (!Array.isArray(values) || values.length === 0) return true;
43
- const negated = values.filter((v) => typeof v === "string" && v.startsWith("!"));
44
- const positive = values.filter((v) => typeof v === "string" && !v.startsWith("!"));
45
- // 反向排除优先:命中任一 `!x` 即不匹配(与 npm 的语义一致)。
46
- if (negated.some((v) => v.slice(1) === actual)) return false;
47
- if (positive.length === 0) return true;
48
- return positive.includes(actual);
49
- };
50
- if (!check(manifest?.os, platform) || !check(manifest?.cpu, arch)) return false;
51
- if (libc === null || libc === undefined) return true;
52
- return check(manifest?.libc, libc);
53
- }
54
-
55
- /**
56
- * 按 Node 的解析规则,从 fromDir 出发找 name 的包目录,但**不越出 rootModulesDir**。
57
- *
58
- * Node 的规则是:从所在目录逐级向上,每级查 `<级>/node_modules/<name>`。所以对
59
- * `root/@scope/pkg` 而言,候选依次是 `root/@scope/pkg/node_modules/...`、
60
- * `root/@scope/node_modules/...`、`root/node_modules/...`(这一级是
61
- * node_modules 套 node_modules,通常不存在),最后到 `dirname(root)/node_modules/...`
62
- * ——正好就是 `root/<name>`。因此循环要允许走到 root 的上一级再停。
63
- * @param {string} fromDir - 从哪个包目录出发解析。
64
- * @param {string} name - 依赖名。
65
- * @param {string} rootModulesDir - 允许查找的边界 node_modules 目录。
66
- * @returns {string|null} 包目录绝对路径,找不到返回 null。
67
- */
68
- export function resolvePackageDir(fromDir, name, rootModulesDir) {
69
- const limit = dirname(rootModulesDir);
70
- let dir = fromDir;
71
- for (;;) {
72
- const candidate = join(dir, "node_modules", name);
73
- if (existsSync(join(candidate, "package.json"))) return candidate;
74
- if (dir === limit || dir === dirname(dir)) return null;
75
- dir = dirname(dir);
76
- }
77
- }
78
-
79
- /** 转成 manifest 里使用的、与平台无关的相对路径(始终用 `/`)。 */
80
- function toRel(root, dir) {
81
- return relative(root, dir).split(sep).join("/");
82
- }
83
-
84
- /**
85
- * 计算一份运行时的依赖闭包。
86
- *
87
- * 只走 dependencies 与 optionalDependencies:
88
- * - 必需依赖缺失 → 记进 gaps(让调用方能如实报「这份闭包不完整」);
89
- * - 可选依赖缺失 → 静默跳过(这正是平台分片包在别的平台上的正常形态);
90
- * - 平台不匹配的包 → 记进 skipped(导出给别的机器时,跨平台的残留不会被带走)。
91
- * @param {object} opts - 选项。
92
- * @param {string} opts.rootModulesDir - 源 node_modules 目录。
93
- * @param {string} [opts.entry] - 入口包名。
94
- * @param {string} [opts.platform] - 目标平台。
95
- * @param {string} [opts.arch] - 目标架构。
96
- * @param {number} [opts.maxPackages] - 条目上限。
97
- * @returns {{rootModulesDir: string, entry: string, platform: string, arch: string,
98
- * packages: {name: string, version: string|null, rel: string, srcDir: string, optional: boolean}[],
99
- * gaps: {name: string, from: string, reason: string}[],
100
- * skipped: {name: string, rel: string, reason: string}[], truncated: boolean}}
101
- */
102
- export function planClosure({
103
- rootModulesDir,
104
- entry = "@huggingface/transformers",
105
- platform = process.platform,
106
- arch = process.arch,
107
- maxPackages = DEFAULT_MAX_PACKAGES
108
- } = {}) {
109
- const packages = [];
110
- const gaps = [];
111
- const skipped = [];
112
- const seen = new Set();
113
-
114
- const entryDir = resolvePackageDir(rootModulesDir, entry, rootModulesDir);
115
- if (entryDir === null) {
116
- gaps.push({ name: entry, from: ".", reason: "入口包不存在" });
117
- return { rootModulesDir, entry, platform, arch, packages, gaps, skipped, truncated: false };
118
- }
119
-
120
- // BFS:队列里放已被解析出来的包目录。用真实目录去重,避免软链接/重复边导致来回走。
121
- const queue = [{ dir: entryDir, rel: toRel(rootModulesDir, entryDir), optional: false }];
122
- let truncated = false;
123
-
124
- while (queue.length > 0) {
125
- const current = queue.shift();
126
- if (seen.has(current.dir)) continue;
127
- seen.add(current.dir);
128
-
129
- const manifest = readJsonSafe(join(current.dir, "package.json"));
130
- if (manifest === undefined) {
131
- gaps.push({ name: current.rel, from: current.rel, reason: "package.json 读不到" });
132
- continue;
133
- }
134
- // 入口包本身不做平台过滤:它是用户明确要的那一个。
135
- if (current.rel !== toRel(rootModulesDir, entryDir) && !matchesPlatform(manifest, platform, arch)) {
136
- skipped.push({ name: manifest.name ?? current.rel, rel: current.rel, reason: "平台不匹配" });
137
- continue;
138
- }
139
- if (packages.length >= maxPackages) {
140
- truncated = true;
141
- break;
142
- }
143
- packages.push({
144
- name: manifest.name ?? current.rel,
145
- version: manifest.version ?? null,
146
- rel: current.rel,
147
- srcDir: current.dir,
148
- optional: current.optional
149
- });
150
-
151
- const required = Object.keys(manifest.dependencies ?? {});
152
- const optional = Object.keys(manifest.optionalDependencies ?? {});
153
- for (const name of [...required, ...optional]) {
154
- const isOptional = optional.includes(name);
155
- const dir = resolvePackageDir(current.dir, name, rootModulesDir);
156
- if (dir === null) {
157
- // 可选依赖缺失是正常形态(平台分片包在别的平台就是这样),不报警。
158
- if (!isOptional) gaps.push({ name, from: current.rel, reason: "必需依赖未找到" });
159
- continue;
160
- }
161
- const rel = toRel(rootModulesDir, dir);
162
- // 不允许越出源树:越出说明解析跑到了宿主其它位置,这份闭包就不是自包含的。
163
- if (rel.startsWith("..")) {
164
- gaps.push({ name, from: current.rel, reason: "解析结果越出源 node_modules" });
165
- continue;
166
- }
167
- queue.push({ dir, rel, optional: isOptional });
168
- }
169
- }
170
-
171
- return { rootModulesDir, entry, platform, arch, packages, gaps, skipped, truncated };
172
- }
1
+ // 依赖闭包遍历 + 平台过滤(issue #131 / PR-A)
2
+ //
3
+ // 用途:adopt(从宿主 node_modules 收编)与 export(导出给别的机器)要回答的是
4
+ // 同一个问题——「一份能跑的运行时,到底需要哪些包,各自该落到哪个位置」。
5
+ //
6
+ // 两个刻意的设计:
7
+ //
8
+ // 1) 按源布局镜像,不压平。入口 transformers.node.mjs 内部是裸导入
9
+ // (`import sharp from "sharp"`、`import * as ONNX_NODE from "onnxruntime-node"`),
10
+ // 靠 Node 逐级向上查找解析。只要目标目录的嵌套结构与源一致,解析行为就一致;
11
+ // 压平会把「同一个包的两个版本分别落在不同层」这类情形悄悄改成另一种结果。
12
+ // 所以 rel 保留 `a/node_modules/b` 这种层级。
13
+ //
14
+ // 2) 跳过 peerDependencies。peer 由宿主提供(例如 @deepseek-ai/* 那些服务),
15
+ // 插件自管的运行时不该把宿主的实现复制进来——复制了反而会和宿主的实例分叉。
16
+ //
17
+ // @module dsh-mneme/runtime/closure
18
+ import { existsSync } from "node:fs";
19
+ import { dirname, join, relative, sep } from "node:path";
20
+ import { readJsonSafe } from "./layout.js";
21
+
22
+ /** 闭包条目数上限:纯属防呆,命中即截断并如实上报,不静默丢包。 */
23
+ export const DEFAULT_MAX_PACKAGES = 400;
24
+
25
+ /**
26
+ * 包是否匹配当前平台。npm 用 package.json 的 os / cpu / libc 字段声明平台,
27
+ * 支持 `["win32"]` 正向匹配与 `["!darwin"]` 反向排除两种写法。
28
+ * 字段缺省表示「不限平台」。
29
+ *
30
+ * libc(glibc / musl)与前两者不同:**只有确实知道本机 libc 时才据此过滤**。
31
+ * 传 null(拿不到证据)就一律放行 —— 反过来「拿不到就当 musl」会把 glibc 变体滤掉,
32
+ * 装出一份必然 import 不起来的运行时,而本仓库没有 Linux 机器能验证这条路径。
33
+ * 少省十几兆,换「不会静默装错」。
34
+ * @param {any} manifest - 包的 package.json(或清单条目)。
35
+ * @param {string} platform - process.platform 取值。
36
+ * @param {string} arch - process.arch 取值。
37
+ * @param {string|null} [libc] - 本机 libc;null/缺省 = 不据此过滤。
38
+ * @returns {boolean} 是否匹配。
39
+ */
40
+ export function matchesPlatform(manifest, platform, arch, libc = null) {
41
+ const check = (values, actual) => {
42
+ if (!Array.isArray(values) || values.length === 0) return true;
43
+ const negated = values.filter((v) => typeof v === "string" && v.startsWith("!"));
44
+ const positive = values.filter((v) => typeof v === "string" && !v.startsWith("!"));
45
+ // 反向排除优先:命中任一 `!x` 即不匹配(与 npm 的语义一致)。
46
+ if (negated.some((v) => v.slice(1) === actual)) return false;
47
+ if (positive.length === 0) return true;
48
+ return positive.includes(actual);
49
+ };
50
+ if (!check(manifest?.os, platform) || !check(manifest?.cpu, arch)) return false;
51
+ if (libc === null || libc === undefined) return true;
52
+ return check(manifest?.libc, libc);
53
+ }
54
+
55
+ /**
56
+ * 按 Node 的解析规则,从 fromDir 出发找 name 的包目录,但**不越出 rootModulesDir**。
57
+ *
58
+ * Node 的规则是:从所在目录逐级向上,每级查 `<级>/node_modules/<name>`。所以对
59
+ * `root/@scope/pkg` 而言,候选依次是 `root/@scope/pkg/node_modules/...`、
60
+ * `root/@scope/node_modules/...`、`root/node_modules/...`(这一级是
61
+ * node_modules 套 node_modules,通常不存在),最后到 `dirname(root)/node_modules/...`
62
+ * ——正好就是 `root/<name>`。因此循环要允许走到 root 的上一级再停。
63
+ * @param {string} fromDir - 从哪个包目录出发解析。
64
+ * @param {string} name - 依赖名。
65
+ * @param {string} rootModulesDir - 允许查找的边界 node_modules 目录。
66
+ * @returns {string|null} 包目录绝对路径,找不到返回 null。
67
+ */
68
+ export function resolvePackageDir(fromDir, name, rootModulesDir) {
69
+ const limit = dirname(rootModulesDir);
70
+ let dir = fromDir;
71
+ for (;;) {
72
+ const candidate = join(dir, "node_modules", name);
73
+ if (existsSync(join(candidate, "package.json"))) return candidate;
74
+ if (dir === limit || dir === dirname(dir)) return null;
75
+ dir = dirname(dir);
76
+ }
77
+ }
78
+
79
+ /** 转成 manifest 里使用的、与平台无关的相对路径(始终用 `/`)。 */
80
+ function toRel(root, dir) {
81
+ return relative(root, dir).split(sep).join("/");
82
+ }
83
+
84
+ /**
85
+ * 计算一份运行时的依赖闭包。
86
+ *
87
+ * 只走 dependencies 与 optionalDependencies:
88
+ * - 必需依赖缺失 → 记进 gaps(让调用方能如实报「这份闭包不完整」);
89
+ * - 可选依赖缺失 → 静默跳过(这正是平台分片包在别的平台上的正常形态);
90
+ * - 平台不匹配的包 → 记进 skipped(导出给别的机器时,跨平台的残留不会被带走)。
91
+ * @param {object} opts - 选项。
92
+ * @param {string} opts.rootModulesDir - 源 node_modules 目录。
93
+ * @param {string} [opts.entry] - 入口包名。
94
+ * @param {string} [opts.platform] - 目标平台。
95
+ * @param {string} [opts.arch] - 目标架构。
96
+ * @param {number} [opts.maxPackages] - 条目上限。
97
+ * @returns {{rootModulesDir: string, entry: string, platform: string, arch: string,
98
+ * packages: {name: string, version: string|null, rel: string, srcDir: string, optional: boolean}[],
99
+ * gaps: {name: string, from: string, reason: string}[],
100
+ * skipped: {name: string, rel: string, reason: string}[], truncated: boolean}}
101
+ */
102
+ export function planClosure({
103
+ rootModulesDir,
104
+ entry = "@huggingface/transformers",
105
+ platform = process.platform,
106
+ arch = process.arch,
107
+ maxPackages = DEFAULT_MAX_PACKAGES
108
+ } = {}) {
109
+ const packages = [];
110
+ const gaps = [];
111
+ const skipped = [];
112
+ const seen = new Set();
113
+
114
+ const entryDir = resolvePackageDir(rootModulesDir, entry, rootModulesDir);
115
+ if (entryDir === null) {
116
+ gaps.push({ name: entry, from: ".", reason: "入口包不存在" });
117
+ return { rootModulesDir, entry, platform, arch, packages, gaps, skipped, truncated: false };
118
+ }
119
+
120
+ // BFS:队列里放已被解析出来的包目录。用真实目录去重,避免软链接/重复边导致来回走。
121
+ const queue = [{ dir: entryDir, rel: toRel(rootModulesDir, entryDir), optional: false }];
122
+ let truncated = false;
123
+
124
+ while (queue.length > 0) {
125
+ const current = queue.shift();
126
+ if (seen.has(current.dir)) continue;
127
+ seen.add(current.dir);
128
+
129
+ const manifest = readJsonSafe(join(current.dir, "package.json"));
130
+ if (manifest === undefined) {
131
+ gaps.push({ name: current.rel, from: current.rel, reason: "package.json 读不到" });
132
+ continue;
133
+ }
134
+ // 入口包本身不做平台过滤:它是用户明确要的那一个。
135
+ if (current.rel !== toRel(rootModulesDir, entryDir) && !matchesPlatform(manifest, platform, arch)) {
136
+ skipped.push({ name: manifest.name ?? current.rel, rel: current.rel, reason: "平台不匹配" });
137
+ continue;
138
+ }
139
+ if (packages.length >= maxPackages) {
140
+ truncated = true;
141
+ break;
142
+ }
143
+ packages.push({
144
+ name: manifest.name ?? current.rel,
145
+ version: manifest.version ?? null,
146
+ rel: current.rel,
147
+ srcDir: current.dir,
148
+ optional: current.optional
149
+ });
150
+
151
+ const required = Object.keys(manifest.dependencies ?? {});
152
+ const optional = Object.keys(manifest.optionalDependencies ?? {});
153
+ for (const name of [...required, ...optional]) {
154
+ const isOptional = optional.includes(name);
155
+ const dir = resolvePackageDir(current.dir, name, rootModulesDir);
156
+ if (dir === null) {
157
+ // 可选依赖缺失是正常形态(平台分片包在别的平台就是这样),不报警。
158
+ if (!isOptional) gaps.push({ name, from: current.rel, reason: "必需依赖未找到" });
159
+ continue;
160
+ }
161
+ const rel = toRel(rootModulesDir, dir);
162
+ // 不允许越出源树:越出说明解析跑到了宿主其它位置,这份闭包就不是自包含的。
163
+ if (rel.startsWith("..")) {
164
+ gaps.push({ name, from: current.rel, reason: "解析结果越出源 node_modules" });
165
+ continue;
166
+ }
167
+ queue.push({ dir, rel, optional: isOptional });
168
+ }
169
+ }
170
+
171
+ return { rootModulesDir, entry, platform, arch, packages, gaps, skipped, truncated };
172
+ }