mirror of
https://github.com/juanfont/headscale.git
synced 2024-12-20 19:09:07 +01:00
Improving code...
This commit is contained in:
parent
5d3c80a192
commit
1b7d99d483
5
app.go
5
app.go
@ -10,6 +10,7 @@ import (
|
||||
"tailscale.com/wgengine/wgcfg"
|
||||
)
|
||||
|
||||
// Config contains the initial Headscale configuration
|
||||
type Config struct {
|
||||
ServerURL string
|
||||
Addr string
|
||||
@ -23,6 +24,7 @@ type Config struct {
|
||||
DBpass string
|
||||
}
|
||||
|
||||
// Headscale represents the base app of the service
|
||||
type Headscale struct {
|
||||
cfg Config
|
||||
dbString string
|
||||
@ -30,6 +32,7 @@ type Headscale struct {
|
||||
privateKey *wgcfg.PrivateKey
|
||||
}
|
||||
|
||||
// NewHeadscale returns the Headscale app
|
||||
func NewHeadscale(cfg Config) (*Headscale, error) {
|
||||
content, err := ioutil.ReadFile(cfg.PrivateKeyPath)
|
||||
if err != nil {
|
||||
@ -54,6 +57,7 @@ func NewHeadscale(cfg Config) (*Headscale, error) {
|
||||
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)
|
||||
@ -67,6 +71,7 @@ func (h *Headscale) Serve() error {
|
||||
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 {
|
||||
|
Loading…
Reference in New Issue
Block a user