feat: add system api

Author: yongxinz

common/captcha/captcha.go

@@ -0,0 +1,48 @@
+package captcha
+
+import (
+	"image/color"
+
+	"github.com/google/uuid"
+	"github.com/mojocn/base64Captcha"
+)
+
+// SetStore 设置store
+func SetStore(s base64Captcha.Store) {
+	base64Captcha.DefaultMemStore = s
+}
+
+// configJsonBody json request body.
+type configJsonBody struct {
+	Id            string
+	CaptchaType   string
+	VerifyValue   string
+	DriverAudio   *base64Captcha.DriverAudio
+	DriverString  *base64Captcha.DriverString
+	DriverChinese *base64Captcha.DriverChinese
+	DriverMath    *base64Captcha.DriverMath
+	DriverDigit   *base64Captcha.DriverDigit
+}
+
+func DriverStringFunc() (id, b64s string, err error) {
+	e := configJsonBody{}
+	e.Id = uuid.New().String()
+	e.DriverString = base64Captcha.NewDriverString(46, 140, 2, 2, 4, "234567890abcdefghjkmnpqrstuvwxyz", &color.RGBA{240, 240, 246, 246}, nil, []string{"wqy-microhei.ttc"})
+	driver := e.DriverString.ConvertFonts()
+	cap := base64Captcha.NewCaptcha(driver, base64Captcha.DefaultMemStore)
+	return cap.Generate()
+}
+
+func DriverDigitFunc() (id, b64s string, err error) {
+	e := configJsonBody{}
+	e.Id = uuid.New().String()
+	e.DriverDigit = base64Captcha.NewDriverDigit(80, 240, 4, 0.7, 80)
+	driver := e.DriverDigit
+	cap := base64Captcha.NewCaptcha(driver, base64Captcha.DefaultMemStore)
+	return cap.Generate()
+}
+
+// Verify 校验验证码
+func Verify(id, code string, clear bool) bool {
+	return base64Captcha.DefaultMemStore.Verify(id, code, clear)
+}

common/cryptx/crypt.go

@@ -0,0 +1,12 @@
+package cryptx
+
+import (
+	"fmt"
+
+	"golang.org/x/crypto/scrypt"
+)
+
+func PasswordEncrypt(salt, password string) string {
+	dk, _ := scrypt.Key([]byte(password), []byte(salt), 32768, 8, 1, 32)
+	return fmt.Sprintf("%x", string(dk))
+}

common/errorx/baseerror.go

@@ -0,0 +1,44 @@
+package errorx
+
+import "github.com/yongxin/zen/common/globalkey"
+
+type CodeError struct {
+	Code int    `json:"code"`
+	Msg  string `json:"msg"`
+}
+
+type CodeErrorResponse struct {
+	Code int    `json:"code"`
+	Msg  string `json:"msg"`
+}
+
+func NewCodeError(code int, msg string) error {
+	return &CodeError{Code: code, Msg: msg}
+}
+
+func NewDefaultError(code int) error {
+	return NewCodeError(code, MapErrMsg(code))
+}
+
+func NewHandlerError(code int, msg string) error {
+	return NewCodeError(code, msg)
+}
+
+func NewSystemError(code int, msg string) error {
+	if globalkey.SysShowSystemError {
+		return NewCodeError(code, msg)
+	} else {
+		return NewCodeError(code, MapErrMsg(code))
+	}
+}
+
+func (e *CodeError) Error() string {
+	return e.Msg
+}
+
+func (e *CodeError) Data() *CodeErrorResponse {
+	return &CodeErrorResponse{
+		Code: e.Code,
+		Msg:  e.Msg,
+	}
+}

common/errorx/errormsg.go

@@ -0,0 +1,109 @@
+package errorx
+
+var errorMsg map[int]string
+
+const (
+	ServerErrorCode              = 1000
+	ParamErrorCode               = 1001
+	CaptchaErrorCode             = 1002
+	AccountErrorCode             = 1003
+	PasswordErrorCode            = 1004
+	NotPermMenuErrorCode         = 1005
+	DeletePermMenuErrorCode      = 1006
+	ParentPermMenuErrorCode      = 1007
+	AddRoleErrorCode             = 1008
+	DeleteRoleErrorCode          = 1009
+	AddDeptErrorCode             = 1010
+	DeleteDeptErrorCode          = 1011
+	AddJobErrorCode              = 1012
+	DeleteJobErrorCode           = 1013
+	AddProfessionErrorCode       = 1014
+	DeleteProfessionErrorCode    = 1015
+	AddUserErrorCode             = 1016
+	DeptHasUserErrorCode         = 1017
+	RoleIsUsingErrorCode         = 1018
+	ParentRoleErrorCode          = 1019
+	ParentDeptErrorCode          = 1020
+	AccountDisableErrorCode      = 1021
+	SetParentIdErrorCode         = 1022
+	SetParentTypeErrorCode       = 1023
+	AddConfigErrorCode           = 1024
+	AddDictionaryErrorCode       = 1025
+	AuthErrorCode                = 1026
+	DeleteDictionaryErrorCode    = 1027
+	JobIsUsingErrorCode          = 1028
+	ProfessionIsUsingErrorCode   = 1029
+	ForbiddenErrorCode           = 1030
+	UpdateRoleUniqueKeyErrorCode = 1031
+	UpdateDeptUniqueKeyErrorCode = 1032
+	AssigningRolesErrorCode      = 1033
+	DeptIdErrorCode              = 1034
+	ProfessionIdErrorCode        = 1035
+	PostIdErrorCode              = 1036
+	ParentRoleIdErrorCode        = 1037
+	ParentDeptIdErrorCode        = 1038
+	ParentPermMenuIdErrorCode    = 1039
+	ParentDictionaryIdErrorCode  = 1040
+	DictionaryIdErrorCode        = 1041
+	PermMenuIdErrorCode          = 1042
+	RoleIdErrorCode              = 1043
+	UserIdErrorCode              = 1044
+	ConfigErrorCode              = 1045
+)
+
+func init() {
+	errorMsg = make(map[int]string)
+	errorMsg[ServerErrorCode] = "服务繁忙，请稍后重试"
+	errorMsg[CaptchaErrorCode] = "验证码错误"
+	errorMsg[AccountErrorCode] = "账号错误"
+	errorMsg[PasswordErrorCode] = "密码错误"
+	errorMsg[NotPermMenuErrorCode] = "权限不足"
+	errorMsg[DeletePermMenuErrorCode] = "该权限菜单存在子级权限菜单"
+	errorMsg[ParentPermMenuErrorCode] = "父级菜单不能为自己"
+	errorMsg[AddRoleErrorCode] = "角色已存在"
+	errorMsg[DeleteRoleErrorCode] = "该角色存在子角色"
+	errorMsg[AddDeptErrorCode] = "部门已存在"
+	errorMsg[DeleteDeptErrorCode] = "该部门存在子部门"
+	errorMsg[AddJobErrorCode] = "岗位已存在"
+	errorMsg[DeleteJobErrorCode] = "该岗位正在使用中"
+	errorMsg[AddProfessionErrorCode] = "职称已存在"
+	errorMsg[DeleteProfessionErrorCode] = "该职称正在使用中"
+	errorMsg[AddUserErrorCode] = "账号已存在"
+	errorMsg[DeptHasUserErrorCode] = "该部门正在使用中"
+	errorMsg[RoleIsUsingErrorCode] = "该角色正在使用中"
+	errorMsg[ParentRoleErrorCode] = "父级角色不能为自己"
+	errorMsg[ParentDeptErrorCode] = "父级部门不能为自己"
+	errorMsg[AccountDisableErrorCode] = "账号已禁用"
+	errorMsg[SetParentIdErrorCode] = "不能设置子级为自己的父级"
+	errorMsg[SetParentTypeErrorCode] = "权限类型不能作为父级菜单"
+	errorMsg[AddConfigErrorCode] = "配置已存在"
+	errorMsg[AddDictionaryErrorCode] = "字典已存在"
+	errorMsg[AuthErrorCode] = "授权已失效，请重新登录"
+	errorMsg[DeleteDictionaryErrorCode] = "该字典集存在配置项"
+	errorMsg[JobIsUsingErrorCode] = "该岗位正在使用中"
+	errorMsg[ProfessionIsUsingErrorCode] = "该职称正在使用中"
+	errorMsg[ForbiddenErrorCode] = "禁止操作"
+	errorMsg[UpdateRoleUniqueKeyErrorCode] = "角色标识已存在"
+	errorMsg[UpdateDeptUniqueKeyErrorCode] = "部门标识已存在"
+	errorMsg[AssigningRolesErrorCode] = "角色不在可控范围"
+	errorMsg[DeptIdErrorCode] = "部门不存在"
+	errorMsg[ProfessionIdErrorCode] = "职称不存在"
+	errorMsg[PostIdErrorCode] = "岗位不存在"
+	errorMsg[ParentRoleIdErrorCode] = "父级角色不存在"
+	errorMsg[ParentDeptIdErrorCode] = "父级部门不存在"
+	errorMsg[ParentPermMenuIdErrorCode] = "父级菜单不存在"
+	errorMsg[ParentDictionaryIdErrorCode] = "字典集不存在"
+	errorMsg[DictionaryIdErrorCode] = "字典不存在"
+	errorMsg[PermMenuIdErrorCode] = "权限菜单不存在"
+	errorMsg[RoleIdErrorCode] = "角色不存在"
+	errorMsg[UserIdErrorCode] = "用户不存在"
+	errorMsg[ConfigErrorCode] = "配置不存在"
+}
+
+func MapErrMsg(errCode int) string {
+	if msg, ok := errorMsg[errCode]; ok {
+		return msg
+	} else {
+		return "服务繁忙，请稍后重试"
+	}
+}

common/globalkey/core.go

@@ -0,0 +1,22 @@
+package globalkey
+
+const (
+	SysPermMenuPrefix          = "/admin/"
+	SysJwtUserId               = "userId"
+	SysPermMenuCachePrefix     = "cache:rms:permmenu:"
+	SysOnlineUserCachePrefix   = "cache:rms:online:"
+	SysLoginCaptchaCachePrefix = "cache:rms:captcha:"
+	SysUserIdCachePrefix       = "cache:rms:sysuser:id:"
+	SysDateFormat              = "2006-01-02 15:04:05"
+	SysNewUserDefaultPassword  = "123456"
+	SysShowSystemError         = true
+	SysProtectPermMenuMaxId    = 44
+	SysProtectDictionaryMaxId  = 4
+	SysSuperUserId             = 1
+	SysSuperRoleId             = 1
+	SysDefaultPermType         = 2
+	SysEnable                  = 1
+	SysDisable                 = 0
+	SysLoginLogType            = 1
+	SysTopParentId             = 0
+)

common/jwtx/jwt.go

@@ -0,0 +1,13 @@
+package jwtx
+
+import "github.com/golang-jwt/jwt"
+
+func GetToken(secretKey string, iat, seconds, userId int64) (string, error) {
+	claims := make(jwt.MapClaims)
+	claims["exp"] = iat + seconds
+	claims["iat"] = iat
+	claims["userId"] = userId
+	token := jwt.New(jwt.SigningMethodHS256)
+	token.Claims = claims
+	return token.SignedString([]byte(secretKey))
+}

common/response/response.go

@@ -0,0 +1,26 @@
+package response
+
+import (
+	"net/http"
+
+	"github.com/zeromicro/go-zero/rest/httpx"
+)
+
+type Body struct {
+	Code int         `json:"code"`
+	Msg  string      `json:"msg"`
+	Data interface{} `json:"data,omitempty"`
+}
+
+func Response(w http.ResponseWriter, resp interface{}, err error) {
+	var body Body
+	if err != nil {
+		body.Code = 0
+		body.Msg = err.Error()
+	} else {
+		body.Code = 200
+		body.Msg = "success"
+		body.Data = resp
+	}
+	httpx.OkJson(w, body)
+}