From b4f7782fd8dedb5d123ca88286f7672ab6198fd3 Mon Sep 17 00:00:00 2001 From: Jeff Emershaw Date: Thu, 22 May 2025 02:55:51 +0000 Subject: [PATCH] support force flag for nodes backfillips --- cmd/headscale/cli/nodes.go | 22 ++++++++++++++-------- 1 file changed, 14 insertions(+), 8 deletions(-) diff --git a/cmd/headscale/cli/nodes.go b/cmd/headscale/cli/nodes.go index fb49f4a3..caac986c 100644 --- a/cmd/headscale/cli/nodes.go +++ b/cmd/headscale/cli/nodes.go @@ -539,19 +539,25 @@ be assigned to nodes.`, output, _ := cmd.Flags().GetString("output") confirm := false - prompt := &survey.Confirm{ - Message: "Are you sure that you want to assign/remove IPs to/from nodes?", + + force, _ := cmd.Flags().GetBool("force") + if !force { + prompt := &survey.Confirm{ + Message: "Are you sure that you want to assign/remove IPs to/from nodes?", + } + err = survey.AskOne(prompt, &confirm) + if err != nil { + return + } } - err = survey.AskOne(prompt, &confirm) - if err != nil { - return - } - if confirm { + + + if confirm || force { ctx, client, conn, cancel := newHeadscaleCLIWithConfig() defer cancel() defer conn.Close() - changes, err := client.BackfillNodeIPs(ctx, &v1.BackfillNodeIPsRequest{Confirmed: confirm}) + changes, err := client.BackfillNodeIPs(ctx, &v1.BackfillNodeIPsRequest{Confirmed: confirm || force }) if err != nil { ErrorOutput( err,