import type { SimulationError } from '@aztec/stdlib/errors';
import { Gas } from '@aztec/stdlib/gas';

import { inspect } from 'util';

import { createSimulationError } from '../../common/errors.js';
import type { Gas as AvmGas } from './avm_gas.js';
import type { ReturnData } from './calldata.js';
import type { AvmRevertReason } from './errors.js';

/**
 * Results of an contract call's execution in the AVM.
 */
export class AvmContractCallResult {
  constructor(
    public reverted: boolean,
    public output: ReturnData,
    public gasLeft: AvmGas,
    public revertReason?: AvmRevertReason,
    public totalInstructions: number = 0, // including nested calls
  ) {}

  toString(): string {
    let resultsStr = `reverted: ${this.reverted}, output: ${this.output.bestEffortReadAll(10)}${this.output.length() > 10 ? ' ...' : ''}, gasLeft: ${inspect(
      this.gasLeft,
    )}, totalInstructions: ${this.totalInstructions}`;
    if (this.revertReason) {
      resultsStr += `, revertReason: ${this.revertReason.message}`;
    }
    return resultsStr;
  }

  finalize(): AvmFinalizedCallResult {
    const revertReason = this.revertReason
      ? createSimulationError(this.revertReason, this.output.bestEffortReadAll())
      : undefined;
    return new AvmFinalizedCallResult(
      this.reverted,
      this.output,
      Gas.from(this.gasLeft),
      revertReason,
      this.totalInstructions,
    );
  }
}

/**
 * Similar to AvmContractCallResult, but with a SimulationError and standard Gas
 * which are useful for consumption external to core AVM simulation.
 */
export class AvmFinalizedCallResult {
  constructor(
    public reverted: boolean,
    public output: ReturnData,
    public gasLeft: Gas,
    public revertReason?: SimulationError,
    public totalInstructions: number = 0, // including nested calls
  ) {}

  toString(): string {
    let resultsStr = `reverted: ${this.reverted}, output: ${this.output.bestEffortReadAll(10)}${this.output.length() > 10 ? ' ...' : ''}, gasLeft: ${inspect(
      this.gasLeft,
    )}, totalInstructions: ${this.totalInstructions}`;
    if (this.revertReason) {
      resultsStr += `, revertReason: ${this.revertReason.message}`;
    }
    return resultsStr;
  }
}