126 lines
4.2 KiB
Go
126 lines
4.2 KiB
Go
package dependencies
|
||
|
||
import (
|
||
"bytes"
|
||
"context"
|
||
"fmt"
|
||
"github.com/huoxue1/qinglong-go/models"
|
||
"github.com/huoxue1/qinglong-go/service/config"
|
||
"github.com/huoxue1/qinglong-go/utils"
|
||
"io"
|
||
"os"
|
||
"strings"
|
||
"time"
|
||
)
|
||
|
||
func AddDep(dep *models.Dependences) {
|
||
if dep.Type == models.NODE {
|
||
addNodeDep(dep)
|
||
} else if dep.Type == models.PYTHON {
|
||
addPythonDep(dep)
|
||
} else {
|
||
addLinuxDep(dep)
|
||
}
|
||
}
|
||
|
||
func DelDep(ids []int) {
|
||
for _, id := range ids {
|
||
dep, err := models.GetDependences(id)
|
||
if err != nil {
|
||
continue
|
||
}
|
||
if dep.Type == models.NODE {
|
||
unInstallDep("pnpm remove "+dep.Name, dep)
|
||
} else if dep.Type == models.PYTHON {
|
||
pip := config.GetKey("PipCmd", "pip")
|
||
unInstallDep(fmt.Sprintf("%s uninstall %s", pip, dep.Name), dep)
|
||
} else {
|
||
unInstallDep("apk uninstall "+dep.Name, dep)
|
||
}
|
||
|
||
}
|
||
}
|
||
|
||
func unInstallDep(command string, dep *models.Dependences) {
|
||
ctx := context.WithValue(context.Background(), "cancel", make(chan int, 1))
|
||
now := time.Now()
|
||
go utils.RunTask(ctx, command, map[string]string{}, func(ctx context.Context) {
|
||
writer := ctx.Value("log").(io.Writer)
|
||
_, _ = writer.Write([]byte(fmt.Sprintf("##开始执行.. %s\n\n", now.Format("2006-01-02 15:04:05"))))
|
||
}, func(ctx context.Context) {
|
||
writer := ctx.Value("log").(io.Writer)
|
||
_, _ = writer.Write([]byte(fmt.Sprintf("\n##执行结束.. %s，耗时%.1f秒\n\n", time.Now().Format("2006-01-02 15:04:05"), time.Now().Sub(now).Seconds())))
|
||
_ = models.DeleteDependences(dep.Id)
|
||
}, os.Stdout)
|
||
}
|
||
|
||
func addNodeDep(dep *models.Dependences) {
|
||
log := ""
|
||
buffer := bytes.NewBufferString(log)
|
||
ctx := context.WithValue(context.Background(), "cancel", make(chan int, 1))
|
||
now := time.Now()
|
||
go utils.RunTask(ctx, fmt.Sprintf("pnpm add %s", dep.Name), map[string]string{}, func(ctx context.Context) {
|
||
dep.Status = 0
|
||
_, _ = models.AddDependences(dep)
|
||
writer := ctx.Value("log").(io.Writer)
|
||
writer.Write([]byte(fmt.Sprintf("##开始执行.. %s\n\n", now.Format("2006-01-02 15:04:05"))))
|
||
}, func(ctx context.Context) {
|
||
writer := ctx.Value("log").(io.Writer)
|
||
writer.Write([]byte(fmt.Sprintf("\n##执行结束.. %s，耗时%.1f秒\n\n", time.Now().Format("2006-01-02 15:04:05"), time.Now().Sub(now).Seconds())))
|
||
dep.Status = 1
|
||
var logs []string
|
||
for _, i2 := range strings.Split(buffer.String(), "\n") {
|
||
logs = append(logs, i2+"\n\n")
|
||
}
|
||
dep.Log = logs
|
||
_ = models.UpdateDependences(dep)
|
||
}, buffer)
|
||
}
|
||
|
||
func addPythonDep(dep *models.Dependences) {
|
||
log := ""
|
||
buffer := bytes.NewBufferString(log)
|
||
ctx := context.WithValue(context.Background(), "cancel", make(chan int, 1))
|
||
now := time.Now()
|
||
pip := config.GetKey("PipCmd", "pip")
|
||
go utils.RunTask(ctx, fmt.Sprintf(pip+" install %s", dep.Name), map[string]string{}, func(ctx context.Context) {
|
||
dep.Status = 0
|
||
_, _ = models.AddDependences(dep)
|
||
writer := ctx.Value("log").(io.Writer)
|
||
writer.Write([]byte(fmt.Sprintf("##开始执行.. %s\n\n", now.Format("2006-01-02 15:04:05"))))
|
||
}, func(ctx context.Context) {
|
||
writer := ctx.Value("log").(io.Writer)
|
||
writer.Write([]byte(fmt.Sprintf("\n##执行结束.. %s，耗时%.1f秒\n\n", time.Now().Format("2006-01-02 15:04:05"), time.Now().Sub(now).Seconds())))
|
||
dep.Status = 1
|
||
var logs []string
|
||
for _, i2 := range strings.Split(buffer.String(), "\n") {
|
||
logs = append(logs, i2+"\n\n")
|
||
}
|
||
dep.Log = logs
|
||
models.UpdateDependences(dep)
|
||
}, buffer)
|
||
}
|
||
|
||
func addLinuxDep(dep *models.Dependences) {
|
||
log := ""
|
||
buffer := bytes.NewBufferString(log)
|
||
ctx := context.WithValue(context.Background(), "cancel", make(chan int, 1))
|
||
now := time.Now()
|
||
go utils.RunTask(ctx, fmt.Sprintf("apk add %s", dep.Name), map[string]string{}, func(ctx context.Context) {
|
||
dep.Status = 0
|
||
_, _ = models.AddDependences(dep)
|
||
writer := ctx.Value("log").(io.Writer)
|
||
writer.Write([]byte(fmt.Sprintf("##开始执行.. %s\n\n", now.Format("2006-01-02 15:04:05"))))
|
||
}, func(ctx context.Context) {
|
||
writer := ctx.Value("log").(io.Writer)
|
||
writer.Write([]byte(fmt.Sprintf("\n##执行结束.. %s，耗时%.1f秒\n\n", time.Now().Format("2006-01-02 15:04:05"), time.Now().Sub(now).Seconds())))
|
||
dep.Status = 1
|
||
var logs []string
|
||
for _, i2 := range strings.Split(buffer.String(), "\n") {
|
||
logs = append(logs, i2+"\n\n")
|
||
}
|
||
dep.Log = logs
|
||
models.UpdateDependences(dep)
|
||
}, buffer)
|
||
}
|