From fc0c519206bb0d65db98275fd5903dc488519096 Mon Sep 17 00:00:00 2001
From: Juan Font Alonso <juanfontalonso@gmail.com>
Date: Thu, 6 May 2021 18:26:01 +0200
Subject: [PATCH] Fill RegisterMethod field in Machine

---
 api.go | 1 +
 cli.go | 1 +
 2 files changed, 2 insertions(+)

diff --git a/api.go b/api.go
index e0c5e753..7b98dfa1 100644
--- a/api.go
+++ b/api.go
@@ -397,6 +397,7 @@ func (h *Headscale) handleNewServer(c *gin.Context, db *gorm.DB, idKey wgcfg.Key
 		m.IPAddress = ip.String()
 		m.NamespaceID = pak.NamespaceID
 		m.AuthKeyID = uint(pak.ID)
+		m.RegisterMethod = "authKey"
 		m.Registered = true
 		db.Save(&m)
 
diff --git a/cli.go b/cli.go
index 0cb53336..f75bb470 100644
--- a/cli.go
+++ b/cli.go
@@ -43,6 +43,7 @@ func (h *Headscale) RegisterMachine(key string, namespace string) error {
 	m.IPAddress = ip.String()
 	m.NamespaceID = ns.ID
 	m.Registered = true
+	m.RegisterMethod = "cli"
 	db.Save(&m)
 	fmt.Println("Machine registered 🎉")
 	return nil