btcctl: add regtest mode to btcctl

cmd/btcctl/config.go

@@ -14,6 +14,7 @@ import (
 	"strings"
 
 	"github.com/btcsuite/btcd/btcjson"
+	"github.com/btcsuite/btcd/chaincfg"
 	"github.com/btcsuite/btcutil"
 	flags "github.com/jessevdk/go-flags"
 )
@@ -98,6 +99,7 @@ type config struct {
 	Proxy          string `long:"proxy" description:"Connect via SOCKS5 proxy (eg. 127.0.0.1:9050)"`
 	ProxyPass      string `long:"proxypass" default-mask:"-" description:"Password for proxy server"`
 	ProxyUser      string `long:"proxyuser" description:"Username for proxy server"`
+	RegressionTest bool   `long:"regtest" description:"Connect to the regression test network"`
 	RPCCert        string `short:"c" long:"rpccert" description:"RPC server certificate chain for validation"`
 	RPCPassword    string `short:"P" long:"rpcpass" default-mask:"-" description:"RPC password"`
 	RPCServer      string `short:"s" long:"rpcserver" description:"RPC server to connect to"`
@@ -111,23 +113,31 @@ type config struct {
 
 // normalizeAddress returns addr with the passed default port appended if
 // there is not already a port specified.
-func normalizeAddress(addr string, useTestNet3, useSimNet, useWallet bool) string {
+func normalizeAddress(addr string, chain *chaincfg.Params, useWallet bool) (string, error) {
 	_, _, err := net.SplitHostPort(addr)
 	if err != nil {
 		var defaultPort string
-		switch {
+		switch chain {
-		case useTestNet3:
+		case &chaincfg.TestNet3Params:
 			if useWallet {
 				defaultPort = "18332"
 			} else {
 				defaultPort = "18334"
 			}
-		case useSimNet:
+		case &chaincfg.SimNetParams:
 			if useWallet {
 				defaultPort = "18554"
 			} else {
 				defaultPort = "18556"
 			}
+		case &chaincfg.RegressionNetParams:
+			if useWallet {
+				// TODO: add port once regtest is supported in btcwallet
+				paramErr := fmt.Errorf("cannot use -wallet with -regtest, btcwallet not yet compatible with regtest")
+				return "", paramErr
+			} else {
+				defaultPort = "18334"
+			}
 		default:
 			if useWallet {
 				defaultPort = "8332"
@@ -136,9 +146,9 @@ func normalizeAddress(addr string, useTestNet3, useSimNet, useWallet bool) strin
 			}
 		}
 
-		return net.JoinHostPort(addr, defaultPort)
+		return net.JoinHostPort(addr, defaultPort), nil
 	}
-	return addr
+	return addr, nil
 }
 
 // cleanAndExpandPath expands environement variables and leading ~ in the
@@ -246,17 +256,27 @@ func loadConfig() (*config, []string, error) {
 		return nil, nil, err
 	}
 
+	// default network is mainnet
+	network := &chaincfg.MainNetParams
+
 	// Multiple networks can't be selected simultaneously.
 	numNets := 0
 	if cfg.TestNet3 {
 		numNets++
+		network = &chaincfg.TestNet3Params
 	}
 	if cfg.SimNet {
 		numNets++
+		network = &chaincfg.SimNetParams
 	}
+	if cfg.RegressionTest {
+		numNets++
+		network = &chaincfg.RegressionNetParams
+	}
+
 	if numNets > 1 {
-		str := "%s: The testnet and simnet params can't be used " +
+		str := "%s: Multiple network params can't be used " +
-			"together -- choose one of the two"
+			"together -- choose one"
 		err := fmt.Errorf(str, "loadConfig")
 		fmt.Fprintln(os.Stderr, err)
 		return nil, nil, err
@@ -273,8 +293,10 @@ func loadConfig() (*config, []string, error) {
 
 	// Add default port to RPC server based on --testnet and --wallet flags
 	// if needed.
-	cfg.RPCServer = normalizeAddress(cfg.RPCServer, cfg.TestNet3,
+	cfg.RPCServer, err = normalizeAddress(cfg.RPCServer, network, cfg.Wallet)
-		cfg.SimNet, cfg.Wallet)
+	if err != nil {
+		return nil, nil, err
+	}
 
 	return &cfg, remainingArgs, nil
 }