From 76515d12d6bbeda26157c49fbffa390536951873 Mon Sep 17 00:00:00 2001
From: Mike Poindexter <staticsnow@gmail.com>
Date: Thu, 29 Aug 2024 23:20:29 -0700
Subject: [PATCH] Fix self notification on expiry update via oidc relogin
 (#2080)

---
 hscontrol/oidc.go | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/hscontrol/oidc.go b/hscontrol/oidc.go
index fe4d357c..72fefac3 100644
--- a/hscontrol/oidc.go
+++ b/hscontrol/oidc.go
@@ -526,7 +526,17 @@ func (h *Headscale) validateNodeForOIDCCallback(
 			util.LogErr(err, "Failed to write response")
 		}
 
-		ctx := types.NotifyCtx(context.Background(), "oidc-expiry", "na")
+		ctx := types.NotifyCtx(context.Background(), "oidc-expiry-self", node.Hostname)
+		h.nodeNotifier.NotifyByNodeID(
+			ctx,
+			types.StateUpdate{
+				Type:        types.StateSelfUpdate,
+				ChangeNodes: []types.NodeID{node.ID},
+			},
+			node.ID,
+		)
+
+		ctx = types.NotifyCtx(context.Background(), "oidc-expiry-peers", node.Hostname)
 		h.nodeNotifier.NotifyWithIgnore(ctx, types.StateUpdateExpire(node.ID, expiry), node.ID)
 
 		return nil, true, nil