qinglong-go/service/dependencies/dependencies.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)
}