mirror of
https://github.com/lejianwen/rustdesk-api.git
synced 2025-11-29 08:33:21 +00:00
358 lines
9.8 KiB
Go
358 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"
|
|
"gorm.io/gorm"
|
|
"strconv"
|
|
)
|
|
|
|
type User struct {
|
|
}
|
|
|
|
// Detail 管理员
|
|
// @Tags 用户
|
|
// @Summary 管理员详情
|
|
// @Description 管理员详情
|
|
// @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 管理员
|
|
// @Tags 用户
|
|
// @Summary 创建管理员
|
|
// @Description 创建管理员
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param body body admin.UserForm true "管理员信息"
|
|
// @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 列表
|
|
// @Tags 用户
|
|
// @Summary 管理员列表
|
|
// @Description 管理员列表
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param page query int false "页码"
|
|
// @Param page_size query int false "页大小"
|
|
// @Param username query int false "账户"
|
|
// @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 编辑
|
|
// @Tags 用户
|
|
// @Summary 管理员编辑
|
|
// @Description 管理员编辑
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param body body admin.UserForm true "用户信息"
|
|
// @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 删除
|
|
// @Tags 用户
|
|
// @Summary 管理员删除
|
|
// @Description 管理员编删除
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param body body admin.UserForm true "用户信息"
|
|
// @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 修改密码
|
|
// @Tags 用户
|
|
// @Summary 修改密码
|
|
// @Description 修改密码
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param body body admin.UserPasswordForm true "用户信息"
|
|
// @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 当前用户
|
|
// @Tags 用户
|
|
// @Summary 当前用户
|
|
// @Description 当前用户
|
|
// @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 修改当前用户密码
|
|
// @Tags 用户
|
|
// @Summary 修改当前用户密码
|
|
// @Description 修改当前用户密码
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param body body admin.ChangeCurPasswordForm true "用户信息"
|
|
// @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)
|
|
// If the password is not empty, the old password is verified
|
|
// otherwise, the old password is not verified
|
|
if !service.AllService.UserService.IsPasswordEmptyByUser(u) {
|
|
oldPwd := service.AllService.UserService.EncryptPassword(f.OldPassword)
|
|
if u.Password != oldPwd {
|
|
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 用户
|
|
// @Summary 我的授权
|
|
// @Description 我的授权
|
|
// @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) {
|
|
q := &admin.GroupUsersQuery{}
|
|
if err := c.ShouldBindJSON(q); err != nil {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "ParamsError")+err.Error())
|
|
return
|
|
}
|
|
u := service.AllService.UserService.CurUser(c)
|
|
gid := u.GroupId
|
|
uid := u.Id
|
|
if service.AllService.UserService.IsAdmin(u) && q.UserId > 0 {
|
|
nu := service.AllService.UserService.InfoById(q.UserId)
|
|
gid = nu.GroupId
|
|
uid = q.UserId
|
|
}
|
|
res := service.AllService.UserService.List(1, 999, func(tx *gorm.DB) {
|
|
tx.Where("group_id = ?", gid)
|
|
})
|
|
var data []*adResp.GroupUsersPayload
|
|
for _, _u := range res.Users {
|
|
gup := &adResp.GroupUsersPayload{}
|
|
gup.FromUser(_u)
|
|
if _u.Id == uid {
|
|
gup.Status = 0
|
|
}
|
|
data = append(data, gup)
|
|
}
|
|
response.Success(c, data)
|
|
}
|
|
|
|
// 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
|
|
}
|
|
u := service.AllService.UserService.Register(f.Username, f.Email, f.Password)
|
|
if u == nil || u.Id == 0 {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "OperationFailed"))
|
|
return
|
|
}
|
|
// 注册成功后自动登录
|
|
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)
|
|
}
|