@strkfarm/sdk 2.0.0-dev.26 → 2.0.0-dev.28

package/src/strategies/universal-adapters/common-adapter.ts

@@ -1,204 +1,207 @@
-import { ContractAddr, Web3Number } from "@/dataTypes";
-import { LeafData } from "@/utils";
-import { Call, hash, num, shortString, uint256 } from "starknet";
-import { AVNU_EXCHANGE, AVNU_MIDDLEWARE, SIMPLE_SANITIZER, toBigInt } from "./adapter-utils";
-import { AdapterLeafType, BaseAdapter, GenerateCallFn, LeafAdapterFn, ManageCall } from "./baseAdapter";
-import { SwapInfo } from "@/modules";
-
-export interface FlashloanCallParams {
-  amount: Web3Number,
-  data: bigint[]
-}
-export interface ApproveCallParams {
-  amount: Web3Number,
-}
-export interface AvnuSwapCallParams {
-  props: SwapInfo
-}
-
-export interface CommonAdapterConfig {
-  id: string,
-  vaultAddress: ContractAddr,
-  vaultAllocator: ContractAddr,
-  manager: ContractAddr,
-  asset: ContractAddr
-}
-
-export class CommonAdapter {
-  config: CommonAdapterConfig;
-
-  constructor(config: CommonAdapterConfig) {
-    this.config = config;
-  }
-
-  protected constructSimpleLeafData(params: {
-    id: string,
-    target: ContractAddr,
-    method: string,
-    packedArguments: bigint[],
-  }, sanitizer: ContractAddr = SIMPLE_SANITIZER): LeafData {
-    const { id, target, method, packedArguments } = params;
-    return {
-      id: BigInt(num.getDecimalString(shortString.encodeShortString(id))),
-      readableId: id,
-      data: [
-        sanitizer.toBigInt(), // sanitizer address
-        target.toBigInt(), // contract
-        toBigInt(hash.getSelectorFromName(method)), // method name
-        BigInt(packedArguments.length),
-        ...packedArguments
-      ]
-    };
-  }
-
-  async getFlashloanAdapter(): Promise<AdapterLeafType<FlashloanCallParams>> {
-    const manageCall = this.getFlashloanCall.bind(this)({amount: Web3Number.fromWei('0', 6), data: []});
-    const packedArguments: bigint[] = [
-      this.config.manager.toBigInt(), // receiver
-      this.config.asset.toBigInt(), // asset
-      toBigInt(0), // is legacy false
-    ];
-    const leaf = this.constructSimpleLeafData({
-      id: this.config.id,
-      target: (await manageCall)[0].call.contractAddress,
-      method: 'flash_loan',
-      packedArguments
-    });
-    return { leaves: [leaf], callConstructor: this.getFlashloanCall.bind(this) };
-  }
-
-  async getFlashloanCall(params: FlashloanCallParams): Promise<ManageCall[]> {
-    const uint256Amount = uint256.bnToUint256(params.amount.toWei());
-    return [{
-      sanitizer: SIMPLE_SANITIZER,
-      call: {
-        contractAddress: this.config.manager,
-        selector: hash.getSelectorFromName('flash_loan'),
-        calldata: [
-          this.config.manager.toBigInt(), // receiver
-          this.config.asset.toBigInt(), // asset
-          toBigInt(uint256Amount.low.toString()), // amount low
-          toBigInt(uint256Amount.high.toString()), // amount high
-          toBigInt(0), // is legacy false
-          BigInt(params.data.length),
-          ...params.data
-        ]
-      }
-    }]
-  }
-
-  getBringLiquidityAdapter(id: string): () => AdapterLeafType<ApproveCallParams> {
-    return () => ({
-      leaves: [this.constructSimpleLeafData({
-        id: id,
-        target: this.config.vaultAddress,
-        method: 'bring_liquidity',
-        packedArguments: []
-      })],
-      callConstructor: this.getBringLiquidityCall().bind(this)
-    });
-  }
-
-  getApproveAdapter(token: ContractAddr, spender: ContractAddr, id: string): () => AdapterLeafType<ApproveCallParams> {
-    return () => ({
-      leaves: [this.constructSimpleLeafData({
-        id: id,
-        target: token,
-        method: 'approve',
-        packedArguments: [
-          spender.toBigInt(), // spender
-        ]
-      })],
-      callConstructor: this.getApproveCall(token, spender).bind(this)
-    });
-  }
-
-  getApproveCall(token: ContractAddr, spender: ContractAddr) {
-    return async (params: ApproveCallParams) => {
-      const uint256Amount = uint256.bnToUint256(params.amount.toWei());
-      console.log(`${CommonAdapter.name}::getApproveCall token: ${token}, spender: ${spender}, amount: ${params.amount}`);
-      return [{
-        sanitizer: SIMPLE_SANITIZER,
-        call: {
-          contractAddress: token,
-          selector: hash.getSelectorFromName('approve'),
-          calldata: [
-            spender.toBigInt(), // spender
-            toBigInt(uint256Amount.low.toString()), // amount low
-            toBigInt(uint256Amount.high.toString()), // amount high
-          ]
-        }
-      }]
-    }
-  }
-
-
-  getBringLiquidityCall(): GenerateCallFn<ApproveCallParams> {
-    return async (params: ApproveCallParams) => {
-      const uint256Amount = uint256.bnToUint256(params.amount.toWei());
-      return [{
-        sanitizer: SIMPLE_SANITIZER,
-        call: {
-          contractAddress: this.config.vaultAddress,
-          selector: hash.getSelectorFromName('bring_liquidity'),
-          calldata: [
-            toBigInt(uint256Amount.low.toString()), // amount low
-            toBigInt(uint256Amount.high.toString()), // amount high
-          ]
-        }
-      }]
-    }
-  }
-
-  // getAvnuAdapter(fromToken: ContractAddr, toToken: ContractAddr, id: string, isMiddleware: boolean): () => AdapterLeafType<AvnuSwapCallParams> {
-  //   return () => ({
-  //     leaf: this.constructSimpleLeafData({
-  //       id: id,
-  //       target: isMiddleware ? AVNU_MIDDLEWARE : AVNU_EXCHANGE,
-  //       method: 'multi_route_swap',
-  //       packedArguments: [
-  //         fromToken.toBigInt(),
-  //         toToken.toBigInt(),
-  //         this.config.vaultAllocator.toBigInt(),
-  //       ]
-  //     }),
-  //     callConstructor: this.getAvnuCall(fromToken, toToken, isMiddleware).bind(this)
-  //   });
-  // }
-
-  // getAvnuCall(fromToken: ContractAddr, toToken: ContractAddr, isMiddleware: boolean): GenerateCallFn<AvnuSwapCallParams> {
-  //   return (params: AvnuSwapCallParams): ManageCall => {
-  //     return {
-  //       sanitizer: SIMPLE_SANITIZER,
-  //       call: {
-  //         contractAddress: isMiddleware ? AVNU_MIDDLEWARE : AVNU_EXCHANGE,
-  //         selector: hash.getSelectorFromName('multi_route_swap'),
-  //         calldata: [
-  //           fromToken.toBigInt(), // sell_token_address
-  //           toBigInt(params.props.token_from_amount.low.toString()), // sell_token_amount low
-  //           toBigInt(params.props.token_from_amount.high.toString()), // sell_token_amount high
-  //           toToken.toBigInt(), // buy_token_address
-  //           toBigInt(params.props.token_to_amount.low.toString()), // buy_token_amount low
-  //           toBigInt(params.props.token_to_amount.high.toString()), // buy_token_amount high
-  //           toBigInt(params.props.token_to_min_amount.low.toString()), // buy_token_min_amount low
-  //           toBigInt(params.props.token_to_min_amount.high.toString()), // buy_token_min_amount high
-  //           this.config.vaultAllocator.toBigInt(), // beneficiary
-  //           toBigInt(0), // integrator_fee_amount_bps
-  //           this.config.vaultAllocator.toBigInt(), // integrator_fee_recipient
-
-  //           // unpack routes
-  //           BigInt(params.props.routes.length),
-  //           ...params.props.routes.map(r => ([
-  //             BigInt(num.hexToDecimalString(r.token_from)),
-  //             BigInt(num.hexToDecimalString(r.token_to)),
-  //             BigInt(num.hexToDecimalString(r.exchange_address)),
-  //             BigInt(r.percent),
-  //             BigInt(r.additional_swap_params.length),
-  //             ...r.additional_swap_params.map(p => BigInt(num.hexToDecimalString(p)))
-  //           ])).flat()
-  //         ]
-  //       }
-  //     }
-  //   }
-  // }
+import { ContractAddr, Web3Number } from "@/dataTypes";
+import { LeafData } from "@/utils";
+import { Call, hash, num, shortString, uint256 } from "starknet";
+import { AVNU_EXCHANGE, AVNU_MIDDLEWARE, SIMPLE_SANITIZER, toBigInt } from "./adapter-utils";
+import { AdapterLeafType, BaseAdapter, GenerateCallFn, LeafAdapterFn, ManageCall } from "./baseAdapter";
+import { SwapInfo } from "@/modules";
+
+export interface FlashloanCallParams {
+  amount: Web3Number,
+  data: bigint[]
+}
+export interface ApproveCallParams {
+  amount: Web3Number,
+}
+export interface AvnuSwapCallParams {
+  props: SwapInfo
+}
+
+export interface CommonAdapterConfig {
+  id: string,
+  vaultAddress: ContractAddr,
+  vaultAllocator: ContractAddr,
+  manager: ContractAddr,
+  asset: ContractAddr
+}
+
+export class CommonAdapter {
+  config: CommonAdapterConfig;
+
+  constructor(config: CommonAdapterConfig) {
+    this.config = config;
+  }
+
+  protected constructSimpleLeafData(params: {
+    id: string,
+    target: ContractAddr,
+    method: string,
+    packedArguments: bigint[],
+  }, sanitizer: ContractAddr = SIMPLE_SANITIZER): LeafData {
+    const { id, target, method, packedArguments } = params;
+    return {
+      id: BigInt(num.getDecimalString(shortString.encodeShortString(id))),
+      readableId: id,
+      data: [
+        sanitizer.toBigInt(), // sanitizer address
+        target.toBigInt(), // contract
+        toBigInt(hash.getSelectorFromName(method)), // method name
+        BigInt(packedArguments.length),
+        ...packedArguments
+      ]
+    };
+  }
+
+  async getFlashloanAdapter(): Promise<AdapterLeafType<FlashloanCallParams>> {
+    const manageCall = this.getFlashloanCall.bind(this)({amount: Web3Number.fromWei('0', 6), data: []});
+    const packedArguments: bigint[] = [
+      this.config.manager.toBigInt(), // receiver
+      this.config.asset.toBigInt(), // asset
+      toBigInt(0), // is legacy false
+    ];
+    const leaf = this.constructSimpleLeafData({
+      id: this.config.id,
+      target: (await manageCall)[0].call.contractAddress,
+      method: 'flash_loan',
+      packedArguments
+    });
+    return { leaves: [leaf], callConstructor: this.getFlashloanCall.bind(this) };
+  }
+
+  async getFlashloanCall(params: FlashloanCallParams): Promise<ManageCall[]> {
+    const uint256Amount = uint256.bnToUint256(params.amount.toWei());
+    return [{
+      proofReadableId: this.config.id,
+      sanitizer: SIMPLE_SANITIZER,
+      call: {
+        contractAddress: this.config.manager,
+        selector: hash.getSelectorFromName('flash_loan'),
+        calldata: [
+          this.config.manager.toBigInt(), // receiver
+          this.config.asset.toBigInt(), // asset
+          toBigInt(uint256Amount.low.toString()), // amount low
+          toBigInt(uint256Amount.high.toString()), // amount high
+          toBigInt(0), // is legacy false
+          BigInt(params.data.length),
+          ...params.data
+        ]
+      }
+    }]
+  }
+
+  getBringLiquidityAdapter(id: string): () => AdapterLeafType<ApproveCallParams> {
+    return () => ({
+      leaves: [this.constructSimpleLeafData({
+        id: id,
+        target: this.config.vaultAddress,
+        method: 'bring_liquidity',
+        packedArguments: []
+      })],
+      callConstructor: this.getBringLiquidityCall(id).bind(this)
+    });
+  }
+
+  getApproveAdapter(token: ContractAddr, spender: ContractAddr, id: string): () => AdapterLeafType<ApproveCallParams> {
+    return () => ({
+      leaves: [this.constructSimpleLeafData({
+        id: id,
+        target: token,
+        method: 'approve',
+        packedArguments: [
+          spender.toBigInt(), // spender
+        ]
+      })],
+      callConstructor: this.getApproveCall(token, spender, id).bind(this)
+    });
+  }
+
+  getApproveCall(token: ContractAddr, spender: ContractAddr, proofReadableId: string) {
+    return async (params: ApproveCallParams) => {
+      const uint256Amount = uint256.bnToUint256(params.amount.toWei());
+      console.log(`${CommonAdapter.name}::getApproveCall token: ${token}, spender: ${spender}, amount: ${params.amount}`);
+      return [{
+        proofReadableId,
+        sanitizer: SIMPLE_SANITIZER,
+        call: {
+          contractAddress: token,
+          selector: hash.getSelectorFromName('approve'),
+          calldata: [
+            spender.toBigInt(), // spender
+            toBigInt(uint256Amount.low.toString()), // amount low
+            toBigInt(uint256Amount.high.toString()), // amount high
+          ]
+        }
+      }]
+    }
+  }
+
+
+  getBringLiquidityCall(proofReadableId: string): GenerateCallFn<ApproveCallParams> {
+    return async (params: ApproveCallParams) => {
+      const uint256Amount = uint256.bnToUint256(params.amount.toWei());
+      return [{
+        proofReadableId,
+        sanitizer: SIMPLE_SANITIZER,
+        call: {
+          contractAddress: this.config.vaultAddress,
+          selector: hash.getSelectorFromName('bring_liquidity'),
+          calldata: [
+            toBigInt(uint256Amount.low.toString()), // amount low
+            toBigInt(uint256Amount.high.toString()), // amount high
+          ]
+        }
+      }]
+    }
+  }
+
+  // getAvnuAdapter(fromToken: ContractAddr, toToken: ContractAddr, id: string, isMiddleware: boolean): () => AdapterLeafType<AvnuSwapCallParams> {
+  //   return () => ({
+  //     leaf: this.constructSimpleLeafData({
+  //       id: id,
+  //       target: isMiddleware ? AVNU_MIDDLEWARE : AVNU_EXCHANGE,
+  //       method: 'multi_route_swap',
+  //       packedArguments: [
+  //         fromToken.toBigInt(),
+  //         toToken.toBigInt(),
+  //         this.config.vaultAllocator.toBigInt(),
+  //       ]
+  //     }),
+  //     callConstructor: this.getAvnuCall(fromToken, toToken, isMiddleware).bind(this)
+  //   });
+  // }
+
+  // getAvnuCall(fromToken: ContractAddr, toToken: ContractAddr, isMiddleware: boolean): GenerateCallFn<AvnuSwapCallParams> {
+  //   return (params: AvnuSwapCallParams): ManageCall => {
+  //     return {
+  //       sanitizer: SIMPLE_SANITIZER,
+  //       call: {
+  //         contractAddress: isMiddleware ? AVNU_MIDDLEWARE : AVNU_EXCHANGE,
+  //         selector: hash.getSelectorFromName('multi_route_swap'),
+  //         calldata: [
+  //           fromToken.toBigInt(), // sell_token_address
+  //           toBigInt(params.props.token_from_amount.low.toString()), // sell_token_amount low
+  //           toBigInt(params.props.token_from_amount.high.toString()), // sell_token_amount high
+  //           toToken.toBigInt(), // buy_token_address
+  //           toBigInt(params.props.token_to_amount.low.toString()), // buy_token_amount low
+  //           toBigInt(params.props.token_to_amount.high.toString()), // buy_token_amount high
+  //           toBigInt(params.props.token_to_min_amount.low.toString()), // buy_token_min_amount low
+  //           toBigInt(params.props.token_to_min_amount.high.toString()), // buy_token_min_amount high
+  //           this.config.vaultAllocator.toBigInt(), // beneficiary
+  //           toBigInt(0), // integrator_fee_amount_bps
+  //           this.config.vaultAllocator.toBigInt(), // integrator_fee_recipient
+
+  //           // unpack routes
+  //           BigInt(params.props.routes.length),
+  //           ...params.props.routes.map(r => ([
+  //             BigInt(num.hexToDecimalString(r.token_from)),
+  //             BigInt(num.hexToDecimalString(r.token_to)),
+  //             BigInt(num.hexToDecimalString(r.exchange_address)),
+  //             BigInt(r.percent),
+  //             BigInt(r.additional_swap_params.length),
+  //             ...r.additional_swap_params.map(p => BigInt(num.hexToDecimalString(p)))
+  //           ])).flat()
+  //         ]
+  //       }
+  //     }
+  //   }
+  // }
 }