headscale/app.go

89 lines
2.1 KiB
Go
Raw Normal View History

2020-06-21 04:32:08 -06:00
package headscale
import (
"fmt"
2021-02-21 15:54:15 -07:00
"os"
"strings"
"sync"
2020-06-21 04:32:08 -06:00
"github.com/gin-gonic/gin"
2021-02-20 15:57:06 -07:00
"tailscale.com/tailcfg"
2021-02-20 14:43:07 -07:00
"tailscale.com/wgengine/wgcfg"
2020-06-21 04:32:08 -06:00
)
2021-02-21 14:14:38 -07:00
// Config contains the initial Headscale configuration
2020-06-21 04:32:08 -06:00
type Config struct {
ServerURL string
Addr string
PrivateKeyPath string
2021-02-20 15:57:06 -07:00
DerpMap *tailcfg.DERPMap
2020-06-21 04:32:08 -06:00
DBhost string
DBport int
DBname string
DBuser string
DBpass string
TLSCertPath string
TLSKeyPath string
2020-06-21 04:32:08 -06:00
}
2021-02-21 14:14:38 -07:00
// Headscale represents the base app of the service
2020-06-21 04:32:08 -06:00
type Headscale struct {
cfg Config
dbString string
publicKey *wgcfg.Key
privateKey *wgcfg.PrivateKey
pollMu sync.Mutex
clientsPolling map[uint64]chan []byte // this is by all means a hackity hack
2020-06-21 04:32:08 -06:00
}
2021-02-21 14:14:38 -07:00
// NewHeadscale returns the Headscale app
2020-06-21 04:32:08 -06:00
func NewHeadscale(cfg Config) (*Headscale, error) {
2021-02-21 15:54:15 -07:00
content, err := os.ReadFile(cfg.PrivateKeyPath)
2020-06-21 04:32:08 -06:00
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
}
h.clientsPolling = make(map[uint64]chan []byte)
2020-06-21 04:32:08 -06:00
return &h, nil
}
2021-02-21 14:14:38 -07:00
// Serve launches a GIN server with the Headscale API
2020-06-21 04:32:08 -06:00
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)
var err error
if h.cfg.TLSCertPath == "" {
if !strings.HasPrefix(h.cfg.ServerURL, "http://") {
fmt.Println("WARNING: listening without TLS but ServerURL does not start with http://")
}
err = r.Run(h.cfg.Addr)
} else {
if !strings.HasPrefix(h.cfg.ServerURL, "https://") {
fmt.Println("WARNING: listening with TLS but ServerURL does not start with https://")
}
err = r.RunTLS(h.cfg.Addr, h.cfg.TLSCertPath, h.cfg.TLSKeyPath)
}
2020-06-21 04:32:08 -06:00
return err
}