Add utitilies for launchpad currency ALTs and accounts

Author: jeffyanta

pkg/code/common/account.go

@@ -13,6 +13,7 @@ import (
 
 	code_data "github.com/code-payments/code-server/pkg/code/data"
 	"github.com/code-payments/code-server/pkg/code/data/account"
+	"github.com/code-payments/code-server/pkg/code/data/currency"
 	"github.com/code-payments/code-server/pkg/code/data/timelock"
 	"github.com/code-payments/code-server/pkg/solana"
 	"github.com/code-payments/code-server/pkg/solana/cvm"
@@ -529,6 +530,64 @@ func ValidateExternalTokenAccount(ctx context.Context, data code_data.Provider,
 	}
 }
 
+type LaunchpadCurrencyAccounts struct {
+	Mint               *Account
+	CurrencyConfig     *Account
+	CurrencyConfigBump uint8
+	LiquidityPool      *Account
+	LiquidityPoolBump  uint8
+	VaultBase          *Account
+	VaultBaseBump      uint8
+	VaultMint          *Account
+	VaultMintBump      uint8
+	FeesBase           *Account
+	FeesMint           *Account
+}
+
+func GetLaunchpadCurrencyAccounts(metadataRecord *currency.MetadataRecord) (*LaunchpadCurrencyAccounts, error) {
+	mint, err := NewAccountFromPublicKeyString(metadataRecord.Mint)
+	if err != nil {
+		return nil, err
+	}
+	currencyConfig, err := NewAccountFromPublicKeyString(metadataRecord.CurrencyConfig)
+	if err != nil {
+		return nil, err
+	}
+	liquidityPool, err := NewAccountFromPublicKeyString(metadataRecord.LiquidityPool)
+	if err != nil {
+		return nil, err
+	}
+	vaultBase, err := NewAccountFromPublicKeyString(metadataRecord.VaultCore)
+	if err != nil {
+		return nil, err
+	}
+	vaultMint, err := NewAccountFromPublicKeyString(metadataRecord.VaultMint)
+	if err != nil {
+		return nil, err
+	}
+	feesBase, err := NewAccountFromPublicKeyString(metadataRecord.FeesCore)
+	if err != nil {
+		return nil, err
+	}
+	feesMint, err := NewAccountFromPublicKeyString(metadataRecord.FeesMint)
+	if err != nil {
+		return nil, err
+	}
+	return &LaunchpadCurrencyAccounts{
+		Mint:               mint,
+		CurrencyConfig:     currencyConfig,
+		CurrencyConfigBump: metadataRecord.CurrencyConfigBump,
+		LiquidityPool:      liquidityPool,
+		LiquidityPoolBump:  metadataRecord.LiquidityPoolBump,
+		VaultBase:          vaultBase,
+		VaultBaseBump:      metadataRecord.VaultCoreBump,
+		VaultMint:          vaultMint,
+		VaultMintBump:      metadataRecord.VaultMintBump,
+		FeesBase:           feesBase,
+		FeesMint:           feesMint,
+	}, nil
+}
+
 func isOnCurve(pubKey ed25519.PublicKey) bool {
 	if len(pubKey) != ed25519.PublicKeySize {
 		return false

pkg/code/transaction/alt.go

@@ -0,0 +1,67 @@
+package transaction
+
+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"
+	"github.com/code-payments/code-server/pkg/solana/system"
+)
+
+// 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")
+	}
+	if err != nil {
+		return solana.AddressLookupTable{}, err
+	}
+
+	vmConfig, err := common.GetVmConfigForMint(ctx, data, mint)
+	if err != nil {
+		return solana.AddressLookupTable{}, err
+	}
+
+	metadataRecord, err := data.GetCurrencyMetadata(ctx, mint.PublicKey().ToBase58())
+	if err != nil {
+		return solana.AddressLookupTable{}, err
+	}
+
+	currencyAccounts, err := common.GetLaunchpadCurrencyAccounts(metadataRecord)
+	if err != nil {
+		return solana.AddressLookupTable{}, err
+	}
+
+	return solana.AddressLookupTable{
+		PublicKey: account.PublicKey().ToBytes(),
+		Addresses: []ed25519.PublicKey{
+			vmConfig.Vm.PublicKey().ToBytes(),
+			vmConfig.Omnibus.PublicKey().ToBytes(),
+			mint.PublicKey().ToBytes(),
+			currencyAccounts.CurrencyConfig.PublicKey().ToBytes(),
+			currencyAccounts.LiquidityPool.PublicKey().ToBytes(),
+			currencyAccounts.VaultBase.PublicKey().ToBytes(),
+			currencyAccounts.VaultMint.PublicKey().ToBytes(),
+			currencyAccounts.FeesBase.PublicKey().ToBytes(),
+			currencyAccounts.FeesMint.PublicKey().ToBytes(),
+			common.CoreMintAccount.PublicKey().ToBytes(),
+			system.RentSysVar,
+			system.RecentBlockhashesSysVar,
+		},
+	}, nil
+}