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>
363 lines
10 KiB
Go
363 lines
10 KiB
Go
package admin
|
|
|
|
import (
|
|
"encoding/json"
|
|
_ "encoding/json"
|
|
"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"
|
|
"github.com/lejianwen/rustdesk-api/v2/service"
|
|
"gorm.io/gorm"
|
|
"strconv"
|
|
)
|
|
|
|
type AddressBook struct {
|
|
}
|
|
|
|
// Detail address book
|
|
// @Tags address book
|
|
// @Summary address book details
|
|
// @Description address book details
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param id path int true "ID"
|
|
// @Success 200 {object} response.Response{data=model.AddressBook}
|
|
// @Failure 500 {object} response.Response
|
|
// @Router /admin/address_book/detail/{id} [get]
|
|
// @Security token
|
|
func (ct *AddressBook) Detail(c *gin.Context) {
|
|
id := c.Param("id")
|
|
iid, _ := strconv.Atoi(id)
|
|
t := service.AllService.AddressBookService.InfoByRowId(uint(iid))
|
|
if t.RowId > 0 {
|
|
response.Success(c, t)
|
|
return
|
|
}
|
|
response.Fail(c, 101, response.TranslateMsg(c, "ItemNotFound"))
|
|
return
|
|
}
|
|
|
|
// Create create address book
|
|
// @Tags address book
|
|
// @Summary create address book
|
|
// @Description create address book
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param body body admin.AddressBookForm true "address book info"
|
|
// @Success 200 {object} response.Response{data=model.AddressBook}
|
|
// @Failure 500 {object} response.Response
|
|
// @Router /admin/address_book/create [post]
|
|
// @Security token
|
|
func (ct *AddressBook) Create(c *gin.Context) {
|
|
f := &admin.AddressBookForm{}
|
|
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
|
|
}
|
|
t := f.ToAddressBook()
|
|
if t.UserId == 0 {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "ParamsError"))
|
|
return
|
|
}
|
|
if t.CollectionId > 0 && !service.AllService.AddressBookService.CheckCollectionOwner(t.UserId, t.CollectionId) {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "ParamsError"))
|
|
return
|
|
}
|
|
|
|
ex := service.AllService.AddressBookService.InfoByUserIdAndIdAndCid(t.UserId, t.Id, t.CollectionId)
|
|
if ex.RowId > 0 {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "ItemExists"))
|
|
return
|
|
}
|
|
|
|
err := service.AllService.AddressBookService.Create(t)
|
|
if err != nil {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "OperationFailed")+err.Error())
|
|
return
|
|
}
|
|
response.Success(c, nil)
|
|
}
|
|
|
|
// BatchCreate batch create address books
|
|
// @Tags address book
|
|
// @Summary batch create address books
|
|
// @Description batch create address books
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param body body admin.AddressBookForm true "address book info"
|
|
// @Success 200 {object} response.Response{data=model.AddressBook}
|
|
// @Failure 500 {object} response.Response
|
|
// @Router /admin/address_book/batchCreate [post]
|
|
// @Security token
|
|
func (ct *AddressBook) BatchCreate(c *gin.Context) {
|
|
f := &admin.AddressBookForm{}
|
|
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
|
|
}
|
|
ul := len(f.UserIds)
|
|
|
|
if ul == 0 {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "ParamsError"))
|
|
return
|
|
}
|
|
if ul > 1 {
|
|
//clear tags in multi-user mode
|
|
f.Tags = []string{}
|
|
//in multi-user mode, can only create in the default address book
|
|
f.CollectionId = 0
|
|
}
|
|
|
|
//create tag
|
|
/*for _, fu := range f.UserIds {
|
|
if fu == 0 {
|
|
continue
|
|
}
|
|
for _, ft := range f.Tags {
|
|
exTag := service.AllService.TagService.InfoByUserIdAndNameAndCollectionId(fu, ft, 0)
|
|
if exTag.Id == 0 {
|
|
service.AllService.TagService.Create(&model.Tag{
|
|
UserId: fu,
|
|
Name: ft,
|
|
})
|
|
}
|
|
}
|
|
}*/
|
|
ts := f.ToAddressBooks()
|
|
for _, t := range ts {
|
|
if t.UserId == 0 {
|
|
continue
|
|
}
|
|
ex := service.AllService.AddressBookService.InfoByUserIdAndIdAndCid(t.UserId, t.Id, t.CollectionId)
|
|
if ex.RowId == 0 {
|
|
service.AllService.AddressBookService.Create(t)
|
|
}
|
|
}
|
|
|
|
response.Success(c, nil)
|
|
}
|
|
|
|
// List list
|
|
// @Tags address book
|
|
// @Summary address book list
|
|
// @Description address book list
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param page query int false "page number"
|
|
// @Param page_size query int false "page size"
|
|
// @Param user_id query int false "userid"
|
|
// @Param is_my query int false "whether it is mine"
|
|
// @Success 200 {object} response.Response{data=model.AddressBookList}
|
|
// @Failure 500 {object} response.Response
|
|
// @Router /admin/address_book/list [get]
|
|
// @Security token
|
|
func (ct *AddressBook) List(c *gin.Context) {
|
|
query := &admin.AddressBookQuery{}
|
|
if err := c.ShouldBindQuery(query); err != nil {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "ParamsError")+err.Error())
|
|
return
|
|
}
|
|
res := service.AllService.AddressBookService.List(query.Page, query.PageSize, func(tx *gorm.DB) {
|
|
tx.Preload("Collection", func(txc *gorm.DB) *gorm.DB {
|
|
return txc.Select("id,name")
|
|
})
|
|
if query.Id != "" {
|
|
tx.Where("id like ?", "%"+query.Id+"%")
|
|
}
|
|
if query.UserId > 0 {
|
|
tx.Where("user_id = ?", query.UserId)
|
|
}
|
|
if query.Username != "" {
|
|
tx.Where("username like ?", "%"+query.Username+"%")
|
|
}
|
|
if query.Hostname != "" {
|
|
tx.Where("hostname like ?", "%"+query.Hostname+"%")
|
|
}
|
|
if query.CollectionId != nil && *query.CollectionId >= 0 {
|
|
tx.Where("collection_id = ?", query.CollectionId)
|
|
}
|
|
})
|
|
|
|
abCIds := make([]uint, 0)
|
|
for _, ab := range res.AddressBooks {
|
|
abCIds = append(abCIds, ab.CollectionId)
|
|
}
|
|
response.Success(c, res)
|
|
}
|
|
|
|
// Update edit
|
|
// @Tags address book
|
|
// @Summary edit address book
|
|
// @Description edit address book
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param body body admin.AddressBookForm true "address book info"
|
|
// @Success 200 {object} response.Response{data=model.AddressBook}
|
|
// @Failure 500 {object} response.Response
|
|
// @Router /admin/address_book/update [post]
|
|
// @Security token
|
|
func (ct *AddressBook) Update(c *gin.Context) {
|
|
f := &admin.AddressBookForm{}
|
|
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
|
|
}
|
|
if f.RowId == 0 {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "ItemNotFound"))
|
|
return
|
|
}
|
|
ex := service.AllService.AddressBookService.InfoByRowId(f.RowId)
|
|
if ex.RowId == 0 {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "ItemNotFound"))
|
|
return
|
|
}
|
|
t := f.ToAddressBook()
|
|
if t.CollectionId > 0 && !service.AllService.AddressBookService.CheckCollectionOwner(t.UserId, t.CollectionId) {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "ParamsError"))
|
|
return
|
|
}
|
|
err := service.AllService.AddressBookService.UpdateAll(t)
|
|
if err != nil {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "OperationFailed")+err.Error())
|
|
return
|
|
}
|
|
response.Success(c, nil)
|
|
}
|
|
|
|
// Delete delete
|
|
// @Tags address book
|
|
// @Summary delete address book
|
|
// @Description delete address book
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param body body admin.AddressBookForm true "address book info"
|
|
// @Success 200 {object} response.Response
|
|
// @Failure 500 {object} response.Response
|
|
// @Router /admin/address_book/delete [post]
|
|
// @Security token
|
|
func (ct *AddressBook) Delete(c *gin.Context) {
|
|
f := &admin.AddressBookForm{}
|
|
if err := c.ShouldBindJSON(f); err != nil {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "ParamsError")+err.Error())
|
|
return
|
|
}
|
|
id := f.RowId
|
|
errList := global.Validator.ValidVar(c, id, "required,gt=0")
|
|
if len(errList) > 0 {
|
|
response.Fail(c, 101, errList[0])
|
|
return
|
|
}
|
|
t := service.AllService.AddressBookService.InfoByRowId(f.RowId)
|
|
if t.RowId == 0 {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "ItemNotFound"))
|
|
return
|
|
}
|
|
err := service.AllService.AddressBookService.Delete(t)
|
|
if err == nil {
|
|
response.Success(c, nil)
|
|
return
|
|
}
|
|
response.Fail(c, 101, response.TranslateMsg(c, "OperationFailed")+err.Error())
|
|
}
|
|
|
|
// ShareByWebClient
|
|
// @Tags address book
|
|
// @Summary share address book
|
|
// @Description share address book
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param body body admin.ShareByWebClientForm true "address book info"
|
|
// @Success 200 {object} response.Response
|
|
// @Failure 500 {object} response.Response
|
|
// @Router /admin/address_book/share [post]
|
|
// @Security token
|
|
func (ct *AddressBook) ShareByWebClient(c *gin.Context) {
|
|
f := &admin.ShareByWebClientForm{}
|
|
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)
|
|
ab := service.AllService.AddressBookService.InfoByUserIdAndId(u.Id, f.Id)
|
|
if ab.RowId == 0 {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "ItemNotFound"))
|
|
return
|
|
}
|
|
m := f.ToShareRecord()
|
|
m.UserId = u.Id
|
|
err := service.AllService.AddressBookService.ShareByWebClient(m)
|
|
if err != nil {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "OperationFailed")+err.Error())
|
|
return
|
|
}
|
|
response.Success(c, &gin.H{
|
|
"share_token": m.ShareToken,
|
|
})
|
|
}
|
|
|
|
func (ct *AddressBook) BatchCreateFromPeers(c *gin.Context) {
|
|
f := &admin.BatchCreateFromPeersForm{}
|
|
if err := c.ShouldBindJSON(f); err != nil {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "ParamsError")+err.Error())
|
|
return
|
|
}
|
|
|
|
if f.UserId == 0 {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "ParamsError"))
|
|
return
|
|
}
|
|
|
|
if f.CollectionId != 0 {
|
|
collection := service.AllService.AddressBookService.CollectionInfoById(f.CollectionId)
|
|
if collection.Id == 0 {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "ItemNotFound"))
|
|
return
|
|
}
|
|
}
|
|
|
|
pl := int64(len(f.PeerIds))
|
|
peers := service.AllService.PeerService.List(1, uint(pl), func(tx *gorm.DB) {
|
|
tx.Where("row_id in ?", f.PeerIds)
|
|
})
|
|
if peers.Total == 0 || pl != peers.Total {
|
|
response.Fail(c, 101, response.TranslateMsg(c, "ItemNotFound"))
|
|
return
|
|
}
|
|
|
|
tags, _ := json.Marshal(f.Tags)
|
|
for _, peer := range peers.Peers {
|
|
ab := service.AllService.AddressBookService.FromPeer(peer)
|
|
ab.Tags = tags
|
|
ab.CollectionId = f.CollectionId
|
|
ab.UserId = f.UserId
|
|
ex := service.AllService.AddressBookService.InfoByUserIdAndIdAndCid(f.UserId, ab.Id, ab.CollectionId)
|
|
if ex.RowId != 0 {
|
|
continue
|
|
}
|
|
service.AllService.AddressBookService.Create(ab)
|
|
}
|
|
response.Success(c, nil)
|
|
}
|