Initial commit

machine.go

@@ -0,0 +1,125 @@
+package headscale
+
+import (
+	"encoding/json"
+	"fmt"
+	"log"
+	"time"
+
+	"github.com/jinzhu/gorm/dialects/postgres"
+	"github.com/tailscale/wireguard-go/wgcfg"
+	"tailscale.com/tailcfg"
+)
+
+type Machine struct {
+	ID         uint64 `gorm:"primary_key"`
+	MachineKey string `gorm:"type:varchar(64);unique_index"`
+	NodeKey    string
+	IPAddress  string
+
+	Registered bool // temp
+	LastSeen   *time.Time
+	Expiry     *time.Time
+
+	HostInfo  postgres.Jsonb
+	Endpoints postgres.Jsonb
+
+	CreatedAt time.Time
+	UpdatedAt time.Time
+	DeletedAt *time.Time
+}
+
+// For the time being this method is rather naive
+func (m Machine) isAlreadyRegistered() bool {
+	return m.Registered
+}
+
+func (m Machine) toNode() (*tailcfg.Node, error) {
+	nKey, err := wgcfg.ParseHexKey(m.NodeKey)
+	if err != nil {
+		return nil, err
+	}
+	mKey, err := wgcfg.ParseHexKey(m.MachineKey)
+	if err != nil {
+		return nil, err
+	}
+	addrs := []wgcfg.CIDR{}
+	allowedIPs := []wgcfg.CIDR{}
+
+	ip, err := wgcfg.ParseCIDR(fmt.Sprintf("%s/32", m.IPAddress))
+	if err != nil {
+		return nil, err
+	}
+	addrs = append(addrs, ip)
+	allowedIPs = append(allowedIPs, ip) // looks like the client expect this
+
+	endpoints := []string{}
+	be, err := m.Endpoints.MarshalJSON()
+	if err != nil {
+		return nil, err
+	}
+	err = json.Unmarshal(be, &endpoints)
+	if err != nil {
+		return nil, err
+	}
+
+	hostinfo := tailcfg.Hostinfo{}
+	hi, err := m.HostInfo.MarshalJSON()
+	if err != nil {
+		return nil, err
+	}
+	err = json.Unmarshal(hi, &hostinfo)
+	if err != nil {
+		return nil, err
+	}
+
+	n := tailcfg.Node{
+		ID:         tailcfg.NodeID(m.ID), // this is the actual ID
+		Name:       "",
+		User:       1,
+		Key:        tailcfg.NodeKey(nKey),
+		KeyExpiry:  *m.Expiry,
+		Machine:    tailcfg.MachineKey(mKey),
+		Addresses:  addrs,
+		AllowedIPs: allowedIPs,
+		Endpoints:  endpoints,
+		// DERP:       "127.3.3.40:4", // wtf?
+
+		Hostinfo: hostinfo,
+		Created:  m.CreatedAt,
+		LastSeen: m.LastSeen,
+
+		KeepAlive:         false,
+		MachineAuthorized: m.Registered,
+	}
+	// n.Key.MarshalText()
+	return &n, nil
+}
+
+func (h *Headscale) getPeers(m Machine) (*[]*tailcfg.Node, error) {
+	db, err := h.db()
+	if err != nil {
+		log.Printf("Cannot open DB: %s", err)
+		return nil, err
+	}
+	defer db.Close()
+
+	// Add user management here?
+	machines := []Machine{}
+	if err = db.Where("machine_key <> ?", m.MachineKey).Find(&machines).Error; err != nil {
+		log.Printf("Error accessing db: %s", err)
+		return nil, err
+	}
+
+	log.Printf("Found %d peers of %s", len(machines), m.MachineKey)
+
+	peers := []*tailcfg.Node{}
+	for _, mn := range machines {
+		peer, err := mn.toNode()
+		if err != nil {
+			return nil, err
+		}
+		peers = append(peers, peer)
+	}
+	return &peers, nil
+}