package service

import (
	"ferry/models/user"
	"ferry/pkg/connection"
	"strings"
)

/*
  @Author : lanyulei
*/

func GetPrincipal(processor []int, processMethod string) (principals string, err error) {
	/*
		person 人员
		persongroup 人员组
		department 部门
		variable 变量
	*/
	var principalList []string
	switch processMethod {
	case "person":
		err = connection.DB.Self.Model(&user.Info{}).
			Where("id in (?)", processor).
			Pluck("nickname", &principalList).Error
		if err != nil {
			return
		}
	case "persongroup":
		err = connection.DB.Self.Model(&user.Group{}).Where("id in (?)", processor).Pluck("nickname", &principalList).Error
		if err != nil {
			return
		}
	case "department":
		err = connection.DB.Self.Model(&user.Dept{}).Where("id in (?)", processor).Pluck("nickname", &principalList).Error
		if err != nil {
			return
		}
	case "variable":
		for _, p := range processor {
			switch p {
			case 1:
				principalList = append(principalList, "创建人")
			case 2:
				principalList = append(principalList, "创建人领导")
			}
		}
	}
	return strings.Join(principalList, ","), nil
}