feat: add swap endpoint to magicswap (#63)

* feat: add magicswap to core api * chore: add changeset
TreasureProject · Jul 24, 2024 · eeb81b0 · eeb81b0
1 parent 3d4fc51
commit eeb81b0
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 4 deletions.
diff --git a/.changeset/grumpy-clouds-melt.md b/.changeset/grumpy-clouds-melt.md
@@ -0,0 +1,5 @@
+---
+"@treasure-dev/tdk-core": minor
+---
+
+Added magicswap helpers to tdk api client
diff --git a/apps/api/src/routes/magicswap.ts b/apps/api/src/routes/magicswap.ts
@@ -186,10 +186,12 @@ export const magicswapRoutes =
           wagmiConfig,
         });
 
-        const poolTokens = pools.reduce((acc, poolToken) => {
-          acc[poolToken.id] = poolToken;
-          return acc;
-        }, {});
+        const poolTokens = pools
+          .flatMap(({ token0, token1 }) => [token0, token1])
+          .reduce((acc, poolToken) => {
+            acc[poolToken.id] = poolToken;
+            return acc;
+          }, {});
 
         const tokenIn = poolTokens[tokenInId];
         const tokenOut = poolTokens[tokenOutId];

diff --git a/packages/core/src/api.ts b/packages/core/src/api.ts
@@ -11,9 +11,12 @@ import type {
   CreateTransactionBody,
   LoginBody,
   LoginReply,
+  PoolReply,
+  PoolsReply,
   ReadCurrentUserSessionsQuerystring,
   ReadCurrentUserSessionsReply,
   ReadLoginPayloadReply,
+  RouteReply,
 } from "../../../apps/api/src/schema";
 import type {
   CreateTransactionReply,
@@ -24,6 +27,10 @@ import type {
   ReadProjectReply,
   ReadTransactionReply,
 } from "../../../apps/api/src/schema";
+import type {
+  RouteBody,
+  SwapBody,
+} from "../../../apps/api/src/schema/magicswap";
 import { DEFAULT_TDK_API_BASE_URI, DEFAULT_TDK_CHAIN_ID } from "./constants";
 
 // @ts-expect-error: Patch BigInt for JSON serialization
@@ -251,4 +258,18 @@ export class TDKAPI {
   harvester = {
     get: (id: string) => this.get<ReadHarvesterReply>(`/harvesters/${id}`),
   };
+
+  magicswap = {
+    getPools: () => this.get<PoolsReply>("/magicswap/pools"),
+    getPool: (id: string) => this.get<PoolReply>(`/magicswap/pools/${id}`),
+    getRoute: (body: RouteBody) =>
+      this.post<RouteBody, RouteReply>("/magicswap/route", body),
+    swap: async (body: SwapBody, waitForCompletion = true) => {
+      const result = await this.post<SwapBody, CreateTransactionReply>(
+        "/magicswap/swap",
+        body,
+      );
+      return waitForCompletion ? this.transaction.wait(result.queueId) : result;
+    },
+  };
 }