import { HardhatRuntimeEnvironment } from "hardhat/types";
import { DeployFunction } from "hardhat-deploy/types";
import { Contract } from "ethers";

/**
 * Deploys a contract named "Vendor" using the deployer account and
 * constructor arguments set to the deployer address
 *
 * @param hre HardhatRuntimeEnvironment object.
 */
const deployVendor: DeployFunction = async function (hre: HardhatRuntimeEnvironment) {
  /*
    On localhost, the deployer account is the one that comes with Hardhat, which is already funded.

    When deploying to live networks (e.g `yarn deploy --network sepolia`), the deployer account
    should have sufficient balance to pay for the gas fees for contract creation.

    You can generate a random account with `yarn generate` which will fill DEPLOYER_PRIVATE_KEY
    with a random private key in the .env file (then used on hardhat.config.ts)
    You can run the `yarn account` command to check your balance in every network.
  */
  const { deployer } = await hre.getNamedAccounts();
  const { deploy } = hre.deployments;
  const yourToken = await hre.ethers.getContract<Contract>("YourToken", deployer);

  /**
   * Student TODO:
   * - Put the address you’re using in the frontend here (leave "" to default to the deployer)
   */
  const FRONTEND_ADDRESS: string = "";

  /**
   * Mode switch:
   * - If true: deploy Vendor and seed it with the token balance
   * - If false: send tokens to your frontend address (or deployer if unset)
   */
  const SEND_TOKENS_TO_VENDOR = false; // Don't switch until Checkpoint 2!

  const recipientAddress = FRONTEND_ADDRESS && FRONTEND_ADDRESS.trim().length > 0 ? FRONTEND_ADDRESS : deployer;

  if (!SEND_TOKENS_TO_VENDOR) {
    // Send the entire initial supply to the wallet you use in the UI (useful when deployer != UI wallet).
    // If FRONTEND_ADDRESS is "", this defaults to the deployer (no-op transfer).
    if (recipientAddress != deployer) {
      await yourToken.transfer(recipientAddress, hre.ethers.parseEther("1000"));
    }
    return;
  } else {
    // Deploy Vendor
    const yourTokenAddress = await yourToken.getAddress();
    await deploy("Vendor", {
      from: deployer,
      // Contract constructor arguments
      args: [yourTokenAddress],
      log: true,
      // autoMine: can be passed to the deploy function to make the deployment process faster on local networks by
      // automatically mining the contract deployment transaction. There is no effect on live networks.
      autoMine: true,
    });
    const vendor = await hre.ethers.getContract<Contract>("Vendor", deployer);
    const vendorAddress = await vendor.getAddress();

    // Transfer tokens to Vendor (seed inventory)
    await yourToken.transfer(vendorAddress, hre.ethers.parseEther("1000"));

    // Make the UI wallet the owner (for withdraw(), etc). Defaults to deployer if unset.
    await vendor.transferOwnership(recipientAddress);
  }
};

export default deployVendor;

// Tags are useful if you have multiple deploy files and only want to run one of them.
// e.g. yarn deploy --tags Vendor
deployVendor.tags = ["Vendor"];