headscale/app.go
2021-02-21 23:54:15 +01:00

109 lines
2.4 KiB
Go

package headscale
import (
"fmt"
"log"
"os"
"github.com/gin-gonic/gin"
"tailscale.com/tailcfg"
"tailscale.com/wgengine/wgcfg"
)
// Config contains the initial Headscale configuration
type Config struct {
ServerURL string
Addr string
PrivateKeyPath string
DerpMap *tailcfg.DERPMap
DBhost string
DBport int
DBname string
DBuser string
DBpass string
}
// Headscale represents the base app of the service
type Headscale struct {
cfg Config
dbString string
publicKey *wgcfg.Key
privateKey *wgcfg.PrivateKey
}
// NewHeadscale returns the Headscale app
func NewHeadscale(cfg Config) (*Headscale, error) {
content, err := os.ReadFile(cfg.PrivateKeyPath)
if err != nil {
return nil, err
}
privKey, err := wgcfg.ParsePrivateKey(string(content))
if err != nil {
return nil, err
}
pubKey := privKey.Public()
h := Headscale{
cfg: cfg,
dbString: fmt.Sprintf("host=%s port=%d dbname=%s user=%s password=%s sslmode=disable", cfg.DBhost,
cfg.DBport, cfg.DBname, cfg.DBuser, cfg.DBpass),
privateKey: privKey,
publicKey: &pubKey,
}
err = h.initDB()
if err != nil {
return nil, err
}
return &h, nil
}
// Serve launches a GIN server with the Headscale API
func (h *Headscale) Serve() error {
r := gin.Default()
r.GET("/key", h.KeyHandler)
r.GET("/register", h.RegisterWebAPI)
r.POST("/machine/:id/map", h.PollNetMapHandler)
r.POST("/machine/:id", h.RegistrationHandler)
// r.LoadHTMLFiles("./frontend/build/index.html")
// r.Use(static.Serve("/", static.LocalFile("./frontend/build", true)))
err := r.Run(h.cfg.Addr)
return err
}
// RegisterMachine is executed from the CLI to register a new Machine using its MachineKey
func (h *Headscale) RegisterMachine(key string) error {
mKey, err := wgcfg.ParseHexKey(key)
if err != nil {
log.Printf("Cannot parse client key: %s", err)
return err
}
db, err := h.db()
if err != nil {
log.Printf("Cannot open DB: %s", err)
return err
}
defer db.Close()
m := Machine{}
if db.First(&m, "machine_key = ?", mKey.HexString()).RecordNotFound() {
log.Printf("Cannot find machine with machine key: %s", mKey.Base64())
return err
}
if m.isAlreadyRegistered() {
fmt.Println("This machine already registered")
return nil
}
ip, err := h.getAvailableIP()
if err != nil {
log.Println(err)
return err
}
m.IPAddress = ip.String()
m.Registered = true
db.Save(&m)
fmt.Println("Machine registered 🎉")
return nil
}