e6956aa001
- React frontend with route-level code splitting - Backend rebranded from Baihu to TaskPool - DB brand migration script and local compatibility
105 lines
2.8 KiB
Go
105 lines
2.8 KiB
Go
package builtininstall
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"runtime"
|
|
|
|
"github.com/engigu/taskpool/cmd/clibase"
|
|
"github.com/engigu/taskpool/internal/utils"
|
|
)
|
|
|
|
func printHelp() {
|
|
clibase.PrintSubCommandUsage("任务池内建依赖安装工具", "taskpool builtininstall", "", nil)
|
|
}
|
|
|
|
// Run 执行内建包安装逻辑
|
|
func Run(args []string) {
|
|
if len(args) > 0 && (args[0] == "-h" || args[0] == "--help") {
|
|
printHelp()
|
|
return
|
|
}
|
|
|
|
fs := flag.NewFlagSet("builtininstall", flag.ExitOnError)
|
|
fs.Usage = printHelp
|
|
|
|
if err := fs.Parse(args); err != nil {
|
|
return
|
|
}
|
|
|
|
fmt.Println(">> [Builtin] 开始为 mise 环境安装内建包...")
|
|
|
|
// 1. 确定内建包路径
|
|
// 优先使用 /www/builtin (Docker 环境),否则尝试相对于二进制文件的当前目录
|
|
builtinPath := "/www/builtin"
|
|
if _, err := os.Stat(builtinPath); os.IsNotExist(err) {
|
|
// 回退到当前目录下的 builtin
|
|
pwd, _ := os.Getwd()
|
|
builtinPath = filepath.Join(pwd, "builtin")
|
|
}
|
|
|
|
if _, err := os.Stat(builtinPath); os.IsNotExist(err) {
|
|
fmt.Printf(">> [Builtin] 错误: 找不到内建包目录: %s\n", builtinPath)
|
|
return
|
|
}
|
|
|
|
// 2. 安装 Node.js 包
|
|
installForLanguage("node", filepath.Join(builtinPath, "nodejs"))
|
|
|
|
// 3. 安装 Python 包
|
|
installForLanguage("python", filepath.Join(builtinPath, "python"))
|
|
|
|
fmt.Println(">> [Builtin] 内建包安装流程完成")
|
|
}
|
|
|
|
func installForLanguage(lang, pkgPath string) {
|
|
if _, err := os.Stat(pkgPath); os.IsNotExist(err) {
|
|
fmt.Printf(">> [Builtin] 警告: %s 的内建包目录不存在: %s\n", lang, pkgPath)
|
|
return
|
|
}
|
|
|
|
versions, err := utils.ListMiseInstalledVersions(lang)
|
|
if err != nil {
|
|
fmt.Printf(">> [Builtin] 错误: 获取 %s 的 mise 版本列表失败: %v\n", lang, err)
|
|
return
|
|
}
|
|
|
|
if len(versions) == 0 {
|
|
fmt.Printf(">> [Builtin] 未发现已安装的 %s 版本,跳过\n", lang)
|
|
return
|
|
}
|
|
|
|
for _, v := range versions {
|
|
fmt.Printf(">> [Builtin] 正在为 %s@%s 安装内建包...\n", lang, v)
|
|
|
|
var subCmdArgs []string
|
|
if lang == "node" {
|
|
// 使用 npm i -g 进行全局安装
|
|
subCmdArgs = []string{"npm", "i", "-g", pkgPath}
|
|
} else {
|
|
// python 改为标准安装 (非 -e),避免 Docker 内软链接可能导致的路径丢失问题
|
|
subCmdArgs = []string{"pip", "install", "--force-reinstall", pkgPath}
|
|
}
|
|
|
|
// 构建参数列表: [mise, exec, lang@v, --, cmd...]
|
|
fullArgs := utils.BuildMiseCommandArgsSimple(subCmdArgs, lang, v)
|
|
|
|
var cmd *exec.Cmd
|
|
if runtime.GOOS == "windows" {
|
|
cmd = exec.Command("cmd", append([]string{"/c"}, fullArgs...)...)
|
|
} else {
|
|
cmd = exec.Command(fullArgs[0], fullArgs[1:]...)
|
|
}
|
|
|
|
out, err := cmd.CombinedOutput()
|
|
if err != nil {
|
|
fmt.Printf(">> [Builtin] 错误: 为 %s@%s 安装失败: %v\n输出: %s\n", lang, v, err, string(out))
|
|
} else {
|
|
fmt.Printf(">> [Builtin] 为 %s@%s 安装成功\n", lang, v)
|
|
}
|
|
}
|
|
}
|