1Panel/core/app/service/group.go

package service

import (
	"bytes"
	"fmt"
	"net/http"
	"strconv"
	"strings"

	"github.com/1Panel-dev/1Panel/core/app/dto"
	"github.com/1Panel-dev/1Panel/core/app/model"
	"github.com/1Panel-dev/1Panel/core/app/repo"
	"github.com/1Panel-dev/1Panel/core/buserr"
	"github.com/1Panel-dev/1Panel/core/global"
	"github.com/1Panel-dev/1Panel/core/utils/req_helper/proxy_local"
	"github.com/1Panel-dev/1Panel/core/utils/xpack"
	"github.com/jinzhu/copier"
)

type GroupService struct{}

type IGroupService interface {
	List(req dto.OperateByType) ([]dto.GroupInfo, error)
	Create(req dto.GroupCreate) error
	Update(req dto.GroupUpdate) error
	Delete(id uint) error
}

func NewIGroupService() IGroupService {
	return &GroupService{}
}

func (u *GroupService) List(req dto.OperateByType) ([]dto.GroupInfo, error) {
	options := []global.DBOption{
		repo.WithOrderBy("is_default desc"),
		repo.WithOrderBy("created_at desc"),
	}
	if len(req.Type) != 0 {
		options = append(options, repo.WithByType(req.Type))
	}
	var (
		groups []model.Group
		err    error
	)
	groups, err = groupRepo.GetList(options...)
	if err != nil {
		return nil, buserr.New("ErrRecordNotFound")
	}
	var dtoUsers []dto.GroupInfo
	for _, group := range groups {
		var item dto.GroupInfo
		if err := copier.Copy(&item, &group); err != nil {
			return nil, buserr.WithDetail("ErrStructTransform", err.Error(), nil)
		}
		dtoUsers = append(dtoUsers, item)
	}
	return dtoUsers, err
}

func (u *GroupService) Create(req dto.GroupCreate) error {
	group, _ := groupRepo.Get(repo.WithByName(req.Name), repo.WithByType(req.Type))
	if group.ID != 0 {
		return buserr.New("ErrRecordExist")
	}
	if err := copier.Copy(&group, &req); err != nil {
		return buserr.WithDetail("ErrStructTransform", err.Error(), nil)
	}
	if err := groupRepo.Create(&group); err != nil {
		return err
	}
	return nil
}

func (u *GroupService) Delete(id uint) error {
	group, _ := groupRepo.Get(repo.WithByID(id))
	if group.ID == 0 {
		return buserr.New("ErrRecordNotFound")
	}
	if group.Type == "script" {
		list, _ := scriptRepo.GetList()
		if len(list) == 0 {
			return groupRepo.Delete(repo.WithByID(id))
		}
		for _, itemData := range list {
			groupIDs := strings.Split(itemData.Groups, ",")
			for _, idItem := range groupIDs {
				groupID, _ := strconv.Atoi(idItem)
				if uint(groupID) == id {
					return buserr.New("ErrGroupIsInUse")
				}
			}
		}
		return groupRepo.Delete(repo.WithByID(id))
	}
	if group.IsDefault {
		return buserr.New("ErrGroupIsDefault")
	}
	defaultGroup, err := groupRepo.Get(repo.WithByType(group.Type), groupRepo.WithByDefault(true))
	if err != nil {
		return err
	}
	switch group.Type {
	case "host":
		err = hostRepo.UpdateGroup(id, defaultGroup.ID)
	case "script":
		err = hostRepo.UpdateGroup(id, defaultGroup.ID)
	case "command":
		err = commandRepo.UpdateGroup(id, defaultGroup.ID)
	case "node":
		err = xpack.UpdateGroup("node", id, defaultGroup.ID)
	case "website":
		bodyItem := []byte(fmt.Sprintf(`{"Group":%v, "NewGroup":%v}`, id, defaultGroup.ID))
		if _, err := proxy_local.NewLocalClient("/api/v2/websites/group/change", http.MethodPost, bytes.NewReader(bodyItem)); err != nil {
			return err
		}
		if err := xpack.UpdateGroup("node", id, defaultGroup.ID); err != nil {
			return err
		}
	default:
		return buserr.New("ErrNotSupportType")
	}
	if err != nil {
		return err
	}
	return groupRepo.Delete(repo.WithByID(id))
}

func (u *GroupService) Update(req dto.GroupUpdate) error {
	if req.IsDefault {
		if err := groupRepo.CancelDefault(req.Type); err != nil {
			return err
		}
	}
	upMap := make(map[string]interface{})
	upMap["name"] = req.Name
	upMap["is_default"] = req.IsDefault

	return groupRepo.Update(req.ID, upMap)
}