mirror of
https://github.com/lejianwen/rustdesk-api.git
synced 2025-12-02 01:53:16 +00:00
58 lines
1.3 KiB
Go
58 lines
1.3 KiB
Go
package config
|
|
|
|
import (
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
const (
|
|
DefaultIdServerPort = 21116
|
|
DefaultRelayServerPort = 21117
|
|
)
|
|
|
|
type Rustdesk struct {
|
|
IdServer string `mapstructure:"id-server"`
|
|
IdServerPort int `mapstructure:"-"`
|
|
RelayServer string `mapstructure:"relay-server"`
|
|
RelayPort int `mapstructure:"-"`
|
|
ApiServer string `mapstructure:"api-server"`
|
|
Key string `mapstructure:"key"`
|
|
KeyFile string `mapstructure:"key-file"`
|
|
Personal int `mapstructure:"personal"`
|
|
//webclient-magic-queryonline
|
|
WebclientMagicQueryonline int `mapstructure:"webclient-magic-queryonline"`
|
|
}
|
|
|
|
func (rd *Rustdesk) LoadKeyFile() {
|
|
// Load key file
|
|
if rd.Key != "" {
|
|
return
|
|
}
|
|
if rd.KeyFile != "" {
|
|
// Load key from file
|
|
b, err := os.ReadFile(rd.KeyFile)
|
|
if err != nil {
|
|
return
|
|
}
|
|
rd.Key = string(b)
|
|
return
|
|
}
|
|
}
|
|
func (rd *Rustdesk) ParsePort() {
|
|
// Parse port
|
|
idres := strings.Split(rd.IdServer, ":")
|
|
if len(idres) == 1 {
|
|
rd.IdServerPort = DefaultIdServerPort
|
|
} else if len(idres) == 2 {
|
|
rd.IdServerPort, _ = strconv.Atoi(idres[1])
|
|
}
|
|
|
|
relayres := strings.Split(rd.RelayServer, ":")
|
|
if len(relayres) == 1 {
|
|
rd.RelayPort = DefaultRelayServerPort
|
|
} else if len(relayres) == 2 {
|
|
rd.RelayPort, _ = strconv.Atoi(relayres[1])
|
|
}
|
|
}
|