feat(eventindexer): introduce disperser (#16694)

Author: cyberhorsey

packages/eventindexer/SgxVerifier.json

@@ -5,9 +5,9 @@ if [ ! -d "../protocol/out" ]; then
     exit 1
 fi
 
-paths=("TaikoL1.sol" "Bridge.sol" "AssignmentHook.sol")
+paths=("TaikoL1.sol" "Bridge.sol" "AssignmentHook.sol" "SgxVerifier.sol" "TaikoToken.sol")
 
-names=("TaikoL1" "Bridge" "AssignmentHook")
+names=("TaikoL1" "Bridge" "AssignmentHook" "SgxVerifier" "TaikoToken")
 
 
 for (( i = 0; i < ${#paths[@]}; ++i ));

packages/eventindexer/TaikoToken.json

@@ -8,6 +8,7 @@ var (
 	commonCategory    = "COMMON"
 	indexerCategory   = "INDEXER"
 	generatorCategory = "GENERATOR"
+	disperserCategory = "DISPERSER"
 )
 
 var (
@@ -71,7 +72,6 @@ var (
 		Value:    6061,
 		EnvVars:  []string{"METRICS_HTTP_PORT"},
 	}
-
 	Layer = &cli.StringFlag{
 		Name:     "layer",
 		Usage:    "Which layer indexing is occurring on",

packages/eventindexer/abigen.sh

@@ -0,0 +1,41 @@
+package flags
+
+import "github.com/urfave/cli/v2"
+
+var (
+	DisperserPrivateKey = &cli.StringFlag{
+		Name:     "disperserPrivateKey",
+		Usage:    "Disperser private key which contains TTKO",
+		Required: true,
+		Category: disperserCategory,
+		EnvVars:  []string{"DISPERSER_PRIVATE_KEY"},
+	}
+	TaikoTokenAddress = &cli.StringFlag{
+		Name:     "taikoTokenAddress",
+		Usage:    "Address of the TaikoToken contract",
+		Required: true,
+		Category: disperserCategory,
+		EnvVars:  []string{"TAIKO_TOKEN_ADDRESS"},
+	}
+	DispersalAmount = &cli.StringFlag{
+		Name:     "taikoTokenAddress",
+		Usage:    "Dispersal amount in wei",
+		Required: true,
+		Category: disperserCategory,
+		EnvVars:  []string{"DISPERSAL_AMOUNT"},
+	}
+	RPCUrl = &cli.StringFlag{
+		Name:     "rpcUrl",
+		Usage:    "RPC URL for the source chain",
+		Required: true,
+		Category: commonCategory,
+		EnvVars:  []string{"RPC_URL"},
+	}
+)
+
+var DisperserFlags = MergeFlags(CommonFlags, []cli.Flag{
+	DisperserPrivateKey,
+	TaikoTokenAddress,
+	DispersalAmount,
+	RPCUrl,
+})

packages/eventindexer/cmd/flags/common.go

@@ -51,6 +51,13 @@ var (
 		Category: indexerCategory,
 		EnvVars:  []string{"ASSIGNMENT_HOOK_ADDRESS"},
 	}
+	SgxVerifierAddress = &cli.StringFlag{
+		Name:     "sgxVerifierAddress",
+		Usage:    "Address of the SGXVerifier contract",
+		Required: false,
+		Category: indexerCategory,
+		EnvVars:  []string{"SGX_VERIFIER_ADDRESS"},
+	}
 	BlockBatchSize = &cli.Uint64Flag{
 		Name:     "blockBatchSize",
 		Usage:    "Block batch size when iterating through blocks",
@@ -90,6 +97,7 @@ var IndexerFlags = MergeFlags(CommonFlags, []cli.Flag{
 	L1TaikoAddress,
 	BridgeAddress,
 	SwapAddresses,
+	SgxVerifierAddress,
 	AssignmentHookAddress,
 	BlockBatchSize,
 	SubscriptionBackoff,

packages/eventindexer/cmd/flags/disperser.go

@@ -9,6 +9,7 @@ import (
 	"github.com/taikoxyz/taiko-mono/packages/eventindexer/api"
 	"github.com/taikoxyz/taiko-mono/packages/eventindexer/cmd/flags"
 	"github.com/taikoxyz/taiko-mono/packages/eventindexer/cmd/utils"
+	"github.com/taikoxyz/taiko-mono/packages/eventindexer/disperser"
 	"github.com/taikoxyz/taiko-mono/packages/eventindexer/generator"
 	"github.com/taikoxyz/taiko-mono/packages/eventindexer/indexer"
 	"github.com/urfave/cli/v2"
@@ -58,6 +59,13 @@ func main() {
 			Description: "Taiko time-series data generator",
 			Action:      utils.SubcommandAction(new(generator.Generator)),
 		},
+		{
+			Name:        "disperser",
+			Flags:       flags.DisperserFlags,
+			Usage:       "Starts the disperser software",
+			Description: "Taiko TTKO disperser",
+			Action:      utils.SubcommandAction(new(disperser.Disperser)),
+		},
 	}
 
 	if err := app.Run(os.Args); err != nil {

packages/eventindexer/cmd/flags/indexer.go

@@ -0,0 +1,98 @@
+package disperser
+
+import (
+	"crypto/ecdsa"
+	"database/sql"
+	"fmt"
+	"math/big"
+
+	"github.com/ethereum-optimism/optimism/op-service/txmgr"
+	"github.com/ethereum/go-ethereum/common"
+	"github.com/ethereum/go-ethereum/crypto"
+	"github.com/taikoxyz/taiko-mono/packages/eventindexer/cmd/flags"
+	"github.com/taikoxyz/taiko-mono/packages/eventindexer/pkg/db"
+	pkgFlags "github.com/taikoxyz/taiko-mono/packages/relayer/pkg/flags"
+	"github.com/urfave/cli/v2"
+	"gorm.io/driver/mysql"
+	"gorm.io/gorm"
+	"gorm.io/gorm/logger"
+)
+
+type DB interface {
+	DB() (*sql.DB, error)
+	GormDB() *gorm.DB
+}
+
+type Config struct {
+	// db configs
+	DatabaseUsername        string
+	DatabasePassword        string
+	DatabaseName            string
+	DatabaseHost            string
+	DatabaseMaxIdleConns    uint64
+	DatabaseMaxOpenConns    uint64
+	DatabaseMaxConnLifetime uint64
+	MetricsHTTPPort         uint64
+	DisperserPrivateKey     *ecdsa.PrivateKey
+	DispersalAmount         *big.Int
+	TaikoTokenAddress       common.Address
+	TxmgrConfigs            *txmgr.CLIConfig
+	RPCURL                  string
+	OpenDBFunc              func() (DB, error)
+}
+
+// NewConfigFromCliContext creates a new config instance from command line flags.
+func NewConfigFromCliContext(c *cli.Context) (*Config, error) {
+	disperserPrivateKey, err := crypto.ToECDSA(
+		common.Hex2Bytes(c.String(flags.DisperserPrivateKey.Name)),
+	)
+	if err != nil {
+		return nil, fmt.Errorf("invalid disperserPrivateKey: %w", err)
+	}
+
+	dispersalAmount, ok := new(big.Int).SetString(c.String(flags.DispersalAmount.Name), 10)
+	if !ok {
+		return nil, fmt.Errorf("Invalid dispersal amount")
+	}
+
+	return &Config{
+		DatabaseUsername:        c.String(flags.DatabaseUsername.Name),
+		DatabasePassword:        c.String(flags.DatabasePassword.Name),
+		DatabaseName:            c.String(flags.DatabaseName.Name),
+		DatabaseHost:            c.String(flags.DatabaseHost.Name),
+		DatabaseMaxIdleConns:    c.Uint64(flags.DatabaseMaxIdleConns.Name),
+		DatabaseMaxOpenConns:    c.Uint64(flags.DatabaseMaxOpenConns.Name),
+		DatabaseMaxConnLifetime: c.Uint64(flags.DatabaseConnMaxLifetime.Name),
+		MetricsHTTPPort:         c.Uint64(flags.MetricsHTTPPort.Name),
+		DisperserPrivateKey:     disperserPrivateKey,
+		RPCURL:                  c.String(flags.RPCUrl.Name),
+		DispersalAmount:         dispersalAmount,
+		TaikoTokenAddress:       common.HexToAddress(c.String(flags.TaikoTokenAddress.Name)),
+		TxmgrConfigs: pkgFlags.InitTxmgrConfigsFromCli(
+			c.String(flags.RPCUrl.Name),
+			disperserPrivateKey,
+			c,
+		),
+		OpenDBFunc: func() (DB, error) {
+			return db.OpenDBConnection(db.DBConnectionOpts{
+				Name:            c.String(flags.DatabaseUsername.Name),
+				Password:        c.String(flags.DatabasePassword.Name),
+				Database:        c.String(flags.DatabaseName.Name),
+				Host:            c.String(flags.DatabaseHost.Name),
+				MaxIdleConns:    c.Uint64(flags.DatabaseMaxIdleConns.Name),
+				MaxOpenConns:    c.Uint64(flags.DatabaseMaxOpenConns.Name),
+				MaxConnLifetime: c.Uint64(flags.DatabaseConnMaxLifetime.Name),
+				OpenFunc: func(dsn string) (*db.DB, error) {
+					gormDB, err := gorm.Open(mysql.Open(dsn), &gorm.Config{
+						Logger: logger.Default.LogMode(logger.Silent),
+					})
+					if err != nil {
+						return nil, err
+					}
+
+					return db.New(gormDB), nil
+				},
+			})
+		},
+	}, nil
+}