import { buildAssert } from '@nomicfoundation/hardhat-chai-matchers/utils';
import { toBeHex } from 'ethers';
import { REVERTED_WITHOUT_REASON_MATCHER } from '../../constants';
import { preventAsyncMatcherChaining } from '../utils';
import { decodeReturnData, getReturnDataFromError } from './utils';

export function supportRevertedWithoutReason(Assertion: Chai.AssertionStatic, chaiUtils: Chai.ChaiUtils) {
    Assertion.addMethod(REVERTED_WITHOUT_REASON_MATCHER, function (this: any) {
        const negated = this.__flags.negate;

        preventAsyncMatcherChaining(this, REVERTED_WITHOUT_REASON_MATCHER, chaiUtils);
        const onSuccess = () => {
            const assert = buildAssert(negated, onSuccess);

            assert(false, `Expected transaction to be reverted without a reason, but it didn't revert`);
        };

        const onError = (error: any) => {
            const assert = buildAssert(negated, onError);

            const returnData = getReturnDataFromError(error);
            const decodedReturnData = decodeReturnData(returnData);

            if (decodedReturnData.kind === 'Error') {
                assert(
                    false,
                    `Expected transaction to be reverted without a reason, but it reverted with reason '${decodedReturnData.reason}'`,
                );
            } else if (decodedReturnData.kind === 'Empty') {
                assert(true, undefined, 'Expected transaction NOT to be reverted without a reason, but it was');
            } else if (decodedReturnData.kind === 'Panic') {
                assert(
                    false,
                    `Expected transaction to be reverted without a reason, but it reverted with panic code ${toBeHex(
                        decodedReturnData.code,
                    )} (${decodedReturnData.description})`,
                );
            } else if (decodedReturnData.kind === 'Custom') {
                assert(
                    false,
                    `Expected transaction to be reverted without a reason, but it reverted with a custom error`,
                );
            } else {
                const _exhaustiveCheck: never = decodedReturnData;
            }
        };

        const derivedPromise = Promise.resolve(this._obj).then(onSuccess, onError);

        this.then = derivedPromise.then.bind(derivedPromise);
        this.catch = derivedPromise.catch.bind(derivedPromise);

        return this;
    });
}