fix(agent): 🐛 fix registration flow from changes in 7591c7a

joshuar · Dec 27, 2024 · 70176ce · 70176ce
1 parent 1e9d3c9
commit 70176ce
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/internal/agent/register.go b/internal/agent/register.go
@@ -31,7 +31,7 @@ func checkRegistration(ctx context.Context, agentUI ui) error {
 	}
 
 	// If not headless, present a UI for the user to configure options.
-	if options.headless {
+	if !options.headless {
 		userInputDoneCh := agentUI.DisplayRegistrationWindow(ctx, registrationOptions)
 		if canceled := <-userInputDoneCh; canceled {
 			return ErrUserCancelledRegistration
@@ -43,12 +43,15 @@ func checkRegistration(ctx context.Context, agentUI ui) error {
 	if err != nil {
 		return fmt.Errorf("device registration failed: %w", err)
 	}
-
 	// Save the returned preferences.
 	if err := preferences.SetHassPreferences(registrationDetails, registrationOptions); err != nil {
 		return fmt.Errorf("saving registration failed: %w", err)
 	}
-
+	// Set registration status.
+	if err := preferences.SetRegistered(true); err != nil {
+		return fmt.Errorf("saving registration failed: %w", err)
+	}
+	// Save preferences to disk.
 	if err := preferences.Save(); err != nil {
 		return fmt.Errorf("saving registration failed: %w", err)
 	}