only crank used oracles option (#356)

* only crank used oracles option

* make pyth requirement a bit more safe

Author: moosecat2

src/bots/pythCranker.ts

@@ -17,6 +17,7 @@ import {
 	DriftClient,
 	getOracleClient,
 	getPythPullOraclePublicKey,
+	getVariant,
 	isOneOfVariant,
 	ONE,
 	OracleClient,
@@ -132,13 +133,25 @@ export class PythCrankerBot implements Bot {
 			.getPerpMarketAccounts()
 			.map((market) => market.marketIndex);
 		const perpMarketConfigs = PerpMarkets[this.globalConfig.driftEnv].filter(
-			(market) => perpMarketIndexes.includes(market.marketIndex)
+			(market) => {
+				const oracleVariant = getVariant(market.oracleSource).toLowerCase();
+				const useMarket = this.crankConfigs.onlyCrankUsedOracles
+					? oracleVariant.includes('pyth') && !oracleVariant.includes('pull')
+					: true;
+				return perpMarketIndexes.includes(market.marketIndex) && useMarket;
+			}
 		);
 		const spotMarketIndexes = this.driftClient
 			.getSpotMarketAccounts()
 			.map((market) => market.marketIndex);
 		const spotMarketConfigs = SpotMarkets[this.globalConfig.driftEnv].filter(
-			(market) => spotMarketIndexes.includes(market.marketIndex)
+			(market) => {
+				const oracleVariant = getVariant(market.oracleSource).toLowerCase();
+				const useMarket = this.crankConfigs.onlyCrankUsedOracles
+					? oracleVariant.includes('pyth') && !oracleVariant.includes('pull')
+					: true;
+				return spotMarketIndexes.includes(market.marketIndex) && useMarket;
+			}
 		);
 
 		for (const marketConfig of perpMarketConfigs) {

src/bots/pythLazerCranker.ts

@@ -8,6 +8,7 @@ import {
 	DevnetSpotMarkets,
 	DriftClient,
 	getPythLazerOraclePublicKey,
+	getVariant,
 	MainnetPerpMarkets,
 	MainnetSpotMarkets,
 	PriorityFeeSubscriber,
@@ -53,13 +54,6 @@ export class PythLazerCrankerBot implements Bot {
 		this.name = crankConfigs.botId;
 		this.dryRun = crankConfigs.dryRun;
 		this.intervalMs = crankConfigs.intervalMs;
-		if (!globalConfig.hermesEndpoint) {
-			throw new Error('Missing hermesEndpoint in global config');
-		}
-
-		if (globalConfig.driftEnv != 'devnet') {
-			throw new Error('Only devnet drift env is supported');
-		}
 
 		const spotMarkets =
 			this.globalConfig.driftEnv === 'mainnet-beta'
@@ -70,13 +64,18 @@ export class PythLazerCrankerBot implements Bot {
 				? MainnetPerpMarkets
 				: DevnetPerpMarkets;
 
-		const allFeedIds = [
-			...spotMarkets.map((market) => market.pythLazerId),
-			...perpMarkets.map((market) => market.pythLazerId),
-		].filter((id) => id !== undefined) as number[];
+		const allFeedIds: number[] = [];
+		for (const market of [...spotMarkets, ...perpMarkets]) {
+			if (
+				(this.crankConfigs.onlyCrankUsedOracles &&
+					!getVariant(market.oracleSource).toLowerCase().includes('lazer')) ||
+				market.pythFeedId == undefined
+			)
+				continue;
+			allFeedIds.push(market.pythLazerId!);
+		}
 		const allFeedIdsSet = new Set(allFeedIds);
 		const feedIdChunks = chunks(Array.from(allFeedIdsSet), 11);
-
 		console.log(feedIdChunks);
 
 		if (!this.globalConfig.lazerEndpoint || !this.globalConfig.lazerToken) {

src/config.ts

@@ -106,17 +106,14 @@ export type PythCrankerBotConfig = BaseBotConfig & {
 			earlyUpdate: PythUpdateConfigs;
 		};
 	};
+	onlyCrankUsedOracles?: boolean;
 };
 
 export type PythLazerCrankerBotConfig = BaseBotConfig & {
 	slotStalenessThresholdRestart: number;
 	txSuccessRateThreshold: number;
 	intervalMs: number;
-	updateConfigs: {
-		[key: string]: {
-			feedId: number;
-		};
-	};
+	onlyCrankUsedOracles?: boolean;
 };
 
 export type SwitchboardCrankerBotConfig = BaseBotConfig & {