thomasandClaude Opus 4.8
16e97c8efa
Translate Chinese comments, log/error messages, validator labels and Swagger annotations to English throughout the source code, generated Swagger docs, config files and CI workflows. Make the English README primary: README.md now holds the English docs and README_EN.md holds the Chinese version, with cross-language links updated accordingly. Note: the generated docs/ swagger files were translated in place; run `go generate ./...` (swag init) to regenerate them from the now-English annotations when a Go toolchain is available. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
349 lines
9.8 KiB
Go
349 lines
9.8 KiB
Go
package admin
|
|
|
|
import (
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/lejianwen/rustdesk-api/v2/global"
|
|
"github.com/lejianwen/rustdesk-api/v2/http/request/admin"
|
|
"github.com/lejianwen/rustdesk-api/v2/http/response"
|
|
adResp "github.com/lejianwen/rustdesk-api/v2/http/response/admin"
|
|
"github.com/lejianwen/rustdesk-api/v2/model"
|
|
"github.com/lejianwen/rustdesk-api/v2/service"
|
|
"github.com/lejianwen/rustdesk-api/v2/utils"
|
|
"gorm.io/gorm"
|
|
"strconv"
|
|
)
|
|
|
|
type User struct {
|
|
}
|
|
|
|
// Detail administrator
|
|
// @Tags user
|
|
// @Summary administrator details
|
|
// @Description administrator details
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param id path int true "ID"
|
|
// @Success 200 {object} response.Response{data=model.User}
|
|
// @Failure 500 {object} response.Response
|
|
// @Router /admin/user/detail/{id} [get]
|
|
// @Security token
|
|
func (ct *User) Detail(c *gin.Context) {
|
|
id := c.Param("id")
|
|
iid, _ := strconv.Atoi(id)
|
|
u := service.AllService.UserService.InfoById(uint(iid))
|
|
if u.Id > 0 {
|
|
response.Success(c, u)
|
|
return
|
|
}
|
|
response.Fail(c, 101, response.TranslateMsg(c, "ItemNotFound"))
|
|
return
|
|
}
|
|
|
|
// Create administrator
|
|
// @Tags user
|
|
// @Summary create administrator
|
|
// @Description create administrator
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param body body admin.UserForm true "administrator info"
|
|
// @Success 200 {object} response.Response{data=model.User}
|
|
// @Failure 500 {object} response.Response
|
|
// @Router /admin/user/create [post]
|
|
// @Security token
|
|
func (ct *User) Create(c *gin.Context) {
|
|
f := &admin.UserForm{}
|
|
if err := c.ShouldBindJSON(f); err != nil {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "ParamsError")+err.Error())
|
|
return
|
|
}
|
|
errList := global.Validator.ValidStruct(c, f)
|
|
if len(errList) > 0 {
|
|
response.Fail(c, 101, errList[0])
|
|
return
|
|
}
|
|
u := f.ToUser()
|
|
err := service.AllService.UserService.Create(u)
|
|
if err != nil {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "OperationFailed")+err.Error())
|
|
return
|
|
}
|
|
response.Success(c, nil)
|
|
}
|
|
|
|
// List list
|
|
// @Tags user
|
|
// @Summary administrator list
|
|
// @Description administrator list
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param page query int false "page number"
|
|
// @Param page_size query int false "page size"
|
|
// @Param username query int false "account"
|
|
// @Success 200 {object} response.Response{data=model.UserList}
|
|
// @Failure 500 {object} response.Response
|
|
// @Router /admin/user/list [get]
|
|
// @Security token
|
|
func (ct *User) List(c *gin.Context) {
|
|
query := &admin.UserQuery{}
|
|
if err := c.ShouldBindQuery(query); err != nil {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "ParamsError")+err.Error())
|
|
return
|
|
}
|
|
res := service.AllService.UserService.List(query.Page, query.PageSize, func(tx *gorm.DB) {
|
|
if query.Username != "" {
|
|
tx.Where("username like ?", "%"+query.Username+"%")
|
|
}
|
|
})
|
|
response.Success(c, res)
|
|
}
|
|
|
|
// Update edit
|
|
// @Tags user
|
|
// @Summary edit administrator
|
|
// @Description edit administrator
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param body body admin.UserForm true "user info"
|
|
// @Success 200 {object} response.Response{data=model.User}
|
|
// @Failure 500 {object} response.Response
|
|
// @Router /admin/user/update [post]
|
|
// @Security token
|
|
func (ct *User) Update(c *gin.Context) {
|
|
f := &admin.UserForm{}
|
|
if err := c.ShouldBindJSON(f); err != nil {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "ParamsError")+err.Error())
|
|
return
|
|
}
|
|
if f.Id == 0 {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "ParamsError"))
|
|
return
|
|
}
|
|
errList := global.Validator.ValidStruct(c, f)
|
|
if len(errList) > 0 {
|
|
response.Fail(c, 101, errList[0])
|
|
return
|
|
}
|
|
u := f.ToUser()
|
|
err := service.AllService.UserService.Update(u)
|
|
if err != nil {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "OperationFailed")+err.Error())
|
|
return
|
|
}
|
|
response.Success(c, nil)
|
|
}
|
|
|
|
// Delete delete
|
|
// @Tags user
|
|
// @Summary delete administrator
|
|
// @Description delete administrator
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param body body admin.UserForm true "user info"
|
|
// @Success 200 {object} response.Response
|
|
// @Failure 500 {object} response.Response
|
|
// @Router /admin/user/delete [post]
|
|
// @Security token
|
|
func (ct *User) Delete(c *gin.Context) {
|
|
f := &admin.UserForm{}
|
|
if err := c.ShouldBindJSON(f); err != nil {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "ParamsError")+err.Error())
|
|
return
|
|
}
|
|
id := f.Id
|
|
errList := global.Validator.ValidVar(c, id, "required,gt=0")
|
|
if len(errList) > 0 {
|
|
response.Fail(c, 101, errList[0])
|
|
return
|
|
}
|
|
u := service.AllService.UserService.InfoById(f.Id)
|
|
if u.Id > 0 {
|
|
err := service.AllService.UserService.Delete(u)
|
|
if err == nil {
|
|
response.Success(c, nil)
|
|
return
|
|
}
|
|
response.Fail(c, 101, err.Error())
|
|
return
|
|
}
|
|
response.Fail(c, 101, response.TranslateMsg(c, "ItemNotFound"))
|
|
}
|
|
|
|
// UpdatePassword change password
|
|
// @Tags user
|
|
// @Summary change password
|
|
// @Description change password
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param body body admin.UserPasswordForm true "user info"
|
|
// @Success 200 {object} response.Response
|
|
// @Failure 500 {object} response.Response
|
|
// @Router /admin/user/updatePassword [post]
|
|
// @Security token
|
|
func (ct *User) UpdatePassword(c *gin.Context) {
|
|
f := &admin.UserPasswordForm{}
|
|
if err := c.ShouldBindJSON(f); err != nil {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "ParamsError")+err.Error())
|
|
return
|
|
}
|
|
errList := global.Validator.ValidStruct(c, f)
|
|
if len(errList) > 0 {
|
|
response.Fail(c, 101, errList[0])
|
|
return
|
|
}
|
|
u := service.AllService.UserService.InfoById(f.Id)
|
|
if u.Id == 0 {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "ItemNotFound"))
|
|
return
|
|
}
|
|
err := service.AllService.UserService.UpdatePassword(u, f.Password)
|
|
if err != nil {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "OperationFailed")+err.Error())
|
|
return
|
|
}
|
|
response.Success(c, nil)
|
|
}
|
|
|
|
// Current current user
|
|
// @Tags user
|
|
// @Summary current user
|
|
// @Description current user
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Success 200 {object} response.Response{data=adResp.LoginPayload}
|
|
// @Failure 500 {object} response.Response
|
|
// @Router /admin/user/current [get]
|
|
// @Security token
|
|
func (ct *User) Current(c *gin.Context) {
|
|
u := service.AllService.UserService.CurUser(c)
|
|
token, _ := c.Get("token")
|
|
t := token.(string)
|
|
responseLoginSuccess(c, u, t)
|
|
}
|
|
|
|
// ChangeCurPwd change the current user's password
|
|
// @Tags user
|
|
// @Summary change the current user's password
|
|
// @Description change the current user's password
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param body body admin.ChangeCurPasswordForm true "user info"
|
|
// @Success 200 {object} response.Response
|
|
// @Failure 500 {object} response.Response
|
|
// @Router /admin/user/changeCurPwd [post]
|
|
// @Security token
|
|
func (ct *User) ChangeCurPwd(c *gin.Context) {
|
|
f := &admin.ChangeCurPasswordForm{}
|
|
if err := c.ShouldBindJSON(f); err != nil {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "ParamsError")+err.Error())
|
|
return
|
|
}
|
|
|
|
errList := global.Validator.ValidStruct(c, f)
|
|
if len(errList) > 0 {
|
|
response.Fail(c, 101, errList[0])
|
|
return
|
|
}
|
|
u := service.AllService.UserService.CurUser(c)
|
|
// Verify the old password only when the account already has one set
|
|
if !service.AllService.UserService.IsPasswordEmptyByUser(u) {
|
|
ok, _, err := utils.VerifyPassword(u.Password, f.OldPassword)
|
|
if err != nil || !ok {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "OldPasswordError"))
|
|
return
|
|
}
|
|
}
|
|
err := service.AllService.UserService.UpdatePassword(u, f.NewPassword)
|
|
if err != nil {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "OperationFailed")+err.Error())
|
|
return
|
|
}
|
|
response.Success(c, nil)
|
|
}
|
|
|
|
// MyOauth
|
|
// @Tags user
|
|
// @Summary my authorizations
|
|
// @Description my authorizations
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Success 200 {object} response.Response{data=[]adResp.UserOauthItem}
|
|
// @Failure 500 {object} response.Response
|
|
// @Router /admin/user/myOauth [get]
|
|
// @Security token
|
|
func (ct *User) MyOauth(c *gin.Context) {
|
|
u := service.AllService.UserService.CurUser(c)
|
|
oal := service.AllService.OauthService.List(1, 100, nil)
|
|
ops := make([]string, 0)
|
|
for _, oa := range oal.Oauths {
|
|
ops = append(ops, oa.Op)
|
|
}
|
|
uts := service.AllService.UserService.UserThirdsByUserId(u.Id)
|
|
var res []*adResp.UserOauthItem
|
|
for _, oa := range oal.Oauths {
|
|
item := &adResp.UserOauthItem{
|
|
Op: oa.Op,
|
|
}
|
|
for _, ut := range uts {
|
|
if ut.Op == oa.Op {
|
|
item.Status = 1
|
|
break
|
|
}
|
|
}
|
|
res = append(res, item)
|
|
}
|
|
response.Success(c, res)
|
|
}
|
|
|
|
// groupUsers
|
|
func (ct *User) GroupUsers(c *gin.Context) {
|
|
aG := service.AllService.GroupService.List(1, 999, nil)
|
|
aU := service.AllService.UserService.List(1, 9999, nil)
|
|
response.Success(c, gin.H{
|
|
"groups": aG.Groups,
|
|
"users": aU.Users,
|
|
})
|
|
}
|
|
|
|
// Register
|
|
func (ct *User) Register(c *gin.Context) {
|
|
if !global.Config.App.Register {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "RegisterClosed"))
|
|
return
|
|
}
|
|
f := &admin.RegisterForm{}
|
|
if err := c.ShouldBindJSON(f); err != nil {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "ParamsError")+err.Error())
|
|
return
|
|
}
|
|
errList := global.Validator.ValidStruct(c, f)
|
|
if len(errList) > 0 {
|
|
response.Fail(c, 101, errList[0])
|
|
return
|
|
}
|
|
regStatus := model.StatusCode(global.Config.App.RegisterStatus)
|
|
// the registration status may not be configured; enabled by default
|
|
if regStatus != model.COMMON_STATUS_DISABLED && regStatus != model.COMMON_STATUS_ENABLE {
|
|
regStatus = model.COMMON_STATUS_ENABLE
|
|
}
|
|
|
|
u := service.AllService.UserService.Register(f.Username, f.Email, f.Password, regStatus)
|
|
if u == nil || u.Id == 0 {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "OperationFailed"))
|
|
return
|
|
}
|
|
if regStatus == model.COMMON_STATUS_DISABLED {
|
|
// requires administrator approval
|
|
response.Fail(c, 101, response.TranslateMsg(c, "RegisterSuccessWaitAdminConfirm"))
|
|
return
|
|
}
|
|
// automatically log in after successful registration
|
|
ut := service.AllService.UserService.Login(u, &model.LoginLog{
|
|
UserId: u.Id,
|
|
Client: model.LoginLogClientWebAdmin,
|
|
Uuid: "",
|
|
Ip: c.ClientIP(),
|
|
Type: model.LoginLogTypeAccount,
|
|
})
|
|
responseLoginSuccess(c, u, ut.Token)
|
|
}
|