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>
92 lines
2.0 KiB
Go
92 lines
2.0 KiB
Go
package service
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/lejianwen/rustdesk-api/v2/model"
|
|
"net"
|
|
"time"
|
|
)
|
|
|
|
type ServerCmdService struct{}
|
|
|
|
// List
|
|
func (is *ServerCmdService) List(page, pageSize uint) (res *model.ServerCmdList) {
|
|
res = &model.ServerCmdList{}
|
|
res.Page = int64(page)
|
|
res.PageSize = int64(pageSize)
|
|
tx := DB.Model(&model.ServerCmd{})
|
|
tx.Count(&res.Total)
|
|
tx.Scopes(Paginate(page, pageSize))
|
|
tx.Find(&res.ServerCmds)
|
|
return
|
|
}
|
|
|
|
// Info
|
|
func (is *ServerCmdService) Info(id uint) *model.ServerCmd {
|
|
u := &model.ServerCmd{}
|
|
DB.Where("id = ?", id).First(u)
|
|
return u
|
|
}
|
|
|
|
// Delete
|
|
func (is *ServerCmdService) Delete(u *model.ServerCmd) error {
|
|
return DB.Delete(u).Error
|
|
}
|
|
|
|
// Create
|
|
func (is *ServerCmdService) Create(u *model.ServerCmd) error {
|
|
res := DB.Create(u).Error
|
|
return res
|
|
}
|
|
|
|
// SendCmd send command
|
|
func (is *ServerCmdService) SendCmd(port int, cmd string, arg string) (string, error) {
|
|
//assemble command
|
|
cmd = cmd + " " + arg
|
|
res, err := is.SendSocketCmd("v6", port, cmd)
|
|
if err == nil {
|
|
return res, nil
|
|
}
|
|
//v6connection failed, trying v4
|
|
res, err = is.SendSocketCmd("v4", port, cmd)
|
|
if err == nil {
|
|
return res, nil
|
|
}
|
|
return "", err
|
|
}
|
|
|
|
// SendSocketCmd
|
|
func (is *ServerCmdService) SendSocketCmd(ty string, port int, cmd string) (string, error) {
|
|
addr := "[::1]"
|
|
tcp := "tcp6"
|
|
if ty == "v4" {
|
|
tcp = "tcp"
|
|
addr = "127.0.0.1"
|
|
}
|
|
conn, err := net.Dial(tcp, fmt.Sprintf("%s:%v", addr, port))
|
|
if err != nil {
|
|
Logger.Debugf("%s connect to id server failed: %v", ty, err)
|
|
return "", err
|
|
}
|
|
defer conn.Close()
|
|
//send command
|
|
_, err = conn.Write([]byte(cmd))
|
|
if err != nil {
|
|
Logger.Debugf("%s send cmd failed: %v", ty, err)
|
|
return "", err
|
|
}
|
|
time.Sleep(100 * time.Millisecond)
|
|
//read response
|
|
buf := make([]byte, 1024)
|
|
n, err := conn.Read(buf)
|
|
if err != nil && err.Error() != "EOF" {
|
|
Logger.Debugf("%s read response failed: %v", ty, err)
|
|
return "", err
|
|
}
|
|
return string(buf[:n]), nil
|
|
}
|
|
|
|
func (is *ServerCmdService) Update(f *model.ServerCmd) error {
|
|
return DB.Model(f).Updates(f).Error
|
|
}
|