@dsh-workbench/core 0.1.7 → 0.1.9

package/dist/host/index.js

@@ -1380,7 +1380,7 @@ function readBundleManager(ctx) {
 }
 //#endregion
 //#region src/shared/api/host/installed-module.ts
-var PACKAGE_NAME = /^(?:@[a-z0-9][a-z0-9._~-]*\/)?[a-z0-9][a-z0-9._~-]*$/;
+var PACKAGE_NAME$1 = /^(?:@[a-z0-9][a-z0-9._~-]*\/)?[a-z0-9][a-z0-9._~-]*$/;
 function asRecord$1(value) {
 	return typeof value === "object" && value !== null ? value : void 0;
 }
@@ -1388,7 +1388,7 @@ function asText$1(value) {
 	return typeof value === "string" ? value : "";
 }
 function isPackageName(value) {
-	return value.length <= 214 && PACKAGE_NAME.test(value);
+	return value.length <= 214 && PACKAGE_NAME$1.test(value);
 }
 function profileDirOf(ctx) {
 	const profile = ctx.get("profileContext");
@@ -1477,6 +1477,118 @@ function readInstalledModule(profileDir, pkg) {
 	};
 }
 //#endregion
+//#region src/shared/api/host/profile-registries.ts
+var REGISTRY_MARKER = "# dsh-workbench 写的：这一行的包只在这个源上（dsh 卸载没有源参数，pnpm 得问得到）";
+function scopeOf(pkg) {
+	const slash = pkg.indexOf("/");
+	return pkg.startsWith("@") && slash > 1 ? pkg.slice(0, slash) : null;
+}
+function normalizeRegistry(url) {
+	const trimmed = url.trim();
+	return trimmed.endsWith("/") ? trimmed : `${trimmed}/`;
+}
+function sameRegistry(a, b) {
+	if (a === null || a === void 0 || b === null || b === void 0) return false;
+	if (a.trim() === "" || b.trim() === "") return false;
+	return normalizeRegistry(a) === normalizeRegistry(b);
+}
+function registryRootOf(name, tarball) {
+	const at = tarball.indexOf(`/${name}/-/`);
+	if (at > 0) return normalizeRegistry(tarball.slice(0, at));
+	try {
+		return normalizeRegistry(new URL(tarball).origin);
+	} catch {
+		return null;
+	}
+}
+var PACKAGE_KEY = /^ {2}'?(\S+)'?:\s*$/;
+var PACKAGE_NAME = /^(?:@[^/\s]+\/)?[^/\s@]+$/;
+var TARBALL = /tarball:\s*(\S+?)\}?\s*$/;
+function sourcesFromLockfile(text) {
+	const found = /* @__PURE__ */ new Map();
+	let name = null;
+	for (const line of text.split("\n")) {
+		const key = PACKAGE_KEY.exec(line);
+		if (key !== null) {
+			const raw = key[1] ?? "";
+			const at = raw.lastIndexOf("@");
+			const candidate = at > 0 ? raw.slice(0, at) : "";
+			name = PACKAGE_NAME.test(candidate) ? candidate : null;
+			continue;
+		}
+		if (name === null) continue;
+		const tarball = TARBALL.exec(line);
+		if (tarball === null) continue;
+		const scope = scopeOf(name);
+		const root = registryRootOf(name, tarball[1] ?? "");
+		name = null;
+		if (scope === null || root === null) continue;
+		const seen = found.get(scope);
+		if (seen === void 0) found.set(scope, root);
+		else if (!sameRegistry(seen, root)) found.set(scope, null);
+	}
+	return [...found.entries()].filter((entry) => entry[1] !== null).map(([scope, registry]) => ({
+		scope,
+		registry
+	}));
+}
+function mergeRegistryConfig(text, entries) {
+	const lines = text.split("\n");
+	const present = (key) => lines.some((line) => line.trim().startsWith(`${key}=`));
+	const appended = [];
+	const added = [];
+	for (const entry of entries) {
+		const key = `${entry.scope}:registry`;
+		if (entry.registry.trim() === "" || present(key)) continue;
+		lines.push(`${key}=${normalizeRegistry(entry.registry)}`);
+		appended.push(`${key}=${normalizeRegistry(entry.registry)}`);
+		added.push(key);
+	}
+	if (appended.length === 0) return {
+		text,
+		added: []
+	};
+	const base = text === "" ? "" : text.endsWith("\n") ? text : `${text}\n`;
+	return {
+		text: `${base}${base.includes("# dsh-workbench 写的：这一行的包只在这个源上（dsh 卸载没有源参数，pnpm 得问得到）") ? "" : `${REGISTRY_MARKER}\n`}${appended.join("\n")}\n`,
+		added
+	};
+}
+function registryFailure(change) {
+	if (change.application !== "failed") return false;
+	if (change.packageResult?.kind === "not-found") return true;
+	const output = change.packageResult?.output ?? change.errorDetail ?? "";
+	return /ERR_PNPM_[A-Z_]*FETCH/.test(output);
+}
+function createRegistryNotebook(dir) {
+	return {
+		sources() {
+			if (dir === null) return [];
+			try {
+				return sourcesFromLockfile(fs.readFileSync(path.join(dir, "pnpm-lock.yaml"), "utf8"));
+			} catch {
+				return [];
+			}
+		},
+		remember(entries) {
+			if (dir === null || entries.length === 0) return [];
+			const file = path.join(dir, ".npmrc");
+			let before;
+			try {
+				before = fs.readFileSync(file, "utf8");
+			} catch {
+				before = "";
+			}
+			const merged = mergeRegistryConfig(before, entries);
+			if (merged.added.length === 0) return [];
+			const temp = `${file}.dsh-workbench.tmp`;
+			fs.writeFileSync(temp, merged.text, { mode: 384 });
+			fs.renameSync(temp, file);
+			return merged.added;
+		}
+	};
+}
+//#endregion
 //#region src/shared/api/host/release.ts
 function parseLedgerVersion(stdout, name) {
 	const parsed = parseJsonLoose(stdout);
@@ -1703,7 +1815,7 @@ function bundleNameOf(change, spec) {
 	return at > 0 ? spec.slice(0, at) : spec;
 }
 function createMarketService(options) {
-	const { manifests, manifestsNow, probe, shell, logger, onInstalled, onUninstalled, isReleased, bundles, installed, modules } = options;
+	const { manifests, manifestsNow, probe, shell, logger, onInstalled, onUninstalled, isReleased, bundles, installed, modules, registries } = options;
 	const command = options.command ?? "opencli";
 	const catalog = () => (manifestsNow?.() ?? manifests).filter((manifest) => (manifest.env?.plugins ?? []).length > 0 || (manifest.env?.libs ?? []).length > 0);
 	function requireEntry(unit) {
@@ -1850,6 +1962,38 @@ function createMarketService(options) {
 			};
 		}
 	}
+	async function rememberInstalledSource(manager, pkg, registry) {
+		if (registries === void 0 || registry === void 0) return;
+		const scope = scopeOf(pkg);
+		if (scope === null) return;
+		const known = await manager.registries().catch(() => null);
+		if (known !== null && sameRegistry(known.resolved, registry)) return;
+		try {
+			const added = registries.remember([{
+				scope,
+				registry
+			}]);
+			if (added.length > 0) logger?.info?.(`[dsh-workbench] 把 ${added.join("、")} 记进 profile 的 .npmrc（卸载只认 pnpm 的配置）`);
+		} catch (error) {
+			logger?.warn?.(`[dsh-workbench] 记这个 scope 的源失败：${String(error)}`);
+		}
+	}
+	async function retryAfterRegistryFix(manager, pkg, change) {
+		if (registries === void 0 || !registryFailure(change)) return change;
+		const known = await manager.registries().catch(() => null);
+		const missing = registries.sources().filter((entry) => !sameRegistry(entry.registry, known?.resolved ?? null));
+		if (missing.length === 0) return change;
+		let added;
+		try {
+			added = registries.remember(missing);
+		} catch (error) {
+			logger?.warn?.(`[dsh-workbench] 卸 ${pkg} 撞上源，但补配置失败：${String(error)}`);
+			return change;
+		}
+		if (added.length === 0) return change;
+		logger?.warn?.(`[dsh-workbench] 卸 ${pkg} 撞上源：补 ${added.join("、")} 后重试一次`);
+		return await manager.remove(pkg).catch(() => null) ?? change;
+	}
 	const service = {
 		async list() {
 			return snapshot();
@@ -1955,7 +2099,8 @@ function createMarketService(options) {
 			};
 		},
 		async installBundle(spec, options) {
-			const change = await requireBundles().install({
+			const manager = requireBundles();
+			const change = await manager.install({
 				spec,
 				...options?.registry === void 0 ? {} : { registry: options.registry },
 				...options?.requestId === void 0 ? {} : { requestId: options.requestId },
@@ -1967,6 +2112,7 @@ function createMarketService(options) {
 				kind: "unknown",
 				pkg
 			};
+			await rememberInstalledSource(manager, pkg, options?.registry);
 			return {
 				change,
 				module,
@@ -1982,8 +2128,9 @@ function createMarketService(options) {
 			const unit = module.kind === "module" ? module.unit : void 0;
 			const manifest = unit === void 0 ? void 0 : catalog().find((item) => item.id === unit);
 			const before = unit === void 0 ? void 0 : await capabilityOf(unit, true);
-			const change = await manager.remove(pkg);
-			if (change === null) throw new Error("插件管理器回的卸载结果看不懂。");
+			const first = await manager.remove(pkg);
+			if (first === null) throw new Error("插件管理器回的卸载结果看不懂。");
+			const change = await retryAfterRegistryFix(manager, pkg, first);
 			if (change.application !== "applied" || unit === void 0) return {
 				change,
 				keptFor: []
@@ -2682,6 +2829,7 @@ var HOST_UNITS = [{
 			bundles: readBundleManager(ctx),
 			installed: (pkg) => readInstalledModule(profileDir, pkg),
 			modules: () => profileModules(profileDir),
+			registries: createRegistryNotebook(profileDir),
 			onUninstalled: (unit) => {
 				panels.release(unit);
 			},