From fd1e4a1dcd4b08868f70f221eee4f2524a69ffdf Mon Sep 17 00:00:00 2001
From: Kristoffer Dalby <kradalby@kradalby.no>
Date: Sun, 27 Feb 2022 18:42:24 +0100
Subject: [PATCH] Generalise registration for openid

---
 oidc.go | 21 +++++++++++----------
 1 file changed, 11 insertions(+), 10 deletions(-)

diff --git a/oidc.go b/oidc.go
index edea32a8..5207c64e 100644
--- a/oidc.go
+++ b/oidc.go
@@ -328,27 +328,28 @@ func (h *Headscale) OIDCCallback(ctx *gin.Context) {
 			return
 		}
 
-		ips, err := h.getAvailableIPs()
+		_, err = h.RegisterMachine(
+			machineKeyStr,
+			namespace.Name,
+			RegisterMethodOIDC,
+			&requestedTime,
+			nil,
+			nil,
+			&now,
+		)
 		if err != nil {
 			log.Error().
 				Caller().
 				Err(err).
-				Msg("could not get an IP from the pool")
+				Msg("could not register machine")
 			ctx.String(
 				http.StatusInternalServerError,
-				"could not get an IP from the pool",
+				"could not register machine",
 			)
 
 			return
 		}
 
-		machine.IPAddresses = ips
-		machine.NamespaceID = namespace.ID
-		machine.Registered = true
-		machine.RegisterMethod = RegisterMethodOIDC
-		machine.LastSuccessfulUpdate = &now
-		machine.Expiry = &requestedTime
-		h.db.Save(&machine)
 	}
 
 	var content bytes.Buffer