ylyx-cli 1.0.16 → 1.1.1

package/lib/config.js

@@ -1,283 +1,325 @@
-const path = require('path');
-const fs = require('fs-extra');
-const { log } = require('./utils');
-
-function getConfigPath() {
-	return path.join(process.cwd(), '.ylyxrc.json');
-}
-
-function normalizeMode(mode) {
-	if (typeof mode !== 'string') return null;
-	const m = mode.trim().toLowerCase();
-	if (m === 'dev' || m === 'prod') return m;
-	return null;
-}
-
-function parseDotEnvFile(envPath) {
-	if (!fs.existsSync(envPath)) return {};
-	const content = fs.readFileSync(envPath, 'utf-8');
-	const lines = content.split(/\r?\n/);
-	const out = {};
-	for (const raw of lines) {
-		const line = raw.trim();
-		if (!line || line.startsWith('#')) continue;
-		const idx = line.indexOf('=');
-		if (idx < 0) continue;
-		const key = line.slice(0, idx).trim();
-		let value = line.slice(idx + 1).trim();
-		if (
-			(value.startsWith('"') && value.endsWith('"')) ||
-			(value.startsWith("'") && value.endsWith("'"))
-		) {
-			value = value.slice(1, -1);
-		}
-		out[key] = value;
-	}
-	return out;
-}
-
-function normalizePublicUrlToDir(publicUrl) {
-	if (!publicUrl || typeof publicUrl !== 'string') return '';
-	let v = publicUrl.trim();
-	v = v.replace(/\\/g, '/');
-	// ignore full urls
-	if (/^https?:\/\//i.test(v)) return '';
-	v = v.replace(/^\/+/, '').replace(/\/+$/, '');
-	return v;
-}
-
-function resolveProdBuildDir(config) {
-	// 1) 优先使用 .ylyxrc.json 的 buildDir（可为相对/绝对路径）
-	if (config.buildDir) {
-		return path.resolve(process.cwd(), config.buildDir);
-	}
-
-	// 2) 默认：读取 .env.production 的 VUE_APP_PUBLIC_URL，拼成 <publicUrlDir>
-	const envProdPath = path.join(process.cwd(), '.env.production');
-	const env = parseDotEnvFile(envProdPath);
-	const publicUrlDir = normalizePublicUrlToDir(env.VUE_APP_PUBLIC_URL);
-	if (publicUrlDir) {
-		return path.resolve(process.cwd(), publicUrlDir);
-	}
-	throw new Error(
-		'无法确定 prod 打包输出目录：请在 .ylyxrc.json 中配置 buildDir，或在 .env.production 中配置 VUE_APP_PUBLIC_URL'
-	);
-}
-
-function readConfigSafe(configPath) {
-	if (!fs.existsSync(configPath)) return {};
-	try {
-		return fs.readJsonSync(configPath);
-	} catch (e) {
-		throw new Error(`读取 .ylyxrc.json 失败，请确认它是有效 JSON：${e.message}`);
-	}
-}
-
-function writeIfAllowed(filePath, content, force) {
-	if (fs.existsSync(filePath) && !force) {
-		log.warn(`已存在，跳过：${path.relative(process.cwd(), filePath)}`);
-		return false;
-	}
-	fs.ensureDirSync(path.dirname(filePath));
-	fs.writeFileSync(filePath, content, 'utf-8');
-	log.success(`已写入：${path.relative(process.cwd(), filePath)}`);
-	return true;
-}
-
-function deriveDefaultVariantContent(baseContent, mode) {
-	const isDev = mode === 'dev';
-
-	let content = baseContent || '';
-	const devRe = /let\s+isDev\s*=\s*[^;]*;/;
-
-	if (devRe.test(content)) {
-		content = content.replace(devRe, `let isDev = ${isDev};`);
-	} else {
-		content = `let isDev = ${isDev};\n` + content;
-	}
-
-	// isProd 已废弃：prod 使用 isDev=false 表示
-	// 删除历史遗留的 isProd 变量与输出字段（若存在）
-	content = content.replace(/let\s+isProd\s*=\s*[^;]*;\s*\r?\n?/g, '');
-	content = content.replace(/window\.config\.IS_PROD\s*=\s*[^;]*;\s*\r?\n?/g, '');
-	// 删除对象形式的 IS_PROD 字段（例如 IS_PROD: isProd,）
-	content = content.replace(/^\s*IS_PROD\s*:\s*[^,}]*,?\s*\r?\n?/gm, '');
-
-	return content;
-}
-
-function copyFile({ src, dest }) {
-	fs.ensureDirSync(path.dirname(dest));
-	if (!fs.existsSync(src)) {
-		log.warn(`未找到源文件：${path.relative(process.cwd(), src)}`);
-		return { ok: false, method: 'missing' };
-	}
-
-	// 如果目标已存在，先删除
-	if (fs.existsSync(dest)) {
-		fs.removeSync(dest);
-	}
-
-	try {
-		fs.copyFileSync(src, dest);
-		return { ok: true };
-	} catch (e) {
-		log.error(`写入失败：${e.message}`);
-		return { ok: false };
-	}
-}
-
-/**
- * 写入/更新 .ylyxrc.json 的 mode 字段（dev|prod），并与已有配置合并
- */
-function setMode(mode) {
-	const normalized = normalizeMode(mode);
-	if (!normalized) {
-		throw new Error(`mode 取值只能是 dev 或 prod（当前: ${mode}）`);
-	}
-
-	const configPath = getConfigPath();
-	const config = readConfigSafe(configPath);
-
-	// 可选：如果当前项目目录存在 public/default-{mode}.js，则复制到 public/default.js
-	// 注意：切换应发生在“项目目录”(process.cwd())，而不是 CLI 自身的 lib 目录(__dirname)
-	// 优先从 .ylyxrc.json 中读取 publicDir，其次使用默认 ./public
-	const configDir = path.resolve(process.cwd(), config.configDir || 'config');
-	const publicDir = path.resolve(process.cwd(), config.publicDir || 'public');
-	const defaultVariant = path.join(configDir, `default-${normalized}.js`);
-	const defaultFile =
-		normalized === 'prod'
-			? path.join(resolveProdBuildDir(config), 'default.js')
-			: path.join(publicDir, 'default.js');
-
-	// 将对应 variant 文件 copy 到目标路径
-	const result = copyFile({
-		src: defaultVariant,
-		dest: defaultFile,
-	});
-
-	if (result.ok) {
-		log.success(
-			`已切换 ${path.relative(process.cwd(), defaultFile)} <- ${path.relative(
-				process.cwd(),
-				defaultVariant
-			)}`
-		);
-	} else if (!result.ok && !fs.existsSync(defaultVariant)) {
-		log.warn(`未找到 ${path.relative(process.cwd(), defaultVariant)}，请先运行 ylyx init config`);
-	}
-
-	config.mode = normalized;
-	fs.writeJsonSync(configPath, config, { spaces: 2 });
-	log.success(`已写入配置：mode=${normalized}`);
-	log.info(`配置文件：${configPath}`);
-}
-
-/**
- * init config：初始化配置目录与示例文件
- * - 写入/补齐 .ylyxrc.json（mode/publicDir/configDir）
- * - 生成 config/default-dev.js 与 config/default-prod.js（基于现有 public/default.js 派生）
- * - 若 public/default.js 不存在，则按当前 mode 生成一份
- */
-function initConfig(options = {}) {
-	const force = !!options.force;
-	const configPath = getConfigPath();
-	const existing = readConfigSafe(configPath);
-
-	const next = {
-		...existing,
-		mode: existing.mode || 'dev',
-		publicDir: existing.publicDir || './public',
-		configDir: existing.configDir || './config',
-		// npm scripts：默认对 dev 与 build:prod 增加前置脚本 pre<name>
-		// 可在 .ylyxrc.json 中改成你的项目脚本名（例如 dev=serve，prod=build）
-		preScripts: existing.preScripts || { dev: 'dev', prod: 'build:prod' },
-	};
-
-	if (options.mode) {
-		const normalized = normalizeMode(options.mode);
-		if (!normalized) throw new Error(`mode 取值只能是 dev 或 prod（当前: ${options.mode}）`);
-		next.mode = normalized;
-	}
-
-	fs.writeJsonSync(configPath, next, { spaces: 2 });
-	log.success(`已初始化配置文件：${path.relative(process.cwd(), configPath)}`);
-
-	const publicDirAbs = path.resolve(process.cwd(), next.publicDir || 'public');
-	const configDirAbs = path.resolve(process.cwd(), next.configDir || 'config');
-	fs.ensureDirSync(publicDirAbs);
-	fs.ensureDirSync(configDirAbs);
-
-	const publicDefault = path.join(publicDirAbs, 'default.js');
-	let baseContent = '';
-	if (fs.existsSync(publicDefault)) {
-		baseContent = fs.readFileSync(publicDefault, 'utf-8');
-	} else {
-		baseContent =
-			"// 自动生成的默认配置（请按项目需要补全）\n" +
-			"let isDev = true;\n" 
-	}
-
-	const devVariantPath = path.join(configDirAbs, 'default-dev.js');
-	const prodVariantPath = path.join(configDirAbs, 'default-prod.js');
-
-	writeIfAllowed(devVariantPath, deriveDefaultVariantContent(baseContent, 'dev'), force);
-	writeIfAllowed(prodVariantPath, deriveDefaultVariantContent(baseContent, 'prod'), force);
-
-	// 如果 public/default.js 不存在，则按当前 mode 生成一份
-	if (!fs.existsSync(publicDefault)) {
-		const src =
-			next.mode === 'prod'
-				? path.join(configDirAbs, 'default-prod.js')
-				: path.join(configDirAbs, 'default-dev.js');
-		if (fs.existsSync(src)) {
-			fs.copyFileSync(src, publicDefault);
-			log.success(
-				`已生成 ${path.relative(process.cwd(), publicDefault)} <- ${path.relative(process.cwd(), src)}`
-			);
-		}
-	} else {
-		log.info(`已存在：${path.relative(process.cwd(), publicDefault)}（未覆盖）`);
-	}
-
-	// 初始化 package.json 的 npm 前置脚本：pre<devScript> / pre<prodScript>
-	const packageJsonPath = path.join(process.cwd(), 'package.json');
-	if (!fs.existsSync(packageJsonPath)) {
-		log.warn('未找到 package.json，已跳过 npm 前置脚本初始化');
-		return;
-	}
-
-	let pkg = {};
-	try {
-		pkg = fs.readJsonSync(packageJsonPath);
-	} catch (e) {
-		throw new Error(`读取 package.json 失败，请确认它是有效 JSON：${e.message}`);
-	}
-
-	pkg.scripts = pkg.scripts || {};
-	const devTarget = (next.preScripts && next.preScripts.dev) || 'dev';
-	const prodTarget = (next.preScripts && next.preScripts.prod) || 'build:prod';
-
-	const desired = [
-		{ name: `pre${devTarget}`, value: 'ylyx config dev' },
-		{ name: `post${prodTarget}`, value: 'ylyx config prod && ylyx deploy' },
-	];
-
-	for (const item of desired) {
-		if (pkg.scripts[item.name] && !force) {
-			log.warn(`package.json scripts 已存在，跳过：${item.name}`);
-			continue;
-		}
-		pkg.scripts[item.name] = item.value;
-		log.success(`已写入 package.json scripts：${item.name}`);
-	}
-
-	fs.writeJsonSync(packageJsonPath, pkg, { spaces: 2 });
-	log.info(`package.json：${path.relative(process.cwd(), packageJsonPath)}`);
-}
-
-module.exports = {
-	setMode,
-	normalizeMode,
-	initConfig,
-};
+const path = require('path');
+const fs = require('fs-extra');
+const inquirer = require('inquirer');
+const { log } = require('./utils');
+const {
+	getConfigPath,
+	readConfigSafe,
+	assertConfigMigrated,
+	findEnvironment,
+	buildMigratedConfig,
+	CONFIG_VERSION,
+} = require('./environments');
+
+function writeIfAllowed(filePath, content, force) {
+	if (fs.existsSync(filePath) && !force) {
+		log.warn(`已存在，跳过：${path.relative(process.cwd(), filePath)}`);
+		return false;
+	}
+	fs.ensureDirSync(path.dirname(filePath));
+	fs.writeFileSync(filePath, content, 'utf-8');
+	log.success(`已写入：${path.relative(process.cwd(), filePath)}`);
+	return true;
+}
+
+/**
+ * 生成 config/default-<envName>.js 的内容：
+ * 注入 `let env = '<envName>';` 与 `let isDev = (envName === 'dev');`（向后兼容旧代码）
+ */
+function deriveDefaultVariantContent(baseContent, envName) {
+	const isDev = envName === 'dev';
+
+	let content = baseContent || '';
+	const envRe = /let\s+env\s*=\s*[^;]*;/;
+	const devRe = /let\s+isDev\s*=\s*[^;]*;/;
+
+	if (envRe.test(content)) {
+		content = content.replace(envRe, `let env = '${envName}';`);
+	} else {
+		content = `let env = '${envName}';\n` + content;
+	}
+
+	if (devRe.test(content)) {
+		content = content.replace(devRe, `let isDev = ${isDev};`);
+	} else {
+		content = `let isDev = ${isDev};\n` + content;
+	}
+
+	// isProd 已废弃：prod 使用 isDev=false 表示
+	// 删除历史遗留的 isProd 变量与输出字段（若存在）
+	content = content.replace(/let\s+isProd\s*=\s*[^;]*;\s*\r?\n?/g, '');
+	content = content.replace(/window\.config\.IS_PROD\s*=\s*[^;]*;\s*\r?\n?/g, '');
+	content = content.replace(/^\s*IS_PROD\s*:\s*[^,}]*,?\s*\r?\n?/gm, '');
+
+	return content;
+}
+
+function copyFile({ src, dest }) {
+	fs.ensureDirSync(path.dirname(dest));
+	if (!fs.existsSync(src)) {
+		log.warn(`未找到源文件：${path.relative(process.cwd(), src)}`);
+		return { ok: false, method: 'missing' };
+	}
+
+	if (fs.existsSync(dest)) {
+		fs.removeSync(dest);
+	}
+
+	try {
+		fs.copyFileSync(src, dest);
+		return { ok: true };
+	} catch (e) {
+		log.error(`写入失败：${e.message}`);
+		return { ok: false };
+	}
+}
+
+/**
+ * 切换到指定环境：把 config/default-<name>.js 复制到该环境的目标位置
+ * - isLocal 环境 -> publicDir/default.js
+ * - 其他环境 -> <环境 buildDir>/default.js
+ */
+function setEnv(name) {
+	const configPath = getConfigPath();
+	const configExists = fs.existsSync(configPath);
+	const config = readConfigSafe(configPath);
+	assertConfigMigrated(config, configExists);
+
+	const env = findEnvironment(config, name);
+
+	const configDir = path.resolve(process.cwd(), config.configDir || 'config');
+	const publicDir = path.resolve(process.cwd(), config.publicDir || 'public');
+	const defaultVariant = path.join(configDir, `default-${env.name}.js`);
+
+	// ponytail: 信任本地 .ylyxrc.json 的路径配置，不做路径穿越校验（本地 CLI 工具，配置由用户自行维护）
+	let targetDir;
+	if (env.isLocal) {
+		targetDir = publicDir;
+	} else {
+		if (!env.buildDir) {
+			throw new Error(`环境 "${env.name}" 未配置 buildDir，无法确定切换目标目录`);
+		}
+		targetDir = path.resolve(process.cwd(), env.buildDir);
+	}
+	const defaultFile = path.join(targetDir, 'default.js');
+
+	const result = copyFile({ src: defaultVariant, dest: defaultFile });
+
+	if (result.ok) {
+		log.success(
+			`已切换 ${path.relative(process.cwd(), defaultFile)} <- ${path.relative(
+				process.cwd(),
+				defaultVariant
+			)}`
+		);
+	} else if (!result.ok && !fs.existsSync(defaultVariant)) {
+		log.warn(`未找到 ${path.relative(process.cwd(), defaultVariant)}，请先运行 ylyx init config`);
+	}
+
+	config.currentEnv = env.name;
+	fs.writeJsonSync(configPath, config, { spaces: 2 });
+	log.success(`已写入配置：currentEnv=${env.name}`);
+	log.info(`配置文件：${configPath}`);
+}
+
+/**
+ * 交互式选择环境并切换：按 isLocal 分组展示，当前环境标注"（当前）"，
+ * 默认光标停在当前环境，选定后复用 setEnv 完成实际切换
+ */
+async function chooseEnvInteractive() {
+	const configPath = getConfigPath();
+	const configExists = fs.existsSync(configPath);
+	const config = readConfigSafe(configPath);
+	assertConfigMigrated(config, configExists);
+
+	const environments = config.environments || [];
+	if (environments.length === 0) {
+		throw new Error('环境列表为空，请先在 .ylyxrc.json 配置 environments 或运行 ylyx init config');
+	}
+
+	const currentEnv = config.currentEnv;
+	const toChoice = (e) => ({
+		name: e.name === currentEnv ? `${e.name}（当前）` : e.name,
+		value: e.name,
+	});
+
+	const localEnvs = environments.filter((e) => e.isLocal);
+	const remoteEnvs = environments.filter((e) => !e.isLocal);
+
+	const choices = [];
+	if (localEnvs.length > 0) {
+		choices.push(new inquirer.Separator('-- 本地环境 --'));
+		choices.push(...localEnvs.map(toChoice));
+	}
+	if (remoteEnvs.length > 0) {
+		choices.push(new inquirer.Separator('-- 部署环境 --'));
+		choices.push(...remoteEnvs.map(toChoice));
+	}
+
+	const ans = await inquirer.prompt([
+		{
+			type: 'list',
+			name: 'env',
+			message: '请选择要切换到的环境',
+			choices,
+			default: currentEnv,
+		},
+	]);
+
+	setEnv(ans.env);
+}
+
+/**
+ * 迁移旧版 .ylyxrc.json（mode/deploy/顶层 buildDir/preScripts）到 environments 结构
+ */
+function migrateConfig() {
+	const configPath = getConfigPath();
+	const existing = readConfigSafe(configPath);
+
+	if (typeof existing.configVersion === 'number' && existing.configVersion >= CONFIG_VERSION) {
+		log.info(`当前配置已是最新版本（configVersion=${existing.configVersion}），无需迁移`);
+		return;
+	}
+
+	const migrated = buildMigratedConfig(existing);
+	fs.writeJsonSync(configPath, migrated, { spaces: 2 });
+
+	const names = migrated.environments.map((e) => e.name).join(', ');
+	log.success(`迁移完成，已生成环境：${names}`);
+	log.warn('请检查 test/prod 环境的 host/username/password 等敏感信息是否正确');
+	log.info('请运行 `ylyx init config --force` 以补写 package.json 的 pre/post 钩子（迁移命令不会修改 package.json）');
+}
+
+/**
+ * init config：初始化配置目录与示例文件
+ * - 写入/补齐 .ylyxrc.json（configVersion/environments/currentEnv/publicDir/configDir）
+ * - 为每个环境生成 config/default-<name>.js（基于现有 public/default.js 派生）
+ * - 若 public/default.js 不存在，则按 currentEnv 生成一份
+ * - 按每个环境的 npmScript 写 package.json 的 pre/post 钩子
+ */
+function initConfig(options = {}) {
+	const force = !!options.force;
+	const configPath = getConfigPath();
+	const existing = readConfigSafe(configPath);
+
+	const hasEnvironments = Array.isArray(existing.environments) && existing.environments.length > 0;
+	if (Array.isArray(existing.environments) && existing.environments.length === 0) {
+		log.info('检测到 .ylyxrc.json 中 environments 为空数组，已使用默认环境（dev/prod）填充');
+	}
+	const environments = hasEnvironments
+		? existing.environments
+		: [
+				{ name: 'dev', isLocal: true, npmScript: 'dev', deployMethod: 'none' },
+				{
+					name: 'prod',
+					npmScript: 'build:prod',
+					deployMethod: 'zip',
+					buildDir: 'dist/app',
+					zipOutDir: './.ylyx-deploy',
+				},
+		  ];
+
+	let currentEnv = existing.currentEnv || 'dev';
+	if (options.env) {
+		const exists = environments.some((e) => e.name === options.env);
+		if (!exists) {
+			throw new Error(
+				`环境不存在: ${options.env}\n可用环境: ${environments.map((e) => e.name).join(', ')}`
+			);
+		}
+		currentEnv = options.env;
+	}
+
+	const next = {
+		...existing,
+		configVersion: CONFIG_VERSION,
+		publicDir: existing.publicDir || './public',
+		configDir: existing.configDir || './config',
+		currentEnv,
+		environments,
+	};
+
+	fs.writeJsonSync(configPath, next, { spaces: 2 });
+	log.success(`已初始化配置文件：${path.relative(process.cwd(), configPath)}`);
+
+	const publicDirAbs = path.resolve(process.cwd(), next.publicDir);
+	const configDirAbs = path.resolve(process.cwd(), next.configDir);
+	fs.ensureDirSync(publicDirAbs);
+	fs.ensureDirSync(configDirAbs);
+
+	const publicDefault = path.join(publicDirAbs, 'default.js');
+	let baseContent = '';
+	if (fs.existsSync(publicDefault)) {
+		baseContent = fs.readFileSync(publicDefault, 'utf-8');
+	} else {
+		baseContent = '// 自动生成的默认配置（请按项目需要补全）\n';
+	}
+
+	for (const env of next.environments) {
+		const variantPath = path.join(configDirAbs, `default-${env.name}.js`);
+		writeIfAllowed(variantPath, deriveDefaultVariantContent(baseContent, env.name), force);
+	}
+
+	if (!fs.existsSync(publicDefault)) {
+		const src = path.join(configDirAbs, `default-${next.currentEnv}.js`);
+		if (fs.existsSync(src)) {
+			fs.copyFileSync(src, publicDefault);
+			log.success(
+				`已生成 ${path.relative(process.cwd(), publicDefault)} <- ${path.relative(process.cwd(), src)}`
+			);
+		}
+	} else {
+		log.info(`已存在：${path.relative(process.cwd(), publicDefault)}（未覆盖）`);
+	}
+
+	const packageJsonPath = path.join(process.cwd(), 'package.json');
+	if (!fs.existsSync(packageJsonPath)) {
+		log.warn('未找到 package.json，已跳过 npm 钩子初始化');
+		return;
+	}
+
+	let pkg = {};
+	try {
+		pkg = fs.readJsonSync(packageJsonPath);
+	} catch (e) {
+		throw new Error(`读取 package.json 失败，请确认它是有效 JSON：${e.message}`);
+	}
+
+	pkg.scripts = pkg.scripts || {};
+
+	for (const env of next.environments) {
+		const npmScript = env.npmScript;
+		if (!npmScript) continue;
+
+		let hookName;
+		let hookValue;
+		if (env.isLocal) {
+			hookName = `pre${npmScript}`;
+			hookValue = `ylyx config ${env.name}`;
+		} else {
+			hookName = `post${npmScript}`;
+			hookValue =
+				env.deployMethod && env.deployMethod !== 'none'
+					? `ylyx config ${env.name} && ylyx deploy --env ${env.name}`
+					: `ylyx config ${env.name}`;
+		}
+
+		if (pkg.scripts[hookName] && !force) {
+			log.warn(`package.json scripts 已存在，跳过：${hookName}`);
+			continue;
+		}
+		pkg.scripts[hookName] = hookValue;
+		log.success(`已写入 package.json scripts：${hookName}`);
+	}
+
+	fs.writeJsonSync(packageJsonPath, pkg, { spaces: 2 });
+	log.info(`package.json：${path.relative(process.cwd(), packageJsonPath)}`);
+}
+
+module.exports = {
+	setEnv,
+	initConfig,
+	migrateConfig,
+	chooseEnvInteractive,
+	deriveDefaultVariantContent,
+};