import {
    TASK_COMPILE_SOLIDITY_GET_DEPENDENCY_GRAPH,
    TASK_FLATTEN_GET_FLATTENED_SOURCE,
} from 'hardhat/builtin-tasks/task-names';
import { Artifacts, CompilerInput, DependencyGraph, HardhatRuntimeEnvironment, ResolvedFile } from 'hardhat/types';
import { isFullyQualifiedName, parseFullyQualifiedName } from 'hardhat/utils/contract-names';
import path from 'path';
import { isBreakableCompilerVersion } from '@matterlabs/hardhat-zksync-solc/dist/src/utils';
import { CONTRACT_NAME_NOT_FOUND, NO_MATCHING_CONTRACT, LIBRARIES_EXPORT_ERROR } from './constants';
import { Bytecode, extractMatchingContractInformation } from './solc/bytecode';
import { ZkSyncVerifyPluginError } from './errors';

export async function inferContractArtifacts(
    artifacts: Artifacts,
    matchingCompilerVersions: string[],
    deployedBytecode: Bytecode,
): Promise<any> {
    const contractMatches = [];
    const fqNames = await artifacts.getAllFullyQualifiedNames();

    for (const fqName of fqNames) {
        const buildInfo: any = await artifacts.getBuildInfo(fqName);

        if (buildInfo === undefined) {
            continue;
        }

        if (!matchingCompilerVersions.includes(buildInfo.solcVersion)) {
            continue;
        }

        const { sourceName, contractName } = parseFullyQualifiedName(fqName);

        const contractInformation = await extractMatchingContractInformation(
            sourceName,
            contractName,
            buildInfo,
            deployedBytecode,
        );
        if (contractInformation !== null) {
            contractMatches.push(contractInformation);
            break;
        }
    }

    if (contractMatches.length === 0) throw new ZkSyncVerifyPluginError(NO_MATCHING_CONTRACT);

    return contractMatches[0];
}

export async function flattenContractFile(hre: HardhatRuntimeEnvironment, filePath: string): Promise<string> {
    return await hre.run(TASK_FLATTEN_GET_FLATTENED_SOURCE, {
        files: [filePath],
    });
}

export async function checkContractName(artifacts: Artifacts, contractFQN: string) {
    if (contractFQN !== undefined) {
        if (!isFullyQualifiedName(contractFQN)) {
            throw new ZkSyncVerifyPluginError(
                `A valid fully qualified name was expected. Fully qualified names look like this: "contracts/AContract.sol:TheContract"
Instead, this name was received: ${contractFQN}`,
            );
        }

        if (!(await artifacts.artifactExists(contractFQN))) {
            throw new ZkSyncVerifyPluginError(`The contract ${contractFQN} is not present in your project.`);
        }
    } else {
        throw new ZkSyncVerifyPluginError(CONTRACT_NAME_NOT_FOUND);
    }
}

export async function getMinimalResolvedFiles(
    hre: HardhatRuntimeEnvironment,
    sourceName: string,
): Promise<ResolvedFile[]> {
    const dependencyGraph: DependencyGraph = await hre.run(TASK_COMPILE_SOLIDITY_GET_DEPENDENCY_GRAPH, {
        sourceNames: [sourceName],
    });

    return dependencyGraph.getResolvedFiles();
}

export function getSolidityStandardJsonInput(
    hre: HardhatRuntimeEnvironment,
    resolvedFiles: ResolvedFile[],
    input: CompilerInput,
): any {
    const standardInput = {
        language: input.language,
        sources: Object.fromEntries(
            resolvedFiles.map((file) => [file.sourceName, { content: file.content.rawContent }]),
        ),
        settings: {},
    };
    standardInput.settings = !isBreakableCompilerVersion(hre.config.zksolc.version)
        ? {
              ...input.settings,
              isSystem: hre.config.zksolc.settings.enableEraVMExtensions ?? false,
              forceEvmla: hre.config.zksolc.settings.forceEVMLA ?? false,
          }
        : {
              ...input.settings,
          };

    return standardInput;
}

export async function getLibraries(librariesModule: string) {
    if (typeof librariesModule !== 'string') {
        return {};
    }

    const librariesModulePath = path.resolve(process.cwd(), librariesModule);

    try {
        const libraries = (await import(librariesModulePath)).default;

        if (typeof libraries !== 'object' || Array.isArray(libraries)) {
            throw new ZkSyncVerifyPluginError(LIBRARIES_EXPORT_ERROR(librariesModule));
        }

        return libraries;
    } catch (error: any) {
        throw new ZkSyncVerifyPluginError(
            `Importing the module for the libraries dictionary failed. Reason: ${error.message}`,
            error,
        );
    }
}