Make swap subsidizer in Geyser configurable (#43)

Author: jeffyanta

pkg/code/async/geyser/backup.go

@@ -142,7 +142,7 @@ func (p *service) backupExternalDepositWorker(serviceCtx context.Context, interv
 
 						log := log.WithField("account", vault.PublicKey().ToBase58())
 
-						err := fixMissingExternalDeposits(tracedCtx, p.data, p.pusher, vault)
+						err := fixMissingExternalDeposits(tracedCtx, p.conf, p.data, p.pusher, vault)
 						if err != nil {
 							m.NoticeError(err)
 							log.WithError(err).Warn("failure fixing missing external deposits")

pkg/code/async/geyser/config.go

@@ -34,6 +34,9 @@ const (
 	MessagingFeeCollectorPublicKeyConfigEnvName = envConfigPrefix + "MESSAGING_FEE_COLLECTOR_PUBLIC_KEY"
 	defaultMessagingFeeCollectorPublicKey       = "invalid" // ensure something valid is set
 
+	SwapSubsidizerPublicKeyConfigEnvName = envConfigPrefix + "SWAP_SUBSIDIZER_PUBLIC_KEY"
+	defaultSwapSubsidizerPublicKey       = "invalid" // ensure something valid is set
+
 	BackupMessagingWorkerIntervalConfigEnvName = envConfigPrefix + "BACKUP_MESSAGING_WORKER_INTERVAL"
 	defaultBackupMessagingWorkerInterval       = 15 * time.Minute // Decrease significantly once feature is live
 )
@@ -52,6 +55,8 @@ type conf struct {
 
 	messagingFeeCollectorPublicKey config.String
 	backupMessagingWorkerInterval  config.Duration
+
+	swapSubsidizerPublicKey config.String
 }
 
 // ConfigProvider defines how config values are pulled
@@ -74,6 +79,8 @@ func WithEnvConfigs() ConfigProvider {
 
 			messagingFeeCollectorPublicKey: env.NewStringConfig(MessagingFeeCollectorPublicKeyConfigEnvName, defaultMessagingFeeCollectorPublicKey),
 			backupMessagingWorkerInterval:  env.NewDurationConfig(BackupMessagingWorkerIntervalConfigEnvName, defaultBackupMessagingWorkerInterval),
+
+			swapSubsidizerPublicKey: env.NewStringConfig(SwapSubsidizerPublicKeyConfigEnvName, defaultSwapSubsidizerPublicKey),
 		}
 	}
 }

pkg/code/async/geyser/external_deposit.go

@@ -39,15 +39,15 @@ var (
 	syncedDepositCache = cache.NewCache(1_000_000)
 )
 
-func fixMissingExternalDeposits(ctx context.Context, data code_data.Provider, pusher push_lib.Provider, vault *common.Account) error {
+func fixMissingExternalDeposits(ctx context.Context, conf *conf, data code_data.Provider, pusher push_lib.Provider, vault *common.Account) error {
 	signatures, err := findPotentialExternalDeposits(ctx, data, vault)
 	if err != nil {
 		return errors.Wrap(err, "error finding potential external deposits")
 	}
 
 	var anyError error
 	for _, signature := range signatures {
-		err := processPotentialExternalDeposit(ctx, data, pusher, signature, vault)
+		err := processPotentialExternalDeposit(ctx, conf, data, pusher, signature, vault)
 		if err != nil {
 			anyError = errors.Wrap(err, "error processing signature for external deposit")
 		}
@@ -115,7 +115,7 @@ func findPotentialExternalDeposits(ctx context.Context, data code_data.Provider,
 	}
 }
 
-func processPotentialExternalDeposit(ctx context.Context, data code_data.Provider, pusher push_lib.Provider, signature string, tokenAccount *common.Account) error {
+func processPotentialExternalDeposit(ctx context.Context, conf *conf, data code_data.Provider, pusher push_lib.Provider, signature string, tokenAccount *common.Account) error {
 	// Avoid reprocessing deposits we've recently seen and processed. Particularly,
 	// the backup process will likely be triggered in frequent bursts, so this is
 	// just an optimization around that.
@@ -199,7 +199,7 @@ func processPotentialExternalDeposit(ctx context.Context, data code_data.Provide
 			return nil
 		}
 
-		isCodeSwap, usdcQuarksSwapped, err := getCodeSwapMetadata(tokenBalances)
+		isCodeSwap, usdcQuarksSwapped, err := getCodeSwapMetadata(ctx, conf, tokenBalances)
 		if err != nil {
 			return errors.Wrap(err, "error getting code swap metadata")
 		}
@@ -381,13 +381,12 @@ func getDeltaQuarksFromTokenBalances(tokenAccount *common.Account, tokenBalances
 	return postQuarkBalance - preQuarkBalance, nil
 }
 
-func getCodeSwapMetadata(tokenBalances *solana.TransactionTokenBalances) (bool, uint64, error) {
+func getCodeSwapMetadata(ctx context.Context, conf *conf, tokenBalances *solana.TransactionTokenBalances) (bool, uint64, error) {
 	// Detect whether this is a Code swap by inspecting whether the swap subsidizer
 	// is included in the transaction.
 	var isCodeSwap bool
 	for _, account := range tokenBalances.Accounts {
-		// todo: configurable
-		if account == "swapBMF2EzkHSn9NDwaSFWMtGC7ZsgzApQv9NSkeUeU" {
+		if account == conf.swapSubsidizerPublicKey.Get(ctx) {
 			isCodeSwap = true
 			break
 		}

pkg/code/async/geyser/handler.go

@@ -144,7 +144,7 @@ func (h *TokenProgramAccountHandler) Handle(ctx context.Context, update *geyserp
 
 	// We've determined this token account is one that we care about. Process
 	// the update as an external deposit.
-	return processPotentialExternalDeposit(ctx, h.data, h.pusher, *update.TxSignature, tokenAccount)
+	return processPotentialExternalDeposit(ctx, h.conf, h.data, h.pusher, *update.TxSignature, tokenAccount)
 }
 
 type TimelockV1ProgramAccountHandler struct {