Files
TaskPool/cmd/builtininstall/builtininstall.go
T
admin e6956aa001 Initial commit: TaskPool React panel
- React frontend with route-level code splitting
- Backend rebranded from Baihu to TaskPool
- DB brand migration script and local compatibility
2026-07-26 08:43:52 +08:00

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)
}
}
}