panel/internal/services/plugin.go

// Package services 插件服务
package services

import (
	"errors"

	"github.com/goravel/framework/facades"

	"github.com/TheTNB/panel/app/models"
	"github.com/TheTNB/panel/internal"
	"github.com/TheTNB/panel/pkg/tools"
	"github.com/TheTNB/panel/types"
)

type PluginImpl struct {
	task internal.Task
}

func NewPluginImpl() *PluginImpl {
	return &PluginImpl{
		task: NewTaskImpl(),
	}
}

// AllInstalled 获取已安装的所有插件
func (r *PluginImpl) AllInstalled() ([]models.Plugin, error) {
	var plugins []models.Plugin
	if err := facades.Orm().Query().Get(&plugins); err != nil {
		return plugins, err
	}

	return plugins, nil
}

// All 获取所有插件
func (r *PluginImpl) All() []types.Plugin {
	var plugins = []types.Plugin{
		types.PluginOpenResty,
		types.PluginMySQL57,
		types.PluginMySQL80,
		types.PluginMySQL84,
		types.PluginPostgreSQL15,
		types.PluginPostgreSQL16,
		types.PluginPHP74,
		types.PluginPHP80,
		types.PluginPHP81,
		types.PluginPHP82,
		types.PluginPHP83,
		types.PluginPHPMyAdmin,
		types.PluginPureFTPd,
		types.PluginRedis,
		types.PluginS3fs,
		types.PluginRsync,
		types.PluginSupervisor,
		types.PluginFail2ban,
		types.PluginFrp,
		types.PluginGitea,
		types.PluginToolBox,
	}

	return plugins
}

// GetBySlug 根据 slug 获取插件
func (r *PluginImpl) GetBySlug(slug string) types.Plugin {
	for _, item := range r.All() {
		if item.Slug == slug {
			return item
		}
	}

	return types.Plugin{}
}

// GetInstalledBySlug 根据 slug 获取已安装的插件
func (r *PluginImpl) GetInstalledBySlug(slug string) models.Plugin {
	var plugin models.Plugin
	if err := facades.Orm().Query().Where("slug", slug).Get(&plugin); err != nil {
		return plugin
	}

	return plugin
}

// Install 安装插件
func (r *PluginImpl) Install(slug string) error {
	plugin := r.GetBySlug(slug)
	installedPlugin := r.GetInstalledBySlug(slug)
	installedPlugins, err := r.AllInstalled()
	if err != nil {
		return err
	}

	if installedPlugin.ID != 0 {
		return errors.New("插件已安装")
	}

	pluginsMap := make(map[string]bool)

	for _, p := range installedPlugins {
		pluginsMap[p.Slug] = true
	}

	for _, require := range plugin.Requires {
		_, requireFound := pluginsMap[require]
		if !requireFound {
			return errors.New("插件 " + slug + " 需要依赖 " + require + " 插件")
		}
	}

	for _, exclude := range plugin.Excludes {
		_, excludeFound := pluginsMap[exclude]
		if excludeFound {
			return errors.New("插件 " + slug + " 不兼容 " + exclude + " 插件")
		}
	}

	var task models.Task
	task.Name = "安装插件 " + plugin.Name
	task.Status = models.TaskStatusWaiting
	task.Shell = plugin.Install + ` >> '/tmp/` + plugin.Slug + `.log' 2>&1`
	task.Log = "/tmp/" + plugin.Slug + ".log"
	if err = facades.Orm().Query().Create(&task); err != nil {
		return errors.New("创建任务失败")
	}

	_ = tools.Remove(task.Log)
	return r.task.Process(task.ID)
}

// Uninstall 卸载插件
func (r *PluginImpl) Uninstall(slug string) error {
	plugin := r.GetBySlug(slug)
	installedPlugin := r.GetInstalledBySlug(slug)
	installedPlugins, err := r.AllInstalled()
	if err != nil {
		return err
	}

	if installedPlugin.ID == 0 {
		return errors.New("插件未安装")
	}

	pluginsMap := make(map[string]bool)

	for _, p := range installedPlugins {
		pluginsMap[p.Slug] = true
	}

	for _, require := range plugin.Requires {
		_, requireFound := pluginsMap[require]
		if !requireFound {
			return errors.New("插件 " + slug + " 需要依赖 " + require + " 插件")
		}
	}

	for _, exclude := range plugin.Excludes {
		_, excludeFound := pluginsMap[exclude]
		if excludeFound {
			return errors.New("插件 " + slug + " 不兼容 " + exclude + " 插件")
		}
	}

	var task models.Task
	task.Name = "卸载插件 " + plugin.Name
	task.Status = models.TaskStatusWaiting
	task.Shell = plugin.Uninstall + " >> /tmp/" + plugin.Slug + ".log 2>&1"
	task.Log = "/tmp/" + plugin.Slug + ".log"
	if err = facades.Orm().Query().Create(&task); err != nil {
		return errors.New("创建任务失败")
	}

	_ = tools.Remove(task.Log)
	return r.task.Process(task.ID)
}

// Update 更新插件
func (r *PluginImpl) Update(slug string) error {
	plugin := r.GetBySlug(slug)
	installedPlugin := r.GetInstalledBySlug(slug)
	installedPlugins, err := r.AllInstalled()
	if err != nil {
		return err
	}

	if installedPlugin.ID == 0 {
		return errors.New("插件未安装")
	}

	pluginsMap := make(map[string]bool)

	for _, p := range installedPlugins {
		pluginsMap[p.Slug] = true
	}

	for _, require := range plugin.Requires {
		_, requireFound := pluginsMap[require]
		if !requireFound {
			return errors.New("插件 " + slug + " 需要依赖 " + require + " 插件")
		}
	}

	for _, exclude := range plugin.Excludes {
		_, excludeFound := pluginsMap[exclude]
		if excludeFound {
			return errors.New("插件 " + slug + " 不兼容 " + exclude + " 插件")
		}
	}

	var task models.Task
	task.Name = "更新插件 " + plugin.Name
	task.Status = models.TaskStatusWaiting
	task.Shell = plugin.Update + " >> /tmp/" + plugin.Slug + ".log 2>&1"
	task.Log = "/tmp/" + plugin.Slug + ".log"
	if err = facades.Orm().Query().Create(&task); err != nil {
		return errors.New("创建任务失败")
	}

	_ = tools.Remove(task.Log)
	return r.task.Process(task.ID)
}