Track ALT address per launchpad currency

Author: jeffyanta

pkg/code/data/currency/model.go

@@ -53,6 +53,8 @@ type MetadataRecord struct {
 	FeesCore   string
 	SellFeeBps uint16
 
+	Alt string
+
 	CreatedBy string
 	CreatedAt time.Time
 }
@@ -142,6 +144,10 @@ func (m *MetadataRecord) Validate() error {
 		return errors.New("invalid buy sell bps")
 	}
 
+	if len(m.Alt) == 0 {
+		return errors.New("alt is required")
+	}
+
 	if len(m.CreatedBy) == 0 {
 		return errors.New("created by is required")
 	}
@@ -188,6 +194,8 @@ func (m *MetadataRecord) Clone() *MetadataRecord {
 		FeesCore:   m.FeesCore,
 		SellFeeBps: m.SellFeeBps,
 
+		Alt: m.Alt,
+
 		CreatedBy: m.CreatedBy,
 		CreatedAt: m.CreatedAt,
 	}
@@ -227,6 +235,8 @@ func (m *MetadataRecord) CopyTo(dst *MetadataRecord) {
 	dst.FeesCore = m.FeesCore
 	dst.SellFeeBps = m.SellFeeBps
 
+	dst.Alt = m.Alt
+
 	dst.CreatedBy = m.CreatedBy
 	dst.CreatedAt = m.CreatedAt
 }

pkg/code/data/currency/postgres/model.go

@@ -82,6 +82,8 @@ type metadataModel struct {
 	FeesCore   string `db:"fees_core"`
 	SellFeeBps uint16 `db:"sell_fee_bps"`
 
+	Alt string `db:"alt"`
+
 	CreatedBy string    `db:"created_by"`
 	CreatedAt time.Time `db:"created_at"`
 }
@@ -125,6 +127,8 @@ func toMetadataModel(obj *currency.MetadataRecord) (*metadataModel, error) {
 		FeesCore:   obj.FeesCore,
 		SellFeeBps: obj.SellFeeBps,
 
+		Alt: obj.Alt,
+
 		CreatedBy: obj.CreatedBy,
 		CreatedAt: obj.CreatedAt,
 	}, nil
@@ -165,6 +169,8 @@ func fromMetadataModel(obj *metadataModel) *currency.MetadataRecord {
 		FeesCore:   obj.FeesCore,
 		SellFeeBps: obj.SellFeeBps,
 
+		Alt: obj.Alt,
+
 		CreatedBy: obj.CreatedBy,
 		CreatedAt: obj.CreatedAt,
 	}
@@ -252,9 +258,9 @@ func (m *metadataModel) dbSave(ctx context.Context, db *sqlx.DB) error {
 	return pgutil.ExecuteInTx(ctx, db, sql.LevelDefault, func(tx *sqlx.Tx) error {
 		err := tx.QueryRowxContext(ctx,
 			`INSERT INTO `+metadataTableName+`
-			(name, symbol, description, image_url, seed, authority, mint, mint_bump, decimals, currency_config, currency_config_bump, liquidity_pool, liquidity_pool_bump, vault_mint, vault_mint_bump, vault_core, vault_core_bump, fees_mint, buy_fee_bps, fees_core, sell_fee_bps, created_by, created_at)
-			VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15, $16, $17, $18, $19, $20, $21, $22, $23)
-			RETURNING id, name, symbol, description, image_url, seed, authority, mint, mint_bump, decimals, currency_config, currency_config_bump, liquidity_pool, liquidity_pool_bump, vault_mint, vault_mint_bump, vault_core, vault_core_bump, fees_mint, buy_fee_bps, fees_core, sell_fee_bps, created_by, created_at`,
+			(name, symbol, description, image_url, seed, authority, mint, mint_bump, decimals, currency_config, currency_config_bump, liquidity_pool, liquidity_pool_bump, vault_mint, vault_mint_bump, vault_core, vault_core_bump, fees_mint, buy_fee_bps, fees_core, sell_fee_bps, alt, created_by, created_at)
+			VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15, $16, $17, $18, $19, $20, $21, $22, $23, $24)
+			RETURNING id, name, symbol, description, image_url, seed, authority, mint, mint_bump, decimals, currency_config, currency_config_bump, liquidity_pool, liquidity_pool_bump, vault_mint, vault_mint_bump, vault_core, vault_core_bump, fees_mint, buy_fee_bps, fees_core, sell_fee_bps, alt, created_by, created_at`,
 			m.Name,
 			m.Symbol,
 			m.Description,
@@ -276,6 +282,7 @@ func (m *metadataModel) dbSave(ctx context.Context, db *sqlx.DB) error {
 			m.BuyFeeBps,
 			m.FeesCore,
 			m.SellFeeBps,
+			m.Alt,
 			m.CreatedBy,
 			m.CreatedAt,
 		).StructScan(m)
@@ -355,7 +362,7 @@ func dbGetAllExchangeRatesForRange(ctx context.Context, db *sqlx.DB, symbol stri
 func dbGetMetadataByMint(ctx context.Context, db *sqlx.DB, mint string) (*metadataModel, error) {
 	res := &metadataModel{}
 	err := db.GetContext(ctx, res,
-		`SELECT id, name, symbol, description, image_url, seed, authority, mint, mint_bump, decimals, currency_config, currency_config_bump, liquidity_pool, liquidity_pool_bump, vault_mint, vault_mint_bump, vault_core, vault_core_bump, fees_mint, buy_fee_bps, fees_core, sell_fee_bps, created_by, created_at
+		`SELECT id, name, symbol, description, image_url, seed, authority, mint, mint_bump, decimals, currency_config, currency_config_bump, liquidity_pool, liquidity_pool_bump, vault_mint, vault_mint_bump, vault_core, vault_core_bump, fees_mint, buy_fee_bps, fees_core, sell_fee_bps, alt, created_by, created_at
 		FROM `+metadataTableName+`
 		WHERE mint = $1`,
 		mint,

pkg/code/data/currency/postgres/store_test.go

@@ -64,6 +64,8 @@ const (
 		fees_core TEXT NOT NULL,
 		sell_fee_bps INTEGER NOT NULL,
 
+		alt TEXT NOT NULL,
+
 		created_by TEXT NOT NULL,
 		created_at TIMESTAMP WITH TIME ZONE NOT NULL
 	);

pkg/code/data/currency/tests/tests.go

@@ -149,6 +149,8 @@ func testMetadataRoundTrip(t *testing.T, s currency.Store) {
 		FeesCore:   "5EcVYL8jHRKeeQqg6eYVBzc73ecH1PFzzaavoQBKRYy5",
 		SellFeeBps: currencycreator.DefaultSellFeeBps,
 
+		Alt: "EkAeTCceLWbmZrAzVZanDJBtHSnkAWndMFgmTnUnVLRR",
+
 		CreatedBy: "jyyy4RpW3X5ApbW5G6vx9ZVPxhoUKGRLbZ4LxC47LYG",
 		CreatedAt: time.Now(),
 	}
@@ -223,6 +225,7 @@ func assertEquivalentMetadataRecords(t *testing.T, obj1, obj2 *currency.Metadata
 	assert.Equal(t, obj1.BuyFeeBps, obj2.BuyFeeBps)
 	assert.Equal(t, obj1.FeesCore, obj2.FeesCore)
 	assert.Equal(t, obj1.SellFeeBps, obj2.SellFeeBps)
+	assert.Equal(t, obj1.Alt, obj2.Alt)
 	assert.Equal(t, obj1.CreatedBy, obj2.CreatedBy)
 	assert.Equal(t, obj1.CreatedAt.Unix(), obj2.CreatedAt.Unix())
 }

pkg/code/transaction/alt.go

@@ -4,8 +4,6 @@ import (
 	"context"
 	"crypto/ed25519"
 
-	"github.com/pkg/errors"
-
 	"github.com/code-payments/code-server/pkg/code/common"
 	code_data "github.com/code-payments/code-server/pkg/code/data"
 	"github.com/code-payments/code-server/pkg/solana"
@@ -15,29 +13,17 @@ import (
 // GetAltForMint gets an address lookup table to operate in a versioned
 // transaction for the provided mint
 func GetAltForMint(ctx context.Context, data code_data.Provider, mint *common.Account) (solana.AddressLookupTable, error) {
-	// todo: This would be tracked in a DB table
-	var account *common.Account
-	var err error
-	switch mint.PublicKey().ToBase58() {
-	case "52MNGpgvydSwCtC2H4qeiZXZ1TxEuRVCRGa8LAfk2kSj":
-		account, err = common.NewAccountFromPublicKeyString("EkAeTCceLWbmZrAzVZanDJBtHSnkAWndMFgmTnUnVLRR")
-	case "497Wy6cY9BjWBiaDHzJ7TcUZqF2gE1Qm7yXtSj1vSr5W":
-		account, err = common.NewAccountFromPublicKeyString("3QLcDkhXMAKuRvCJuc6kcye4w6yyHaDs1dYcktcB1pRA")
-	case "2o4PFbDZ73BihFraknfVTQeUtELKAeVUL4oa6bkrYU3A":
-		account, err = common.NewAccountFromPublicKeyString("4bPdZB23pPYSg49H3fEMLSaqarQayvhpRJatxgv1P2JP")
-	default:
-		return solana.AddressLookupTable{}, errors.New("unsupported currency")
-	}
+	metadataRecord, err := data.GetCurrencyMetadata(ctx, mint.PublicKey().ToBase58())
 	if err != nil {
 		return solana.AddressLookupTable{}, err
 	}
 
-	vmConfig, err := common.GetVmConfigForMint(ctx, data, mint)
+	account, err := common.NewAccountFromPublicKeyString(metadataRecord.Alt)
 	if err != nil {
 		return solana.AddressLookupTable{}, err
 	}
 
-	metadataRecord, err := data.GetCurrencyMetadata(ctx, mint.PublicKey().ToBase58())
+	vmConfig, err := common.GetVmConfigForMint(ctx, data, mint)
 	if err != nil {
 		return solana.AddressLookupTable{}, err
 	}