opbot refunder first pass

contrib/opbot/botmd/botmd.go

@@ -2,10 +2,19 @@ package botmd
 
 import (
 	"context"
+	"fmt"
 	"github.com/slack-io/slacker"
 	"github.com/synapsecns/sanguine/contrib/opbot/config"
 	"github.com/synapsecns/sanguine/contrib/opbot/signoz"
+	"github.com/synapsecns/sanguine/core/dbcommon"
 	"github.com/synapsecns/sanguine/core/metrics"
+	signerConfig "github.com/synapsecns/sanguine/ethergo/signer/config"
+	"github.com/synapsecns/sanguine/ethergo/signer/signer"
+	"github.com/synapsecns/sanguine/ethergo/submitter"
+	submitterdb "github.com/synapsecns/sanguine/ethergo/submitter/db"
+	cctpSql "github.com/synapsecns/sanguine/services/cctp-relayer/db/sql"
+	omnirpcClient "github.com/synapsecns/sanguine/services/omnirpc/client"
+	"golang.org/x/sync/errgroup"
 )
 
 // Bot represents the bot server.
@@ -15,6 +24,10 @@ type Bot struct {
 	cfg           config.Config
 	signozClient  *signoz.Client
 	signozEnabled bool
+	rpcClient     omnirpcClient.RPCClient
+	signer        signer.Signer
+	submitter     submitter.TransactionSubmitter
+	db            submitterdb.SubmitterDBFactory
 }
 
 // NewBot creates a new bot server.
@@ -32,8 +45,10 @@ func NewBot(handler metrics.Handler, cfg config.Config) Bot {
 		bot.signozEnabled = true
 	}
 
+	bot.rpcClient = omnirpcClient.NewOmnirpcClient(cfg.OmniRPCURL, handler, omnirpcClient.WithCaptureReqRes())
+
 	bot.addMiddleware(bot.tracingMiddleware(), bot.metricsMiddleware())
-	bot.addCommands(bot.traceCommand(), bot.rfqLookupCommand())
+	bot.addCommands(bot.traceCommand(), bot.rfqLookupCommand(), bot.rfqRefund())
 
 	return bot
 }
@@ -53,6 +68,33 @@ func (b *Bot) addCommands(commands ...*slacker.CommandDefinition) {
 // Start starts the bot server.
 // nolint: wrapcheck
 func (b *Bot) Start(ctx context.Context) error {
+	var err error
+	b.signer, err = signerConfig.SignerFromConfig(ctx, b.cfg.Signer)
+	if err != nil {
+		return fmt.Errorf("failed to create signer: %w", err)
+	}
+
+	dbType, err := dbcommon.DBTypeFromString(b.cfg.Database.Type)
+	if err != nil {
+		return fmt.Errorf("could not get db type: %w", err)
+	}
+
+	store, err := cctpSql.Connect(ctx, dbType, b.cfg.Database.DSN, b.handler)
+	if err != nil {
+		return fmt.Errorf("could not connect to database: %w", err)
+	}
+
+	b.submitter = submitter.NewTransactionSubmitter(b.handler, b.signer, b.rpcClient, store.SubmitterDB(), &b.cfg.SubmitterConfig)
+
+	g, ctx := errgroup.WithContext(ctx)
+	g.Go(func() error {
+		return b.submitter.Start(ctx)
+	})
+
+	g.Go(func() error {
+		return b.server.Listen(ctx)
+	})
+
 	// nolint: wrapcheck
-	return b.server.Listen(ctx)
+	return g.Wait()
 }

contrib/opbot/botmd/commands.go

@@ -4,15 +4,21 @@ package botmd
 
 import (
 	"fmt"
+	"github.com/ethereum/go-ethereum/accounts/abi/bind"
 	"github.com/ethereum/go-ethereum/common"
+	"github.com/ethereum/go-ethereum/core/types"
 	"github.com/hako/durafmt"
 	"github.com/slack-go/slack"
 	"github.com/slack-io/slacker"
 	"github.com/synapsecns/sanguine/contrib/opbot/signoz"
 	"github.com/synapsecns/sanguine/ethergo/chaindata"
+	rfqClient "github.com/synapsecns/sanguine/services/rfq/api/client"
+	"github.com/synapsecns/sanguine/services/rfq/contracts/fastbridge"
 	"github.com/synapsecns/sanguine/services/rfq/relayer/relapi"
 	"log"
+	"math/big"
 	"regexp"
+	"strconv"
 	"strings"
 	"sync"
 	"time"
@@ -239,6 +245,80 @@ func (b *Bot) rfqLookupCommand() *slacker.CommandDefinition {
 		}}
 }
 
+func (b *Bot) rfqRefund() *slacker.CommandDefinition {
+	return &slacker.CommandDefinition{
+		Command:     "refund <tx> <origin_chainid>",
+		Description: "refund a quote request",
+		Examples:    []string{"refund 0x1234"},
+		Handler: func(ctx *slacker.CommandContext) {
+			client, err := rfqClient.NewUnauthenticatedClient(b.handler, b.cfg.RFQApiURL)
+			if err != nil {
+				log.Printf("error creating rfq client: %v\n", err)
+				return
+			}
+
+			contracts, err := client.GetRFQContracts(ctx.Context())
+			if err != nil {
+				log.Printf("error fetching rfq contracts: %v\n", err)
+				return
+			}
+
+			tx := stripLinks(ctx.Request().Param("tx"))
+			originChainIDStr := ctx.Request().Param("origin_chainid")
+
+			originChainID, err := strconv.Atoi(originChainIDStr)
+			if err != nil {
+				_, err := ctx.Response().Reply("origin_chainid must be a number")
+				if err != nil {
+					log.Println(err)
+				}
+				return
+			}
+
+			chainClient, err := b.rpcClient.GetChainClient(ctx.Context(), originChainID)
+			if err != nil {
+				log.Printf("error getting chain client: %v\n", err)
+				return
+			}
+
+			contractAddress, ok := contracts.Contracts[uint32(originChainID)]
+			if !ok {
+				_, err := ctx.Response().Reply("contract address not found")
+				if err != nil {
+					log.Println(err)
+				}
+				return
+			}
+
+			fastBridgeHandle, err := fastbridge.NewFastBridge(common.HexToAddress(contractAddress), chainClient)
+			if err != nil {
+				log.Printf("error creating fast bridge: %v\n", err)
+				return
+			}
+
+			for _, relayer := range b.cfg.RelayerURLS {
+				relClient := relapi.NewRelayerClient(b.handler, relayer)
+				qr, err := relClient.GetQuoteRequestByTXID(ctx.Context(), tx)
+				if err != nil {
+					log.Printf("error fetching quote request: %v\n", err)
+					continue
+				}
+
+				nonce, err := b.submitter.SubmitTransaction(ctx.Context(), big.NewInt(int64(originChainID)), func(transactor *bind.TransactOpts) (tx *types.Transaction, err error) {
+					return fastBridgeHandle.Refund(transactor, common.Hex2Bytes(qr.QuoteRequestRaw))
+				})
+				if err != nil {
+					log.Printf("error submitting refund: %v\n", err)
+					continue
+				}
+
+				_, err = ctx.Response().Reply(fmt.Sprintf("refund submitted with nonce %d", nonce))
+				return
+			}
+		},
+	}
+}
+
 func toExplorerSlackLink(ogHash string) string {
 	rfqHash := strings.ToUpper(ogHash)
 	// cut off 0x

contrib/opbot/config/config.go

@@ -1,6 +1,11 @@
 // Package config provides a simple way to read and write configuration files.
 package config
 
+import (
+	"github.com/synapsecns/sanguine/ethergo/signer/config"
+	submitterConfig "github.com/synapsecns/sanguine/ethergo/submitter/config"
+)
+
 // Config represents the configuration of the application.
 type Config struct {
 	// SlackBotToken is the token of the slack bot.
@@ -18,4 +23,20 @@ type Config struct {
 	SignozBaseURL string `yaml:"signoz_base_url"`
 	// RelayerURLS is the list of RFQ relayer URLs.
 	RelayerURLS []string `yaml:"rfq_relayer_urls"`
+	// RFQApiURL is the URL of the RFQ API.
+	RFQApiURL string `yaml:"rfq_api_url"`
+	// OmniRPCURL is the URL of the Omni RPC.
+	OmniRPCURL string `yaml:"omni_rpc_url"`
+	// Signer is the signer config.
+	Signer config.SignerConfig `yaml:"signer"`
+	// SubmitterConfig is the submitter config.
+	SubmitterConfig submitterConfig.Config `yaml:"submitter_config"`
+	// Database is the database config.
+	Database DatabaseConfig `yaml:"database"`
+}
+
+// DatabaseConfig represents the configuration for the database.
+type DatabaseConfig struct {
+	Type string `yaml:"type"`
+	DSN  string `yaml:"dsn"` // Data Source Name
 }

contrib/opbot/go.mod

@@ -271,7 +271,6 @@ replace (
 	// later versions give errors on uint64 being too high.
 	github.com/brianvoe/gofakeit/v6 => github.com/brianvoe/gofakeit/v6 v6.9.0
 	github.com/gogo/protobuf => github.com/regen-network/protobuf v1.3.3-alpha.regen.1
-	github.com/slack-go/slack => github.com/slack-go/slack v0.12.2
 	github.com/slack-io/slacker => github.com/slack-io/slacker v0.1.1-0.20240701203341-bd3ee211e9d2
 	github.com/synapsecns/sanguine/core => ./../../core
 	github.com/synapsecns/sanguine/ethergo => ./../../ethergo

contrib/opbot/go.sum

@@ -990,8 +990,8 @@ github.com/sirupsen/logrus v1.9.3 h1:dueUQJ1C2q9oE3F7wvmSGAaVtTmUizReu6fjN8uqzbQ
 github.com/sirupsen/logrus v1.9.3/go.mod h1:naHLuLoDiP4jHNo9R0sCBMtWGeIprob74mVsIT4qYEQ=
 github.com/skeema/knownhosts v1.2.2 h1:Iug2P4fLmDw9f41PB6thxUkNUkJzB5i+1/exaj40L3A=
 github.com/skeema/knownhosts v1.2.2/go.mod h1:xYbVRSPxqBZFrdmDyMmsOs+uX1UZC3nTN3ThzgDxUwo=
-github.com/slack-go/slack v0.12.2 h1:x3OppyMyGIbbiyFhsBmpf9pwkUzMhthJMRNmNlA4LaQ=
-github.com/slack-go/slack v0.12.2/go.mod h1:hlGi5oXA+Gt+yWTPP0plCdRKmjsDxecdHxYQdlMQKOw=
+github.com/slack-go/slack v0.13.0 h1:7my/pR2ubZJ9912p9FtvALYpbt0cQPAqkRy2jaSI1PQ=
+github.com/slack-go/slack v0.13.0/go.mod h1:hlGi5oXA+Gt+yWTPP0plCdRKmjsDxecdHxYQdlMQKOw=
 github.com/slack-io/commander v0.0.0-20231120025847-9fd78b4b2d54 h1:aRc+G2mUb697z6bR09Roq6kP08suJulgNo00SGhAsfM=
 github.com/slack-io/commander v0.0.0-20231120025847-9fd78b4b2d54/go.mod h1:aHmXZnL/ELKlfMybblXnnCl+IeHQ+gMb++DT7ujIGlA=
 github.com/slack-io/proper v0.0.0-20231119200853-f78ba4fc878f h1:wiEJBKJKvMOeE9KtLV7iwtHOsNXDBZloL+FPlkZ6vnA=

contrib/opbot/sql/base/base.go

@@ -0,0 +1,38 @@
+package base
+
+import (
+	"github.com/synapsecns/sanguine/core/metrics"
+	submitterDB "github.com/synapsecns/sanguine/ethergo/submitter/db"
+	"github.com/synapsecns/sanguine/ethergo/submitter/db/txdb"
+	"gorm.io/gorm"
+)
+
+// Store is a store that implements an underlying gorm db.
+type Store struct {
+	db             *gorm.DB
+	metrics        metrics.Handler
+	submitterStore submitterDB.Service
+}
+
+// NewStore creates a new store.
+func NewStore(db *gorm.DB, metrics metrics.Handler) *Store {
+	txDB := txdb.NewTXStore(db, metrics)
+	return &Store{db: db, metrics: metrics, submitterStore: txDB}
+}
+
+// DB gets the database object for mutation outside of the lib.
+func (s Store) DB() *gorm.DB {
+	return s.db
+}
+
+// SubmitterDB gets the submitter database object for mutation outside of the lib.
+func (s Store) SubmitterDB() submitterDB.Service {
+	return s.submitterStore
+}
+
+// GetAllModels gets all models to migrate.
+// see: https://medium.com/@SaifAbid/slice-interfaces-8c78f8b6345d for an explanation of why we can't do this at initialization time
+func GetAllModels() (allModels []interface{}) {
+	allModels = append(allModels, txdb.GetAllModels()...)
+	return allModels
+}

contrib/opbot/sql/base/doc.go

@@ -0,0 +1,2 @@
+// Package base contains the base sql implementation
+package base

contrib/opbot/sql/doc.go

@@ -0,0 +1,2 @@
+// Package sql provides a common interface for starting sql-lite databases
+package sql

contrib/opbot/sql/mysql/doc.go

@@ -0,0 +1,2 @@
+// Package mysql  contains a mysql db
+package mysql

contrib/opbot/sql/mysql/store.go

@@ -0,0 +1,66 @@
+package mysql
+
+import (
+	"context"
+	"fmt"
+	"github.com/synapsecns/sanguine/contrib/opbot/sql/base"
+	submitterDB "github.com/synapsecns/sanguine/ethergo/submitter/db"
+	"time"
+
+	"github.com/ipfs/go-log"
+	common_base "github.com/synapsecns/sanguine/core/dbcommon"
+	"github.com/synapsecns/sanguine/core/metrics"
+	"gorm.io/driver/mysql"
+	"gorm.io/gorm"
+	"gorm.io/gorm/schema"
+)
+
+// Logger is the mysql logger.
+var logger = log.Logger("synapse-mysql")
+
+// NewMysqlStore creates a new mysql store for a given data store.
+func NewMysqlStore(ctx context.Context, dbURL string, handler metrics.Handler) (*Store, error) {
+	logger.Debug("create mysql store")
+
+	gdb, err := gorm.Open(mysql.Open(dbURL), &gorm.Config{
+		Logger:               common_base.GetGormLogger(logger),
+		FullSaveAssociations: true,
+		NamingStrategy:       NamingStrategy,
+		NowFunc:              time.Now,
+	})
+
+	if err != nil {
+		return nil, fmt.Errorf("could not create mysql connection: %w", err)
+	}
+
+	sqlDB, err := gdb.DB()
+	if err != nil {
+		return nil, fmt.Errorf("could not get sql db: %w", err)
+	}
+
+	// fixes a timeout issue https://stackoverflow.com/a/42146536
+	sqlDB.SetMaxIdleConns(MaxIdleConns)
+	sqlDB.SetConnMaxLifetime(time.Hour)
+
+	handler.AddGormCallbacks(gdb)
+
+	err = gdb.WithContext(ctx).AutoMigrate(base.GetAllModels()...)
+	if err != nil {
+		return nil, fmt.Errorf("could not migrate on mysql: %w", err)
+	}
+
+	return &Store{base.NewStore(gdb, handler)}, nil
+}
+
+// Store is the mysql store. It extends the bsae store for mysql queries.
+type Store struct {
+	*base.Store
+}
+
+// MaxIdleConns is exported here for testing. Tests execute too slowly with a reconnect each time.
+var MaxIdleConns = 10
+
+// NamingStrategy is for table prefixes.
+var NamingStrategy = schema.NamingStrategy{}
+
+var _ submitterDB.SubmitterDBFactory = &Store{}

contrib/opbot/sql/sqlite/doc.go

@@ -0,0 +1,2 @@
+// Package sqlite implements the sqlite package
+package sqlite