update to arkd v0.7.0 + arkd setup on startup (#225)

Author: louisinger

cmd/nigiri/faucet.go

@@ -6,6 +6,7 @@ import (
 	"errors"
 	"fmt"
 	"io/ioutil"
+	"math"
 	"net/http"
 	"os/exec"
 	"path/filepath"
@@ -24,6 +25,7 @@ var faucet = cli.Command{
 	Action:    faucetAction,
 	Flags: []cli.Flag{
 		&liquidFlag,
+		&arkFlag,
 	},
 }
 
@@ -59,6 +61,19 @@ func faucetAction(ctx *cli.Context) error {
 		return err
 	}
 
+	isArk := ctx.Bool("ark")
+	if isArk {
+		amount := 1.0
+		if ctx.Args().Len() >= 2 {
+			amount, err = strconv.ParseFloat(ctx.Args().Get(1), 64)
+			if err != nil {
+				return fmt.Errorf("invalid amount: %v", err)
+			}
+		}
+
+		return faucetArk(ctx.Args().First(), amount)
+	}
+
 	address := ctx.Args().First()
 	if address == "cln" {
 		jsonOut, err := outputCommand("docker", "exec", "cln", "lightning-cli", "--network="+network, "newaddr")
@@ -146,3 +161,65 @@ func outputCommand(name string, arg ...string) ([]byte, error) {
 	}
 	return b, nil
 }
+
+func faucetArk(address string, amount float64) error {
+	amount = math.Round(amount * 100000000) // convert to satoshis
+	bashCmd := exec.Command("docker", "exec", "-t", "ark", "ark", "balance")
+	output, err := bashCmd.Output()
+	if err != nil {
+		return fmt.Errorf("failed to get balance: %w", err)
+	}
+
+	// Parse the JSON response to extract offchain balance
+	var balanceData map[string]interface{}
+	if err := json.Unmarshal(output, &balanceData); err != nil {
+		return fmt.Errorf("failed to parse balance JSON: %w", err)
+	}
+
+	// Extract offchain balance
+	offchainBalance, ok := balanceData["offchain_balance"].(map[string]interface{})
+	if !ok {
+		return fmt.Errorf("failed to get offchain_balance from response")
+	}
+
+	total, ok := offchainBalance["total"].(float64)
+	if !ok {
+		return fmt.Errorf("failed to get total from offchain_balance")
+	}
+
+	if total < amount {
+		bashCmd = exec.Command("docker", "exec", "-t", "ark", "arkd", "wallet", "address")
+		output, err := bashCmd.Output()
+		if err != nil {
+			return fmt.Errorf("failed to get wallet address: %w", err)
+		}
+		address := strings.TrimSpace(string(output))
+
+		// Fund the address using nigiri faucet
+		bashCmd = exec.Command("nigiri", "faucet", address)
+		if err := bashCmd.Run(); err != nil {
+			return fmt.Errorf("failed to fund wallet address: %w", err)
+		}
+
+		bashCmd = exec.Command("docker", "exec", "-t", "ark", "arkd", "note", "--amount", fmt.Sprintf("%d", int64(amount)))
+		output, err = bashCmd.Output()
+		if err != nil {
+			return fmt.Errorf("failed to create arkd note: %w", err)
+		}
+		noteStr := strings.TrimSpace(string(output))
+
+		bashCmd = exec.Command("docker", "exec", "-t", "ark", "ark", "redeem-notes", "-n", noteStr, "--password", "secret")
+		if err := bashCmd.Run(); err != nil {
+			return fmt.Errorf("failed to redeem note: %w", err)
+		}
+	}
+
+	bashCmd = exec.Command("docker", "exec", "-t", "ark", "ark", "send", "--to", address, "--amount", fmt.Sprintf("%d", int64(amount)), "--password", "secret")
+	output, err = bashCmd.Output()
+	if err != nil {
+		return fmt.Errorf("failed to send ark: %w", err)
+	}
+	fmt.Println(string(output))
+
+	return nil
+}

cmd/nigiri/resources/docker-compose.yml

@@ -234,29 +234,45 @@ services:
       - ./volumes/lightningd:/.lightning
       - ./volumes/bitcoin:/etc/bitcoin
 
+  ark-wallet:
+    container_name: ark-wallet
+    image: ghcr.io/arkade-os/arkd-wallet:v0.7.0
+    depends_on:
+      - bitcoin
+    environment:
+      ARKD_WALLET_LOG_LEVEL: "5"
+      ARKD_WALLET_BITCOIND_RPC_USER: admin1
+      ARKD_WALLET_BITCOIND_RPC_PASS: 123
+      ARKD_WALLET_BITCOIND_RPC_HOST: bitcoin:18443
+      ARKD_WALLET_DATADIR: "./data/regtest"
+      ARKD_WALLET_NETWORK: "regtest"
+    volumes:
+      - ./volumes/ark/wallet:/app/data
+    ports:
+      - "6060:6060"
+
   ark:
     container_name: ark
-    image: ghcr.io/arkade-os/ark:v0.6.2
-    user: "${UID:-1000}:${GID:-1000}"
+    image: ghcr.io/arkade-os/arkd:v0.7.0
     depends_on:
       - bitcoin
+      - ark-wallet
     environment:
-      ARK_ROUND_INTERVAL: "15"
-      ARK_LOG_LEVEL: "5"
-      ARK_NETWORK: "regtest"
-      ARK_PORT: "7070"
-      ARK_NO_TLS: "true"
-      ARK_NO_MACAROONS: "true"
-      ARK_TX_BUILDER_TYPE: "covenantless"
-      ARK_BITCOIND_RPC_USER: "admin1"
-      ARK_BITCOIND_RPC_PASS: "123"
-      ARK_BITCOIND_RPC_HOST: "bitcoin:18443"
-      ARK_BITCOIND_ZMQ_BLOCK: "tcp://bitcoin:28332"
-      ARK_BITCOIND_ZMQ_TX: "tcp://bitcoin:28333"
-      ARK_DATADIR: "/app/data"
+      ARKD_LOG_LEVEL: "5"
+      ARKD_NO_MACAROONS: "true"
+      ARKD_VTXO_TREE_EXPIRY: "100" # blocks
+      ARKD_SCHEDULER_TYPE: "block"
+      ARKD_UNILATERAL_EXIT_DELAY: "512"
+      ARKD_BOARDING_EXIT_DELAY: "1024"
+      ARKD_DATADIR: "./data/regtest"
+      ARKD_WALLET_ADDR: "ark-wallet:6060"
+      ARKD_ESPLORA_URL: "http://chopsticks:3000"
+      ARKD_VTXO_MIN_AMOUNT: "1"
+      ARKD_LIVE_STORE_TYPE: "inmemory"
+      ARKD_EVENT_DB_TYPE: "badger"
+      ARKD_DB_TYPE: "sqlite"
     volumes:
       - ./volumes/ark/data:/app/data
-      - ./volumes/ark/wallet:/app/wallet-data
     ports:
       - "7070:7070"
     restart: unless-stopped

cmd/nigiri/start.go

@@ -5,9 +5,11 @@ import (
 	"errors"
 	"fmt"
 	"os"
+	"os/exec"
 	"path/filepath"
 	"strconv"
 	"strings"
+	"time"
 
 	"github.com/logrusorgru/aurora"
 	"github.com/urfave/cli/v2"
@@ -141,7 +143,7 @@ func startAction(ctx *cli.Context) error {
 	}
 
 	if effectiveFlags.Ark {
-		services = append(services, "ark")
+		services = append(services, "ark", "ark-wallet")
 	}
 
 	bashCmd := runDockerCompose(composePath, append([]string{"up", "-d"}, services...)...)
@@ -195,5 +197,72 @@ func startAction(ctx *cli.Context) error {
 		)
 	}
 
+	if effectiveFlags.Ark {
+		done := make(chan bool)
+		go spinner(done, "setting up arkd...")
+
+		time.Sleep(4 * time.Second) // give time for the container to start
+		if err := setupArk(); err != nil {
+			return fmt.Errorf("failed to setup Ark: %w", err)
+		}
+
+		done <- true
+		fmt.Println("✓ arkd setup completed successfully!")
+	}
+
+	return nil
+}
+
+func spinner(done chan bool, message string) {
+	frames := []string{"⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"}
+	i := 0
+	for {
+		select {
+		case <-done:
+			fmt.Printf("\r%s\r", strings.Repeat(" ", len(message)+3))
+			return
+		default:
+			fmt.Printf("\r%s %s", frames[i], message)
+			time.Sleep(100 * time.Millisecond)
+			i = (i + 1) % len(frames)
+		}
+	}
+}
+
+func setupArk() error {
+	bashCmd := exec.Command("docker", "exec", "-t", "ark", "arkd", "wallet", "create", "--password", "secret")
+	bashCmd.Run()
+
+	bashCmd = exec.Command("docker", "exec", "-t", "ark", "arkd", "wallet", "unlock", "--password", "secret")
+	if err := bashCmd.Run(); err != nil {
+		return fmt.Errorf("failed to unlock wallet: %w", err)
+	}
+
+	bashCmd = exec.Command("docker", "exec", "-t", "ark", "arkd", "wallet", "status")
+	if err := bashCmd.Run(); err != nil {
+		return fmt.Errorf("failed to check wallet status: %w", err)
+	}
+
+	time.Sleep(10 * time.Second)
+
+	bashCmd = exec.Command("docker", "exec", "-t", "ark", "ark", "init", "--network", "regtest", "--password", "secret", "--server-url", "localhost:7070", "--explorer", "http://chopsticks:3000")
+	bashCmd.Run() // Ignore error as wallet might already exist
+
+	// faucet arkd wallet
+	bashCmd = exec.Command("docker", "exec", "-t", "ark", "arkd", "wallet", "address")
+	output, err := bashCmd.Output()
+	if err != nil {
+		return fmt.Errorf("failed to get wallet address: %w", err)
+	}
+	address := strings.TrimSpace(string(output))
+
+	// Fund the address using nigiri faucet
+	for i := 0; i < 10; i++ {
+		bashCmd = exec.Command("nigiri", "faucet", address)
+		if err := bashCmd.Run(); err != nil {
+			return fmt.Errorf("failed to fund wallet address: %w", err)
+		}
+	}
+
 	return nil
 }