lib/contract/rpc_calls.ex

defmodule Soroban.Contract.RPCCalls do
  @moduledoc """
  Exposes the functions to execute the simulate and send_transaction endpoints
  """
  alias Soroban.RPC

  alias Soroban.RPC.{
    GetLatestLedgerResponse,
    SendTransactionResponse,
    SimulateTransactionResponse
  }

  alias Stellar.TxBuild
  alias Stellar.TxBuild.{BumpFootprintExpiration, RestoreFootprint}
  alias Stellar.TxBuild.SorobanTransactionData, as: TxSorobanTransactionData
  alias StellarBase.XDR.{SorobanTransactionData, UInt32}

  alias Stellar.TxBuild.{
    Account,
    BaseFee,
    InvokeHostFunction,
    SequenceNumber,
    Signature,
    SorobanAuthorizationEntry
  }

  @type validation :: {:ok, any()}
  @type account :: Account.t()
  @type auths :: list(String.t()) | nil
  @type auth_secret_key :: String.t() | list() | nil
  @type envelope_xdr :: String.t()
  @type footprint_operations :: BumpFootprintExpiration.t() | RestoreFootprint.t()
  @type operation :: InvokeHostFunction.t() | footprint_operations()
  @type simulate_response :: {:ok, SimulateTransactionResponse.t()}
  @type send_response :: {:ok, SendTransactionResponse.t()}
  @type signature :: Signature.t()
  @type sequence_number :: SequenceNumber.t()

  @spec simulate(
          operation :: operation(),
          source_account :: account(),
          sequence_number :: sequence_number()
        ) :: simulate_response()
  def simulate(_operation, _source_account, _sequence_number, soroban_data \\ nil)

  def simulate(operation, source_account, sequence_number, nil) do
    {:ok, envelop_xdr} =
      source_account
      |> TxBuild.new(sequence_number: sequence_number)
      |> TxBuild.add_operation(operation)
      |> TxBuild.envelope()

    RPC.simulate_transaction(envelop_xdr)
  end

  def simulate(operation, source_account, sequence_number, soroban_data) do
    soroban_data = TxSorobanTransactionData.to_xdr(soroban_data)

    {:ok, envelop_xdr} =
      source_account
      |> TxBuild.new(sequence_number: sequence_number)
      |> TxBuild.add_operation(operation)
      |> TxBuild.set_soroban_data(soroban_data)
      |> TxBuild.envelope()

    RPC.simulate_transaction(envelop_xdr)
  end

  @spec send_transaction(
          simulate_response :: simulate_response(),
          source_account :: account(),
          sequence_number :: sequence_number(),
          signature :: signature(),
          operation :: operation(),
          auth_secret_key :: auth_secret_key(),
          extra_fee_rate :: float()
        ) :: send_response() | simulate_response()
  def send_transaction(
        _simulate_transaction,
        _source_account,
        _sequence_number,
        _signature,
        _invoke_host_function_op,
        auth_secret_key \\ nil,
        extra_fee_rate \\ 0.0
      )

  def send_transaction(
        {:ok,
         %SimulateTransactionResponse{
           transaction_data: transaction_data,
           min_resource_fee: min_resource_fee,
           results: [%{auth: auth}]
         }},
        source_account,
        sequence_number,
        signature,
        %InvokeHostFunction{} = operation,
        auth_secret_keys,
        extra_fee_rate
      ) do
    with %InvokeHostFunction{} = invoke_host_function_op <-
           set_host_function_auth(operation, auth, auth_secret_keys) do
      {transaction_data, min_resource_fee} =
        process_transaction_response(
          transaction_data,
          String.to_integer(min_resource_fee),
          auth_secret_keys
        )

      %BaseFee{fee: base_fee} = BaseFee.new()
      fee = BaseFee.new(base_fee + min_resource_fee + round(min_resource_fee * extra_fee_rate))

      {:ok, envelope_xdr} =
        source_account
        |> TxBuild.new(sequence_number: sequence_number)
        |> TxBuild.add_operation(invoke_host_function_op)
        |> TxBuild.set_base_fee(fee)
        |> TxBuild.set_soroban_data(transaction_data)
        |> TxBuild.sign(signature)
        |> TxBuild.envelope()

      RPC.send_transaction(envelope_xdr)
    end
  end

  def send_transaction(
        {:ok,
         %SimulateTransactionResponse{
           transaction_data: transaction_data,
           min_resource_fee: min_resource_fee,
           results: results
         }},
        source_account,
        sequence_number,
        signature,
        operation,
        _auth_secret_keys,
        extra_fee_rate
      )
      when is_list(results) do
    with {:ok, operation} <- validate_operation(operation) do
      {transaction_data, min_resource_fee} =
        process_transaction_response(
          transaction_data,
          String.to_integer(min_resource_fee),
          nil
        )

      %BaseFee{fee: base_fee} = BaseFee.new()
      fee = BaseFee.new(base_fee + min_resource_fee + round(min_resource_fee * extra_fee_rate))

      {:ok, envelope_xdr} =
        source_account
        |> TxBuild.new(sequence_number: sequence_number)
        |> TxBuild.add_operation(operation)
        |> TxBuild.set_base_fee(fee)
        |> TxBuild.set_soroban_data(transaction_data)
        |> TxBuild.sign(signature)
        |> TxBuild.envelope()

      RPC.send_transaction(envelope_xdr)
    end
  end

  def send_transaction(
        {:ok, %SimulateTransactionResponse{}} = response,
        _source_account,
        _sequence_number,
        _signature,
        _invoke_host_function_op,
        _auth_secret_key,
        _extra_fee_rate
      ),
      do: response

  @spec retrieve_unsigned_xdr(
          simulate_response :: simulate_response(),
          source_account :: account(),
          sequence_number :: sequence_number(),
          invoke_host_function_op :: operation(),
          extra_fee_rate :: float()
        ) :: envelope_xdr() | simulate_response()
  def retrieve_unsigned_xdr(
        _simulate_response,
        _source_account,
        _sequence_number,
        _invoke_host_function_op,
        extra_fee_rate \\ 0.0
      )

  def retrieve_unsigned_xdr(
        {:ok,
         %SimulateTransactionResponse{
           transaction_data: transaction_data,
           min_resource_fee: min_resource_fee,
           results: [%{auth: auth}]
         }},
        source_account,
        sequence_number,
        invoke_host_function_op,
        extra_fee_rate
      ) do
    invoke_host_function_op = set_host_function_auth(invoke_host_function_op, auth, [])

    {transaction_data, min_resource_fee} =
      process_transaction_response(
        transaction_data,
        String.to_integer(min_resource_fee),
        nil
      )

    %BaseFee{fee: base_fee} = BaseFee.new()
    fee = BaseFee.new(base_fee + min_resource_fee + round(min_resource_fee * extra_fee_rate))

    {:ok, envelope_xdr} =
      source_account
      |> TxBuild.new(sequence_number: sequence_number)
      |> TxBuild.add_operation(invoke_host_function_op)
      |> TxBuild.set_base_fee(fee)
      |> TxBuild.set_soroban_data(transaction_data)
      |> TxBuild.envelope()

    envelope_xdr
  end

  def retrieve_unsigned_xdr(
        {:ok, %SimulateTransactionResponse{}} = response,
        _source_account,
        _sequence_number,
        _invoke_host_function_op,
        _extra_fee_rate
      ),
      do: response

  @spec set_host_function_auth(
          invoke_host_function :: operation(),
          auths :: auths(),
          auth_secret_key :: auth_secret_key()
        ) :: operation() | {:error, atom()}
  defp set_host_function_auth(
         invoke_host_function_op,
         nil,
         _auth_secret_key
       ),
       do: invoke_host_function_op

  defp set_host_function_auth(
         %InvokeHostFunction{} = invoke_host_function_op,
         auths,
         []
       ),
       do: InvokeHostFunction.set_auth(invoke_host_function_op, auths)

  defp set_host_function_auth(
         %InvokeHostFunction{} = invoke_host_function_op,
         auths,
         nil
       ),
       do: InvokeHostFunction.set_auth(invoke_host_function_op, auths)

  defp set_host_function_auth(
         %InvokeHostFunction{} = invoke_host_function_op,
         auths,
         auth_secret_keys
       )
       when length(auths) == length(auth_secret_keys) do
    {:ok, %GetLatestLedgerResponse{sequence: latest_ledger}} = RPC.get_latest_ledger()

    authorizations =
      auth_secret_keys
      |> Enum.zip(auths)
      |> Enum.map(fn {auth_secret_key, auth} ->
        SorobanAuthorizationEntry.sign_xdr(auth, auth_secret_key, latest_ledger)
      end)

    InvokeHostFunction.set_auth(invoke_host_function_op, authorizations)
  end

  defp set_host_function_auth(_invoke_host_function_op, _auths, _auth_secret_keys),
    do: {:error, :invalid_auth_secret_keys_length}

  @spec validate_operation(operation :: footprint_operations()) :: validation()
  defp validate_operation(%BumpFootprintExpiration{} = operation), do: {:ok, operation}
  defp validate_operation(%RestoreFootprint{} = operation), do: {:ok, operation}

  # This function is needed since when the function invoker is not the function authorizer
  # the transaction data returns min_resource_fee and instructions with wrong values.
  # More info: https://discord.com/channels/897514728459468821/1112853306881081354
  @spec process_transaction_response(
          transaction_data :: String.t(),
          min_resource_fee :: non_neg_integer(),
          auth_secret_key :: auth_secret_key()
        ) :: {SorobanTransactionData.t(), non_neg_integer()}
  defp process_transaction_response(transaction_data, min_resource_fee, nil),
    do: {transaction_data, min_resource_fee}

  defp process_transaction_response(transaction_data, min_resource_fee, _auth_secret_key) do
    {%{
       resources: %{instructions: %{datum: datum}} = resources
     } = soroban_data,
     ""} =
      transaction_data
      |> Base.decode64!()
      |> SorobanTransactionData.decode_xdr!()

    new_instructions = UInt32.new(datum + round(datum * 0.25))
    new_resources = %{resources | instructions: new_instructions}
    soroban_data = %{soroban_data | resources: new_resources}
    min_resource_fee = min_resource_fee + round(min_resource_fee * 0.1)
    {soroban_data, min_resource_fee}
  end
end