import { Address, UniqueArray, DelayedTransactions, LT_COLLATOR } from './utils';
import {
  AbiParam,
  AbiFunction,
  AbiEvent,
  AbiFields,
  FullContractState,
  RawTokensObject,
  TokensObject,
  Transaction,
  Message,
  AbiFunctionName,
  AbiEventName,
  AbiFieldName,
  AbiFunctionInputs,
  DecodedAbiFunctionOutputs,
  DecodedAbiFunctionInputs,
  DecodedAbiFields,
  AbiFunctionInputsWithDefault,
  DecodedAbiEventData,
  TransactionId,
  serializeTokensObject,
  parseTransaction,
  parseTokensObject,
  serializeTransaction,
  IgnoreTransactionTreeSimulationError,
  serializeIgnoreCode,
  AbiGetterName,
  AbiGetter,
  AbiGetterInputsWithDefault,
  DecodedAbiGetterOutputs,
  AbiGetterInputs,
  SignatureContext,
} from './models';
import { Stream, Subscriber } from './stream';
import { ProviderApiResponse, ProviderRpcClient } from './index';

/**
 * @category Contract
 */
export class Contract<Abi> {
  private readonly _provider: ProviderRpcClient;
  private readonly _abi: string;
  private readonly _address: Address;
  private readonly _methods: ContractMethods<Abi>;
  private readonly _getters: ContractGetters<Abi>;
  private readonly _fields: ContractFields<Abi>;

  public readonly methodsAbi: { [K in AbiFunctionName<Abi>]: AbiFunction<Abi, K> };
  public readonly gettersAbi: { [K in AbiGetterName<Abi>]: AbiGetter<Abi, K> };
  public readonly eventsAbi: { [K in AbiEventName<Abi>]: AbiEvent<Abi, K> };
  public readonly fieldsAbi: AbiFields<Abi>;

  constructor(provider: ProviderRpcClient, abi: Abi, address: Address) {
    type ContractFunction = { name: string; inputs?: AbiParam[]; outputs?: AbiParam[] };

    if (!Array.isArray((abi as any).functions)) {
      throw new Error('Invalid abi. Functions array required');
    }
    if (!Array.isArray((abi as any).events)) {
      throw new Error('Invalid abi. Events array required');
    }

    this._provider = provider;
    this._abi = JSON.stringify(abi);
    this.methodsAbi = ((abi as any).functions as ContractFunction[]).reduce(
      (functions, item) => {
        if (item.inputs == null) {
          item.inputs = [];
        }
        if (item.outputs == null) {
          item.outputs = [];
        }

        (functions as any)[item.name] = item;
        return functions;
      },
      {} as typeof Contract.prototype.methodsAbi,
    );

    this.gettersAbi = (((abi as any).getters ?? []) as ContractFunction[]).reduce(
      (getters, item) => {
        if (item.inputs == null) {
          item.inputs = [];
        }
        if (item.outputs == null) {
          item.outputs = [];
        }

        (getters as any)[item.name] = item;
        return getters;
      },
      {} as typeof Contract.prototype.gettersAbi,
    );

    this.eventsAbi = ((abi as any).events as ContractFunction[]).reduce(
      (events, item) => {
        if (item.inputs == null) {
          item.inputs = [];
        }
        (events as any)[item.name] = item;
        return events;
      },
      {} as typeof Contract.prototype.eventsAbi,
    );
    this.fieldsAbi = (abi as any).fields;

    this._address = address;

    this._methods = new Proxy(
      {},
      {
        get: <K extends AbiFunctionName<Abi>>(_object: Record<string, unknown>, method: K) => {
          const rawAbi = this.methodsAbi[method];
          return (params: TokensObject = {}) =>
            new ContractMethodImpl(this._provider, rawAbi, this._abi, this._address, method, params);
        },
      },
    ) as unknown as ContractMethods<Abi>;

    this._getters = new Proxy(
      {},
      {
        get: <K extends AbiGetterName<Abi>>(_object: Record<string, unknown>, getter: K) => {
          const rawAbi = this.gettersAbi[getter];
          return (params: TokensObject = {}) =>
            new ContractGetterImpl(this._provider, rawAbi, this._abi, this._address, getter, params);
        },
      },
    ) as unknown as ContractGetters<Abi>;

    this._fields = new Proxy(
      {},
      {
        get: <K extends AbiFieldName<Abi>>(_object: Record<string, unknown>, field: K) => {
          return async (params: GetContractFieldsParams = {}) => {
            await this._provider.ensureInitialized();
            const { fields, state } = await this._provider.rawApi.getContractFields({
              address: this._address.toString(),
              abi: this._abi,
              cachedState: params?.cachedState,
              allowPartial: params?.allowPartial == null ? false : params.allowPartial,
            });
            if (fields == null) {
              if (state == null) {
                throw new Error('Account does not exist');
              } else if (!state.isDeployed) {
                throw new Error('Account is not deployed');
              } else {
                throw new Error('Invalid account data');
              }
            }
            const parsedFields = parseTokensObject(this.fieldsAbi as AbiParam[], fields) as DecodedAbiFields<Abi>;
            if (parsedFields == null || !Object.prototype.hasOwnProperty.call(parsedFields, field)) {
              throw new Error('Unknown field');
            }
            return parsedFields[field];
          };
        },
      },
    ) as unknown as ContractFields<Abi>;
  }

  public get methods(): ContractMethods<Abi> {
    return this._methods;
  }

  public get getters(): ContractGetters<Abi> {
    return this._getters;
  }

  public get fields(): ContractFields<Abi> {
    return this._fields;
  }

  public get address(): Address {
    return this._address;
  }

  public get abi(): string {
    return this._abi;
  }

  /**
   * Requests contract data
   *
   * ---
   * Required permissions: `basic`
   */
  public async getFullState(): Promise<ProviderApiResponse<'getFullContractState'>> {
    await this._provider.ensureInitialized();
    return (await this._provider.rawApi.getFullContractState({
      address: this.address.toString(),
    })) as ProviderApiResponse<'getFullContractState'>;
  }

  /**
   * Unpacks all fields from the contract state using the specified ABI
   *
   * ---
   * Required permissions: `basic`
   */
  public async getFields(args: GetContractFieldsParams = {}): Promise<{
    fields?: DecodedAbiFields<Abi>;
    state?: FullContractState;
  }> {
    await this._provider.ensureInitialized();
    const { fields, state } = await this._provider.rawApi.getContractFields({
      address: this.address.toString(),
      abi: this._abi,
      cachedState: args.cachedState,
      allowPartial: args.allowPartial == null ? false : args.allowPartial,
    });
    return {
      fields:
        fields != null ? (parseTokensObject(this.fieldsAbi as AbiParam[], fields) as DecodedAbiFields<Abi>) : undefined,
      state,
    };
  }

  /**
   * Creates new contract transactions stream
   *
   * @param subscriber
   */
  public transactions(subscriber: Subscriber): Stream<unknown, Transaction> {
    return subscriber.transactions(this._address).flatMap(({ transactions }) => transactions);
  }

  /**
   * Creates new contract events stream
   *
   * @param subscriber
   */
  public events(subscriber: Subscriber): Stream<unknown, DecodedEventWithTransaction<Abi, AbiEventName<Abi>>> {
    return subscriber
      .transactions(this._address)
      .flatMap(({ transactions }) => transactions)
      .flatMap(tx =>
        this.decodeTransactionEvents({ transaction: tx }).then(events => {
          events.forEach(event => ((event as DecodedEventWithTransaction<Abi, AbiEventName<Abi>>).transaction = tx));
          return events as DecodedEventWithTransaction<Abi, AbiEventName<Abi>>[];
        }),
      );
  }

  public async waitForEvent<E extends AbiEventName<Abi> = AbiEventName<Abi>>(
    args: WaitForEventParams<Abi, E> = {},
  ): Promise<DecodedEvent<Abi, E> | undefined> {
    const { range, filter } = args;
    const filterFn =
      typeof filter === 'string'
        ? ({ event }: DecodedEventWithTransaction<Abi, AbiEventName<Abi>>) => event === filter
        : filter;

    let subscriber = args.subscriber;
    const hasTempSubscriber = subscriber == null;
    if (subscriber == null) {
      subscriber = new this._provider.Subscriber();
    }

    const event = await (
      range?.fromLt != null || range?.fromUtime != null
        ? subscriber.oldTransactions(this._address, range).merge(subscriber.transactions(this._address))
        : subscriber.transactions(this.address)
    )
      .flatMap(item => item.transactions)
      .takeWhile(
        item =>
          range == null ||
          ((range.fromLt == null || LT_COLLATOR.compare(item.id.lt, range.fromLt) > 0) &&
            (range.fromUtime == null || item.createdAt > range.fromUtime) &&
            (range.toLt == null || LT_COLLATOR.compare(item.id.lt, range.toLt) < 0) &&
            (range.toUtime == null || item.createdAt < range.toUtime)),
      )
      .flatMap(tx =>
        this.decodeTransactionEvents({ transaction: tx }).then(events => {
          events.forEach(event => ((event as DecodedEventWithTransaction<Abi, AbiEventName<Abi>>).transaction = tx));
          return events as DecodedEventWithTransaction<Abi, AbiEventName<Abi>>[];
        }),
      )
      .filterMap(async event => {
        if (filterFn == null || (await filterFn(event))) {
          return event;
        } else {
          return undefined;
        }
      })
      .first();

    hasTempSubscriber && (await subscriber.unsubscribe());

    return event;
  }

  public async getPastEvents<E extends AbiEventName<Abi> = AbiEventName<Abi>>(
    args: GetPastEventParams<Abi, E>,
  ): Promise<EventsBatch<Abi, E>> {
    const { range, filter, limit } = args;
    const filterFn =
      typeof filter === 'string'
        ? ({ event }: DecodedEventWithTransaction<Abi, AbiEventName<Abi>>) => event === filter
        : filter;

    const result: DecodedEventWithTransaction<Abi, E>[] = [];
    let currentContinuation = args?.continuation;

    outer: while (true) {
      const { transactions, continuation } = await this._provider.getTransactions({
        address: this._address,
        continuation: currentContinuation,
      });
      if (transactions.length === null) {
        break;
      }

      const filteredTransactions = transactions.filter(
        item =>
          (range?.fromLt == null || LT_COLLATOR.compare(item.id.lt, range.fromLt) > 0) &&
          (range?.fromUtime == null || item.createdAt > range.fromUtime) &&
          (range?.toLt == null || LT_COLLATOR.compare(item.id.lt, range.toLt) < 0) &&
          (range?.toUtime == null || item.createdAt < range.toUtime),
      );

      if (filteredTransactions.length > 0) {
        const parsedEvents = await Promise.all(
          filteredTransactions.map(async tx => {
            return {
              tx,
              events: await this.decodeTransactionEvents({ transaction: tx }).then(events => {
                events.forEach(
                  event => ((event as DecodedEventWithTransaction<Abi, AbiEventName<Abi>>).transaction = tx),
                );
                return events as DecodedEventWithTransaction<Abi, AbiEventName<Abi>>[];
              }),
            };
          }),
        );

        for (let { tx, events } of parsedEvents) {
          if (filterFn != null) {
            events = await Promise.all(events.map(async event => ((await filterFn(event)) ? event : undefined))).then(
              events => events.filter((event): event is Awaited<DecodedEventWithTransaction<Abi, E>> => event != null),
            );
          }

          currentContinuation = tx.id; // update continuation in case of early break

          for (const event of events) {
            if (limit != null && result.length >= limit) {
              break outer;
            }
            result.push(event);
          }

          if (limit != null && result.length >= limit) {
            break outer;
          }
        }
      }

      currentContinuation = continuation;
      if (currentContinuation == null) {
        break;
      }
    }

    return { events: result, continuation: currentContinuation };
  }

  public async decodeTransaction(
    args: DecodeTransactionParams<Abi>,
  ): Promise<DecodedTransaction<Abi, AbiFunctionName<Abi>> | undefined> {
    await this._provider.ensureInitialized();
    try {
      const result = await this._provider.rawApi.decodeTransaction({
        transaction: serializeTransaction(args.transaction),
        abi: this._abi,
        method: args.methods,
      });
      if (result == null) {
        return undefined;
      }

      const { method, input, output } = result;

      const rawAbi = (this.methodsAbi as any)[method];

      return {
        method,
        input: rawAbi.inputs != null ? parseTokensObject(rawAbi.inputs, input) : {},
        output: rawAbi.outputs != null ? parseTokensObject(rawAbi.outputs, output) : {},
      } as DecodedTransaction<Abi, AbiFunctionName<Abi>>;
    } catch (_) {
      return undefined;
    }
  }

  public async decodeTransactionEvents(
    args: DecodeTransactionEventsParams,
  ): Promise<DecodedEvent<Abi, AbiEventName<Abi>>[]> {
    await this._provider.ensureInitialized();
    try {
      const { events } = await this._provider.rawApi.decodeTransactionEvents({
        transaction: serializeTransaction(args.transaction),
        abi: this._abi,
      });

      const result: DecodedEvent<Abi, AbiEventName<Abi>>[] = [];

      for (const { event, data } of events) {
        const rawAbi = (this.eventsAbi as any)[event];

        result.push({
          event,
          data: rawAbi.inputs != null ? parseTokensObject(rawAbi.inputs, data) : {},
        } as DecodedEvent<Abi, AbiEventName<Abi>>);
      }

      return result;
    } catch (_) {
      return [];
    }
  }

  public async decodeInputMessage(
    args: DecodeInputParams<Abi>,
  ): Promise<DecodedInput<Abi, AbiFunctionName<Abi>> | undefined> {
    await this._provider.ensureInitialized();
    try {
      const result = await this._provider.rawApi.decodeInput({
        abi: this._abi,
        body: args.body,
        internal: args.internal,
        method: args.methods,
      });
      if (result == null) {
        return undefined;
      }

      const { method, input } = result;

      const rawAbi = (this.methodsAbi as any)[method];

      return {
        method,
        input: rawAbi.inputs != null ? parseTokensObject(rawAbi.inputs, input) : {},
      } as DecodedInput<Abi, AbiFunctionName<Abi>>;
    } catch (_) {
      return undefined;
    }
  }

  public async decodeOutputMessage(
    args: DecodeOutputParams<Abi>,
  ): Promise<DecodedOutput<Abi, AbiFunctionName<Abi>> | undefined> {
    await this._provider.ensureInitialized();
    try {
      const result = await this._provider.rawApi.decodeOutput({
        abi: this._abi,
        body: args.body,
        method: args.methods,
      });
      if (result == null) {
        return undefined;
      }

      const { method, output } = result;

      const rawAbi = (this.methodsAbi as any)[method];

      return {
        method,
        output: rawAbi.outputs != null ? parseTokensObject(rawAbi.outputs, output) : {},
      } as DecodedOutput<Abi, AbiFunctionName<Abi>>;
    } catch (_) {
      return undefined;
    }
  }

  public async decodeEvent(args: DecodeEventParams<Abi>): Promise<DecodedEvent<Abi, AbiEventName<Abi>> | undefined> {
    await this._provider.ensureInitialized();
    try {
      const result = await this._provider.rawApi.decodeEvent({
        abi: this.abi,
        body: args.body,
        event: args.events,
      });
      if (result == null) {
        return undefined;
      }

      const { event, data } = result;

      const rawAbi = (this.eventsAbi as any)[event];

      return {
        event,
        data: rawAbi.inputs != null ? parseTokensObject(rawAbi.inputs, data) : {},
      } as DecodedEvent<Abi, AbiEventName<Abi>>;
    } catch (_) {
      return undefined;
    }
  }
}

/**
 * @category Contract
 */
export class TvmException extends Error {
  constructor(public readonly code: number) {
    super(`TvmException: ${code}`);
  }
}

/**
 * @category Contract
 */
export type DelayedMessageExecution = {
  /**
   * External message hash
   */
  messageHash: string;
  /**
   * Message expiration timestamp
   */
  expireAt: number;
  /**
   * Transaction promise (it will be rejected if the message has expired)
   */
  transaction: Promise<Transaction>;
};

/**
 * @category Contract
 */
export type ContractMethods<C> = {
  [K in AbiFunctionName<C>]: (
    params: AbiFunctionInputsWithDefault<C, K>,
  ) => ContractMethod<AbiFunctionInputs<C, K>, DecodedAbiFunctionOutputs<C, K>>;
};

/**
 * @category Contract
 */
export type ContractGetters<C> = {
  [K in AbiGetterName<C>]: (
    params: AbiGetterInputsWithDefault<C, K>,
  ) => ContractGetter<AbiGetterInputs<C, K>, DecodedAbiGetterOutputs<C, K>>;
};

/**
 * @category Contract
 */
export type ContractFields<C> = {
  [K in AbiFieldName<C>]: (params?: GetContractFieldsParams) => Promise<DecodedAbiFields<C>[K]>;
};

/**
 * @category Contract
 */
export interface ContractMethod<I, O> {
  /**
   * Target contract address
   */
  readonly address: Address;
  readonly abi: string;
  readonly method: string;
  readonly params: I;

  /**
   * Sends internal message and returns wallet transaction
   *
   * @param args
   */
  send(args: SendInternalParams): Promise<Transaction>;

  /**
   * Sends internal message without waiting for the transaction
   *
   * @param args
   */
  sendDelayed(args: SendInternalParams): Promise<DelayedMessageExecution>;

  /**
   * Sends internal message and waits for the new transaction on target address
   *
   * @param args
   */
  sendWithResult(
    args: SendInternalParams,
  ): Promise<{ parentTransaction: Transaction; childTransaction: Transaction; output?: O }>;

  /**
   * Estimates wallet fee for calling this method as an internal message
   *
   * @param args
   */
  estimateFees(args: SendInternalParams): Promise<string>;

  /**
   * Sends external message and returns contract transaction with parsed output
   *
   * @param args
   */
  sendExternal(args: SendExternalParams): Promise<{ transaction: Transaction; output?: O }>;

  /**
   * Sends external message without waiting for the transaction
   *
   * @param args
   */
  sendExternalDelayed(args: SendExternalDelayedParams): Promise<DelayedMessageExecution>;

  /**
   * Executes only a compute phase locally
   *
   * @param args
   */
  call(args?: CallParams): Promise<O>;

  /**
   * Encodes this method as an external message and
   * executes all transaction phases locally, producing a new state
   */
  executeExternal(args: ExecuteExternalParams): Promise<ExecutorOutput<O>>;

  /**
   * Encodes this method as an internal message and
   * executes all transaction phases locally, producing a new state
   */
  executeInternal(args: ExecuteInternalParams): Promise<ExecutorOutput<O>>;

  /**
   * Encodes method call as BOC
   */
  encodeInternal(): Promise<string>;
}

class ContractMethodImpl implements ContractMethod<any, any> {
  readonly params: RawTokensObject;

  constructor(
    readonly provider: ProviderRpcClient,
    readonly functionAbi: { inputs: AbiParam[]; outputs: AbiParam[] },
    readonly abi: string,
    readonly address: Address,
    readonly method: string,
    params: TokensObject,
  ) {
    this.params = serializeTokensObject(params);
  }

  async send(args: SendInternalParams): Promise<Transaction> {
    await this.provider.ensureInitialized();
    const { transaction } = await this.provider.rawApi.sendMessage({
      sender: args.from.toString(),
      recipient: this.address.toString(),
      amount: args.amount,
      bounce: args.bounce == null ? true : args.bounce,
      payload: {
        abi: this.abi,
        method: this.method,
        params: this.params,
      },
      stateInit: args.stateInit,
      ignoredActionPhaseCodes: args.ignoredActionPhaseCodes?.map(serializeIgnoreCode),
      ignoredComputePhaseCodes: args.ignoredComputePhaseCodes?.map(serializeIgnoreCode),
    });
    return parseTransaction(transaction);
  }

  async sendDelayed(args: SendInternalParams): Promise<DelayedMessageExecution> {
    await this.provider.ensureInitialized();

    const transactions = new DelayedTransactions();

    const subscription = await this.provider.subscribe('messageStatusUpdated');
    subscription.on('data', data => {
      if (!data.address.equals(args.from)) {
        return;
      }
      transactions.fillTransaction(data.hash, data.transaction);
    });

    const { message } = await this.provider.rawApi
      .sendMessageDelayed({
        sender: args.from.toString(),
        recipient: this.address.toString(),
        amount: args.amount,
        bounce: args.bounce == null ? true : args.bounce,
        payload: {
          abi: this.abi,
          method: this.method,
          params: this.params,
        },
        stateInit: args.stateInit,
        ignoredActionPhaseCodes: args.ignoredActionPhaseCodes?.map(serializeIgnoreCode),
        ignoredComputePhaseCodes: args.ignoredComputePhaseCodes?.map(serializeIgnoreCode),
      })
      .catch(e => {
        subscription.unsubscribe().catch(console.error);
        throw e;
      });

    const transaction = transactions
      .waitTransaction(this.address, message.hash)
      .finally(() => subscription.unsubscribe().catch(console.error));

    return {
      messageHash: message.hash,
      expireAt: message.expireAt,
      transaction,
    };
  }

  async sendWithResult(
    args: SendInternalWithResultParams,
  ): Promise<{ parentTransaction: Transaction; childTransaction: Transaction; output?: any }> {
    await this.provider.ensureInitialized();
    let subscriber = args.subscriber;
    const hasTempSubscriber = subscriber == null;
    if (subscriber == null) {
      subscriber = new this.provider.Subscriber();
    }

    try {
      // Parent transaction from wallet
      let parentTransaction: { transaction: Transaction; possibleMessages: Message[] } | undefined = undefined;

      // Child transaction promise
      let resolveChildTransactionPromise: ((transaction: Transaction) => void) | undefined;
      const childTransactionPromise = new Promise<Transaction>(resolve => {
        resolveChildTransactionPromise = tx => resolve(tx);
      });

      // Array for collecting transactions on target before parent transaction promise resolution
      const possibleChildren: Transaction[] = [];

      // Subscribe to this account
      subscriber
        .transactions(this.address)
        .flatMap(batch => batch.transactions)
        // Listen only messages from sender
        .filter(item => item.inMessage.src?.equals(args.from) || false)
        .on(tx => {
          if (parentTransaction == null) {
            // If we don't known whether the message was sent just collect all transactions from the sender
            possibleChildren.push(tx);
          } else if (parentTransaction.possibleMessages.findIndex(msg => msg.hash == tx.inMessage.hash) >= 0) {
            // Resolve promise if transaction was found
            resolveChildTransactionPromise?.(tx);
          }
        });

      // Send message
      const transaction = await this.send(args);
      // Extract all outgoing messages from the parent transaction to this contract
      const possibleMessages = transaction.outMessages.filter(msg => msg.dst?.equals(this.address) || false);

      // Update stream state
      parentTransaction = {
        transaction,
        possibleMessages,
      };

      // Check whether child transaction was already found
      const alreadyReceived = possibleChildren.find(tx => {
        return possibleMessages.findIndex(msg => msg.hash == tx.inMessage.hash) >= 0;
      });
      if (alreadyReceived != null) {
        resolveChildTransactionPromise?.(alreadyReceived);
      }

      const childTransaction = await childTransactionPromise;

      // Parse output
      let output: any = undefined;
      try {
        const result = await this.provider.rawApi.decodeTransaction({
          transaction: serializeTransaction(childTransaction),
          abi: this.abi,
          method: this.method,
        });
        if (result != null) {
          output = this.functionAbi.outputs != null ? parseTokensObject(this.functionAbi.outputs, result.output) : {};
        }
      } catch (e) {
        console.error(e);
      }

      // Done
      return {
        parentTransaction: parentTransaction.transaction,
        childTransaction,
        output,
      };
    } finally {
      hasTempSubscriber && (await subscriber.unsubscribe());
    }
  }

  async estimateFees(args: SendInternalParams): Promise<string> {
    await this.provider.ensureInitialized();
    const { fees } = await this.provider.rawApi.estimateFees({
      sender: args.from.toString(),
      recipient: this.address.toString(),
      amount: args.amount,
      payload: {
        abi: this.abi,
        method: this.method,
        params: this.params,
      },
      stateInit: args.stateInit,
    });
    return fees;
  }

  async sendExternal(args: SendExternalParams): Promise<{ transaction: Transaction; output?: any }> {
    await this.provider.ensureInitialized();
    const method =
      args.withoutSignature === true
        ? this.provider.rawApi.sendUnsignedExternalMessage
        : this.provider.rawApi.sendExternalMessage;

    const { transaction, output } = await method({
      publicKey: (args as any).publicKey,
      recipient: this.address.toString(),
      stateInit: args.stateInit,
      payload: {
        abi: this.abi,
        method: this.method,
        params: this.params,
      },
      local: args.local,
      executorParams: args.executorParams
        ? {
            disableSignatureCheck: args.executorParams.disableSignatureCheck,
            overrideBalance: args.executorParams.overrideBalance,
          }
        : undefined,
    });

    return {
      transaction: parseTransaction(transaction),
      output: output != null ? parseTokensObject(this.functionAbi.outputs, output) : undefined,
    };
  }

  async sendExternalDelayed(args: SendExternalDelayedParams): Promise<DelayedMessageExecution> {
    await this.provider.ensureInitialized();

    const transactions = new DelayedTransactions();

    const subscription = await this.provider.subscribe('messageStatusUpdated');
    subscription.on('data', data => {
      if (!data.address.equals(this.address)) {
        return;
      }
      transactions.fillTransaction(data.hash, data.transaction);
    });

    const { message } = await this.provider.rawApi
      .sendExternalMessageDelayed({
        publicKey: args.publicKey,
        recipient: this.address.toString(),
        stateInit: args.stateInit,
        payload: {
          abi: this.abi,
          method: this.method,
          params: this.params,
        },
      })
      .catch(e => {
        subscription.unsubscribe().catch(console.error);
        throw e;
      });

    const transaction = transactions
      .waitTransaction(this.address, message.hash)
      .finally(() => subscription.unsubscribe().catch(console.error));

    return {
      messageHash: message.hash,
      expireAt: message.expireAt,
      transaction,
    };
  }

  async call(args: CallParams = {}): Promise<any> {
    await this.provider.ensureInitialized();
    const { output, code } = await this.provider.rawApi.runLocal({
      address: this.address.toString(),
      cachedState: args.cachedState,
      responsible: args.responsible,
      functionCall: {
        abi: this.abi,
        method: this.method,
        params: this.params,
      },
      withSignatureId: args.withSignatureId,
      withSignatureContext: args.withSignatureContext,
      libraries: args.libraries,
    });

    if (output == null || code != 0) {
      throw new TvmException(code);
    } else {
      return parseTokensObject(this.functionAbi.outputs, output);
    }
  }

  /**
   * Encodes this method as an external message and
   * executes all transaction phases locally, producing a new state
   */
  async executeExternal(args: ExecuteExternalParams): Promise<ExecutorOutput<any>> {
    await this.provider.ensureInitialized();
    const { transaction, newState, output } = await this.provider.rawApi.executeLocal({
      address: this.address.toString(),
      cachedState: args.cachedState,
      stateInit: args.stateInit,
      payload: {
        abi: this.abi,
        method: this.method,
        params: this.params,
      },
      messageHeader: {
        type: 'external',
        publicKey: args.publicKey,
        withoutSignature: args.withoutSignature,
      },
      executorParams:
        args.executorParams != null
          ? {
              disableSignatureCheck: args.executorParams.disableSignatureCheck,
              overrideBalance: args.executorParams.overrideBalance,
            }
          : undefined,
    });

    return {
      transaction: parseTransaction(transaction),
      newState,
      output: output !== undefined ? parseTokensObject(this.functionAbi.outputs, output) : undefined,
    } as ExecutorOutput<any>;
  }

  /**
   * Encodes this method as an internal message and
   * executes all transaction phases locally, producing a new state
   */
  async executeInternal(args: ExecuteInternalParams): Promise<ExecutorOutput<any>> {
    await this.provider.ensureInitialized();
    const { transaction, newState, output } = await this.provider.rawApi.executeLocal({
      address: this.address.toString(),
      cachedState: args.cachedState,
      stateInit: args.stateInit,
      payload: {
        abi: this.abi,
        method: this.method,
        params: this.params,
      },
      messageHeader: {
        type: 'internal',
        sender: args.sender.toString(),
        amount: args.amount,
        bounce: args.bounce != null ? args.bounce : false,
        bounced: args.bounced,
      },
      executorParams:
        args.executorParams != null
          ? {
              disableSignatureCheck: args.executorParams.disableSignatureCheck,
              overrideBalance: args.executorParams.overrideBalance,
            }
          : undefined,
    });

    return {
      transaction: parseTransaction(transaction),
      newState,
      output: output !== undefined ? parseTokensObject(this.functionAbi.outputs, output) : undefined,
    } as ExecutorOutput<any>;
  }

  async encodeInternal(): Promise<string> {
    await this.provider.ensureInitialized();
    const { boc } = await this.provider.rawApi.encodeInternalInput({
      abi: this.abi,
      method: this.method,
      params: this.params,
    });
    return boc;
  }
}

/**
 * @category Contract
 */
export interface ContractGetter<I, O> {
  /**
   * Target contract address
   */
  readonly address: Address;
  readonly abi: string;
  readonly getter: string;
  readonly params: I;

  /**
   * Runs getter and returns parsed output
   *
   * @param args
   */
  run(args?: RunGetterParams): Promise<O>;
}

class ContractGetterImpl implements ContractGetter<any, any> {
  readonly params: RawTokensObject;

  constructor(
    readonly provider: ProviderRpcClient,
    readonly getterAbi: { inputs: AbiParam[]; outputs: AbiParam[] },
    readonly abi: string,
    readonly address: Address,
    readonly getter: string,
    params: TokensObject,
  ) {
    this.params = serializeTokensObject(params);
  }

  async run(args: RunGetterParams = {}): Promise<any> {
    await this.provider.ensureInitialized();
    const { output, code } = await this.provider.rawApi.runGetter({
      address: this.address.toString(),
      cachedState: args.cachedState,
      getterCall: {
        abi: this.abi,
        getter: this.getter,
        params: this.params,
      },
      withSignatureId: args.withSignatureId,
      withSignatureContext: args.withSignatureContext,
      libraries: args.libraries,
    });

    if (output == null || code != 0) {
      throw new TvmException(code);
    } else {
      return parseTokensObject(this.getterAbi.outputs, output);
    }
  }
}

/**
 * @category Contract
 */
export type GetContractFieldsParams = {
  /**
   * Cached contract state
   */
  cachedState?: FullContractState;
  /**
   * Don't fail if something is left in a cell after unpacking
   */
  allowPartial?: boolean;
};

/**
 * @category Contract
 */
export type SendInternalParams = {
  /**
   * Preferred wallet address.
   * It is the same address as the `accountInteraction.address`, but it must be explicitly provided
   */
  from: Address;
  /**
   * Amount of nano EVER to send
   */
  amount: string;
  /**
   * @default true
   */
  bounce?: boolean;
  /**
   * Optional base64 encoded TVC
   *
   * NOTE: If the selected contract do not support stateInit in the internal message,
   * an error is returned
   */
  stateInit?: string;
  /**
   * Optional compute phase error codes to be ignored during transaction tree simulation
   */
  ignoredComputePhaseCodes?: IgnoreTransactionTreeSimulationError[];
  /**
   * Optional action phase error codes to be ignored during transaction tree simulation
   */
  ignoredActionPhaseCodes?: IgnoreTransactionTreeSimulationError[];
};

/**
 * @category Contract
 */
export type SendInternalWithResultParams = SendInternalParams & {
  /**
   * Existing subscriber
   */
  subscriber?: Subscriber;
};

/**
 * @category Contract
 */
export type SendExternalParams = (
  | {
      /**
       * Whether to prepare this message without signature. Default: false
       */
      withoutSignature: true;
    }
  | {
      /**
       * The public key of the preferred account.
       * It is the same publicKey as the `accountInteraction.publicKey`, but it must be explicitly provided
       */
      publicKey: string;
      /**
       * Whether to prepare this message without signature. Default: false
       */
      withoutSignature?: false;
    }
) & {
  /**
   * Optional base64 encoded TVC
   */
  stateInit?: string;
  /**
   * Whether to run this message locally. Default: false
   */
  local?: boolean;
  /**
   * Optional executor parameters used during local contract execution
   */
  executorParams?: {
    /**
     * If `true`, signature verification always succeds
     */
    disableSignatureCheck?: boolean;
    /**
     * Explicit account balance in nano EVER
     */
    overrideBalance?: string | number;
  };
};

/**
 * @category Contract
 */
export type SendExternalDelayedParams = {
  /**
   * The public key of the preferred account.
   * It is the same publicKey as the `accountInteraction.publicKey`, but it must be explicitly provided
   */
  publicKey: string;
  /**
   * Optional base64 encoded TVC
   */
  stateInit?: string;
};

/**
 * @category Contract
 */
export type CallParams = {
  /**
   * Cached contract state
   */
  cachedState?: FullContractState;
  /**
   * Whether to run the method locally as responsible.
   *
   * This will use internal message with unlimited account balance.
   */
  responsible?: boolean;
  /**
   * Whether to use the signature id during signature verification (true by default).
   * - If `true`, uses the signature id of the selected network (if the capability is enabled).
   * - If `false`, forces signature check to ignore any signature id.
   * - If `number`, uses the specified number as a signature id.
   * @deprecated
   * Use `withSignatureContext` instead.
   */
  withSignatureId?: boolean | number;
  /**
   * Optional advanced signature configuration.
   *
   * In most cases you **do not need to set this manually**.
   * The wallet or network configuration will choose the correct mode.
   *
   * This field controls how the data is prepared before signing:
   *
   * - `empty` — signs the data as-is.
   *   Use for simple or legacy signing.
   *
   * - `signatureId` — signs the data together with a network-specific id.
   *   Prevents signatures from being reused on another network.
   *
   * - `signatureDomainL2` — same as signature id but with a different prefix.
   *   This should be used where the `SignatureDomain` capability is enabled.
   *
   * If provided, this field overrides `withSignatureId`.
   */
  withSignatureContext?: SignatureContext;
  /**
   * Optional libraries map
   **/
  libraries?: { [K: string]: string };
};

/**
 * @category Contract
 */
export type RunGetterParams = {
  /**
   * Cached contract state
   */
  cachedState?: FullContractState;
  /**
   * Whether to use the signature id during signature verification (true by default).
   * - If `true`, uses the signature id of the selected network (if the capability is enabled).
   * - If `false`, forces signature check to ignore any signature id.
   * - If `number`, uses the specified number as a signature id.
   * @deprecated
   * Use `withSignatureContext` instead.
   */
  withSignatureId?: boolean | number;
  /**
   * Optional advanced signature configuration.
   *
   * In most cases you **do not need to set this manually**.
   * The wallet or network configuration will choose the correct mode.
   *
   * This field controls how the data is prepared before signing:
   *
   * - `empty` — signs the data as-is.
   *   Use for simple or legacy signing.
   *
   * - `signatureId` — signs the data together with a network-specific id.
   *   Prevents signatures from being reused on another network.
   *
   * - `signatureDomainL2` — same as signature id but with a different prefix.
   *   This should be used where the `SignatureDomain` capability is enabled.
   *
   * If provided, this field overrides `withSignatureId`.
   */
  withSignatureContext?: SignatureContext;
  /**
   * Optional libraries map
   **/
  libraries?: { [K: string]: string };
};

/**
 * @category Contract
 */
export type ExecuteExternalParams = {
  /**
   * Cached contract state
   */
  cachedState?: FullContractState;
  /**
   * The public key of the signer.
   */
  publicKey: string;
  /**
   * Optional base64 encoded TVC
   */
  stateInit?: string;
  /**
   * Optional executor parameters used during local contract execution
   */
  executorParams?: {
    /**
     * If `true`, signature verification always succeeds
     */
    disableSignatureCheck?: boolean;
    /**
     * Explicit account balance in nano EVER
     */
    overrideBalance?: string | number;
  };
  /**
   * Whether to prepare this message without signature. Default: false
   */
  withoutSignature?: boolean;
};

/**
 * @category Contract
 */
export type ExecuteInternalParams = {
  /**
   * Cached contract state
   */
  cachedState?: FullContractState;
  /**
   * Optional base64 encoded TVC
   */
  stateInit?: string;
  /**
   * Message source address
   */
  sender: Address;
  /**
   * Amount of nano EVER to attach to the message
   */
  amount: string;
  /**
   * Whether to bounce message back on error. Default: false
   */
  bounce?: boolean;
  /**
   * Whether the constructed message is bounced. Default: false
   */
  bounced?: boolean;
  /**
   * Optional executor parameters used during local contract execution
   */
  executorParams?: {
    /**
     * If `true`, signature verification always succeeds
     */
    disableSignatureCheck?: boolean;
    /**
     * Explicit account balance in nano EVER
     */
    overrideBalance?: string | number;
  };
};

/**
 * @category Contract
 */
export type GetPastEventParams<Abi, E extends AbiEventName<Abi>> = {
  filter?: E | EventsFilter<Abi, AbiEventName<Abi>>;
  range?: EventsRange;
  limit?: number;
  continuation?: TransactionId;
};

/**
 * @category Contract
 */
export type WaitForEventParams<Abi, E extends AbiEventName<Abi>> = {
  filter?: E | EventsFilter<Abi, E>;
  range?: EventsRange;
  subscriber?: Subscriber;
};

/**
 * @category Contract
 */
export type EventsBatch<Abi, E extends AbiEventName<Abi>> = {
  events: DecodedEventWithTransaction<Abi, E>[];
  continuation?: TransactionId;
};

/**
 * @category Contract
 */
export type EventsFilter<Abi, E extends AbiEventName<Abi> = AbiEventName<Abi>> = (
  event: DecodedEventWithTransaction<Abi, E>,
) => Promise<boolean> | boolean;

/**
 * @category Contract
 */
export type DecodedEventWithTransaction<Abi, E extends AbiEventName<Abi>> = DecodedEvent<Abi, E> & {
  transaction: Transaction;
};

/**
 * @category Contract
 */
export type EventsRange = {
  fromLt?: string;
  fromUtime?: number;
  toLt?: string;
  toUtime?: number;
};

/**
 * @category Contract
 */
export type DecodeTransactionParams<Abi> = {
  transaction: Transaction;
  methods: UniqueArray<AbiFunctionName<Abi>[]>;
};

/**
 * @category Contract
 */
export type DecodedTransaction<Abi, T> =
  T extends AbiFunctionName<Abi>
    ? { method: T; input: DecodedAbiFunctionInputs<Abi, T>; output: DecodedAbiFunctionOutputs<Abi, T> }
    : never;

/**
 * @category Contract
 */
export type DecodeInputParams<Abi> = {
  body: string;
  methods: UniqueArray<AbiFunctionName<Abi>[]>;
  internal: boolean;
};

/**
 * @category Contract
 */
export type DecodedInput<Abi, T> =
  T extends AbiFunctionName<Abi> ? { method: T; input: DecodedAbiFunctionInputs<Abi, T> } : never;

/**
 * @category Contract
 */
export type DecodeOutputParams<Abi> = {
  /**
   * Base64 encoded message body BOC
   */
  body: string;
  methods: UniqueArray<AbiFunctionName<Abi>[]>;
};

/**
 * @category Contract
 */
export type DecodeEventParams<Abi> = {
  /**
   * Base64 encoded message body BOC
   */
  body: string;
  events: UniqueArray<AbiEventName<Abi>[]>;
};

/**
 * @category Contract
 */
export type DecodedOutput<Abi, T> =
  T extends AbiFunctionName<Abi> ? { method: T; output: DecodedAbiFunctionOutputs<Abi, T> } : never;

/**
 * @category Contract
 */
export type DecodeTransactionEventsParams = {
  transaction: Transaction;
};

/**
 * @category Contract
 */
export type DecodedEvent<Abi, T> =
  T extends AbiEventName<Abi> ? { event: T; data: DecodedAbiEventData<Abi, T> } : never;

/**
 * @category Contract
 */
export type ExecutorOutput<O> = {
  /**
   * Executed transaction
   */
  transaction: Transaction;
  /**
   * Contract state after the executed transaction
   */
  newState: FullContractState | undefined;
  /**
   * Parsed function call output
   */
  output: O | undefined;
};