route.ts

import { CHAIN } from "@/utils/constants";
import { CompassApiSDK } from "@compass-labs/api-sdk";
import { type UnsignedTransaction } from "@compass-labs/api-sdk/models/components";
import { createPublicClient, createWalletClient, http } from "viem";
import { privateKeyToAccount } from "viem/accounts";
import { base } from "viem/chains";

export async function POST(request: Request) {
  try {
    const { vaultAddress, amount, token, isAll } = await request.json();

    if (!vaultAddress || !amount || !token || typeof isAll !== "boolean") {
      throw new Error("Missing body parameters.");
    }

    const account = privateKeyToAccount(
      process.env.PRIVATE_KEY as `0x${string}`
    );

    const walletClient = createWalletClient({
      account,
      chain: base,
      transport: http(process.env.RPC_URL),
    });

    const publicClient = createPublicClient({
      chain: base,
      transport: http(process.env.RPC_URL),
    });

    const compassApiSDK = new CompassApiSDK({
      apiKeyAuth: process.env.COMPASS_API_KEY,
    });

    const withdraw = await compassApiSDK.earn.earnManage({
      owner: account.address,
      chain: CHAIN,
      venue: {
        type: "VAULT",
        vaultAddress,
      },
      action: "WITHDRAW",
      amount: isAll ? "ALL" : amount,
    });

    const transaction = withdraw.transaction as UnsignedTransaction;

    if (!transaction) {
      throw new Error("No transaction returned from earnManage");
    }

    const withdrawTxHash = await walletClient.sendTransaction({
      ...(transaction as any),
      value: BigInt(transaction.value),
      gas: transaction.gas ? BigInt(transaction.gas) : undefined,
      maxFeePerGas: BigInt(transaction.maxFeePerGas),
      maxPriorityFeePerGas: BigInt(transaction.maxPriorityFeePerGas),
    });

    const tx = await publicClient.waitForTransactionReceipt({
      hash: withdrawTxHash,
    });

    console.log("tx", tx);

    if (tx.status !== "success") {
      throw new Error("Withdraw transaction reverted.");
    }

    return new Response("", {
      status: 200,
    });
  } catch (error) {
    console.log("error", error);
    if (error instanceof Error) {
      return new Response(JSON.stringify({ error: error.message }), {
        status: 400,
      });
    } else {
      return new Response("Error", {
        status: 400,
      });
    }
  }
}