import type { SolidityParameterType } from "../../types/module.js";
import type { DeploymentState } from "../execution/types/deployment-state.js";

import { ExecutionResultType } from "../execution/types/execution-result.js";
import { ExecutionSateType } from "../execution/types/execution-state.js";
import { assertIgnitionInvariant } from "../utils/assertions.js";

export function findResultForFutureById(
  deploymentState: DeploymentState,
  futureId: string,
): SolidityParameterType {
  const exState = deploymentState.executionStates[futureId];

  assertIgnitionInvariant(
    exState !== undefined,
    `Expected execution state for ${futureId} to exist, but it did not`,
  );

  assertIgnitionInvariant(
    exState.type === ExecutionSateType.DEPLOYMENT_EXECUTION_STATE ||
      exState.type === ExecutionSateType.STATIC_CALL_EXECUTION_STATE ||
      exState.type === ExecutionSateType.CONTRACT_AT_EXECUTION_STATE ||
      exState.type === ExecutionSateType.READ_EVENT_ARGUMENT_EXECUTION_STATE ||
      exState.type === ExecutionSateType.ENCODE_FUNCTION_CALL_EXECUTION_STATE,
    `Expected execution state for ${futureId} to be support result lookup, but instead it was ${exState.type}`,
  );

  if (exState.type === ExecutionSateType.CONTRACT_AT_EXECUTION_STATE) {
    return exState.contractAddress;
  }

  assertIgnitionInvariant(
    exState.result !== undefined,
    `Expected execution state for ${futureId} to have a result, but it did not`,
  );

  if (
    exState.type === ExecutionSateType.READ_EVENT_ARGUMENT_EXECUTION_STATE ||
    exState.type === ExecutionSateType.ENCODE_FUNCTION_CALL_EXECUTION_STATE
  ) {
    return exState.result;
  }

  assertIgnitionInvariant(
    exState.result.type === ExecutionResultType.SUCCESS,
    `Cannot access the result of ${futureId}, it was not a deployment success or static call success`,
  );

  switch (exState.type) {
    case ExecutionSateType.DEPLOYMENT_EXECUTION_STATE:
      return exState.result.address;
    case ExecutionSateType.STATIC_CALL_EXECUTION_STATE: {
      return exState.result.value;
    }
  }
}