chore: update tools converter

Author: shreyasbhat0

tools/converter.go

@@ -2,18 +2,26 @@ package main
 
 import (
 	"bufio"
+	"encoding/json"
 	"flag"
 	"fmt"
 	"log"
 	"os"
 	"strings"
 
 	"github.com/btcsuite/btcutil/bech32"
+	"github.com/cometbft/cometbft/crypto"
+	"github.com/cosmos/cosmos-sdk/types"
 )
 
+// GetModuleAddress computes the address for a given module name with a specific prefix.
+func GetModuleAddress(moduleName string) types.AccAddress {
+	return types.AccAddress(crypto.AddressHash([]byte(moduleName)))
+}
+
 // convertAddresses takes a list of addresses and converts them from oldPrefix to newPrefix.
-func convertAddresses(addressList []string, oldPrefix, newPrefix string) ([]string, error) {
-	var convertedAddresses []string
+func convertAddresses(addressList []string, oldPrefix, newPrefix string) (map[string]string, error) {
+	convertedMap := make(map[string]string)
 
 	for _, address := range addressList {
 		hrp, data, err := bech32.Decode(address)
@@ -26,18 +34,16 @@ func convertAddresses(addressList []string, oldPrefix, newPrefix string) ([]stri
 			return nil, fmt.Errorf("expected prefix '%s', but got '%s' for address '%s'", oldPrefix, hrp, address)
 		}
 
-		// Encode with the new prefix
 		mainnetAddress, err := bech32.Encode(newPrefix, data)
 		if err != nil {
 			return nil, fmt.Errorf("failed to encode address %s with new prefix %s: %w", address, newPrefix, err)
 		}
 
-		convertedAddresses = append(convertedAddresses, mainnetAddress)
+		convertedMap[address] = mainnetAddress
 	}
-	return convertedAddresses, nil
+	return convertedMap, nil
 }
 
-// readAddressesFromFile reads a file line by line and returns a slice of addresses.
 func readAddressesFromFile(filename string) ([]string, error) {
 	var addresses []string
 
@@ -59,23 +65,21 @@ func readAddressesFromFile(filename string) ([]string, error) {
 	return addresses, nil
 }
 
-// writeAddressesToFile writes a slice of addresses to a file, one per line.
-func writeAddressesToFile(filename string, addresses []string) error {
+// writeAddressesToJSONFile writes a map of old-to-new addresses to a JSON file.
+func writeAddressesToJSONFile(filename string, addressMap map[string]string) error {
 	file, err := os.Create(filename)
 	if err != nil {
 		return fmt.Errorf("failed to create file %s: %w", filename, err)
 	}
 	defer file.Close()
 
-	writer := bufio.NewWriter(file)
-	for _, address := range addresses {
-		_, err := writer.WriteString(address + "\n")
-		if err != nil {
-			return fmt.Errorf("failed to write to file %s: %w", filename, err)
-		}
+	encoder := json.NewEncoder(file)
+	encoder.SetIndent("", "  ")
+	if err := encoder.Encode(addressMap); err != nil {
+		return fmt.Errorf("failed to write JSON to file %s: %w", filename, err)
 	}
 
-	return writer.Flush()
+	return nil
 }
 
 func main() {
@@ -84,19 +88,38 @@ func main() {
 	newPrefix := flag.String("newPrefix", "", "New prefix to replace the old prefix")
 	addresses := flag.String("addresses", "", "Comma-separated list of addresses to convert")
 	addressFile := flag.String("file", "", "File containing list of addresses to convert (one per line)")
+	moduleName := flag.String("module", "", "Module name to generate the address for")
 
 	// Parse command-line arguments
 	flag.Parse()
 
-	// Check for required arguments
+	cfg := types.GetConfig()
+
+	if *newPrefix == "arkeo" {
+		cfg.SetBech32PrefixForAccount("arkeo", "arkeopub")
+	} else if *newPrefix == "tarkeo" {
+		cfg.SetBech32PrefixForAccount("tarkeo", "tarkeopub")
+	} else {
+		log.Fatalf("Unsupported module prefix '%s'. Supported prefixes are 'arkeo' and 'tarkeo'.", *newPrefix)
+	}
+
+	// Module address generation operation
+	if *moduleName != "" {
+
+		moduleAddr := GetModuleAddress(*moduleName)
+
+		fmt.Printf("Generated address for module '%s': %s\n", *moduleName, moduleAddr)
+		return
+	}
+
+	// Address conversion operation
 	if *oldPrefix == "" || *newPrefix == "" {
 		log.Fatal("Usage: go run converter.go -oldPrefix=<oldPrefix> -newPrefix=<newPrefix> -addresses=<comma-separated addresses> or -file=<filename>")
 	}
 
 	var addressList []string
 	var err error
 
-	// Determine source of addresses (file or list)
 	if *addressFile != "" {
 		addressList, err = readAddressesFromFile(*addressFile)
 		if err != nil {
@@ -108,18 +131,16 @@ func main() {
 		log.Fatal("Please provide addresses either through -addresses or -file option.")
 	}
 
-	// Convert addresses
-	convertedAddresses, err := convertAddresses(addressList, *oldPrefix, *newPrefix)
+	convertedMap, err := convertAddresses(addressList, *oldPrefix, *newPrefix)
 	if err != nil {
 		log.Fatalf("Error converting addresses: %v", err)
 	}
 
-	// Write converted addresses to file
-	outputFile := "converted.txt"
-	err = writeAddressesToFile(outputFile, convertedAddresses)
+	outputFile := "converted.json"
+	err = writeAddressesToJSONFile(outputFile, convertedMap)
 	if err != nil {
 		log.Fatalf("Error writing converted addresses to file: %v", err)
 	}
 
-	fmt.Printf("Converted addresses written to %s\n", outputFile)
+	fmt.Printf("Converted addresses written to %s", outputFile)
 }