@simon_he/pi/dist/index.mjs

Project-aware CLI that detects npm, pnpm, yarn, bun, Go, Rust and Python projects, then routes installs, scripts, builds and workspace commands automatically.

import { S as help, _ as getSupportedPkgToolNames, a as pinit, b as resolvePkgTool, c as printPrunDoctor, d as getCcommand, f as pci, g as getPkgToolStatus, h as forgetPkgToolPreference, i as pio, l as printPrunInit, m as loading, n as pu, o as pil, p as pi, r as pix, s as pfind, t as pui, u as prun, v as printPkgToolCandidates, x as pa, y as printPkgToolStatus } from "./pui-DDvoNMqM.mjs";
import path from "node:path";
import process from "node:process";
import fg from "fast-glob";
import { spaceFormat } from "lazy-js-utils";
import { hasPkg, isGo, isRust, jsShell, useNodeWorker } from "lazy-js-utils/node";
import color from "picocolors";

//#region src/index.ts
let rootPath = process.cwd();
const runMap = {
	pi,
	pix,
	pa,
	pu,
	pui,
	pil,
	pci,
	prun,
	pinit,
	pfind,
	pio
};
const isZh = process.env.PI_Lang === "zh";
const pkgToolFlagCommands = new Set([
	"pi",
	"pil",
	"pci"
]);
const supportedPkgTools = new Set(getSupportedPkgToolNames());
function getExecName(argv1 = process.argv[1] ?? "") {
	return path.basename(argv1).replace(/(?:\.(?:mjs|cjs|js|cmd|ps1))+$/i, "");
}
function parsePkgToolFlags(argv) {
	const hasInspectFlag = argv.includes("--show-tool") || argv.includes("--list-tools");
	let chooseTool = false;
	let forgetTool = false;
	let listTools = false;
	let showTool = false;
	let showToolJson = false;
	let preferredTool = "";
	let invalidPreferredTool = "";
	const normalizedArgv = [];
	for (let i = 0; i < argv.length; i++) {
		const arg = argv[i];
		if (arg === "--forget-tool") {
			forgetTool = true;
			continue;
		}
		if (arg === "--show-tool") {
			showTool = true;
			continue;
		}
		if (arg === "--list-tools") {
			listTools = true;
			continue;
		}
		if (arg === "--json" && hasInspectFlag) {
			showToolJson = true;
			continue;
		}
		if (arg === "--choose-tool") {
			chooseTool = true;
			const next = argv[i + 1];
			if (next && !next.startsWith("-") && supportedPkgTools.has(next)) {
				preferredTool = next;
				i++;
			}
			continue;
		}
		if (arg.startsWith("--choose-tool=")) {
			chooseTool = true;
			const value = arg.slice(14);
			if (supportedPkgTools.has(value)) preferredTool = value;
			else invalidPreferredTool = value;
			continue;
		}
		normalizedArgv.push(arg);
	}
	return {
		chooseTool,
		forgetTool,
		invalidPreferredTool,
		listTools,
		normalizedArgv,
		preferredTool,
		showTool,
		showToolJson
	};
}
async function setup() {
	const exec = getExecName(process.argv[1]);
	const argv = process.argv.slice(2);
	await help(argv);
	if (exec === "prun" && argv[0] === "--init") {
		printPrunInit(argv.slice(1));
		return;
	}
	if ((exec === "prun" || exec === "pfind") && argv[0] === "--doctor") {
		printPrunDoctor();
		return;
	}
	const supportsPkgToolFlags = pkgToolFlagCommands.has(exec);
	const { chooseTool, forgetTool, invalidPreferredTool, listTools, normalizedArgv, preferredTool, showTool, showToolJson } = supportsPkgToolFlags ? parsePkgToolFlags(argv) : {
		chooseTool: false,
		forgetTool: false,
		invalidPreferredTool: "",
		listTools: false,
		normalizedArgv: argv,
		preferredTool: "",
		showTool: false,
		showToolJson: false
	};
	if (invalidPreferredTool) {
		console.error(color.red(isZh ? `不支持直接指定 ${invalidPreferredTool}，可选值为: ${getSupportedPkgToolNames().join(", ")}` : `Unsupported tool "${invalidPreferredTool}". Valid values: ${getSupportedPkgToolNames().join(", ")}`));
		process.exitCode = 1;
		return;
	}
	if (chooseTool) process.env.PI_FORCE_PICK_TOOL = "1";
	else delete process.env.PI_FORCE_PICK_TOOL;
	if (forgetTool) process.env.PI_FORGET_PICK_TOOL = "1";
	else delete process.env.PI_FORGET_PICK_TOOL;
	if (preferredTool) process.env.PI_PREFERRED_TOOL = preferredTool;
	else delete process.env.PI_PREFERRED_TOOL;
	let params = spaceFormat(normalizedArgv.join(" ")).trim();
	const hasPackage = await hasPkg(rootPath);
	if (supportsPkgToolFlags && (chooseTool || forgetTool || showTool || listTools)) {
		if (!hasPackage) {
			console.log(color.yellow(isZh ? "当前命令仅在 Node 项目的包管理场景下可用。" : "This option is only available for package-manager selection in Node projects."));
			return;
		}
		if (showTool || listTools) {
			if (forgetTool && !chooseTool) {
				const removed = await forgetPkgToolPreference();
				console.log(removed ? color.green(isZh ? "已清除当前 workspace 保存的包管理器选择。" : "Cleared the saved package-manager choice for this workspace.") : color.yellow(isZh ? "当前 workspace 没有保存的包管理器选择。" : "No saved package-manager choice was found for this workspace."));
			}
			if (chooseTool) await resolvePkgTool();
			const status = await getPkgToolStatus();
			if (listTools) printPkgToolCandidates(status, { json: showToolJson });
			else printPkgToolStatus(status, { json: showToolJson });
			return;
		}
		if (normalizedArgv.length === 0) {
			if (forgetTool && !chooseTool) {
				const removed = await forgetPkgToolPreference();
				console.log(removed ? color.green(isZh ? "已清除当前 workspace 保存的包管理器选择。" : "Cleared the saved package-manager choice for this workspace.") : color.yellow(isZh ? "当前 workspace 没有保存的包管理器选择。" : "No saved package-manager choice was found for this workspace."));
				return;
			}
			await resolvePkgTool();
			return;
		}
	}
	if (!hasPackage) {
		if (await isGo(rootPath)) {
			if (exec === "pi") {
				const loading_status = await loading(`${isZh ? "正在为您安装" : "Installing"} ${params} ...\n`);
				const { status } = params ? await useNodeWorker(`go get ${params}`) : await useNodeWorker("go mod tidy");
				if (status === 0) loading_status.succeed(color.green(isZh ? "安装成功! 😊" : "Installed successfully! 😊"));
				else loading_status.fail(color.red(isZh ? "安装失败 😭" : "Failed to install 😭"));
			} else if (exec === "pui") {
				const loading_status = await loading(`${isZh ? "正在为您卸载" : "Uninstalling"} ${params} ...\n`);
				const { status } = await useNodeWorker(`go clean ${params}`);
				if (status === 0) loading_status.succeed(color.green(isZh ? "卸载成功! 😊" : "Uninstalled successfully! 😊"));
				else loading_status.fail(color.red(isZh ? "卸载失败 😭" : "Failed to uninstall 😭"));
			} else if (exec === "prun") {
				const target = (await fg(params ? params.endsWith(".go") ? [`**/${params}`] : [`**/${params}.go`, `**/${params}/main.go`] : "main.go"))[0];
				if (target) return await jsShell(`go run ${target}`, "inherit");
				const { ccommand } = getCcommand();
				return ccommand(params);
			} else if (exec === "pinit") await jsShell(`go mod init ${params}`, "inherit");
			else if (exec === "pbuild") await jsShell(`go build ${params}`, "inherit");
			else console.log(color.red(isZh ? "当前指令还不支持" : "The commands is not supported"));
			process.exit();
		}
		let projectPath = "";
		if (params && !await isRust()) {
			projectPath = params.split(" ")[0];
			rootPath = path.resolve(rootPath, projectPath);
			params = params.replace(projectPath, "").trim();
		}
		if (await isRust(rootPath)) {
			if (exec === "pi") {
				const loading_status = await loading(`${isZh ? "正在为您安装" : "Installing"} ${params} ...\n`);
				const { status } = await useNodeWorker(`cargo install ${params}${projectPath ? `--manifest-path=./${projectPath}/Cargo.toml` : ""}`);
				if (status === 0) loading_status.succeed(color.green(isZh ? "安装成功! 😊" : "Installed successfully! 😊"));
				else loading_status.fail(color.red(isZh ? "安装失败 😭" : "Failed to install 😭"));
			} else if (exec === "pui") {
				const loading_status = await loading(`${isZh ? "正在为您卸载" : "Uninstalling"} ${params} ...\n`);
				const { status } = await useNodeWorker(`cargo uninstall ${params}${projectPath ? `--manifest-path=./${projectPath}/Cargo.toml` : ""}`);
				if (status === 0) loading_status.succeed(color.green(isZh ? "卸载成功! 😊" : "Uninstalled successfully! 😊"));
				else loading_status.fail(color.red(isZh ? "卸载失败 😭" : "Failed to uninstall 😭"));
			} else if (exec === "prun") await jsShell(`cargo run ${params}${projectPath ? `--manifest-path=./${projectPath}/Cargo.toml` : ""}`, "inherit");
			else if (exec === "pinit") await jsShell(`cargo init ${params}${projectPath ? `--manifest-path=./${projectPath}/Cargo.toml` : ""}`, "inherit");
			else if (exec === "pbuild") await jsShell(`cargo build ${params}${projectPath ? `--manifest-path=./${projectPath}/Cargo.toml` : ""}`, "inherit");
			else console.log(color.red(isZh ? "当前指令还不支持" : "The commands is not supported"));
			process.exit();
		}
	}
	const handler = runMap[exec];
	if (!handler) {
		if (exec === "pbuild") console.log(color.yellow(isZh ? "pbuild 仅支持 Go/Rust 项目（go build / cargo build）。" : "pbuild is only supported for Go/Rust projects (go build / cargo build)."));
		else console.log(color.yellow(isZh ? "命令不存在, 请执行 pi -h 查看帮助" : "The command does not exist, please execute pi -h to view the help"));
		return;
	}
	const pkg = normalizedArgv.filter((v) => !v.startsWith("-")).join(" ");
	await handler(params, pkg);
}

//#endregion
export { setup };