mirror of
https://github.com/psviderski/uncloud.git
synced 2026-08-26 19:13:34 +00:00
71 lines
1.7 KiB
Go
71 lines
1.7 KiB
Go
package config
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"net/netip"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/psviderski/uncloud/internal/secret"
|
|
)
|
|
|
|
const (
|
|
DefaultSSHUser = "root"
|
|
DefaultSSHPort = 22
|
|
)
|
|
|
|
type MachineConnection struct {
|
|
SSH SSHDestination `yaml:"ssh,omitempty"`
|
|
SSHKeyFile string `yaml:"ssh_key_file,omitempty"`
|
|
// TCP is the address and port of the machine's API server.
|
|
// The pointer is used to omit the field when not set. Otherwise, yaml marshalling includes an empty object.
|
|
TCP *netip.AddrPort `yaml:"tcp,omitempty"`
|
|
Host string `yaml:"host,omitempty"`
|
|
PublicKey secret.Secret `yaml:"public_key,omitempty"`
|
|
}
|
|
|
|
func (c MachineConnection) String() string {
|
|
if c.SSH != "" {
|
|
return string(c.SSH)
|
|
} else if c.TCP != nil && c.TCP.IsValid() {
|
|
return fmt.Sprintf("tcp://%s", c.TCP)
|
|
}
|
|
return "unknown connection"
|
|
}
|
|
|
|
// SSHDestination represents an SSH destination string in the canonical form of "user@host:port".
|
|
// The default user "root" and port 22 can be omitted.
|
|
type SSHDestination string
|
|
|
|
func NewSSHDestination(user, host string, port int) SSHDestination {
|
|
dst := host
|
|
if port != 0 && port != DefaultSSHPort {
|
|
dst = net.JoinHostPort(host, strconv.Itoa(port))
|
|
}
|
|
if user == "" {
|
|
user = DefaultSSHUser
|
|
}
|
|
dst = user + "@" + dst
|
|
return SSHDestination(dst)
|
|
}
|
|
|
|
func (d SSHDestination) Parse() (user string, host string, port int, err error) {
|
|
host = string(d)
|
|
if strings.Contains(host, "@") {
|
|
user, host, _ = strings.Cut(host, "@")
|
|
}
|
|
if user == "" {
|
|
user = DefaultSSHUser
|
|
}
|
|
h, p, sErr := net.SplitHostPort(host)
|
|
if sErr == nil {
|
|
host = h
|
|
port, err = strconv.Atoi(p)
|
|
}
|
|
if port == 0 {
|
|
port = DefaultSSHPort
|
|
}
|
|
return
|
|
}
|