feat: add deps envs manager
This commit is contained in:
engigu
@@ -0,0 +1,251 @@
|
||||
package deps_env
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"encoding/json"
|
||||
"os/exec"
|
||||
"strings"
|
||||
|
||||
"baihu/internal/logger"
|
||||
)
|
||||
|
||||
// CondaManager Conda 运行时管理器
|
||||
type CondaManager struct {
|
||||
condaPath string
|
||||
}
|
||||
|
||||
// NewCondaManager 创建 Conda 管理器
|
||||
func NewCondaManager() *CondaManager {
|
||||
return &CondaManager{}
|
||||
}
|
||||
|
||||
// GetType 获取运行时类型
|
||||
func (cm *CondaManager) GetType() string {
|
||||
return "conda"
|
||||
}
|
||||
|
||||
// IsAvailable 检查 Conda 是否可用
|
||||
func (cm *CondaManager) IsAvailable() bool {
|
||||
path, err := cm.findCondaPath()
|
||||
if err != nil {
|
||||
return false
|
||||
}
|
||||
cm.condaPath = path
|
||||
return true
|
||||
}
|
||||
|
||||
// findCondaPath 查找 conda 可执行文件路径
|
||||
func (cm *CondaManager) findCondaPath() (string, error) {
|
||||
// 尝试常见的 conda 路径
|
||||
paths := []string{"conda", "micromamba", "/opt/conda/bin/conda", "/root/miniconda3/bin/conda", "/root/anaconda3/bin/conda"}
|
||||
for _, p := range paths {
|
||||
if path, err := exec.LookPath(p); err == nil {
|
||||
return path, nil
|
||||
}
|
||||
}
|
||||
return "", exec.ErrNotFound
|
||||
}
|
||||
|
||||
// getCondaPath 获取 conda 路径
|
||||
func (cm *CondaManager) getCondaPath() string {
|
||||
if cm.condaPath == "" {
|
||||
cm.findCondaPath()
|
||||
}
|
||||
return cm.condaPath
|
||||
}
|
||||
|
||||
// condaEnvJSON conda env list --json 的输出结构
|
||||
type condaEnvJSON struct {
|
||||
Envs []string `json:"envs"`
|
||||
}
|
||||
|
||||
// ListEnvs 列出所有 Conda 环境
|
||||
func (cm *CondaManager) ListEnvs() ([]RuntimeEnv, error) {
|
||||
condaPath := cm.getCondaPath()
|
||||
if condaPath == "" {
|
||||
return nil, exec.ErrNotFound
|
||||
}
|
||||
|
||||
cmd := exec.Command(condaPath, "env", "list", "--json")
|
||||
output, err := cmd.Output()
|
||||
if err != nil {
|
||||
logger.Errorf("Failed to list conda envs: %v", err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var envJSON condaEnvJSON
|
||||
if err := json.Unmarshal(output, &envJSON); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var envs []RuntimeEnv
|
||||
for _, envPath := range envJSON.Envs {
|
||||
name := extractEnvName(envPath)
|
||||
// 过滤以 . 开头的环境
|
||||
if strings.HasPrefix(name, ".") {
|
||||
continue
|
||||
}
|
||||
envs = append(envs, RuntimeEnv{
|
||||
Name: name,
|
||||
Path: envPath,
|
||||
Active: false,
|
||||
})
|
||||
}
|
||||
|
||||
return envs, nil
|
||||
}
|
||||
|
||||
// extractEnvName 从路径中提取环境名称
|
||||
func extractEnvName(envPath string) string {
|
||||
parts := strings.Split(envPath, "/")
|
||||
if len(parts) > 0 {
|
||||
name := parts[len(parts)-1]
|
||||
// 如果是 base 环境，路径可能是 /opt/conda 这样的
|
||||
if name == "conda" || name == "miniconda3" || name == "anaconda3" {
|
||||
return "base"
|
||||
}
|
||||
return name
|
||||
}
|
||||
return envPath
|
||||
}
|
||||
|
||||
// CreateEnv 创建 Conda 环境
|
||||
func (cm *CondaManager) CreateEnv(name string, version string) error {
|
||||
condaPath := cm.getCondaPath()
|
||||
if condaPath == "" {
|
||||
return exec.ErrNotFound
|
||||
}
|
||||
|
||||
args := []string{"create", "-n", name, "-y"}
|
||||
if version != "" {
|
||||
args = append(args, "python="+version)
|
||||
}
|
||||
|
||||
cmd := exec.Command(condaPath, args...)
|
||||
output, err := cmd.CombinedOutput()
|
||||
if err != nil {
|
||||
logger.Errorf("Failed to create conda env: %v, output: %s", err, string(output))
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// DeleteEnv 删除 Conda 环境
|
||||
func (cm *CondaManager) DeleteEnv(name string) error {
|
||||
condaPath := cm.getCondaPath()
|
||||
if condaPath == "" {
|
||||
return exec.ErrNotFound
|
||||
}
|
||||
|
||||
if name == "base" {
|
||||
return nil // 不允许删除 base 环境
|
||||
}
|
||||
|
||||
cmd := exec.Command(condaPath, "env", "remove", "-n", name, "-y")
|
||||
output, err := cmd.CombinedOutput()
|
||||
if err != nil {
|
||||
logger.Errorf("Failed to delete conda env: %v, output: %s", err, string(output))
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// ListPackages 列出环境中的包
|
||||
func (cm *CondaManager) ListPackages(envName string) ([]RuntimePackage, error) {
|
||||
condaPath := cm.getCondaPath()
|
||||
if condaPath == "" {
|
||||
return nil, exec.ErrNotFound
|
||||
}
|
||||
|
||||
args := []string{"list"}
|
||||
if envName != "" && envName != "base" {
|
||||
args = append(args, "-n", envName)
|
||||
}
|
||||
|
||||
cmd := exec.Command(condaPath, args...)
|
||||
output, err := cmd.Output()
|
||||
if err != nil {
|
||||
logger.Errorf("Failed to list packages: %v", err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return parseCondaList(string(output)), nil
|
||||
}
|
||||
|
||||
// parseCondaList 解析 conda list 输出
|
||||
func parseCondaList(output string) []RuntimePackage {
|
||||
var packages []RuntimePackage
|
||||
scanner := bufio.NewScanner(strings.NewReader(output))
|
||||
|
||||
for scanner.Scan() {
|
||||
line := scanner.Text()
|
||||
// 跳过注释和空行
|
||||
if strings.HasPrefix(line, "#") || strings.TrimSpace(line) == "" {
|
||||
continue
|
||||
}
|
||||
|
||||
fields := strings.Fields(line)
|
||||
if len(fields) >= 2 {
|
||||
name := fields[0]
|
||||
|
||||
pkg := RuntimePackage{
|
||||
Name: name,
|
||||
Version: fields[1],
|
||||
}
|
||||
if len(fields) >= 4 {
|
||||
pkg.Channel = fields[3]
|
||||
}
|
||||
packages = append(packages, pkg)
|
||||
}
|
||||
}
|
||||
|
||||
return packages
|
||||
}
|
||||
|
||||
// InstallPackage 安装包
|
||||
func (cm *CondaManager) InstallPackage(envName string, packageName string) error {
|
||||
condaPath := cm.getCondaPath()
|
||||
if condaPath == "" {
|
||||
return exec.ErrNotFound
|
||||
}
|
||||
|
||||
args := []string{"install", "-y"}
|
||||
if envName != "" && envName != "base" {
|
||||
args = append(args, "-n", envName)
|
||||
}
|
||||
args = append(args, packageName)
|
||||
|
||||
cmd := exec.Command(condaPath, args...)
|
||||
output, err := cmd.CombinedOutput()
|
||||
if err != nil {
|
||||
logger.Errorf("Failed to install package: %v, output: %s", err, string(output))
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// UninstallPackage 卸载包
|
||||
func (cm *CondaManager) UninstallPackage(envName string, packageName string) error {
|
||||
condaPath := cm.getCondaPath()
|
||||
if condaPath == "" {
|
||||
return exec.ErrNotFound
|
||||
}
|
||||
|
||||
args := []string{"remove", "-y"}
|
||||
if envName != "" && envName != "base" {
|
||||
args = append(args, "-n", envName)
|
||||
}
|
||||
args = append(args, packageName)
|
||||
|
||||
cmd := exec.Command(condaPath, args...)
|
||||
output, err := cmd.CombinedOutput()
|
||||
if err != nil {
|
||||
logger.Errorf("Failed to uninstall package: %v, output: %s", err, string(output))
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
@@ -0,0 +1,72 @@
|
||||
package deps_env
|
||||
|
||||
// RuntimeEnv 运行时环境信息
|
||||
type RuntimeEnv struct {
|
||||
Name string `json:"name"`
|
||||
Path string `json:"path"`
|
||||
Version string `json:"version"`
|
||||
Active bool `json:"active"`
|
||||
}
|
||||
|
||||
// RuntimePackage 包信息
|
||||
type RuntimePackage struct {
|
||||
Name string `json:"name"`
|
||||
Version string `json:"version"`
|
||||
Channel string `json:"channel,omitempty"`
|
||||
}
|
||||
|
||||
// RuntimeManager 运行时管理器接口
|
||||
type RuntimeManager interface {
|
||||
// GetType 获取运行时类型
|
||||
GetType() string
|
||||
// IsAvailable 检查运行时是否可用
|
||||
IsAvailable() bool
|
||||
// ListEnvs 列出所有环境
|
||||
ListEnvs() ([]RuntimeEnv, error)
|
||||
// CreateEnv 创建环境
|
||||
CreateEnv(name string, version string) error
|
||||
// DeleteEnv 删除环境
|
||||
DeleteEnv(name string) error
|
||||
// ListPackages 列出环境中的包
|
||||
ListPackages(envName string) ([]RuntimePackage, error)
|
||||
// InstallPackage 安装包
|
||||
InstallPackage(envName string, packageName string) error
|
||||
// UninstallPackage 卸载包
|
||||
UninstallPackage(envName string, packageName string) error
|
||||
}
|
||||
|
||||
// RuntimeService 运行时服务
|
||||
type RuntimeService struct {
|
||||
managers map[string]RuntimeManager
|
||||
}
|
||||
|
||||
// NewRuntimeService 创建运行时服务
|
||||
func NewRuntimeService() *RuntimeService {
|
||||
rs := &RuntimeService{
|
||||
managers: make(map[string]RuntimeManager),
|
||||
}
|
||||
// 注册 Conda 管理器
|
||||
rs.RegisterManager(NewCondaManager())
|
||||
return rs
|
||||
}
|
||||
|
||||
// RegisterManager 注册运行时管理器
|
||||
func (rs *RuntimeService) RegisterManager(manager RuntimeManager) {
|
||||
rs.managers[manager.GetType()] = manager
|
||||
}
|
||||
|
||||
// GetManager 获取指定类型的管理器
|
||||
func (rs *RuntimeService) GetManager(runtimeType string) RuntimeManager {
|
||||
return rs.managers[runtimeType]
|
||||
}
|
||||
|
||||
// GetAvailableRuntimes 获取可用的运行时列表
|
||||
func (rs *RuntimeService) GetAvailableRuntimes() []string {
|
||||
var available []string
|
||||
for name, manager := range rs.managers {
|
||||
if manager.IsAvailable() {
|
||||
available = append(available, name)
|
||||
}
|
||||
}
|
||||
return available
|
||||
}
|
||||
Reference in New Issue
Block a user