refactor code

Author: 0xbcdev

cmd/node/auto_backup_priv_validator_state.go

@@ -32,9 +32,10 @@ const (
 
 func GetAutoBackupPrivValidatorStateCmd() *cobra.Command {
 	var cmd = &cobra.Command{
-		Use:   commandAutoBackupPrivValidatorState + " [node_home]",
-		Short: "Designed to be run as a service, it will automatically backup the `priv_validator_state.json`, and kill the node process if the content of the file is decreased",
-		Args:  cobra.ExactArgs(1),
+		Use:     commandAutoBackupPrivValidatorState + " [node_home]",
+		Aliases: []string{"auto-backup-pvs", "auto-backup-priv-validator-state"},
+		Short:   "Designed to be run as a service, it will automatically backup the `priv_validator_state.json`, and kill the node process if the content of the file is decreased",
+		Args:    cobra.ExactArgs(1),
 		Run: func(cmd *cobra.Command, args []string) {
 			utils.MustNotUserRoot()
 
@@ -93,39 +94,7 @@ func GetAutoBackupPrivValidatorStateCmd() *cobra.Command {
 			fmt.Println("INF: binary to kill:", binaryNameToKill, "at", binaryPathToKill)
 
 			if cmd.Flags().Changed(flagGenSetup) {
-				const serviceFileName = "auto-backup-priv-validator-state"
-				fmt.Println("Input chain name (eg: Cosmos Hub):")
-				chainName := utils.ReadText(false)
-				fmt.Println("Mainnet or Testnet?")
-				networkType := utils.ReadText(false)
-				fmt.Println("INF: setup guide:")
-				fmt.Println("1. Create service file")
-				fmt.Println("> sudo vi /etc/systemd/system/" + serviceFileName + ".service")
-				fmt.Printf(`[Unit]
-Description=Auto backup priv_validator_state.json for Validator on %s %s
-After=network.target
-#
-[Service]
-User=%s
-ExecStart=%s/go/bin/%s start %s --%s %s --%s %d
-RestartSec=1
-Restart=on-failure
-LimitNOFILE=1024
-#
-[Install]
-WantedBy=multi-user.target
-`, chainName, networkType, currentUser.Username, currentUser.HomeDir, constants.BINARY_NAME, nodeHomeDirectory, flagBinaryKillByAutoBackup, binaryPathToKill, flagKeep, keepRecent)
-				fmt.Println("2. Setup visudo")
-				fmt.Println("> sudo visudo")
-				fmt.Printf(strings.ReplaceAll(strings.ReplaceAll(`# Allow user @USER@ to manage @SVC@ service
-@USER@ ALL= NOPASSWD: /usr/bin/systemctl start @SVC@
-@USER@ ALL= NOPASSWD: /usr/bin/systemctl stop @SVC@
-@USER@ ALL= NOPASSWD: /usr/bin/systemctl restart @SVC@
-@USER@ ALL= NOPASSWD: /usr/bin/systemctl enable @SVC@ # Do not allow disable
-@USER@ ALL= NOPASSWD: /usr/bin/systemctl status @SVC@
-`, "@USER@", currentUser.Username), "@SVC@", serviceFileName))
-				fmt.Println("3. Enable service to automatically run at startup")
-				fmt.Println("> sudo systemctl daemon-reload && sudo systemctl enable " + serviceFileName + ".service")
+				genSetupThenExit(nodeHomeDirectory, binaryPathToKill, keepRecent, currentUser)
 				return
 			}
 
@@ -501,3 +470,40 @@ func loadLatestBackupPrivValidatorStateOrExitWithErr(backupDstPath string) types
 
 	return *pvs
 }
+
+func genSetupThenExit(nodeHomeDirectory, binaryPathToKill string, keepRecent int, currentUser *user.User) {
+	const serviceFileName = "auto-backup-pvs"
+	fmt.Println("Input chain name (eg: Cosmos Hub):")
+	chainName := utils.ReadText(false)
+	fmt.Println("Mainnet or Testnet?")
+	networkType := utils.ReadText(false)
+	fmt.Println("INF: setup guide:")
+	fmt.Println("1. Create service file")
+	fmt.Println("> sudo vi /etc/systemd/system/" + serviceFileName + ".service")
+	fmt.Printf(`[Unit]
+Description=Auto backup priv_validator_state.json for Validator on %s %s
+After=network.target
+#
+[Service]
+User=%s
+ExecStart=%s/go/bin/%s start %s --%s %s --%s %d
+RestartSec=1
+Restart=on-failure
+LimitNOFILE=1024
+#
+[Install]
+WantedBy=multi-user.target
+`, chainName, networkType, currentUser.Username, currentUser.HomeDir, constants.BINARY_NAME, nodeHomeDirectory, flagBinaryKillByAutoBackup, binaryPathToKill, flagKeep, keepRecent)
+	fmt.Println("2. Setup visudo")
+	fmt.Println("> sudo visudo")
+	fmt.Printf(strings.ReplaceAll(strings.ReplaceAll(`# Allow user @USER@ to manage @SVC@ service
+@USER@ ALL= NOPASSWD: /usr/bin/systemctl start @SVC@
+@USER@ ALL= NOPASSWD: /usr/bin/systemctl stop @SVC@
+@USER@ ALL= NOPASSWD: /usr/bin/systemctl restart @SVC@
+@USER@ ALL= NOPASSWD: /usr/bin/systemctl enable @SVC@ # Do not allow disable
+@USER@ ALL= NOPASSWD: /usr/bin/systemctl status @SVC@
+`, "@USER@", currentUser.Username), "@SVC@", serviceFileName))
+	fmt.Println("3. Enable service to automatically run at startup")
+	fmt.Println("> sudo systemctl daemon-reload && sudo systemctl enable " + serviceFileName + ".service")
+	os.Exit(0)
+}