Relayer: CLI tool

cmd/native-relayer/cli/config.go

@@ -0,0 +1,26 @@
+package cli
+
+import "time"
+
+// Configuration struct
+type Config struct {
+	NativeRPC                string        `mapstructure:"native_rpc"`
+	NativeGRPC               string        `mapstructure:"native_grpc"`
+	StartBlockHeight         int           `mapstructure:"start_block_height"`
+	IkaRPC                   string        `mapstructure:"ika_rpc"`
+	IkaSignerMnemonic        string        `mapstructure:"ika_signer_mnemonic"`
+	IkaNativeLcPackage       string        `mapstructure:"ika_native_lc_package"`
+	IkaNativeLcModule        string        `mapstructure:"ika_native_lc_module"`
+	IkaNativeLcFunction      string        `mapstructure:"ika_native_lc_function"`
+	IkaGasAcc                string        `mapstructure:"ika_gas_acc"`
+	IkaGasBudget             string        `mapstructure:"ika_gas_budget"`
+	BtcRPCHost               string        `mapstructure:"btc_rpc_host"`
+	BtcRPCUser               string        `mapstructure:"btc_rpc_user"`
+	BtcRPCPass               string        `mapstructure:"btc_rpc_pass"`
+	BtcConfirmationThreshold uint8         `mapstructure:"btc_confirmation_threshold"`
+	HTTPPostMode             bool          `mapstructure:"http_post_mode"`
+	DisableTLS               bool          `mapstructure:"disable_tls"`
+	ProcessTxsInterval       time.Duration `mapstructure:"process_txs_interval"`
+	ConfirmTxsInterval       time.Duration `mapstructure:"confirm_txs_interval"`
+	DBFile                   string        `mapstructure:"db_file"`
+}

cmd/native-relayer/cli/root.go

@@ -0,0 +1,35 @@
+package cli
+
+import (
+	"os"
+	"sync"
+
+	"github.com/gonative-cc/relayer/nbtc"
+	"github.com/rs/zerolog/log"
+	"github.com/spf13/cobra"
+)
+
+var (
+	verbose bool // Flag for verbose mode
+
+	rootCmd = &cobra.Command{
+		Use:   "relayer-cli",
+		Short: "CLI tool for managing the relayer",
+	}
+
+	relayerInstance *nbtc.Relayer
+	relayerWg       sync.WaitGroup
+)
+
+func Execute() {
+	if err := rootCmd.Execute(); err != nil {
+		log.Error().Err(err).Msg("CLI execution failed")
+		os.Exit(1)
+	}
+}
+
+func init() {
+	rootCmd.PersistentFlags().BoolVarP(&verbose, "verbose", "v", false, "Enable verbose logging")
+	rootCmd.AddCommand(startCmd)
+	rootCmd.AddCommand(stopCmd)
+}

cmd/native-relayer/cli/start.go

@@ -0,0 +1,129 @@
+package cli
+
+import (
+	"context"
+	"os"
+	"os/signal"
+	"syscall"
+
+	"github.com/block-vision/sui-go-sdk/signer"
+	"github.com/block-vision/sui-go-sdk/sui"
+	"github.com/btcsuite/btcd/rpcclient"
+	"github.com/gonative-cc/relayer/bitcoin"
+	"github.com/gonative-cc/relayer/dal"
+	"github.com/gonative-cc/relayer/ika"
+	"github.com/gonative-cc/relayer/ika2btc"
+	"github.com/gonative-cc/relayer/native2ika"
+	"github.com/gonative-cc/relayer/nbtc"
+	"github.com/rs/zerolog/log"
+	"github.com/spf13/cobra"
+)
+
+var startCmd = &cobra.Command{
+	Use:   "start",
+	Short: "Starts the relayer",
+	Run: func(cmd *cobra.Command, args []string) {
+		config, err := loadConfig()
+		if err != nil {
+			log.Error().Err(err).Msg("Error loading config")
+			os.Exit(1)
+		}
+
+		// TODO:Set up logging based on verbose flag
+
+		db, err := dal.NewDB(config.DBFile)
+		if err != nil {
+			log.Error().Err(err).Msg("Error creating database")
+			os.Exit(1)
+		}
+		if err := db.InitDB(); err != nil {
+			log.Error().Err(err).Msg("Error initializing database")
+			os.Exit(1)
+		}
+
+		suiClient := sui.NewSuiClient(config.IkaRPC).(*sui.Client)
+		if suiClient == nil {
+			log.Error().Err(err).Msg("Error creating Sui client")
+			os.Exit(1)
+		}
+		ikaSigner, err := signer.NewSignertWithMnemonic(config.IkaSignerMnemonic)
+		if err != nil {
+			log.Error().Err(err).Msg("Error creating signer with mnemonic")
+			os.Exit(1)
+		}
+
+		ikaClient, err := ika.NewClient(
+			suiClient,
+			ikaSigner,
+			ika.SuiCtrCall{
+				Package:  config.IkaNativeLcPackage,
+				Module:   config.IkaNativeLcModule,
+				Function: config.IkaNativeLcFunction,
+			},
+			ika.SuiCtrCall{
+				Package:  config.IkaNativeLcPackage,
+				Module:   config.IkaNativeLcModule,
+				Function: config.IkaNativeLcFunction,
+			},
+			config.IkaGasAcc,
+			config.IkaGasBudget,
+		)
+		if err != nil {
+			log.Error().Err(err).Msg("Error creating IKA client")
+			os.Exit(1)
+		}
+
+		btcProcessor, err := ika2btc.NewProcessor(
+			rpcclient.ConnConfig{
+				Host:         config.BtcRPCHost,
+				User:         config.BtcRPCUser,
+				Pass:         config.BtcRPCPass,
+				HTTPPostMode: config.HTTPPostMode,
+				DisableTLS:   config.DisableTLS,
+			},
+			config.BtcConfirmationThreshold,
+			db,
+		)
+		if err != nil {
+			log.Error().Err(err).Msg("Error creating Bitcoin processor")
+			os.Exit(1)
+		}
+		btcProcessor.BtcClient = &bitcoin.MockClient{}
+
+		nativeProcessor := native2ika.NewProcessor(ikaClient, db)
+
+		relayer, err := nbtc.NewRelayer(
+			nbtc.RelayerConfig{
+				ProcessTxsInterval:    config.ProcessTxsInterval,
+				ConfirmTxsInterval:    config.ConfirmTxsInterval,
+				ConfirmationThreshold: config.BtcConfirmationThreshold,
+			},
+			db,
+			nativeProcessor,
+			btcProcessor,
+		)
+		if err != nil {
+			log.Error().Err(err).Msg("Error creating relayer")
+			os.Exit(1)
+		}
+
+		relayerWg.Add(1)
+		go func() {
+			defer relayerWg.Done()
+			if err := relayer.Start(context.Background()); err != nil {
+				log.Error().Err(err).Msg("Relayer encountered an error")
+			}
+		}()
+
+		relayerInstance = relayer
+
+		c := make(chan os.Signal, 1)
+		signal.Notify(c, os.Interrupt, syscall.SIGTERM)
+
+		<-c
+		log.Info().Msg("Stopping the relayer...")
+		relayerInstance.Stop()
+		relayerWg.Wait()
+		log.Info().Msg("Relayer stopped.")
+	},
+}

cmd/native-relayer/cli/stop.go

@@ -0,0 +1,23 @@
+package cli
+
+import (
+	"github.com/rs/zerolog/log"
+	"github.com/spf13/cobra"
+)
+
+var stopCmd = &cobra.Command{
+	Use:   "stop",
+	Short: "Stops the relayer",
+	Run: func(cmd *cobra.Command, args []string) {
+		if relayerInstance == nil {
+			log.Info().Msg("Relayer is not running.")
+			return
+		}
+		log.Info().Msg("Stopping the relayer...")
+		relayerInstance.Stop()
+		relayerWg.Wait()
+		relayerInstance = nil
+		log.Info().Msg("Relayer stopped.")
+
+	},
+}

cmd/native-relayer/cli/utils.go

@@ -0,0 +1,40 @@
+package cli
+
+import (
+	"flag"
+	"fmt"
+
+	"github.com/rs/zerolog/log"
+	"github.com/spf13/viper"
+)
+
+func loadConfig() (*Config, error) {
+	var configFile string
+	flag.StringVar(&configFile, "config", "", "Path to the config file")
+	flag.Parse()
+
+	viper.SetConfigName("config")
+	viper.SetConfigType("yaml")
+	viper.AddConfigPath(".")
+	viper.AddConfigPath("$HOME/.native-relayer")
+
+	if configFile != "" {
+		viper.SetConfigFile(configFile)
+	}
+
+	if err := viper.ReadInConfig(); err != nil {
+		if _, ok := err.(viper.ConfigFileNotFoundError); !ok {
+			return nil, fmt.Errorf("error reading config file: %w", err)
+		}
+	}
+
+	var config Config
+	if err := viper.Unmarshal(&config); err != nil {
+		return nil, fmt.Errorf("error unmarshalling config: %w", err)
+	}
+
+	log.Info().Msg("Loaded Configuration:")
+	log.Info().Interface("config", config).Msg("")
+
+	return &config, nil
+}

cmd/native-relayer/config.yaml

@@ -0,0 +1,29 @@
+# Relayer Configuration
+
+# Native
+native_rpc: "127.0.0.1:26657"  
+native_grpc: "127.0.0.1:9090"       
+
+# IKA
+ika_rpc: "127.0.0.1:0001"          
+ika_signer_mnemonic: "" 
+ika_native_lc_package: ""                
+ika_native_lc_module: ""               
+ika_native_lc_function: ""  
+ika_gas_acc: ""                            
+ika_gas_budget: ""                         
+
+# Bitcoin
+btc_rpc_host: "your-bitcoin-rpc.com"           
+btc_rpc_user: "your-bitcoin-rpc-user"          
+btc_rpc_pass: "your-bitcoin-rpc-password"      
+btc_confirmation_threshold: 6  
+http_post_mode: true
+disable_tls: false           
+
+# Relayer intervals in seconds
+process_txs_interval: 5s                      
+confirm_txs_interval: 7s                   
+
+# Database
+db_file: "relayer.db"     

cmd/native-relayer/main.go

@@ -0,0 +1,7 @@
+package main
+
+import "github.com/gonative-cc/relayer/cmd/native-relayer/cli"
+
+func main() {
+	cli.Execute()
+}