import { bytesToHex, randomBytes } from "@noble/hashes/utils";
import type { Hex } from "viem";
import { parseEventLogs } from "../event/actions/parse-logs.js";
import { createdEvent } from "../extensions/tokens/__generated__/ERC20Entrypoint/events/Created.js";
import { createById } from "../extensions/tokens/__generated__/ERC20Entrypoint/write/createById.js";
import type { TransactionReceipt } from "../transaction/types.js";
import { padHex, toHex } from "../utils/encoding/hex.js";
import { DEFAULT_DEVELOPER_ADDRESS } from "./constants.js";
import { getDeployedEntrypointERC20 } from "./get-entrypoint-erc20.js";
import {
  encodeInitParams,
  encodePoolConfig,
  generateSalt,
} from "./token-utils.js";
import type { CreateTokenOptions } from "./types.js";

export async function getTokenAddressFromReceipt(receipt: TransactionReceipt) {
  const assetEvent = createdEvent();
  const decodedEvent = parseEventLogs({
    events: [assetEvent],
    logs: receipt.logs,
  });

  if (decodedEvent.length === 0 || !decodedEvent[0]) {
    throw new Error(
      `No AssetCreated event found in transaction: ${receipt.transactionHash}`,
    );
  }

  return decodedEvent[0]?.args.asset;
}

export async function prepareCreateToken(options: CreateTokenOptions) {
  const { client, params, account, launchConfig } = options;

  const creator = params.owner || account.address;
  const encodedInitData = await encodeInitParams({
    client,
    creator,
    params,
  });

  const salt: Hex = generateSalt(options.salt || bytesToHex(randomBytes(31)));

  const entrypoint = await getDeployedEntrypointERC20(options);

  let hookData: Hex = "0x";
  let contractId = padHex(toHex("ERC20Asset"), { size: 32 });
  if (launchConfig?.kind === "pool") {
    hookData = encodePoolConfig(launchConfig.config);
    contractId = padHex(toHex("ERC20Asset_Pool"), { size: 32 });
  }

  const transaction = createById({
    contract: entrypoint,
    contractId,
    params: {
      data: encodedInitData,
      hookData,
      developer: options.developerAddress || DEFAULT_DEVELOPER_ADDRESS,
      salt,
    },
    creator,
  });

  return transaction;
}