Merge pull request #138 from JackalLabs/marston/terminate

add shutdown command

jprov/jprovd/init_provider.go

@@ -1,7 +1,10 @@
 package main
 
 import (
+	"bufio"
 	"fmt"
+	"os"
+	"strings"
 
 	"github.com/JackalLabs/jackal-provider/jprov/crypto"
 	"github.com/JackalLabs/jackal-provider/jprov/utils"
@@ -10,6 +13,28 @@ import (
 	"github.com/spf13/cobra"
 )
 
+func askForConfirmation(s string) bool {
+	reader := bufio.NewReader(os.Stdin)
+
+	for {
+		fmt.Printf("%s [y/n]: ", s)
+
+		response, err := reader.ReadString('\n')
+		if err != nil {
+			fmt.Println("failed to read string")
+			return false
+		}
+
+		response = strings.ToLower(strings.TrimSpace(response))
+
+		if response == "y" || response == "yes" {
+			return true
+		} else if response == "n" || response == "no" {
+			return false
+		}
+	}
+}
+
 func CmdInitProvider() *cobra.Command {
 	cmd := &cobra.Command{
 		Use:   "init [ip] [totalspace] [keybase-identity]",
@@ -55,3 +80,47 @@ func CmdInitProvider() *cobra.Command {
 
 	return cmd
 }
+
+func CmdShutdownProvider() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "terminate",
+		Short: "Permanently remove provider from network and get deposit back",
+		Long:  "Permanently remove provider from network and get deposit back.",
+		Args:  cobra.ExactArgs(0),
+		RunE: func(cmd *cobra.Command, args []string) (err error) {
+			if !askForConfirmation("Terminate Provider Permanently?") {
+				return nil
+			}
+
+			clientCtx, err := client.GetClientTxContext(cmd)
+			if err != nil {
+				fmt.Println(err)
+				return err
+			}
+
+			address, err := crypto.GetAddress(clientCtx)
+			if err != nil {
+				fmt.Println(err)
+				return err
+			}
+
+			fmt.Printf("Terminating provider: %s\n", address)
+			msg := types.NewMsgShutdownProvider(
+				address,
+			)
+			if err := msg.ValidateBasic(); err != nil {
+				fmt.Println(err)
+				return err
+			}
+			res, err := utils.SendTx(clientCtx, cmd.Flags(), "", msg)
+			if err != nil {
+				fmt.Println(err)
+				return err
+			}
+			fmt.Println(res.RawLog)
+			return err
+		},
+	}
+
+	return cmd
+}

jprov/jprovd/root.go

@@ -79,6 +79,7 @@ func NewRootCmd() *cobra.Command {
 		VersionCmd(),
 		NetworkCmd(),
 		BlanketCmd(),
+		CmdShutdownProvider(),
 	)
 
 	return rootCmd