ylyx-cli 1.0.15 → 1.1.0

package/bin/ylyx.js

@@ -4,7 +4,7 @@ const { program } = require('commander');
 const { generate } = require('../lib/index');
 const { listTemplates, getTemplateInfo, addTemplate } = require('../lib/template');
 const { interactiveCreate } = require('../lib/interactive');
-const { setMode, initConfig } = require('../lib/config');
+const { setEnv, initConfig, migrateConfig, chooseEnvInteractive } = require('../lib/config');
 const { deploy } = require('../lib/deploy');
 const pkg = require('../package.json');
 
@@ -18,21 +18,15 @@ program
 	});
 
 program
-	.command('config [mode]')
-	.description('写入/更新 .ylyxrc.json 配置（目前支持：mode=dev|prod）')
-	.option('-m, --mode <mode>', '运行模式：dev | prod')
-	.option('--symlink', '切换 default.js 时强制使用 symlink（dev 默认）')
-	.option('--copy', '切换 default.js 时强制使用 copy')
-	.action((modeArg, options) => {
+	.command('config [name]')
+	.description('写入/更新 .ylyxrc.json 配置（切换到指定环境；不传参数时交互式选择）')
+	.action(async (name) => {
 		try {
-			const mode = options.mode || modeArg;
-			if (!mode) {
-				console.error('❌ 缺少参数：mode（dev|prod）');
-				console.error('示例：ylyx config --mode dev   或   ylyx config dev');
-				process.exit(1);
+			if (name) {
+				setEnv(name);
+			} else {
+				await chooseEnvInteractive();
 			}
-			const method = options.copy ? 'copy' : options.symlink ? 'symlink' : undefined;
-			setMode(mode, { method });
 		} catch (error) {
 			console.error('❌ 配置失败:', error.message);
 			process.exit(1);
@@ -44,7 +38,7 @@ program
 	.description('初始化相关资源')
 	.command('config')
 	.description('初始化配置（生成 .ylyxrc.json / config/default-*.js 等）')
-	.option('-m, --mode <mode>', '初始 mode：dev | prod（默认 dev）')
+	.option('-e, --env <name>', '初始 currentEnv（默认 dev）')
 	.option('-f, --force', '强制覆盖已存在文件')
 	.action((options) => {
 		try {
@@ -55,6 +49,20 @@ program
 		}
 	});
 
+program
+	.command('migrate')
+	.description('迁移相关资源')
+	.command('config')
+	.description('迁移旧版 .ylyxrc.json（mode/deploy）到 environments 结构')
+	.action(() => {
+		try {
+			migrateConfig();
+		} catch (error) {
+			console.error('❌ 迁移失败:', error.message);
+			process.exit(1);
+		}
+	});
+
 program
 	.command('generate <template-name>')
 	.alias('g')
@@ -113,15 +121,15 @@ program
 
 program
 	.command('deploy')
-	.description('部署：递归上传本地目录到远端目录（读取当前项目 .ylyxrc.json 的 deploy 字段）')
-	.option('--env <env>', '部署环境：test|prod（不传会交互选择）')
-	.option('--build-dir <name>', '压缩文件名使用的 buildDir（默认取 .ylyxrc.json 的 buildDir）')
-	.option('--host <host>', '覆盖 deploy.host')
-	.option('--port <port>', '覆盖 deploy.port')
-	.option('-u, --username <username>', '覆盖 deploy.username')
-	.option('-p, --password <password>', '覆盖 deploy.password')
-	.option('--local <path>', '覆盖 deploy.localDir（默认 ./EXTERNAL_DIGIC）')
-	.option('--remote <path>', '覆盖 deploy.remoteDir')
+	.description('部署：递归上传本地目录到远端目录（读取当前项目 .ylyxrc.json 的 environments 数组）')
+	.option('--env <name>', '部署环境名称（不传会交互选择）')
+	.option('--build-dir <name>', '压缩文件名使用的 buildDir（默认取该环境的 buildDir）')
+	.option('--host <host>', '覆盖该环境的 host')
+	.option('--port <port>', '覆盖该环境的 port')
+	.option('-u, --username <username>', '覆盖该环境的 username')
+	.option('-p, --password <password>', '覆盖该环境的 password')
+	.option('--local <path>', '覆盖该环境的 localDir（默认取该环境的 buildDir）')
+	.option('--remote <path>', '覆盖该环境的 remoteDir')
 	.option('--zip', '部署后把 localDir 压缩成 zip（参考 compressing.zip.compressDir）')
 	.option('--zip-out <path>', 'zip 输出目录（默认：./.ylyx-deploy）')
 	.action(async (options) => {

package/lib/config.js

@@ -1,77 +1,15 @@
 const path = require('path');
 const fs = require('fs-extra');
+const inquirer = require('inquirer');
 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}`);
-	}
-}
+const {
+	getConfigPath,
+	readConfigSafe,
+	assertConfigMigrated,
+	findEnvironment,
+	buildMigratedConfig,
+	CONFIG_VERSION,
+} = require('./environments');
 
 function writeIfAllowed(filePath, content, force) {
 	if (fs.existsSync(filePath) && !force) {
@@ -84,12 +22,23 @@ function writeIfAllowed(filePath, content, force) {
 	return true;
 }
 
-function deriveDefaultVariantContent(baseContent, mode) {
-	const isDev = mode === 'dev';
+/**
+ * 生成 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 {
@@ -100,127 +49,201 @@ function deriveDefaultVariantContent(baseContent, mode) {
 	// 删除历史遗留的 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 ensureSymlinkOrCopy({ src, dest, preferSymlink }) {
+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);
 	}
 
-	if (preferSymlink) {
-		try {
-			// Windows/macOS/Linux 都支持；Windows 可能因为权限失败，失败后回退 copy
-			fs.ensureSymlinkSync(src, dest, 'file');
-			return { ok: true, method: 'symlink' };
-		} catch (e) {
-			log.warn(`创建 symlink 失败，将降级为 copy：${e.message}`);
-		}
-	}
-
 	try {
 		fs.copyFileSync(src, dest);
-		return { ok: true, method: 'copy' };
+		return { ok: true };
 	} catch (e) {
 		log.error(`写入失败：${e.message}`);
-		return { ok: false, method: 'copy_failed' };
+		return { ok: false };
 	}
 }
 
 /**
- * 写入/更新 .ylyxrc.json 的 mode 字段（dev|prod），并与已有配置合并
+ * 切换到指定环境：把 config/default-<name>.js 复制到该环境的目标位置
+ * - isLocal 环境 -> publicDir/default.js
+ * - 其他环境 -> <环境 buildDir>/default.js
  */
-function setMode(mode, options = {}) {
-	const normalized = normalizeMode(mode);
-	if (!normalized) {
-		throw new Error(`mode 取值只能是 dev 或 prod（当前: ${mode}）`);
-	}
-
+function setEnv(name) {
 	const configPath = getConfigPath();
+	const configExists = fs.existsSync(configPath);
 	const config = readConfigSafe(configPath);
+	assertConfigMigrated(config, configExists);
+
+	const env = findEnvironment(config, name);
 
-	// 可选：如果当前项目目录存在 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');
-
-	// 默认策略：dev 使用 symlink（实时同步），prod 使用 copy（稳定）
-	const method = options.method || (normalized === 'dev' ? 'symlink' : 'copy');
-	const result = ensureSymlinkOrCopy({
-		src: defaultVariant,
-		dest: defaultFile,
-		preferSymlink: method === 'symlink',
-	});
+	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) {
-		const label = result.method === 'symlink' ? 'symlink' : 'copy';
 		log.success(
-			`已(${label})切换 ${path.relative(process.cwd(), defaultFile)} <- ${path.relative(
+			`已切换 ${path.relative(process.cwd(), defaultFile)} <- ${path.relative(
 				process.cwd(),
 				defaultVariant
 			)}`
 		);
-		if (normalized === 'dev' && result.method === 'symlink') {
-			log.info('dev 模式实时同步已启用：修改 default-dev.js 会立即反映到 default.js');
-		}
-	} else if (result.method === 'missing') {
+	} else if (!result.ok && !fs.existsSync(defaultVariant)) {
 		log.warn(`未找到 ${path.relative(process.cwd(), defaultVariant)}，请先运行 ylyx init config`);
 	}
 
-	config.mode = normalized;
+	config.currentEnv = env.name;
 	fs.writeJsonSync(configPath, config, { spaces: 2 });
-	log.success(`已写入配置：mode=${normalized}`);
+	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（mode/publicDir/configDir）
- * - 生成 config/default-dev.js 与 config/default-prod.js（基于现有 public/default.js 派生）
- * - 若 public/default.js 不存在，则按当前 mode 生成一份
+ * - 写入/补齐 .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,
-		mode: existing.mode || 'dev',
+		configVersion: CONFIG_VERSION,
 		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' },
+		currentEnv,
+		environments,
 	};
 
-	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');
+	const publicDirAbs = path.resolve(process.cwd(), next.publicDir);
+	const configDirAbs = path.resolve(process.cwd(), next.configDir);
 	fs.ensureDirSync(publicDirAbs);
 	fs.ensureDirSync(configDirAbs);
 
@@ -229,23 +252,16 @@ function initConfig(options = {}) {
 	if (fs.existsSync(publicDefault)) {
 		baseContent = fs.readFileSync(publicDefault, 'utf-8');
 	} else {
-		baseContent =
-			"// 自动生成的默认配置（请按项目需要补全）\n" +
-			"let isDev = true;\n" 
+		baseContent = '// 自动生成的默认配置（请按项目需要补全）\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);
+	for (const env of next.environments) {
+		const variantPath = path.join(configDirAbs, `default-${env.name}.js`);
+		writeIfAllowed(variantPath, deriveDefaultVariantContent(baseContent, env.name), 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');
+		const src = path.join(configDirAbs, `default-${next.currentEnv}.js`);
 		if (fs.existsSync(src)) {
 			fs.copyFileSync(src, publicDefault);
 			log.success(
@@ -256,10 +272,9 @@ function initConfig(options = {}) {
 		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 前置脚本初始化');
+		log.warn('未找到 package.json，已跳过 npm 钩子初始化');
 		return;
 	}
 
@@ -271,21 +286,30 @@ function initConfig(options = {}) {
 	}
 
 	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 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}`;
+		}
 
-	for (const item of desired) {
-		if (pkg.scripts[item.name] && !force) {
-			log.warn(`package.json scripts 已存在，跳过：${item.name}`);
+		if (pkg.scripts[hookName] && !force) {
+			log.warn(`package.json scripts 已存在，跳过：${hookName}`);
 			continue;
 		}
-		pkg.scripts[item.name] = item.value;
-		log.success(`已写入 package.json scripts：${item.name}`);
+		pkg.scripts[hookName] = hookValue;
+		log.success(`已写入 package.json scripts：${hookName}`);
 	}
 
 	fs.writeJsonSync(packageJsonPath, pkg, { spaces: 2 });
@@ -293,7 +317,9 @@ function initConfig(options = {}) {
 }
 
 module.exports = {
-	setMode,
-	normalizeMode,
+	setEnv,
 	initConfig,
+	migrateConfig,
+	chooseEnvInteractive,
+	deriveDefaultVariantContent,
 };