Merge pull request #367 from drift-labs/master

change swift url based on driftenv

Author: moosecat2

src/experimental-bots/swift/makerExample.ts

@@ -20,14 +20,20 @@ import { simulateAndGetTxWithCUs } from '../../utils';
 export class SwiftMaker {
 	interval: NodeJS.Timeout | null = null;
 	private ws: WebSocket | null = null;
+	private swiftUrl: string;
 	private heartbeatTimeout: NodeJS.Timeout | null = null;
 	private readonly heartbeatIntervalMs = 80_000;
 	constructor(
 		private driftClient: DriftClient,
 		private userMap: UserMap,
 		runtimeSpec: RuntimeSpec,
 		private dryRun?: boolean
-	) {}
+	) {
+		this.swiftUrl =
+			runtimeSpec.driftEnv === 'mainnet-beta'
+				? 'wss://swift.drift.trade/ws'
+				: 'wss://master.swift.drift.trade/ws';
+	}
 
 	async init() {
 		await this.subscribeWs();
@@ -36,7 +42,7 @@ export class SwiftMaker {
 	async subscribeWs() {
 		const keypair = this.driftClient.wallet.payer!;
 		const ws = new WebSocket(
-			`wss://master.swift.drift.trade/ws?pubkey=` + keypair.publicKey.toBase58()
+			this.swiftUrl + '?pubkey=' + keypair.publicKey.toBase58()
 		);
 
 		ws.on('open', async () => {

src/experimental-bots/swift/takerExample.ts

@@ -16,12 +16,18 @@ const CONFIRM_TIMEOUT = 30_000;
 
 export class SwiftTaker {
 	interval: NodeJS.Timeout | null = null;
+	swiftUrl: string;
 
 	constructor(
 		private driftClient: DriftClient,
 		runtimeSpec: RuntimeSpec,
 		private intervalMs: number
-	) {}
+	) {
+		this.swiftUrl =
+			runtimeSpec.driftEnv === 'mainnet-beta'
+				? 'https://swift.drift.trade/orders'
+				: 'https://master.swift.drift.trade/orders';
+	}
 
 	async init() {
 		await this.startInterval();
@@ -76,8 +82,7 @@ export class SwiftTaker {
 			);
 
 			const response = await axios.default.post(
-				// 'http://0.0.0.0:3000/orders',
-				'https://master.swift.drift.trade/orders',
+				this.swiftUrl,
 				{
 					market_index: marketIndex,
 					market_type: 'perp',