From ec36ec6fed7f9893af80c1bf2b68871c56e01688 Mon Sep 17 00:00:00 2001
From: kompotkot <sumarokovsergei@yandex.ru>
Date: Tue, 25 Jun 2024 14:17:38 +0000
Subject: [PATCH] Moonstream API mantle support

---
 moonstreamapi/configs/sample.env              |  2 +
 .../moonstreamapi/admin/subscription_types.py | 44 +++++++++++++++++++
 .../moonstreamapi/providers/__init__.py       |  4 ++
 .../providers/moonworm_provider.py            | 16 +++++++
 .../moonstreamapi/providers/transactions.py   | 14 ++++++
 moonstreamapi/moonstreamapi/settings.py       | 13 ++++++
 moonstreamapi/moonstreamapi/web3_provider.py  |  8 +++-
 moonstreamapi/requirements.txt                |  2 +-
 moonstreamapi/setup.py                        |  2 +-
 9 files changed, 102 insertions(+), 3 deletions(-)

diff --git a/moonstreamapi/configs/sample.env b/moonstreamapi/configs/sample.env
index 824071a8..eae85085 100644
--- a/moonstreamapi/configs/sample.env
+++ b/moonstreamapi/configs/sample.env
@@ -34,6 +34,8 @@ export MOONSTREAM_AVALANCHE_WEB3_PROVIDER_URI="https://<connection_path_uri_to_n
 export MOONSTREAM_AVALANCHE_FUJI_WEB3_PROVIDER_URI="https://<connection_path_uri_to_node>"
 export MOONSTREAM_BLAST_WEB3_PROVIDER_URI="https://<connection_path_uri_to_node>"
 export MOONSTREAM_BLAST_SEPOLIA_WEB3_PROVIDER_URI="https://<connection_path_uri_to_node>"
+export MOONSTREAM_MANTLE_WEB3_PROVIDER_URI="https://<connection_path_uri_to_node>"
+export MOONSTREAM_MANTLE_SEPOLIA_WEB3_PROVIDER_URI="https://<connection_path_uri_to_node>"
 
 export MOONSTREAM_QUERIES_JOURNAL_ID="<bugout_journal_id_where_store_queries_for_executing>"
 export MOONSTREAM_USAGE_REPORTS_JOURNAL_ID="<bugout_journal_id_where_save_generated_reports>"
diff --git a/moonstreamapi/moonstreamapi/admin/subscription_types.py b/moonstreamapi/moonstreamapi/admin/subscription_types.py
index fb303beb..f79efaad 100644
--- a/moonstreamapi/moonstreamapi/admin/subscription_types.py
+++ b/moonstreamapi/moonstreamapi/admin/subscription_types.py
@@ -260,6 +260,28 @@ CANONICAL_SUBSCRIPTION_TYPES = {
         stripe_price_id=None,
         active=False,
     ),
+    "mantle_smartcontract": SubscriptionTypeResourceData(
+        id="mantle_smartcontract",
+        name="Mantle smartcontract",
+        blockchain="mantle",
+        choices=["input:address", "tag:erc721"],
+        description="Contracts events and tx_calls of contract of Mantle blockchain.",
+        icon_url="https://static.simiotics.com/moonstream/assets/mantle-logo.png",
+        stripe_product_id=None,
+        stripe_price_id=None,
+        active=True,
+    ),
+    "mantle_sepolia_smartcontract": SubscriptionTypeResourceData(
+        id="mantle_sepolia_smartcontract",
+        name="Mantle Sepolia smartcontract",
+        blockchain="mantle_sepolia",
+        choices=["input:address", "tag:erc721"],
+        description="Contracts events and tx_calls of contract of Mantle Sepolia blockchain.",
+        icon_url="https://static.simiotics.com/moonstream/assets/mantle-sepolia-logo.png",
+        stripe_product_id=None,
+        stripe_price_id=None,
+        active=True,
+    ),
     "mumbai_blockchain": SubscriptionTypeResourceData(
         id="mumbai_blockchain",
         name="Mumbai transactions",
@@ -470,6 +492,28 @@ CANONICAL_SUBSCRIPTION_TYPES = {
         stripe_price_id=None,
         active=True,
     ),
+    "mantle_blockchain": SubscriptionTypeResourceData(
+        id="mantle_blockchain",
+        name="Mantle smartcontract",
+        blockchain="mantle",
+        choices=["input:address", "tag:erc721"],
+        description="Mantle chain transactions subscription.",
+        icon_url="https://static.simiotics.com/moonstream/assets/mantle-logo.png",
+        stripe_product_id=None,
+        stripe_price_id=None,
+        active=False,
+    ),
+    "mantle_sepolia_blockchain": SubscriptionTypeResourceData(
+        id="mantle_sepolia_blockchain",
+        name="Mantle Sepolia transactions",
+        blockchain="mantle_sepolia",
+        choices=["input:address", "tag:erc721"],
+        description="Mantle Sepolia chain transactions subscription.",
+        icon_url="https://static.simiotics.com/moonstream/assets/mantle-logo.png",
+        stripe_product_id=None,
+        stripe_price_id=None,
+        active=False,
+    ),
 }
 
 
diff --git a/moonstreamapi/moonstreamapi/providers/__init__.py b/moonstreamapi/moonstreamapi/providers/__init__.py
index eab6a2dd..454d79c3 100644
--- a/moonstreamapi/moonstreamapi/providers/__init__.py
+++ b/moonstreamapi/moonstreamapi/providers/__init__.py
@@ -65,6 +65,8 @@ event_providers: Dict[str, Any] = {
     moonworm_provider.AvalancheFujiMoonwormProvider.event_type: moonworm_provider.AvalancheFujiMoonwormProvider,
     moonworm_provider.BlastMoonwormProvider.event_type: moonworm_provider.BlastMoonwormProvider,
     moonworm_provider.BlastSepoliaMoonwormProvider.event_type: moonworm_provider.BlastSepoliaMoonwormProvider,
+    moonworm_provider.MantleMoonwormProvider.event_type: moonworm_provider.MantleMoonwormProvider,
+    moonworm_provider.MantleSepoliaMoonwormProvider.event_type: moonworm_provider.MantleSepoliaMoonwormProvider,
     transactions.EthereumTransactions.event_type: transactions.EthereumTransactions,
     transactions.PolygonTransactions.event_type: transactions.PolygonTransactions,
     transactions.ProofOfPlayApexTransactions.event_type: transactions.ProofOfPlayApexTransactions,
@@ -83,6 +85,8 @@ event_providers: Dict[str, Any] = {
     transactions.AvalancheFujiTransactions.event_type: transactions.AvalancheFujiTransactions,
     transactions.BlastTransactions.event_type: transactions.BlastTransactions,
     transactions.BlastSepoliaTransactions.event_type: transactions.BlastSepoliaTransactions,
+    transactions.MantleTransactions.event_type: transactions.MantleTransactions,
+    transactions.MantleSepoliaTransactions.event_type: transactions.MantleSepoliaTransactions,
     bugout.polygon_whalewatch_provider.event_type: bugout.polygon_whalewatch_provider,
     bugout.ethereum_txpool_provider.event_type: bugout.ethereum_txpool_provider,
     bugout.ethereum_whalewatch_provider.event_type: bugout.ethereum_whalewatch_provider,
diff --git a/moonstreamapi/moonstreamapi/providers/moonworm_provider.py b/moonstreamapi/moonstreamapi/providers/moonworm_provider.py
index 3d14b17c..16986e64 100644
--- a/moonstreamapi/moonstreamapi/providers/moonworm_provider.py
+++ b/moonstreamapi/moonstreamapi/providers/moonworm_provider.py
@@ -35,6 +35,8 @@ avalanche_event_type = "avalanche_blockchain"
 avalanche_fuji_sepolia_event_type = "avalanche_fuji_sepolia_blockchain"
 blast_event_type = "blast_blockchain"
 blast_sepolia_event_type = "blast_sepolia_blockchain"
+mantle_event_type = "mantle_blockchain"
+mantle_sepolia_event_type = "mantle_sepolia_blockchain"
 
 allowed_tags = ["tag:erc721"]
 
@@ -526,3 +528,17 @@ ProofOfPlayApexMoonwormProvider = MoonwormProvider(
     description="Provider for reviving transactions from Proof of Play Apex tables.",
     streamboaundary_range_limit=2 * 60 * 60,
 )
+
+MantleMoonwormProvider = MoonwormProvider(
+    event_type="mantle_smartcontract",
+    blockchain=AvailableBlockchainType("mantle"),
+    description="Provider for reviving transactions from Mantle tables.",
+    streamboaundary_range_limit=2 * 60 * 60,
+)
+
+MantleSepoliaMoonwormProvider = MoonwormProvider(
+    event_type="mantle_sepolia_smartcontract",
+    blockchain=AvailableBlockchainType("mantle_sepolia"),
+    description="Provider for reviving transactions from Mantle Sepolia tables.",
+    streamboaundary_range_limit=2 * 60 * 60,
+)
diff --git a/moonstreamapi/moonstreamapi/providers/transactions.py b/moonstreamapi/moonstreamapi/providers/transactions.py
index 1179a243..c7674b09 100644
--- a/moonstreamapi/moonstreamapi/providers/transactions.py
+++ b/moonstreamapi/moonstreamapi/providers/transactions.py
@@ -573,3 +573,17 @@ ProofOfPlayApexTransactions = TransactionsProvider(
     description="Provider for resiving transactions from Proof of Play Apex tables.",
     streamboaundary_range_limit=2 * 60 * 60,
 )
+
+MantleTransactions = TransactionsProvider(
+    event_type="mantle_blockchain",
+    blockchain=AvailableBlockchainType("mantle"),
+    description="Provider for resiving transactions from Mantle tables.",
+    streamboaundary_range_limit=2 * 60 * 60,
+)
+
+MantleSepoliaTransactions = TransactionsProvider(
+    event_type="mantle_sepolia_blockchain",
+    blockchain=AvailableBlockchainType("mantle_sepolia"),
+    description="Provider for resiving transactions from Mantle Sepolia tables.",
+    streamboaundary_range_limit=2 * 60 * 60,
+)
diff --git a/moonstreamapi/moonstreamapi/settings.py b/moonstreamapi/moonstreamapi/settings.py
index a8ba7b0d..b1269805 100644
--- a/moonstreamapi/moonstreamapi/settings.py
+++ b/moonstreamapi/moonstreamapi/settings.py
@@ -228,6 +228,19 @@ if MOONSTREAM_PROOFOFPLAY_APEX_WEB3_PROVIDER_URI == "":
     raise Exception(
         "MOONSTREAM_PROOFOFPLAY_APEX_WEB3_PROVIDER_URI env variable is not set"
     )
+MOONSTREAM_MANTLE_WEB3_PROVIDER_URI = os.environ.get(
+    "MOONSTREAM_MANTLE_WEB3_PROVIDER_URI", ""
+)
+if MOONSTREAM_MANTLE_WEB3_PROVIDER_URI == "":
+    raise Exception("MOONSTREAM_MANTLE_WEB3_PROVIDER_URI env variable is not set")
+
+MOONSTREAM_MANTLE_SEPOLIA_WEB3_PROVIDER_URI = os.environ.get(
+    "MOONSTREAM_MANTLE_SEPOLIA_WEB3_PROVIDER_URI", ""
+)
+if MOONSTREAM_MANTLE_SEPOLIA_WEB3_PROVIDER_URI == "":
+    raise Exception(
+        "MOONSTREAM_MANTLE_SEPOLIA_WEB3_PROVIDER_URI env variable is not set"
+    )
 
 ## QueryAPI
 
diff --git a/moonstreamapi/moonstreamapi/web3_provider.py b/moonstreamapi/moonstreamapi/web3_provider.py
index ca74e014..7ebc15aa 100644
--- a/moonstreamapi/moonstreamapi/web3_provider.py
+++ b/moonstreamapi/moonstreamapi/web3_provider.py
@@ -13,14 +13,16 @@ from web3.providers.rpc import HTTPProvider
 
 from .settings import (
     MOONSTREAM_AMOY_WEB3_PROVIDER_URI,
-    MOONSTREAM_ARBITRUM_ONE_WEB3_PROVIDER_URI,
     MOONSTREAM_ARBITRUM_NOVA_WEB3_PROVIDER_URI,
+    MOONSTREAM_ARBITRUM_ONE_WEB3_PROVIDER_URI,
     MOONSTREAM_ARBITRUM_SEPOLIA_WEB3_PROVIDER_URI,
     MOONSTREAM_AVALANCHE_FUJI_WEB3_PROVIDER_URI,
     MOONSTREAM_AVALANCHE_WEB3_PROVIDER_URI,
     MOONSTREAM_BLAST_SEPOLIA_WEB3_PROVIDER_URI,
     MOONSTREAM_BLAST_WEB3_PROVIDER_URI,
     MOONSTREAM_ETHEREUM_WEB3_PROVIDER_URI,
+    MOONSTREAM_MANTLE_SEPOLIA_WEB3_PROVIDER_URI,
+    MOONSTREAM_MANTLE_WEB3_PROVIDER_URI,
     MOONSTREAM_MUMBAI_WEB3_PROVIDER_URI,
     MOONSTREAM_POLYGON_WEB3_PROVIDER_URI,
     MOONSTREAM_PROOFOFPLAY_APEX_WEB3_PROVIDER_URI,
@@ -109,6 +111,10 @@ def connect(
             web3_uri = MOONSTREAM_BLAST_SEPOLIA_WEB3_PROVIDER_URI
         elif blockchain_type == AvailableBlockchainType.PROOFOFPLAY_APEX:
             web3_uri = MOONSTREAM_PROOFOFPLAY_APEX_WEB3_PROVIDER_URI
+        elif blockchain_type == AvailableBlockchainType.MANTLE:
+            web3_uri = MOONSTREAM_MANTLE_WEB3_PROVIDER_URI
+        elif blockchain_type == AvailableBlockchainType.MANTLE_SEPOLIA:
+            web3_uri = MOONSTREAM_MANTLE_SEPOLIA_WEB3_PROVIDER_URI
         else:
             raise Exception("Wrong blockchain type provided for web3 URI")
 
diff --git a/moonstreamapi/requirements.txt b/moonstreamapi/requirements.txt
index 57840006..623bb0e1 100644
--- a/moonstreamapi/requirements.txt
+++ b/moonstreamapi/requirements.txt
@@ -37,7 +37,7 @@ lru-dict==1.1.8
 Mako==1.2.3
 MarkupSafe==2.1.1
 moonstream==0.1.1
-moonstreamdb==0.4.4
+moonstreamdb==0.4.5
 moonstreamdb-v3==0.0.9
 multiaddr==0.0.9
 multidict==6.0.2
diff --git a/moonstreamapi/setup.py b/moonstreamapi/setup.py
index 2e3567be..b6b5b47f 100644
--- a/moonstreamapi/setup.py
+++ b/moonstreamapi/setup.py
@@ -16,7 +16,7 @@ setup(
         "bugout>=0.2.15",
         "fastapi",
         "moonstream",
-        "moonstreamdb>=0.4.4",
+        "moonstreamdb>=0.4.5",
         "moonstreamdb-v3>=0.0.9",
         "humbug",
         "pydantic==1.10.2",