myriation/headscale
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 1 Wiki Activity

Merge pull request #171 from juanfont/force-flag

Browse source 
Added --force flag on node delete
This commit is contained in:
Juan Font 2021-10-16 14:21:35 +02:00 committed by GitHub
commit a9a1a8fb3c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 11 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
cmd/headscale/cli/nodes.go
View file

@ -144,15 +144,18 @@ var deleteNodeCmd = &cobra.Command{
} }
confirm := false confirm := false
prompt := &survey.Confirm{ force, _ := cmd.Flags().GetBool("force")
Message: fmt.Sprintf("Do you want to remove the node %s?", m.Name), if !force {
} prompt := &survey.Confirm{
err = survey.AskOne(prompt, &confirm) Message: fmt.Sprintf("Do you want to remove the node %s?", m.Name),
if err != nil { }
return err = survey.AskOne(prompt, &confirm)
if err != nil {
return
}
} }
if confirm { if confirm || force {
err = h.DeleteMachine(m) err = h.DeleteMachine(m)
if strings.HasPrefix(output, "json") { if strings.HasPrefix(output, "json") {
JsonOutput(map[string]string{"Result": "Node deleted"}, err, output) JsonOutput(map[string]string{"Result": "Node deleted"}, err, output)

1
cmd/headscale/cli/root.go
View file

@ -9,6 +9,7 @@ import (
func init() { func init() {
rootCmd.PersistentFlags().StringP("output", "o", "", "Output format. Empty for human-readable, 'json' or 'json-line'") rootCmd.PersistentFlags().StringP("output", "o", "", "Output format. Empty for human-readable, 'json' or 'json-line'")
rootCmd.PersistentFlags().Bool("force", false, "Disable prompts and forces the execution")
} }
var rootCmd = &cobra.Command{ var rootCmd = &cobra.Command{