feat: complete mise env base code
This commit is contained in:
engigu
@@ -2,12 +2,10 @@ package services
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"os/exec"
|
||||
"strings"
|
||||
|
||||
"github.com/engigu/baihu-panel/internal/database"
|
||||
"github.com/engigu/baihu-panel/internal/logger"
|
||||
"github.com/engigu/baihu-panel/internal/models"
|
||||
"github.com/engigu/baihu-panel/internal/services/deps"
|
||||
)
|
||||
|
||||
type DependencyService struct{}
|
||||
@@ -17,21 +15,25 @@ func NewDependencyService() *DependencyService {
|
||||
}
|
||||
|
||||
// List 获取依赖列表
|
||||
func (s *DependencyService) List(depType string) ([]models.Dependency, error) {
|
||||
var deps []models.Dependency
|
||||
func (s *DependencyService) List(language, langVersion string) ([]models.Dependency, error) {
|
||||
var results []models.Dependency
|
||||
query := database.DB
|
||||
if depType != "" {
|
||||
query = query.Where("type = ?", depType)
|
||||
if language != "" {
|
||||
query = query.Where("language = ?", language)
|
||||
}
|
||||
err := query.Order("id desc").Find(&deps).Error
|
||||
return deps, err
|
||||
if langVersion != "" {
|
||||
query = query.Where("lang_version = ?", langVersion)
|
||||
}
|
||||
err := query.Order("id desc").Find(&results).Error
|
||||
return results, err
|
||||
}
|
||||
|
||||
// Create 创建依赖记录
|
||||
func (s *DependencyService) Create(dep *models.Dependency) error {
|
||||
// 检查是否已存在
|
||||
// 检查是否已存在（名称、语言、版本 必须完全匹配）
|
||||
var existing models.Dependency
|
||||
if err := database.DB.Where("name = ? AND type = ?", dep.Name, dep.Type).First(&existing).Error; err == nil {
|
||||
err := database.DB.Where("name = ? AND language = ? AND lang_version = ?", dep.Name, dep.Language, dep.LangVersion).First(&existing).Error
|
||||
if err == nil {
|
||||
return errors.New("依赖已存在")
|
||||
}
|
||||
return database.DB.Create(dep).Error
|
||||
@@ -44,134 +46,51 @@ func (s *DependencyService) Delete(id int) error {
|
||||
|
||||
// Install 安装依赖
|
||||
func (s *DependencyService) Install(dep *models.Dependency) error {
|
||||
var cmd *exec.Cmd
|
||||
var packageSpec string
|
||||
|
||||
if dep.Version != "" {
|
||||
if dep.Type == "py" {
|
||||
packageSpec = dep.Name + "==" + dep.Version
|
||||
} else {
|
||||
packageSpec = dep.Name + "@" + dep.Version
|
||||
}
|
||||
} else {
|
||||
packageSpec = dep.Name
|
||||
m := deps.GetManager(dep.Language)
|
||||
if m == nil {
|
||||
return errors.New("不支持的依赖类型: " + dep.Language)
|
||||
}
|
||||
|
||||
switch dep.Type {
|
||||
case "py":
|
||||
cmd = exec.Command("pip", "install", packageSpec)
|
||||
case "node":
|
||||
cmd = exec.Command("npm", "install", "-g", packageSpec)
|
||||
default:
|
||||
return errors.New("不支持的依赖类型")
|
||||
}
|
||||
|
||||
logger.Infof("Installing %s package: %s", dep.Type, packageSpec)
|
||||
output, err := cmd.CombinedOutput()
|
||||
dep.Log = string(output)
|
||||
|
||||
if err != nil {
|
||||
logger.Errorf("Install failed: %v, output: %s", err, string(output))
|
||||
return errors.New("安装失败: " + string(output))
|
||||
}
|
||||
logger.Infof("Install success: %s", packageSpec)
|
||||
|
||||
return nil
|
||||
return m.Install(dep)
|
||||
}
|
||||
|
||||
// Uninstall 卸载依赖
|
||||
func (s *DependencyService) Uninstall(dep *models.Dependency) error {
|
||||
var cmd *exec.Cmd
|
||||
|
||||
switch dep.Type {
|
||||
case "py":
|
||||
cmd = exec.Command("pip", "uninstall", "-y", dep.Name)
|
||||
case "node":
|
||||
cmd = exec.Command("npm", "uninstall", "-g", dep.Name)
|
||||
default:
|
||||
return errors.New("不支持的依赖类型")
|
||||
m := deps.GetManager(dep.Language)
|
||||
if m == nil {
|
||||
return errors.New("不支持的依赖类型: " + dep.Language)
|
||||
}
|
||||
|
||||
logger.Infof("Uninstalling %s package: %s", dep.Type, dep.Name)
|
||||
output, err := cmd.CombinedOutput()
|
||||
if err != nil {
|
||||
logger.Errorf("Uninstall failed: %v, output: %s", err, string(output))
|
||||
return errors.New("卸载失败: " + string(output))
|
||||
}
|
||||
|
||||
return nil
|
||||
return m.Uninstall(dep)
|
||||
}
|
||||
|
||||
// GetInstalledPackages 获取已安装的包列表
|
||||
func (s *DependencyService) GetInstalledPackages(depType string) ([]models.Dependency, error) {
|
||||
var packages []models.Dependency
|
||||
|
||||
switch depType {
|
||||
case "py":
|
||||
return s.getPipPackages()
|
||||
case "node":
|
||||
return s.getNpmPackages()
|
||||
default:
|
||||
return packages, errors.New("不支持的依赖类型")
|
||||
func (s *DependencyService) GetInstalledPackages(language, langVersion string) ([]models.Dependency, error) {
|
||||
m := deps.GetManager(language)
|
||||
if m == nil {
|
||||
return nil, errors.New("不支持的依赖类型: " + language)
|
||||
}
|
||||
return m.GetInstalledPackages(language, langVersion)
|
||||
}
|
||||
|
||||
// getPipPackages 获取 pip 已安装的包
|
||||
func (s *DependencyService) getPipPackages() ([]models.Dependency, error) {
|
||||
cmd := exec.Command("pip", "list", "--format=freeze")
|
||||
output, err := cmd.Output()
|
||||
// GetInstallCommand 获取安装命令
|
||||
func (s *DependencyService) GetInstallCommand(dep *models.Dependency) (string, error) {
|
||||
m := deps.GetManager(dep.Language)
|
||||
if m == nil {
|
||||
return "", errors.New("不支持的依赖类型: " + dep.Language)
|
||||
}
|
||||
return m.GetInstallCommand(dep)
|
||||
}
|
||||
|
||||
// GetReinstallAllCommand 获取全部重装命令
|
||||
func (s *DependencyService) GetReinstallAllCommand(language, langVersion string) (string, error) {
|
||||
m := deps.GetManager(language)
|
||||
if m == nil {
|
||||
return "", errors.New("不支持的依赖类型: " + language)
|
||||
}
|
||||
|
||||
deps_list, err := s.List(language, langVersion)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
return "", err
|
||||
}
|
||||
|
||||
var packages []models.Dependency
|
||||
lines := strings.Split(string(output), "\n")
|
||||
for _, line := range lines {
|
||||
line = strings.TrimSpace(line)
|
||||
if line == "" {
|
||||
continue
|
||||
}
|
||||
parts := strings.SplitN(line, "==", 2)
|
||||
pkg := models.Dependency{
|
||||
Name: parts[0],
|
||||
Type: "py",
|
||||
}
|
||||
if len(parts) > 1 {
|
||||
pkg.Version = parts[1]
|
||||
}
|
||||
packages = append(packages, pkg)
|
||||
}
|
||||
|
||||
return packages, nil
|
||||
}
|
||||
|
||||
// getNpmPackages 获取 npm 全局安装的包
|
||||
func (s *DependencyService) getNpmPackages() ([]models.Dependency, error) {
|
||||
cmd := exec.Command("npm", "list", "-g", "--depth=0", "--json")
|
||||
output, err := cmd.Output()
|
||||
if err != nil {
|
||||
// npm list 在没有包时也会返回错误，忽略
|
||||
}
|
||||
|
||||
var packages []models.Dependency
|
||||
// 简单解析，不用 json 库
|
||||
lines := strings.Split(string(output), "\n")
|
||||
for _, line := range lines {
|
||||
line = strings.TrimSpace(line)
|
||||
if strings.Contains(line, `"version"`) {
|
||||
continue
|
||||
}
|
||||
if strings.HasPrefix(line, `"`) && strings.Contains(line, ":") {
|
||||
// 格式: "package-name": {
|
||||
name := strings.Trim(strings.Split(line, ":")[0], `" `)
|
||||
if name != "" && name != "dependencies" {
|
||||
packages = append(packages, models.Dependency{
|
||||
Name: name,
|
||||
Type: "node",
|
||||
})
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return packages, nil
|
||||
return m.GetReinstallAllCommand(deps_list)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user