feat(bot): additional exchange accounts support

bludnic · Aug 20, 2024 · 06c1722 · 06c1722
1 parent 8c7b600
commit 06c1722
Show file tree

Hide file tree

Showing 7 changed files with 49 additions and 5 deletions.
diff --git a/packages/backtesting/src/backtesting.ts b/packages/backtesting/src/backtesting.ts
@@ -45,6 +45,7 @@ export class Backtesting<T extends IBotConfiguration<T>> {
     this.processor = createStrategyRunner({
       store: this.store,
       exchange: this.exchange,
+      additionalExchanges: [], // @todo not supported yet
       botConfig,
       botTemplate,
     });

diff --git a/packages/bot-processor/src/strategy-runner.ts b/packages/bot-processor/src/strategy-runner.ts
@@ -28,17 +28,25 @@ export class StrategyRunner<T extends IBotConfiguration> {
     private control: IBotControl,
     private botConfig: T,
     private exchange: IExchange,
+    private additionalExchanges: IExchange[],
     private botTemplate: BotTemplate<T>,
   ) {}
 
   async start(state: BotState) {
-    const context = createContext(this.control, this.botConfig, this.exchange, "start", state);
+    const context = createContext(
+      this.control,
+      this.botConfig,
+      this.exchange,
+      this.additionalExchanges,
+      "start",
+      state,
+    );
 
     await this.runTemplate(context);
   }
 
   async stop(state: BotState) {
-    const context = createContext(this.control, this.botConfig, this.exchange, "stop", state);
+    const context = createContext(this.control, this.botConfig, this.exchange, this.additionalExchanges, "stop", state);
 
     await this.runTemplate(context);
   }
@@ -53,6 +61,7 @@ export class StrategyRunner<T extends IBotConfiguration> {
       this.control,
       this.botConfig,
       this.exchange,
+      this.additionalExchanges,
       "process",
       state,
       market,
@@ -89,12 +98,13 @@ export class StrategyRunner<T extends IBotConfiguration> {
 export function createStrategyRunner<T extends IBotConfiguration>(options: {
   store: IStore;
   exchange: IExchange;
+  additionalExchanges: IExchange[];
   botConfig: T;
   botTemplate: BotTemplate<T>;
 }) {
-  const { botConfig, store, exchange, botTemplate } = options;
+  const { botConfig, store, exchange, additionalExchanges, botTemplate } = options;
 
   const botControl = new BotControl(store, botConfig);
 
-  return new StrategyRunner(botControl, botConfig, exchange, botTemplate);
+  return new StrategyRunner(botControl, botConfig, exchange, additionalExchanges, botTemplate);
 }
diff --git a/packages/bot-processor/src/types/bot/bot-context.type.ts b/packages/bot-processor/src/types/bot/bot-context.type.ts
@@ -9,6 +9,7 @@ export type TBotContext<T extends IBotConfiguration, S extends BotState = BotSta
    * Default exchange instance.
    */
   exchange: IExchange;
+  additionalExchanges: IExchange[];
   /**
    * Bot control panel
    */

diff --git a/packages/bot-processor/src/utils/createContext.ts b/packages/bot-processor/src/utils/createContext.ts
@@ -6,6 +6,7 @@ export function createContext<T extends IBotConfiguration>(
   control: IBotControl,
   config: T,
   exchange: IExchange,
+  additionalExchanges: IExchange[],
   command: "start" | "stop" | "process", // @todo add type in file
   state: BotState,
   market: MarketData = {
@@ -18,6 +19,7 @@ export function createContext<T extends IBotConfiguration>(
     control,
     config,
     exchange,
+    additionalExchanges,
     command,
     onStart: command === "start",
     onStop: command === "stop",

diff --git a/...prisma/src/migrations/20240820174038_bot_with_additional_exchanges_accounts/migration.sql b/...prisma/src/migrations/20240820174038_bot_with_additional_exchanges_accounts/migration.sql
@@ -0,0 +1,13 @@
+-- CreateTable
+CREATE TABLE "_AdditionalExchangeAccounts" (
+    "A" INTEGER NOT NULL,
+    "B" INTEGER NOT NULL,
+    CONSTRAINT "_AdditionalExchangeAccounts_A_fkey" FOREIGN KEY ("A") REFERENCES "Bot" ("id") ON DELETE CASCADE ON UPDATE CASCADE,
+    CONSTRAINT "_AdditionalExchangeAccounts_B_fkey" FOREIGN KEY ("B") REFERENCES "ExchangeAccount" ("id") ON DELETE CASCADE ON UPDATE CASCADE
+);
+
+-- CreateIndex
+CREATE UNIQUE INDEX "_AdditionalExchangeAccounts_AB_unique" ON "_AdditionalExchangeAccounts"("A", "B");
+
+-- CreateIndex
+CREATE INDEX "_AdditionalExchangeAccounts_B_index" ON "_AdditionalExchangeAccounts"("B");
diff --git a/packages/prisma/src/schema.prisma b/packages/prisma/src/schema.prisma
@@ -120,7 +120,10 @@ model ExchangeAccount {
 
   smartTrades SmartTrade[]
   orders      Order[]
-  bots        Bot[]
+
+  // Will include the only the bots that uses this account as primary
+  primaryBots Bot[]
+  bots        Bot[] @relation(name: "AdditionalExchangeAccounts")
 
   expired Boolean @default(false)
 }
@@ -209,6 +212,8 @@ model Bot {
   exchangeAccount   ExchangeAccount @relation(fields: [exchangeAccountId], references: [id])
   exchangeAccountId Int
 
+  additionalExchangeAccounts ExchangeAccount[] @relation(name: "AdditionalExchangeAccounts")
+
   owner   User     @relation(fields: [ownerId], references: [id])
   ownerId Int
   logs    BotLog[]

diff --git a/packages/processing/src/bot/bot.processing.ts b/packages/processing/src/bot/bot.processing.ts
@@ -163,7 +163,18 @@ export class BotProcessing {
       },
     });
 
+    const additionalExchangeAccounts = await xprisma.exchangeAccount.findMany({
+      where: {
+        bots: {
+          some: { id: this.bot.id },
+        },
+      },
+    });
+
     const exchange = exchangeProvider.fromAccount(exchangeAccount);
+    const additionalExchanges = additionalExchangeAccounts.map((exchangeAccount) =>
+      exchangeProvider.fromAccount(exchangeAccount),
+    );
 
     const configuration: IBotConfiguration = {
       id: this.bot.id,
@@ -178,6 +189,7 @@ export class BotProcessing {
     const processor = createStrategyRunner({
       store: storeAdapter,
       exchange,
+      additionalExchanges,
       botConfig: configuration,
       botTemplate: strategyFn,
     });