@morpho-dev/router 0.1.17 → 0.1.18

package/README.md

@@ -19,17 +19,17 @@ pnpm add -g @morpho-dev/router
 Examples:
 
 ```bash
-# Router with seeded random offers
-router start --seed 100
-
 # Router with postgres database and custom port
 router start --db-endpoint <db url> --port 8080
 
-# Router with offers loaded from file
-router start --file offers.json
-
 # Start Router indexer (Default indexer,api)
 router start --services indexer
+
+# Router Mock with seeded random offers
+router mock --seed 100
+
+# Router with offers loaded from file
+router mock --file offers.json
 ```
 
 <!-- AUTO-GENERATED: CLI-HELP START -->

package/dist/cli.js

@@ -41,7 +41,7 @@ var __export = (target, all) => {
 // package.json
 var package_default = {
   name: "@morpho-dev/router",
-  version: "0.1.17",
+  version: "0.1.18",
   description: "Router package for Morpho protocol"};
 
 // src/core/Chain.ts
@@ -142,8 +142,12 @@ var chains = {
       [
         "0xA0b86991c6218b36c1d19D4a2e9Eb0cE3606eB48",
         // USDC
-        "0x6B175474E89094C44Da98b954EedeAC495271d0F"
+        "0x6B175474E89094C44Da98b954EedeAC495271d0F",
         // DAI
+        "0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2",
+        // WETH
+        "0x2260FAC5E5542a773Aa44fBCfeDf7C193bc2C599"
+        // WBTC
       ].map((address) => address.toLowerCase())
     ),
     morpho: "0x0000000000000000000000000000000000000000",
@@ -166,8 +170,12 @@ var chains = {
       [
         "0x833589fCD6eDb6E08f4c7C32D4f71b54bdA02913",
         // USDC
-        "0x50c5725949A6F0c72E6C4a641F24049A917DB0Cb"
+        "0x50c5725949A6F0c72E6C4a641F24049A917DB0Cb",
         // DAI
+        "0x4200000000000000000000000000000000000006",
+        // WETH
+        "0x2260FAC5E5542a773Aa44fBCfeDf7C193bc2C599"
+        // WBTC
       ].map((address) => address.toLowerCase())
     ),
     morpho: "0x0000000000000000000000000000000000000000",
@@ -190,8 +198,12 @@ var chains = {
       [
         "0xA0b86991c6218b36c1d19D4a2e9Eb0cE3606eB48",
         // USDC
-        "0x6B175474E89094C44Da98b954EedeAC495271d0F"
+        "0x6B175474E89094C44Da98b954EedeAC495271d0F",
         // DAI
+        "0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2",
+        // WETH
+        "0x2260FAC5E5542a773Aa44fBCfeDf7C193bc2C599"
+        // WBTC
       ].map((address) => address.toLowerCase())
     ),
     morpho: "0x11a002d45db720ed47a80d2f3489cba5b833eaf5",
@@ -215,8 +227,12 @@ var chains = {
       [
         "0xA0b86991c6218b36c1d19D4a2e9Eb0cE3606eB48",
         // USDC
-        "0x6B175474E89094C44Da98b954EedeAC495271d0F"
+        "0x6B175474E89094C44Da98b954EedeAC495271d0F",
         // DAI
+        "0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2",
+        // WETH
+        "0x2260FAC5E5542a773Aa44fBCfeDf7C193bc2C599"
+        // WBTC
       ].map((address) => address.toLowerCase())
     ),
     morpho: "0x23DFBc4B8B80C14CC5e25011B8491f268395BAd6",
@@ -1215,39 +1231,92 @@ function fromSnakeCase3(input) {
 function toSnakeCase2(offer) {
   return toSnakeCase(offer);
 }
-function random2() {
-  const loanToken = privateKeyToAccount(generatePrivateKey()).address;
-  const maturity = from3("end_of_month");
-  const expiry = from3("end_of_week") - 1;
-  const lltv = from(0.965);
+function random2(config) {
+  const chain = config?.chains ? config.chains[Math.floor(Math.random() * config.chains.length)] : chains.ethereum;
+  const loanToken = config?.loanTokens ? config.loanTokens[Math.floor(Math.random() * config.loanTokens.length)] : privateKeyToAccount(generatePrivateKey()).address;
+  const collateralCandidates = config?.collateralTokens ? config.collateralTokens.filter((a) => a !== loanToken) : [privateKeyToAccount(generatePrivateKey()).address];
+  const collateralAsset = collateralCandidates[Math.floor(Math.random() * collateralCandidates.length)];
+  const maturityOption = weightedChoice([
+    ["end_of_month", 1],
+    ["end_of_next_month", 1]
+  ]);
+  const maturity = config?.maturity ?? from3(maturityOption);
+  const lltv = from(
+    weightedChoice([
+      [0.385, 1],
+      [0.5, 1],
+      [0.625, 2],
+      [0.77, 8],
+      [0.86, 10],
+      [0.915, 8],
+      [0.945, 6],
+      [0.965, 4],
+      [0.98, 2]
+    ])
+  );
+  const buy = config?.buy !== void 0 ? config.buy : Math.random() > 0.5;
+  const ONE = 1000000000000000000n;
+  const qMin = buy ? 16 : 4;
+  const qMax = buy ? 32 : 16;
+  const len = qMax - qMin + 1;
+  const ratePairs = Array.from(
+    { length: len },
+    (_, idx) => {
+      const q = qMin + idx;
+      const scaledRate = BigInt(q) * (ONE / 4n);
+      const weight = buy ? 1 + idx : 1 + (len - 1 - idx);
+      return [scaledRate, weight];
+    }
+  );
+  const rate = config?.rate ?? weightedChoice(ratePairs);
+  const loanTokenDecimals = config?.assetsDecimals?.[loanToken] ?? 18;
+  const unit = BigInt(10) ** BigInt(loanTokenDecimals);
+  const amountBase = BigInt(100 + Math.floor(Math.random() * (1e6 - 100 + 1)));
+  const assetsScaled = config?.assets ?? amountBase * unit;
+  const consumed2 = config?.consumed !== void 0 ? config.consumed : Math.random() < 0.8 ? 0n : assetsScaled * BigInt(1 + Math.floor(Math.random() * 900)) / 1000n;
+  const callbackBySide = (() => {
+    if (buy) return { address: zeroAddress, data: "0x", gasLimit: 0n };
+    const sellCallbackAddress = WhitelistedCallbackAddresses["sell_erc20_callback" /* SellERC20Callback */][0].toLowerCase();
+    const amount = assetsScaled * 1000000000000000000000n;
+    const data = encodeSellERC20Callback({
+      collaterals: [collateralAsset],
+      amounts: [amount]
+    });
+    return { address: sellCallbackAddress, data, gasLimit: 500000n };
+  })();
   const offer = from5({
-    offering: privateKeyToAccount(generatePrivateKey()).address,
-    assets: BigInt(Math.floor(Math.random() * 1e6)),
-    rate: BigInt(Math.floor(Math.random() * 1e6)),
+    offering: config?.offering ?? privateKeyToAccount(generatePrivateKey()).address,
+    assets: assetsScaled,
+    rate,
     maturity,
-    expiry,
-    start: expiry - 10,
+    expiry: config?.expiry ?? maturity - 1,
+    start: config?.start ?? maturity - 10,
     nonce: BigInt(Math.floor(Math.random() * 1e6)),
-    buy: Math.random() > 0.5,
-    chainId: 1n,
+    buy,
+    chainId: chain.id,
     loanToken,
-    collaterals: [
+    collaterals: config?.collaterals ?? [
       from2({
-        asset: zeroAddress,
+        asset: collateralAsset,
         oracle: zeroAddress,
         lltv
       })
     ],
-    callback: {
-      address: zeroAddress,
-      data: "0x",
-      gasLimit: 0n
-    },
-    consumed: 0n,
+    callback: config?.callback ?? callbackBySide,
+    consumed: consumed2,
     blockNumber: Math.floor(Math.random() * Number.MAX_SAFE_INTEGER)
   });
   return offer;
 }
+var weightedChoice = (pairs) => {
+  const total = pairs.reduce((sum, [, weight]) => sum + weight, 0);
+  let roll = Math.random() * total;
+  for (const [value, weight] of pairs) {
+    roll -= weight;
+    if (roll < 0) return value;
+  }
+  return pairs[0][0];
+};
 var domain = (chainId) => ({
   chainId,
   verifyingContract: zeroAddress
@@ -6173,6 +6242,7 @@ var RouterCmd = class _RouterCmd extends Command {
     command.setOptionValue("rpcUrl", rpcUrl);
     command.setOptionValue("routerServices", routerServices);
     command.setOptionValue("logger", logger);
+    command.setOptionValue("chain", chain);
     return { routerServices };
   }
   // allows to use the logger defined in the setup function for the action of the command
@@ -6187,12 +6257,116 @@ var RouterCmd = class _RouterCmd extends Command {
   }
 };
 
-// src/cli/commands/start.ts
+// src/cli/commands/mock.ts
+var mockCmd = new RouterCmd("mock");
+mockCmd.description("Start Router mock.").addOption(
+  new Option("--seed <n>", "Seed random offers to router").argParser((v) => Number(v)).default(0)
+).addOption(new Option("--file <path>", "Seed offers from a JSON file")).action(
+  async (opts) => {
+    const {
+      port,
+      routerServices: { liquidityStore, offerStore, routerApi },
+      logger,
+      chain
+    } = opts;
+    const stops = [];
+    if (opts.seed > 0) {
+      const offers2 = Array.from(Array(opts.seed ?? 0).keys()).map((_i) => ({
+        offer: Offer_exports.random({
+          chains: [chain],
+          loanTokens: Array.from(chain.whitelistedAssets.values()),
+          collateralTokens: Array.from(chain.whitelistedAssets.values()),
+          assetsDecimals: {
+            "0xA0b86991c6218b36c1d19D4a2e9Eb0cE3606eB48": 6,
+            // USDC
+            "0x6B175474E89094C44Da98b954EedeAC495271d0F": 18,
+            // DAI
+            "0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2": 18,
+            // WETH
+            "0x2260FAC5E5542a773Aa44fBCfeDf7C193bc2C599": 8,
+            // WBTC
+            "0x4200000000000000000000000000000000000006": 18,
+            // WETH
+            "0x833589fCD6eDb6E08f4c7C32D4f71b54bdA02913": 6,
+            // USDC
+            "0x50c5725949A6F0c72E6C4a641F24049A917DB0Cb": 18
+            // DAI
+          }
+        }),
+        status: "valid"
+      }));
+      await offerStore.createMany(offers2);
+      for (const offer of offers2) {
+        await liquidityStore.save({
+          pools: [
+            {
+              id: offer.offer.hash,
+              amount: offer.offer.assets
+            }
+          ],
+          offerPools: [
+            {
+              offerHash: offer.offer.hash,
+              poolId: offer.offer.hash,
+              amount: offer.offer.assets
+            }
+          ],
+          links: []
+        });
+      }
+      logger.info({ msg: `Offers seeded`, count: opts.seed });
+    }
+    if (opts.file) {
+      const offers2 = await getOffersFromFile(opts.file);
+      await offerStore.createMany(offers2.map((ro) => ({ offer: ro })));
+      for (const offer of offers2) {
+        await liquidityStore.save({
+          pools: [
+            {
+              id: offer.hash,
+              amount: offer.assets
+            }
+          ],
+          offerPools: [
+            {
+              offerHash: offer.hash,
+              poolId: offer.hash,
+              amount: offer.assets
+            }
+          ],
+          links: []
+        });
+      }
+      logger.info({ msg: `Offers seeded from file`, count: offers2.length, file: opts.file });
+    }
+    await routerApi.serve({ port });
+    logger.info({ msg: `API started`, url: `http://localhost:${port}` });
+    const shutdown = async (signal) => {
+      logger.warn({ msg: `Shutdown signal`, signal });
+      try {
+        stops.forEach((stop) => stop());
+      } catch {
+      }
+      process.exit(0);
+    };
+    process.once("SIGINT", () => void shutdown("SIGINT"));
+    process.once("SIGTERM", () => void shutdown("SIGTERM"));
+    process.stdin.resume();
+  }
+);
+async function getOffersFromFile(file) {
+  const raw = await readFile(file, "utf8");
+  const json = JSON.parse(raw);
+  const items = Array.isArray(json) ? json : json.offers ?? [];
+  if (!Array.isArray(items) || items.length === 0) {
+    console.log("No offers found in file");
+    return [];
+  }
+  return items.map((o) => Offer_exports.from(o));
+}
 var startCmd = new RouterCmd("start");
 startCmd.description("Start Router services.").addOption(
   new Option("--block-window <n>", "Block window to get logs from").argParser((v) => Number(v)).default(100)
-).addOption(
-  new Option("--seed <n>", "Seed random offers to router").argParser((v) => Number(v)).default(0)
 ).addOption(new Option("--file <path>", "Seed offers from a JSON file")).addOption(
   new Option("--max-block-number <n>", "Block number at which to stop indexing").argParser(
     (v) => Number(v)
@@ -6205,24 +6379,10 @@ startCmd.description("Start Router services.").addOption(
   async (opts) => {
     const {
       port,
-      routerServices: { offerStore, indexer, routerApi },
+      routerServices: { indexer, routerApi },
       logger
     } = opts;
     const stops = [];
-    if (opts.seed > 0) {
-      await offerStore.createMany(
-        Array.from(Array(opts.seed ?? 0).keys()).map((_i) => ({
-          offer: Offer_exports.random(),
-          status: "valid"
-        }))
-      );
-      logger.info({ msg: `Offers seeded`, count: opts.seed });
-    }
-    if (opts.file) {
-      const offers2 = await getOffersFromFile(opts.file);
-      await offerStore.createMany(offers2.map((ro) => ({ offer: ro })));
-      logger.info({ msg: `Offers seeded from file`, count: offers2.length, file: opts.file });
-    }
     const selectedServices = new Set(opts.services);
     if (selectedServices.has("indexer")) {
       stops.push(await indexer.start());
@@ -6245,16 +6405,6 @@ startCmd.description("Start Router services.").addOption(
     process.stdin.resume();
   }
 );
-async function getOffersFromFile(file) {
-  const raw = await readFile(file, "utf8");
-  const json = JSON.parse(raw);
-  const items = Array.isArray(json) ? json : json.offers ?? [];
-  if (!Array.isArray(items) || items.length === 0) {
-    console.log("No offers found in file");
-    return [];
-  }
-  return items.map((o) => Offer_exports.from(o));
-}
 var tunnelCmd = new Command("tunnel");
 tunnelCmd.description("Expose the local Router via ngrok").addOption(
   new Option("--port <n>").env("ROUTER_PORT").default(7891).argParser((v) => Number(v))
@@ -6286,6 +6436,7 @@ cli.name(package_default.name.split("/")[1] || "cli").description(package_defaul
 cli.addCommand(startCmd);
 cli.addCommand(tunnelCmd);
 cli.addCommand(mempoolCmd);
+cli.addCommand(mockCmd);
 cli.command("__commands", { hidden: true }).description("internal: list commands").action(() => {
   const names2 = cli.commands.map((c) => c.name());
   console.log(JSON.stringify(names2));