feat: save and display aexn extensions (#710)

Author: jyeshe

lib/ae_mdw/aex9.ex

@@ -4,7 +4,7 @@ defmodule AeMdw.Aex9 do
   """
 
   alias AeMdw.Collection
-  alias AeMdw.Contracts.AexnContract
+  alias AeMdw.AexnContracts
   alias AeMdw.Database
   alias AeMdw.Db.Format
   alias AeMdw.Db.Model
@@ -113,7 +113,7 @@ defmodule AeMdw.Aex9 do
           {:ok, balances_cursor() | nil, [aex9_balance()], balances_cursor() | nil}
           | {:error, Error.t()}
   def fetch_balances(contract_pk, pagination, cursor) do
-    if AexnContract.is_aex9?(contract_pk) do
+    if AexnContracts.is_aex9?(contract_pk) do
       {amounts, _height_hash} = Db.aex9_balances!(contract_pk)
       accounts_pks = amounts |> Map.keys() |> Enum.sort()
       cursor = deserialize_balances_cursor(cursor)
@@ -390,7 +390,7 @@ defmodule AeMdw.Aex9 do
   end
 
   defp validate_aex9(contract_pk) do
-    if AexnContract.is_aex9?(contract_pk) do
+    if AexnContracts.is_aex9?(contract_pk) do
       :ok
     else
       {:error,

lib/ae_mdw/contracts/aexn_contract.ex renamed to lib/ae_mdw/aexn_contracts.ex

@@ -1,4 +1,4 @@
-defmodule AeMdw.Contracts.AexnContract do
+defmodule AeMdw.AexnContracts do
   @moduledoc """
   AEX-N detection and common calls to interact with the contract.
   """
@@ -30,7 +30,7 @@ defmodule AeMdw.Contracts.AexnContract do
   @spec is_aex141?(pubkey()) :: boolean()
   def is_aex141?(pubkey) when is_binary(pubkey) do
     with {:ok, {type_info, _compiler_vsn, _source_hash}} <- Contract.get_info(pubkey),
-         true <- has_all_aex141_signatures?(type_info),
+         true <- valid_aex141_signatures?(type_info),
          {:ok, extensions} <- call_contract(pubkey, "aex141_extensions") do
       has_valid_aex141_extensions?(extensions, type_info)
     else
@@ -39,15 +39,50 @@ defmodule AeMdw.Contracts.AexnContract do
     end
   end
 
-  def is_aex141?(_no_fcode), do: false
+  @spec has_aex141_signatures?(pubkey()) :: boolean()
+  def has_aex141_signatures?(pubkey) do
+    case Contract.get_info(pubkey) do
+      {:ok, {type_info, _compiler_vsn, _source_hash}} ->
+        valid_aex141_signatures?(type_info)
+
+      {:error, _reason} ->
+        false
+    end
+  end
+
+  @spec has_valid_aex141_extensions?(Model.aexn_extensions(), pubkey() | Contract.type_info()) ::
+          boolean()
+  def has_valid_aex141_extensions?(extensions, pubkey) when is_binary(pubkey) do
+    case Contract.get_info(pubkey) do
+      {:ok, {type_info, _compiler_vsn, _source_hash}} ->
+        has_valid_aex141_extensions?(extensions, type_info)
+
+      {:error, _reason} ->
+        false
+    end
+  end
 
-  @spec call_meta_info(pubkey()) :: {:ok, Model.aexn_meta_info()} | :not_found
+  def has_valid_aex141_extensions?(extensions, {:fcode, functions, _hash_names, _code}) do
+    Enum.all?(extensions, &valid_aex141_extension?(&1, functions))
+  end
+
+  def has_valid_aex141_extensions?(_extensions, _no_fcode), do: false
+
+  @spec call_meta_info(pubkey()) :: {:ok, Model.aexn_meta_info()} | :error
   def call_meta_info(contract_pk) do
     with {:ok, {:tuple, meta_info_tuple}} <- call_contract(contract_pk, "meta_info", []) do
       {:ok, decode_meta_info(meta_info_tuple)}
     end
   end
 
+  @spec call_extensions(Model.aexn_type(), pubkey()) :: {:ok, Model.aexn_extensions()} | :error
+  def call_extensions(aexn_type, pubkey) do
+    case aexn_type do
+      :aex9 -> call_contract(pubkey, "aex9_extensions")
+      :aex141 -> call_contract(pubkey, "aex141_extensions")
+    end
+  end
+
   #
   # Private functions
   #
@@ -58,9 +93,9 @@ defmodule AeMdw.Contracts.AexnContract do
       {:ok, return} ->
         {:ok, return}
 
-      {:error, _call_error} ->
-        Log.warn("#{method} call error for #{enc_ct(contract_pk)}")
-        :not_found
+      {:error, call_error} ->
+        Log.warn("#{method} call error for #{enc_ct(contract_pk)}: #{inspect(call_error)}")
+        :error
     end
   end
 
@@ -69,6 +104,7 @@ defmodule AeMdw.Contracts.AexnContract do
   defp decode_meta_info({name, symbol, variant_url, variant_type}) do
     url =
       case variant_url do
+        {:variant, [0, 1], 1, {url}} -> url
         {:variant, [0, 1], 1, url} -> url
         _other -> nil
       end
@@ -90,15 +126,15 @@ defmodule AeMdw.Contracts.AexnContract do
     end)
   end
 
-  defp has_all_aex141_signatures?({:fcode, functions, _hash_names, _code}) do
+  defp valid_aex141_signatures?({:fcode, functions, _hash_names, _code}) do
     valid_base_signatures? =
       AeMdw.Node.aex141_signatures()
       |> has_all_signatures?(functions)
 
     valid_base_signatures? and valid_aex141_metadata?(functions)
   end
 
-  defp has_all_aex141_signatures?(_no_fcode), do: false
+  defp valid_aex141_signatures?(_no_fcode), do: false
 
   @option_string {:variant, [tuple: [], tuple: [:string]]}
   @option_metadata_map {:variant, [tuple: [], tuple: [{:map, :string, :string}]]}
@@ -119,10 +155,6 @@ defmodule AeMdw.Contracts.AexnContract do
     end
   end
 
-  defp has_valid_aex141_extensions?(extensions, {:fcode, functions, _hash_names, _code}) do
-    Enum.all?(extensions, &valid_aex141_extension?(&1, functions))
-  end
-
   defp valid_aex141_extension?("mintable", functions) do
     case Map.get(functions, @mint_hash) do
       nil ->
@@ -143,4 +175,6 @@ defmodule AeMdw.Contracts.AexnContract do
       match?({_code, {[:address], :integer}, _body}, functions[@check_swap_hash]) and
       match?({_code, {[], {:map, :address, :string}}, _body}, functions[@swapped_hash])
   end
+
+  defp valid_aex141_extension?(_any, _functions), do: false
 end

lib/ae_mdw/db/contract.ex

@@ -4,7 +4,7 @@ defmodule AeMdw.Db.Contract do
   """
   alias AeMdw.Collection
   alias AeMdw.Contract
-  alias AeMdw.Contracts.AexnContract
+  alias AeMdw.AexnContracts
   alias AeMdw.Database
   alias AeMdw.Db.Model
   alias AeMdw.Db.Origin
@@ -35,9 +35,10 @@ defmodule AeMdw.Db.Contract do
           Model.aexn_type(),
           Model.aexn_meta_info(),
           pubkey(),
-          integer()
+          integer(),
+          Model.aexn_extensions()
         ) :: state()
-  def aexn_creation_write(state, aexn_type, aexn_meta_info, contract_pk, txi) do
+  def aexn_creation_write(state, aexn_type, aexn_meta_info, contract_pk, txi, extensions) do
     name = elem(aexn_meta_info, 0)
     symbol = elem(aexn_meta_info, 1)
 
@@ -48,7 +49,8 @@ defmodule AeMdw.Db.Contract do
       Model.aexn_contract(
         index: {aexn_type, contract_pk},
         txi: txi,
-        meta_info: aexn_meta_info
+        meta_info: aexn_meta_info,
+        extensions: extensions
       )
 
     state
@@ -246,11 +248,11 @@ defmodule AeMdw.Db.Contract do
 
   @spec which_aexn_contract_pubkey(pubkey(), pubkey()) :: pubkey() | nil
   def which_aexn_contract_pubkey(contract_pk, addr) do
-    if AexnContract.is_aex9?(contract_pk) do
+    if AexnContracts.is_aex9?(contract_pk) do
       contract_pk
     else
       # remotely called contract is aex9?
-      if addr != contract_pk and AexnContract.is_aex9?(addr), do: addr
+      if addr != contract_pk and AexnContracts.is_aex9?(addr), do: addr
     end
   end
 
@@ -356,17 +358,18 @@ defmodule AeMdw.Db.Contract do
 
   @spec update_aex9_state(State.t(), pubkey(), block_index(), txi()) :: State.t()
   def update_aex9_state(state, contract_pk, {kbi, mbi} = block_index, txi) do
-    if AexnContract.is_aex9?(contract_pk) do
+    if AexnContracts.is_aex9?(contract_pk) do
       with false <- State.exists?(state, Model.AexnContract, {:aex9, contract_pk}),
-           {:ok, aex9_meta_info} <- AexnContract.call_meta_info(contract_pk) do
+           {:ok, extensions} <- AexnContracts.call_extensions(:aex9, contract_pk),
+           {:ok, aex9_meta_info} <- AexnContracts.call_meta_info(contract_pk) do
         AsyncTasks.Producer.enqueue(:derive_aex9_presence, [contract_pk, kbi, mbi, txi])
-        aexn_creation_write(state, :aex9, aex9_meta_info, contract_pk, txi)
+        aexn_creation_write(state, :aex9, aex9_meta_info, contract_pk, txi, extensions)
       else
         true ->
           AsyncTasks.Producer.enqueue(:update_aex9_state, [contract_pk], [block_index, txi])
           state
 
-        :not_found ->
+        :error ->
           state
       end
     else

lib/ae_mdw/db/model.ex

@@ -308,17 +308,20 @@ defmodule AeMdw.Db.Model do
   @type aex9_meta_info :: {aexn_name(), aexn_symbol(), non_neg_integer()}
   @type aex141_meta_info :: {aexn_name(), aexn_symbol(), String.t(), atom()}
   @type aexn_meta_info :: aex9_meta_info() | aex141_meta_info()
+  @type aexn_extensions :: [String.t()]
 
   @type aexn_contract ::
           record(:aexn_contract,
             index: {aexn_type(), Db.pubkey()},
             txi: Txs.txi(),
-            meta_info: aexn_meta_info()
+            meta_info: aexn_meta_info(),
+            extensions: aexn_extensions()
           )
   @aexn_contract_defaults [
     index: nil,
     txi: -1,
-    meta_info: nil
+    meta_info: nil,
+    extensions: []
   ]
   defrecord :aexn_contract, @aexn_contract_defaults
 

lib/ae_mdw/db/mutations/aexn_create_contract_mutation.ex

@@ -5,6 +5,7 @@ defmodule AeMdw.Db.AexnCreateContractMutation do
 
   alias AeMdw.Blocks
   alias AeMdw.Db.Contract, as: DBContract
+  alias AeMdw.Db.Model
   alias AeMdw.Db.State
   alias AeMdw.Sync.AsyncTasks
   alias AeMdw.Txs
@@ -15,7 +16,8 @@ defmodule AeMdw.Db.AexnCreateContractMutation do
     :contract_pk,
     :aexn_meta_info,
     :block_index,
-    :create_txi
+    :create_txi,
+    :extensions
   ]
 
   @typep aexn_type :: AeMdw.Db.Model.aexn_type()
@@ -27,23 +29,26 @@ defmodule AeMdw.Db.AexnCreateContractMutation do
             contract_pk: pubkey(),
             aexn_meta_info: aexn_meta_info(),
             block_index: Blocks.block_index(),
-            create_txi: Txs.txi()
+            create_txi: Txs.txi(),
+            extensions: Model.aexn_extensions()
           }
 
   @spec new(
           aexn_type(),
           pubkey(),
           aexn_meta_info(),
           Blocks.block_index(),
-          Txs.txi()
+          Txs.txi(),
+          Model.aexn_extensions()
         ) :: t()
-  def new(aexn_type, contract_pk, aexn_meta_info, block_index, create_txi) do
+  def new(aexn_type, contract_pk, aexn_meta_info, block_index, create_txi, extensions) do
     %__MODULE__{
       aexn_type: aexn_type,
       contract_pk: contract_pk,
       aexn_meta_info: aexn_meta_info,
       block_index: block_index,
-      create_txi: create_txi
+      create_txi: create_txi,
+      extensions: extensions
     }
   end
 
@@ -54,12 +59,20 @@ defmodule AeMdw.Db.AexnCreateContractMutation do
           contract_pk: contract_pk,
           aexn_meta_info: aexn_meta_info,
           block_index: {kbi, mbi},
-          create_txi: create_txi
+          create_txi: create_txi,
+          extensions: extensions
         },
         state
       ) do
     state =
-      DBContract.aexn_creation_write(state, aexn_type, aexn_meta_info, contract_pk, create_txi)
+      DBContract.aexn_creation_write(
+        state,
+        aexn_type,
+        aexn_meta_info,
+        contract_pk,
+        create_txi,
+        extensions
+      )
 
     if aexn_type == :aex9 do
       AsyncTasks.Producer.enqueue(:derive_aex9_presence, [contract_pk, kbi, mbi, create_txi])

lib/ae_mdw/db/mutations/contract_call_mutation.ex

@@ -5,7 +5,7 @@ defmodule AeMdw.Db.ContractCallMutation do
 
   alias AeMdw.Blocks
   alias AeMdw.Contract
-  alias AeMdw.Contracts.AexnContract
+  alias AeMdw.AexnContracts
   alias AeMdw.Db.Contract, as: DBContract
   alias AeMdw.Db.Origin
   alias AeMdw.Db.State
@@ -60,7 +60,7 @@ defmodule AeMdw.Db.ContractCallMutation do
         state
       ) do
     # update balance on any aex9 call
-    if AexnContract.is_aex9?(contract_pk) do
+    if AexnContracts.is_aex9?(contract_pk) do
       AsyncTasks.Producer.enqueue(:update_aex9_state, [contract_pk], [block_index, txi])
     end
 

lib/ae_mdw/db/sync/contract.ex

@@ -4,7 +4,7 @@ defmodule AeMdw.Db.Sync.Contract do
   """
   alias AeMdw.Blocks
   alias AeMdw.Contract
-  alias AeMdw.Contracts.AexnContract
+  alias AeMdw.AexnContracts
   alias AeMdw.Db.IntCallsMutation
   alias AeMdw.Db.Model
   alias AeMdw.Db.Mutation
@@ -112,21 +112,26 @@ defmodule AeMdw.Db.Sync.Contract do
   def aexn_create_contract_mutation(contract_pk, block_index, txi) do
     aexn_type =
       cond do
-        AexnContract.is_aex9?(contract_pk) -> :aex9
-        AexnContract.is_aex141?(contract_pk) -> :aex141
+        AexnContracts.is_aex9?(contract_pk) -> :aex9
+        AexnContracts.has_aex141_signatures?(contract_pk) -> :aex141
         true -> nil
       end
 
     with true <- aexn_type != nil,
-         {:ok, aexn_meta_info} <-
-           AexnContract.call_meta_info(contract_pk) do
-      AexnCreateContractMutation.new(
-        aexn_type,
-        contract_pk,
-        aexn_meta_info,
-        block_index,
-        txi
-      )
+         {:ok, aexn_extensions} <- AexnContracts.call_extensions(aexn_type, contract_pk),
+         {:ok, aexn_meta_info} <- AexnContracts.call_meta_info(contract_pk) do
+      if aexn_type == :aex9 or
+           (aexn_type == :aex141 and
+              AexnContracts.has_valid_aex141_extensions?(aexn_extensions, contract_pk)) do
+        AexnCreateContractMutation.new(
+          aexn_type,
+          contract_pk,
+          aexn_meta_info,
+          block_index,
+          txi,
+          aexn_extensions
+        )
+      end
     else
       _false_or_notfound ->
         nil