Merge pull request #1 from kehiy/main

feat: --total flag, help user to stake all of account balance

Author: Ja7ad

README.md

@@ -29,6 +29,7 @@ Pactus bond tools for multiple stake by config, you can stake many validators in
 - `validators` is list of validators for stake **note: if your address is first time for validator need set public key in `pub`**
 
 2. run tools with `./staker -config ./cfg.json -password foobar`
-- `-config` config file path default is `./cfg.json`
-- `-password` is optional for wallet password
-- `-server` is for custom node rpc address
+- `--config` config file path default is `./cfg.json`
+- `--password` is optional for wallet password
+- `--server` is for custom node rpc address
+- `--total` is a flag that ignore amount it config and it will stake whole of account balance

go.mod

@@ -1 +1,3 @@
-module staker
+module staker
+
+go 1.21.1

main.go

@@ -7,12 +7,15 @@ import (
 	"log"
 	"os"
 	"os/exec"
+	"regexp"
 	"os/signal"
 	"strconv"
 	"sync"
 	"syscall"
 )
 
+var blanceRegex = regexp.MustCompile(`balance: (\d+\.\d+)`)
+
 type Cfg struct {
 	PactusWalletExecPath string      `json:"pactus_wallet_exec_path"`
 	WalletPath           string      `json:"wallet_path"`
@@ -30,6 +33,8 @@ func main() {
 	cfgPath := flag.String("config", "./cfg.json", "config file path")
 	password := flag.String("password", "", "pactus wallet password")
 	rpc := flag.String("server", "", "custom node rpc")
+	total := flag.Bool("total", false, "determine that all balance of account will be staked")
+
 	flag.Parse()
 
 	b, err := os.ReadFile(*cfgPath)
@@ -45,6 +50,22 @@ func main() {
 
 	amount := strconv.FormatFloat(cfg.Amount, 'g', -1, 64)
 
+	if *total {
+		args := make([]string, 0)
+		args = append(args, "address", "balance", cfg.WalletAddress)
+		out, err := exec.Command(cfg.PactusWalletExecPath, args...).Output()
+		if err != nil {
+			log.Fatalf("err: %s, msg: %s", err.Error(), string(out))
+		}
+
+		match := blanceRegex.FindStringSubmatch(string(out))
+		if len(match) > 1 {
+			amount = match[1]
+		} else {
+			log.Fatalf("err: can't find the address balance, msg: %s", string(out))
+		}
+	}
+
 	var wg sync.WaitGroup
 	ctx, cancel := context.WithCancel(context.Background())
 
@@ -53,6 +74,7 @@ func main() {
 		os.Exit(0)
 	}()
 
+
 	for _, val := range cfg.Validators {
 		args := make([]string, 0)
 		args = append(args, "--path", cfg.WalletPath, "tx", "bond")