import { Interface } from "@ethersproject/abi";
import { Contract } from "@ethersproject/contracts";
import { keccak256 as solidityKeccak256 } from "@ethersproject/solidity";
/**
 * Balancer Vault partial ABI interface.
 *
 * This definition only contains the Vault methods that are used by GPv2 Vault
 * relayer. It is copied here to avoid relying on build artifacts.
 */
export const VAULT_INTERFACE = new Interface([
  "function manageUserBalance((uint8, address, uint256, address, address)[])",
  "function batchSwap(uint8, (bytes32, uint256, uint256, uint256, bytes)[], address[], (address, bool, address, bool), int256[], uint256)",
]);

/**
 * Grants the required roles to the specified Vault relayer.
 *
 * This method is intended to be called by the Balancer Vault admin, and **not**
 * traders. It is included in the exported TypeScript library for completeness
 * and "documentation".
 *
 * @param authorizer The Vault authorizer contract that manages access.
 * @param vaultAddress The address to the Vault.
 * @param vaultRelayerAddress The address to the GPv2 Vault relayer contract.
 */
export async function grantRequiredRoles(
  authorizer: Contract,
  vaultAddress: string,
  vaultRelayerAddress: string,
): Promise<void> {
  for (const name in VAULT_INTERFACE.functions) {
    await authorizer.grantRole(
      solidityKeccak256(
        ["uint256", "bytes4"],
        [vaultAddress, VAULT_INTERFACE.getSighash(name)],
      ),
      vaultRelayerAddress,
    );
  }
}