headscale/cli.go

44 lines
985 B
Go
Raw Normal View History

2021-04-08 23:57:31 +02:00
package headscale
import (
2021-05-08 13:27:53 +02:00
"errors"
2021-04-08 23:57:31 +02:00
2021-06-24 15:44:19 +02:00
"gorm.io/gorm"
2021-06-25 18:57:08 +02:00
"tailscale.com/types/wgkey"
2021-04-08 23:57:31 +02:00
)
// RegisterMachine is executed from the CLI to register a new Machine using its MachineKey
2021-05-08 13:27:53 +02:00
func (h *Headscale) RegisterMachine(key string, namespace string) (*Machine, error) {
2021-04-08 23:57:31 +02:00
ns, err := h.GetNamespace(namespace)
if err != nil {
2021-05-08 13:27:53 +02:00
return nil, err
2021-04-08 23:57:31 +02:00
}
2021-06-25 18:57:08 +02:00
mKey, err := wgkey.ParseHex(key)
2021-04-08 23:57:31 +02:00
if err != nil {
2021-05-08 13:27:53 +02:00
return nil, err
2021-04-08 23:57:31 +02:00
}
2021-07-04 21:40:46 +02:00
2021-04-08 23:57:31 +02:00
m := Machine{}
2021-07-04 21:40:46 +02:00
if result := h.db.First(&m, "machine_key = ?", mKey.HexString()); errors.Is(result.Error, gorm.ErrRecordNotFound) {
2021-05-08 13:27:53 +02:00
return nil, errors.New("Machine not found")
2021-04-08 23:57:31 +02:00
}
2021-10-10 17:22:42 +08:00
h.updateMachineExpiry(&m) // update the machine's expiry before bailing if its already registered
2021-04-08 23:57:31 +02:00
if m.isAlreadyRegistered() {
2021-05-08 13:27:53 +02:00
return nil, errors.New("Machine already registered")
2021-04-08 23:57:31 +02:00
}
ip, err := h.getAvailableIP()
if err != nil {
2021-05-08 13:27:53 +02:00
return nil, err
2021-04-08 23:57:31 +02:00
}
m.IPAddress = ip.String()
m.NamespaceID = ns.ID
m.Registered = true
2021-05-06 18:26:01 +02:00
m.RegisterMethod = "cli"
2021-07-04 21:40:46 +02:00
h.db.Save(&m)
2021-10-10 17:22:42 +08:00
2021-05-08 13:27:53 +02:00
return &m, nil
2021-04-08 23:57:31 +02:00
}