import { expect, test } from "vitest";

import { TEST_CLIENT } from "~test/test-clients.js";
import { ANVIL_PKEY_A } from "~test/test-wallets.js";
import { typedData } from "~test/typed-data.js";
import { ANVIL_CHAIN } from "../../test/src/chains.js";
import { sendTransaction } from "../exports/thirdweb.js";
import { prepareTransaction } from "../transaction/prepare-transaction.js";
import { toUnits } from "../utils/units.js";
import { privateKeyToAccount } from "./private-key.js";
import { getWalletBalance } from "./utils/getWalletBalance.js";

test("default", () => {
  expect(
    privateKeyToAccount({ client: TEST_CLIENT, privateKey: ANVIL_PKEY_A }),
  ).toMatchInlineSnapshot(`
    {
      "address": "0xf39Fd6e51aad88F6F4ce6aB8827279cffFb92266",
      "sendTransaction": [Function],
      "signAuthorization": [Function],
      "signMessage": [Function],
      "signTransaction": [Function],
      "signTypedData": [Function],
    }
  `);
});

test("sign message", async () => {
  const account = privateKeyToAccount({
    client: TEST_CLIENT,
    privateKey: ANVIL_PKEY_A,
  });
  expect(
    await account.signMessage({ message: "hello world" }),
  ).toMatchInlineSnapshot(
    '"0xa461f509887bd19e312c0c58467ce8ff8e300d3c1a90b608a760c5b80318eaf15fe57c96f9175d6cd4daad4663763baa7e78836e067d0163e9a2ccf2ff753f5b1b"',
  );
});

test("sign transaction", async () => {
  const account = privateKeyToAccount({
    client: TEST_CLIENT,
    privateKey: ANVIL_PKEY_A,
  });

  if (!account.signTransaction) {
    // throwing an error instead of testing for a definition to make the TS compiler happy on the subsequent expect
    throw new Error("signTransaction should be defined for local accounts");
  }

  expect(
    await account.signTransaction({
      chainId: 1,
      gas: 21000n,
      maxFeePerGas: toUnits("20", 9),
      to: "0x70997970c51812dc3a010c7d01b50e0d17dc79c8",
      value: toUnits("1", 18),
    }),
  ).toMatchInlineSnapshot(
    '"0x02f86f0180808504a817c8008252089470997970c51812dc3a010c7d01b50e0d17dc79c8880de0b6b3a764000080c001a0f40a2d2ae9638056cafbe9083c7125edc8555e0e715db0984dd859a5c6dfac57a020f36fd0b32bef4d6d75c62f220e59c5fb60c244ca3b361e750985ee5c3a0931"',
  );
});

test("sign typed data", async () => {
  const account = privateKeyToAccount({
    client: TEST_CLIENT,
    privateKey: ANVIL_PKEY_A,
  });
  expect(
    await account.signTypedData({ ...typedData.basic, primaryType: "Mail" }),
  ).toMatchInlineSnapshot(
    '"0x32f3d5975ba38d6c2fba9b95d5cbed1febaa68003d3d588d51f2de522ad54117760cfc249470a75232552e43991f53953a3d74edf6944553c6bef2469bb9e5921b"',
  );
});

test("send transaction", async () => {
  const account = privateKeyToAccount({
    client: TEST_CLIENT,
    privateKey: ANVIL_PKEY_A,
  });
  const startingBalance = await getWalletBalance({
    address: account.address,
    chain: ANVIL_CHAIN,
    client: TEST_CLIENT,
  });
  const startingBalanceRecipient = await getWalletBalance({
    address: "0x70997970c51812dc3a010c7d01b50e0d17dc79c8",
    chain: ANVIL_CHAIN,
    client: TEST_CLIENT,
  });
  const tx = prepareTransaction({
    chain: ANVIL_CHAIN,
    client: TEST_CLIENT,
    to: "0x70997970c51812dc3a010c7d01b50e0d17dc79c8",
    value: toUnits("1", 18),
  });

  expect(
    await sendTransaction({
      account,
      transaction: tx,
    }),
  ).toBeDefined();

  const endingBalance = await getWalletBalance({
    address: account.address,
    chain: ANVIL_CHAIN,
    client: TEST_CLIENT,
  });
  const endingBalanceRecipient = await getWalletBalance({
    address: "0x70997970c51812dc3a010c7d01b50e0d17dc79c8",
    chain: ANVIL_CHAIN,
    client: TEST_CLIENT,
  });
  expect(endingBalance.value).toBeLessThan(startingBalance.value);
  expect(endingBalanceRecipient.value).toBeGreaterThan(
    startingBalanceRecipient.value,
  );
});