Add solana config and refactor global config

Author: rickyrombo

config/config.go

@@ -1,6 +1,7 @@
 package config
 
 import (
+	"log"
 	"os"
 
 	_ "github.com/joho/godotenv/autoload"
@@ -15,6 +16,8 @@ type Config struct {
 	PythonUpstreams               []string
 	NetworkTakeRate               float64
 	StakingBridgeUsdcPayoutWallet string
+	SolanaConfig                  SolanaConfig
+	AntiAbuseOracles              []string
 }
 
 var Cfg = Config{
@@ -25,21 +28,35 @@ var Cfg = Config{
 	AxiomDataset:                  os.Getenv("axiomDataset"),
 	NetworkTakeRate:               10,
 	StakingBridgeUsdcPayoutWallet: "7vGA3fcjvxa3A11MAxmyhFtYowPLLCNyvoxxgN3NN2Vf",
+	SolanaConfig:                  SolCfg,
 }
 
 func init() {
-	if os.Getenv("ENV") == "stage" {
+	switch env := os.Getenv("ENV"); env {
+	case "dev":
+		fallthrough
+	case "development":
+		Cfg.AntiAbuseOracles = []string{"http://audius-protocol-discovery-provider-1"}
+	case "stage":
+		fallthrough
+	case "staging":
+		Cfg.AntiAbuseOracles = []string{"https://discoveryprovider.staging.audius.co"}
 		Cfg.PythonUpstreams = []string{
 			"https://discoveryprovider.staging.audius.co",
 			"https://discoveryprovider2.staging.audius.co",
 			"https://discoveryprovider3.staging.audius.co",
 			"https://discoveryprovider5.staging.audius.co",
 		}
-	} else {
+	case "prod":
+		fallthrough
+	case "production":
+		Cfg.AntiAbuseOracles = []string{"https://discoveryprovider.audius.co"}
 		Cfg.PythonUpstreams = []string{
 			"https://discoveryprovider.audius.co",
 			"https://discoveryprovider2.audius.co",
 			"https://discoveryprovider3.audius.co",
 		}
+	default:
+		log.Fatalf("Unknown environment: %s", env)
 	}
 }

config/solana_config.go

@@ -0,0 +1,101 @@
+package config
+
+import (
+	"log"
+	"os"
+	"strings"
+
+	"bridgerton.audius.co/api/spl/programs/claimable_tokens"
+	"bridgerton.audius.co/api/spl/programs/reward_manager"
+	"github.com/gagliardetto/solana-go"
+)
+
+type SolanaConfig struct {
+	RpcProviders []string
+	FeePayers    []solana.Wallet
+	SolanaRelay  string
+
+	MintAudio solana.PublicKey
+
+	RewardManagerProgramID   solana.PublicKey
+	RewardManagerState       solana.PublicKey
+	RewardManagerLookupTable solana.PublicKey
+
+	ClaimableTokensProgramID solana.PublicKey
+}
+
+var SolCfg = SolanaConfig{
+	RpcProviders: strings.Split(os.Getenv("solanaRpcProviders"), ","),
+}
+
+const (
+	// Dev
+	DevSolanaRelay              = "http://audius-protocol-discovery-provider-1/solana/relay"
+	DevMintAudio                = "37RCjhgV1qGV2Q54EHFScdxZ22ydRMdKMtVgod47fDP3"
+	DevRewardManagerProgramID   = "testLsJKtyABc9UXJF8JWFKf1YH4LmqCWBC42c6akPb"
+	DevRewardManagerState       = "DJPzVothq58SmkpRb1ATn5ddN2Rpv1j2TcGvM3XsHf1c"
+	DevRewardManagerLookupTable = "GNHKVSmHvoRBt1JJCxz7RSMfzDQGDGhGEjmhHyxb3K5J"
+	DevClaimableTokensProgramID = "testHKV1B56fbvop4w6f2cTGEub9dRQ2Euta5VmqdX9"
+
+	// Stage
+	StageSolanaRelay              = "https://discoveryprovider.staging.audius.co/solana/relay"
+	StageMintAudio                = "BELGiMZQ34SDE6x2FUaML2UHDAgBLS64xvhXjX5tBBZo"
+	StageRewardManagerProgramID   = "CDpzvz7DfgbF95jSSCHLX3ERkugyfgn9Fw8ypNZ1hfXp"
+	StageRewardManagerState       = "GaiG9LDYHfZGqeNaoGRzFEnLiwUT7WiC6sA6FDJX9ZPq"
+	StageRewardManagerLookupTable = "ChFCWjeFxM6SRySTfT46zXn2K7m89TJsft4HWzEtkB4J"
+	StageClaimableTokensProgramID = "2sjQNmUfkV6yKKi4dPR8gWRgtyma5aiymE3aXL2RAZww"
+
+	// Prod
+	ProdSolanaRelay              = "https://discoveryprovider.audius.co/solana/relay"
+	ProdMintAudio                = "9LzCMqDgTKYz9Drzqnpgee3SGa89up3a247ypMj2xrqM"
+	ProdRewardManagerProgramID   = "DDZDcYdQFEMwcu2Mwo75yGFjJ1mUQyyXLWzhZLEVFcei"
+	ProdRewardManagerState       = "71hWFVYokLaN1PNYzTAWi13EfJ7Xt9VbSWUKsXUT8mxE"
+	ProdRewardManagerLookupTable = "4UQwpGupH66RgQrWRqmPM9Two6VJEE68VZ7GeqZ3mvVv"
+	ProdClaimableTokensProgramID = "Ewkv3JahEFRKkcJmpoKB7pXbnUHwjAyXiwEo4ZY2rezQ"
+)
+
+func init() {
+	walletKeys := strings.Split(os.Getenv("solanaFeePayerKeys"), ",")
+	SolCfg.FeePayers = make([]solana.Wallet, len(walletKeys))
+	for i, privkeyString := range walletKeys {
+		privkey := solana.MustPrivateKeyFromBase58(privkeyString)
+		SolCfg.FeePayers[i] = solana.Wallet{
+			PrivateKey: privkey,
+		}
+	}
+
+	switch env := os.Getenv("ENV"); env {
+	case "dev":
+		fallthrough
+	case "development":
+		SolCfg.SolanaRelay = DevSolanaRelay
+		SolCfg.MintAudio = solana.MustPublicKeyFromBase58(DevMintAudio)
+		SolCfg.RewardManagerProgramID = solana.MustPublicKeyFromBase58(DevRewardManagerProgramID)
+		SolCfg.RewardManagerState = solana.MustPublicKeyFromBase58(DevRewardManagerState)
+		SolCfg.RewardManagerLookupTable = solana.MustPublicKeyFromBase58(DevRewardManagerLookupTable)
+		SolCfg.ClaimableTokensProgramID = solana.MustPublicKeyFromBase58(DevClaimableTokensProgramID)
+	case "stage":
+		fallthrough
+	case "staging":
+		SolCfg.SolanaRelay = StageSolanaRelay
+		SolCfg.MintAudio = solana.MustPublicKeyFromBase58(StageMintAudio)
+		SolCfg.RewardManagerProgramID = solana.MustPublicKeyFromBase58(StageRewardManagerProgramID)
+		SolCfg.RewardManagerState = solana.MustPublicKeyFromBase58(StageRewardManagerState)
+		SolCfg.RewardManagerLookupTable = solana.MustPublicKeyFromBase58(StageRewardManagerLookupTable)
+		SolCfg.ClaimableTokensProgramID = solana.MustPublicKeyFromBase58(StageClaimableTokensProgramID)
+	case "prod":
+		fallthrough
+	case "production":
+		SolCfg.SolanaRelay = ProdSolanaRelay
+		SolCfg.MintAudio = solana.MustPublicKeyFromBase58(ProdMintAudio)
+		SolCfg.RewardManagerProgramID = solana.MustPublicKeyFromBase58(ProdRewardManagerProgramID)
+		SolCfg.RewardManagerState = solana.MustPublicKeyFromBase58(ProdRewardManagerState)
+		SolCfg.RewardManagerLookupTable = solana.MustPublicKeyFromBase58(ProdRewardManagerLookupTable)
+		SolCfg.ClaimableTokensProgramID = solana.MustPublicKeyFromBase58(ProdClaimableTokensProgramID)
+	default:
+		log.Fatalf("Unknown environment: %s", env)
+	}
+
+	reward_manager.SetProgramID(SolCfg.RewardManagerProgramID)
+	claimable_tokens.SetProgramID(SolCfg.ClaimableTokensProgramID)
+}