mirror of
				https://github.com/TeaOSLab/EdgeAdmin.git
				synced 2025-11-04 05:00:25 +08:00 
			
		
		
		
	
		
			
				
	
	
		
			124 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			124 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
package users
 | 
						|
 | 
						|
import (
 | 
						|
	"github.com/TeaOSLab/EdgeAdmin/internal/utils/numberutils"
 | 
						|
	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
 | 
						|
	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
 | 
						|
	"github.com/iwind/TeaGo/actions"
 | 
						|
	"github.com/iwind/TeaGo/maps"
 | 
						|
	"github.com/xlzd/gotp"
 | 
						|
)
 | 
						|
 | 
						|
type CreatePopupAction struct {
 | 
						|
	actionutils.ParentAction
 | 
						|
}
 | 
						|
 | 
						|
func (this *CreatePopupAction) Init() {
 | 
						|
	this.Nav("", "", "")
 | 
						|
}
 | 
						|
 | 
						|
func (this *CreatePopupAction) RunGet(params struct{}) {
 | 
						|
	this.Show()
 | 
						|
}
 | 
						|
 | 
						|
func (this *CreatePopupAction) RunPost(params struct {
 | 
						|
	Username  string
 | 
						|
	Pass1     string
 | 
						|
	Pass2     string
 | 
						|
	Fullname  string
 | 
						|
	Mobile    string
 | 
						|
	Tel       string
 | 
						|
	Email     string
 | 
						|
	Remark    string
 | 
						|
	ClusterId int64
 | 
						|
 | 
						|
	// OTP
 | 
						|
	OtpOn bool
 | 
						|
 | 
						|
	Must *actions.Must
 | 
						|
	CSRF *actionutils.CSRF
 | 
						|
}) {
 | 
						|
	params.Must.
 | 
						|
		Field("username", params.Username).
 | 
						|
		Require("请输入用户名").
 | 
						|
		Match(`^[a-zA-Z0-9_]+$`, "用户名中只能含有英文、数字和下划线")
 | 
						|
 | 
						|
	checkUsernameResp, err := this.RPC().UserRPC().CheckUserUsername(this.AdminContext(), &pb.CheckUserUsernameRequest{
 | 
						|
		UserId:   0,
 | 
						|
		Username: params.Username,
 | 
						|
	})
 | 
						|
	if err != nil {
 | 
						|
		this.ErrorPage(err)
 | 
						|
		return
 | 
						|
	}
 | 
						|
	if checkUsernameResp.Exists {
 | 
						|
		this.FailField("username", "此用户名已经被占用,请换一个")
 | 
						|
	}
 | 
						|
 | 
						|
	params.Must.
 | 
						|
		Field("pass1", params.Pass1).
 | 
						|
		Require("请输入密码").
 | 
						|
		Field("pass2", params.Pass2).
 | 
						|
		Require("请再次输入确认密码").
 | 
						|
		Equal(params.Pass1, "两次输入的密码不一致")
 | 
						|
 | 
						|
	params.Must.
 | 
						|
		Field("fullname", params.Fullname).
 | 
						|
		Require("请输入全名")
 | 
						|
 | 
						|
	if params.ClusterId <= 0 {
 | 
						|
		this.Fail("请选择关联集群")
 | 
						|
	}
 | 
						|
 | 
						|
	if len(params.Mobile) > 0 {
 | 
						|
		params.Must.
 | 
						|
			Field("mobile", params.Mobile).
 | 
						|
			Mobile("请输入正确的手机号")
 | 
						|
	}
 | 
						|
	if len(params.Email) > 0 {
 | 
						|
		params.Must.
 | 
						|
			Field("email", params.Email).
 | 
						|
			Email("请输入正确的电子邮箱")
 | 
						|
	}
 | 
						|
 | 
						|
	createResp, err := this.RPC().UserRPC().CreateUser(this.AdminContext(), &pb.CreateUserRequest{
 | 
						|
		Username:      params.Username,
 | 
						|
		Password:      params.Pass1,
 | 
						|
		Fullname:      params.Fullname,
 | 
						|
		Mobile:        params.Mobile,
 | 
						|
		Tel:           params.Tel,
 | 
						|
		Email:         params.Email,
 | 
						|
		Remark:        params.Remark,
 | 
						|
		Source:        "admin:" + numberutils.FormatInt64(this.AdminId()),
 | 
						|
		NodeClusterId: params.ClusterId,
 | 
						|
	})
 | 
						|
	if err != nil {
 | 
						|
		this.ErrorPage(err)
 | 
						|
		return
 | 
						|
	}
 | 
						|
 | 
						|
	var userId = createResp.UserId
 | 
						|
 | 
						|
	defer this.CreateLogInfo("创建用户 %d", userId)
 | 
						|
 | 
						|
	// OTP
 | 
						|
	if params.OtpOn {
 | 
						|
		_, err = this.RPC().LoginRPC().UpdateLogin(this.AdminContext(), &pb.UpdateLoginRequest{Login: &pb.Login{
 | 
						|
			Id:   0,
 | 
						|
			Type: "otp",
 | 
						|
			ParamsJSON: maps.Map{
 | 
						|
				"secret": gotp.RandomSecret(16), // TODO 改成可以设置secret长度
 | 
						|
			}.AsJSON(),
 | 
						|
			IsOn:    true,
 | 
						|
			AdminId: 0,
 | 
						|
			UserId:  userId,
 | 
						|
		}})
 | 
						|
		if err != nil {
 | 
						|
			this.ErrorPage(err)
 | 
						|
			return
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	this.Success()
 | 
						|
}
 |