159 lines
4.6 KiB
Python
159 lines
4.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
JavaScript NestJS 后端项目初始化模块
|
|
"""
|
|
|
|
import subprocess
|
|
import sys
|
|
import shutil
|
|
|
|
|
|
def run_command_interactive(cmd, cwd, description):
|
|
"""运行命令并支持用户交互"""
|
|
print(f"\n{'=' * 60}")
|
|
print(f"执行: {description}")
|
|
print(f"{'=' * 60}")
|
|
print(f"💡 提示: 此步骤可能需要您进行交互操作")
|
|
print(f"{'=' * 60}\n")
|
|
|
|
try:
|
|
result = subprocess.run(cmd, cwd=cwd, shell=True)
|
|
|
|
print(f"\n{'=' * 60}")
|
|
if result.returncode == 0:
|
|
print(f"✅ {description} - 完成")
|
|
else:
|
|
print(f"❌ {description} - 失败 (错误码: {result.returncode})")
|
|
print(f"{'=' * 60}\n")
|
|
|
|
return result.returncode == 0
|
|
|
|
except Exception as e:
|
|
print(f"\n❌ 执行失败: {str(e)}")
|
|
return False
|
|
|
|
|
|
def run_command_with_progress(cmd, cwd, description):
|
|
"""运行命令并显示实时输出"""
|
|
try:
|
|
print(f"\n{'=' * 60}")
|
|
print(f"执行: {description}")
|
|
print(f"{'=' * 60}")
|
|
|
|
process = subprocess.Popen(
|
|
cmd,
|
|
cwd=cwd,
|
|
shell=True,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.STDOUT,
|
|
bufsize=0
|
|
)
|
|
|
|
for line in process.stdout:
|
|
try:
|
|
decoded_line = line.decode('utf-8', errors='replace')
|
|
except:
|
|
try:
|
|
decoded_line = line.decode('gbk', errors='replace')
|
|
except:
|
|
decoded_line = line.decode('latin-1', errors='replace')
|
|
print(decoded_line, end='')
|
|
|
|
process.wait()
|
|
|
|
if process.returncode == 0:
|
|
print(f"\n{'=' * 60}")
|
|
print(f"✅ {description} - 完成")
|
|
print(f"{'=' * 60}\n")
|
|
return True
|
|
else:
|
|
print(f"\n{'=' * 60}")
|
|
print(f"❌ {description} - 失败 (错误码: {process.returncode})")
|
|
print(f"{'=' * 60}\n")
|
|
return False
|
|
|
|
except Exception as e:
|
|
print(f"\n❌ 执行失败: {str(e)}")
|
|
return False
|
|
|
|
|
|
def init_nestjs_project(backend_dir, project_name):
|
|
"""初始化 NestJS 项目"""
|
|
print("\n🚀 初始化 NestJS 项目...")
|
|
print("\n⚠️ 重要提示:")
|
|
print(" 1. NestJS CLI 会询问使用哪个包管理器，建议选择 npm")
|
|
print(" 2. 创建过程可能需要几分钟，请耐心等待\n")
|
|
|
|
# 删除空目录，因为 NestJS CLI 需要创建目录
|
|
backend_dir.rmdir()
|
|
|
|
# 使用 NestJS CLI 创建项目
|
|
success = run_command_interactive(
|
|
f"npx @nestjs/cli new {backend_dir.name} --skip-git",
|
|
backend_dir.parent,
|
|
"创建 NestJS 项目"
|
|
)
|
|
|
|
if not success:
|
|
# 如果失败，重新创建目录
|
|
backend_dir.mkdir(exist_ok=True)
|
|
print("\n💡 提示: 请确保已安装 Node.js 和 npm")
|
|
print(" 下载地址: https://nodejs.org/")
|
|
sys.exit(1)
|
|
|
|
# 检查目录是否创建成功
|
|
if not backend_dir.exists():
|
|
print("❌ 项目目录创建失败")
|
|
sys.exit(1)
|
|
|
|
# 修改默认端口为 8080 并添加 CORS
|
|
main_ts = backend_dir / "src" / "main.ts"
|
|
if main_ts.exists():
|
|
main_ts.write_text(f'''import {{ NestFactory }} from '@nestjs/core';
|
|
import {{ AppModule }} from './app.module';
|
|
|
|
async function bootstrap() {{
|
|
const app = await NestFactory.create(AppModule);
|
|
|
|
// 启用 CORS
|
|
app.enableCors();
|
|
|
|
// 设置全局前缀
|
|
app.setGlobalPrefix('api');
|
|
|
|
await app.listen(8080);
|
|
console.log('🚀 服务器启动: http://localhost:8080');
|
|
console.log('📍 健康检查: http://localhost:8080/api/health');
|
|
}}
|
|
bootstrap();
|
|
''', encoding='utf-8')
|
|
|
|
# 修改 AppController 添加健康检查
|
|
app_controller = backend_dir / "src" / "app.controller.ts"
|
|
if app_controller.exists():
|
|
app_controller.write_text(f'''import {{ Controller, Get }} from '@nestjs/common';
|
|
import {{ AppService }} from './app.service';
|
|
|
|
@Controller()
|
|
export class AppController {{
|
|
constructor(private readonly appService: AppService) {{}}
|
|
|
|
@Get('health')
|
|
health() {{
|
|
return {{
|
|
status: 'ok',
|
|
message: 'Welcome to {project_name} API',
|
|
}};
|
|
}}
|
|
|
|
@Get('hello')
|
|
getHello(): string {{
|
|
return this.appService.getHello();
|
|
}}
|
|
}}
|
|
''', encoding='utf-8')
|
|
|
|
print("\n✅ NestJS 项目初始化成功")
|
|
print("💡 启动命令: npm run start:dev (开发模式) 或 npm run start (生产模式)")
|
|
print("📚 NestJS 文档: https://docs.nestjs.com/")
|