import type {
  UserOperationFeeOptions,
  UserOperationOverrides,
  UserOperationStruct,
} from "../../types.js";
import { deepHexlify, resolveProperties } from "../../utils/index.js";
import { applyUserOpOverrideOrFeeOption } from "../../utils/userop.js";
import type { MiddlewareClient } from "../actions.js";
import type { ClientMiddlewareFn } from "../types.js";

/**
 * Description default gas estimator middleware for `SmartAccountClient`
 * You can override this middleware with your custom gas estimator middleware
 * by passing it to the client constructor
 *
 * @param {MiddlewareClient} client smart account client instance to apply the middleware to
 * @returns {ClientMiddlewareFn} middleware execution function used to estimate gas for user operations
 */
export const defaultGasEstimator: <C extends MiddlewareClient>(
  client: C,
) => ClientMiddlewareFn =
  (client: MiddlewareClient): ClientMiddlewareFn =>
  async (struct, { account, overrides, feeOptions }) => {
    const request = deepHexlify(await resolveProperties(struct));

    const estimates = await client.estimateUserOperationGas(
      request,
      account.getEntryPoint().address,
      overrides?.stateOverride,
    );

    const callGasLimit = applyUserOpOverrideOrFeeOption(
      estimates.callGasLimit,
      overrides?.callGasLimit,
      feeOptions?.callGasLimit,
    );
    const verificationGasLimit = applyUserOpOverrideOrFeeOption(
      estimates.verificationGasLimit,
      overrides?.verificationGasLimit,
      feeOptions?.verificationGasLimit,
    );
    const preVerificationGas = applyUserOpOverrideOrFeeOption(
      estimates.preVerificationGas,
      overrides?.preVerificationGas,
      feeOptions?.preVerificationGas,
    );

    struct.callGasLimit = callGasLimit;
    struct.verificationGasLimit = verificationGasLimit;
    struct.preVerificationGas = preVerificationGas;

    const entryPoint = account.getEntryPoint();
    if (entryPoint.version === "0.7.0") {
      const paymasterVerificationGasLimit = applyUserOpOverrideOrFeeOption(
        estimates.paymasterVerificationGasLimit,
        (overrides as UserOperationOverrides<"0.7.0">)
          ?.paymasterVerificationGasLimit,
        (feeOptions as UserOperationFeeOptions<"0.7.0">)
          ?.paymasterVerificationGasLimit,
      );
      const uo_v7 = struct as UserOperationStruct<"0.7.0">;

      uo_v7.paymasterVerificationGasLimit = paymasterVerificationGasLimit;

      uo_v7.paymasterPostOpGasLimit =
        uo_v7.paymasterPostOpGasLimit ?? (uo_v7.paymaster ? "0x0" : undefined);
    }

    return struct;
  };