Merge pull request ilyarolf#111 from ilyarolf/develop

Develop

Author: ilyarolf

.env

@@ -8,12 +8,14 @@ DB_NAME = ""
 DB_ENCRYPTION = ""
 DB_PASS = ""
 NGROK_TOKEN = ""
-PAGE_ENTRIES = ""
-BOT_LANGUAGE = ""
-MULTIBOT = ""
-ETHPLORER_API_KEY = ""
-CURRENCY = ""
+PAGE_ENTRIES = "8"
+BOT_LANGUAGE = "en"
+MULTIBOT = "false"
+CURRENCY = "USD"
 RUNTIME_ENVIRONMENT = "DEV"
 WEBHOOK_SECRET_TOKEN = ""
-REDIS_HOST = ""
-REDIS_PASSWORD = ""
+KRYPTO_EXPRESS_API_KEY = ""
+KRYPTO_EXPRESS_API_URL = "https://kryptoexpress.pro/api"
+KRYPTO_EXPRESS_API_SECRET = ""
+REDIS_PASSWORD = ""
+REDIS_HOST = ""

bot.py

@@ -12,12 +12,14 @@
 from db import create_db_and_tables
 import uvicorn
 from fastapi.responses import JSONResponse
+from processing.processing import processing_router
 from services.notification import NotificationService
 
 redis = Redis(host=config.REDIS_HOST, password=config.REDIS_PASSWORD)
 bot = Bot(config.TOKEN, default=DefaultBotProperties(parse_mode=ParseMode.HTML))
 dp = Dispatcher(storage=RedisStorage(redis))
 app = FastAPI()
+app.include_router(processing_router)
 
 
 @app.post(config.WEBHOOK_PATH)

callbacks.py

@@ -4,6 +4,7 @@
 from aiogram.filters.callback_data import CallbackData
 
 from enums.bot_entity import BotEntity
+from enums.cryptocurrency import Cryptocurrency
 from utils.localizator import Localizator
 
 
@@ -160,8 +161,8 @@ def create(level: int, statistics_entity: StatisticsEntity | None = None,
 
 
 class WalletCallback(BaseCallback, prefix="wallet"):
-    pass
+    cryptocurrency: Cryptocurrency | None
 
     @staticmethod
-    def create(level: int):
-        return WalletCallback(level=level)
+    def create(level: int, cryptocurrency: Cryptocurrency | None = None):
+        return WalletCallback(level=level, cryptocurrency=cryptocurrency)

config.py

@@ -27,8 +27,10 @@
 PAGE_ENTRIES = int(os.environ.get("PAGE_ENTRIES"))
 BOT_LANGUAGE = os.environ.get("BOT_LANGUAGE")
 MULTIBOT = os.environ.get("MULTIBOT", False) == 'true'
-ETHPLORER_API_KEY = os.environ.get("ETHPLORER_API_KEY")
 CURRENCY = Currency(os.environ.get("CURRENCY"))
+KRYPTO_EXPRESS_API_KEY = os.environ.get("KRYPTO_EXPRESS_API_KEY")
+KRYPTO_EXPRESS_API_URL = os.environ.get("KRYPTO_EXPRESS_API_URL")
+KRYPTO_EXPRESS_API_SECRET = os.environ.get("KRYPTO_EXPRESS_API_SECRET")
 WEBHOOK_SECRET_TOKEN = os.environ.get("WEBHOOK_SECRET_TOKEN")
 REDIS_HOST = os.environ.get("REDIS_HOST")
 REDIS_PASSWORD = os.environ.get("REDIS_PASSWORD")

crypto_api/CryptoApiWrapper.py

@@ -0,0 +1,64 @@
+import aiohttp
+import config
+from enums.cryptocurrency import Cryptocurrency
+from enums.withdraw_type import WithdrawType
+from models.withdrawal import WithdrawalDTO
+
+
+class CryptoApiWrapper:
+    LTC_API_BASENAME_TX = "https://litecoinspace.org/tx/"
+    BTC_API_BASENAME_TX = "https://mempool.space/tx/"
+    SOL_API_BASENAME_TX = "https://solscan.io/tx/"
+    ETH_API_BASENAME_TX = "https://etherscan.io/tx/"
+    BNB_API_BASENAME_TX = "https://bscscan.com/tx/"
+
+    @staticmethod
+    async def fetch_api_request(url: str, params: dict | None = None, method: str = "GET", data: str | None = None,
+                                headers: dict | None = None) -> dict:
+        async with aiohttp.ClientSession() as session:
+            async with session.request(method, url, params=params, data=data, headers=headers) as response:
+                if response.status == 200:
+                    data = await response.json()
+                    return data
+
+    @staticmethod
+    async def get_crypto_prices() -> dict:
+        url = f"https://api.coingecko.com/api/v3/simple/price"
+        params = {
+            "ids": "bitcoin,litecoin,solana,ethereum,binancecoin",
+            "vs_currencies": "usd,eur,gbp,jpy,cad"
+        }
+        return await CryptoApiWrapper.fetch_api_request(url, params)
+
+    @staticmethod
+    async def get_wallet_balance() -> dict:
+        url = f"{config.KRYPTO_EXPRESS_API_URL}/wallet"
+        headers = {
+            "X-Api-Key": config.KRYPTO_EXPRESS_API_KEY
+        }
+        response = await CryptoApiWrapper.fetch_api_request(
+            url,
+            headers=headers
+        )
+        return {k: v for k, v in response.items() if v > 0}
+
+    @staticmethod
+    async def withdrawal(cryptocurrency: Cryptocurrency, to_address: str, only_calculate: bool) -> WithdrawalDTO:
+        url = f"{config.KRYPTO_EXPRESS_API_URL}/wallet/withdrawal"
+        headers = {
+            "X-Api-Key": config.KRYPTO_EXPRESS_API_KEY,
+            "Content-Type": "application/json"
+        }
+        body = WithdrawalDTO(
+            withdrawType=WithdrawType.ALL,
+            cryptoCurrency=cryptocurrency.name,
+            toAddress=to_address,
+            onlyCalculate=only_calculate
+        )
+        response = await CryptoApiWrapper.fetch_api_request(
+            url,
+            method="POST",
+            data=body.model_dump_json(),
+            headers=headers
+        )
+        return WithdrawalDTO.model_validate(response)

docker-compose.yml

@@ -19,26 +19,8 @@ services:
     build:
       context: .
     container_name: 'AiogramShopBot'
-    environment:
-      WEBHOOK_PATH: "/"
-      WEBAPP_HOST: "0.0.0.0" # Don't touch this
-      WEBAPP_PORT: 5000 # Here your port
-      TOKEN: "1234567890:QWER.....TYI" # Here your bot token from botfather
-      NGROK_TOKEN: 'NGROK_TOKEN_HERE' # Here your ngrok token from ngrok.com
-      ADMIN_ID_LIST: "12345678,87654321" # Telegram ID's for admins;
-      SUPPORT_LINK: "https://t.me/YourUsername"
-      DB_NAME: "database.db" # Here your database name
-      DB_ENCRYPTION: "false" # DB encryption option
-      DB_PASS: "1234567890" # Here your database password (Required only if DB_ENCRYPTION-"true”)
-      PAGE_ENTRIES: 8 # Items per page
-      BOT_LANGUAGE: "en" # The name of your file from the l10n folder without the .json suffix
-      MULTIBOT: "false" # Allows the use of a multibot
-      ETHPLORER_API_KEY: "" # API key from Ethplorer
-      CURRENCY: "USD" # fiat currency
-      RUNTIME_ENVIRONMENT: "PROD"
-      WEBHOOK_SECRET_TOKEN: "1234567890" # Any string you want
-      REDIS_HOST: "redis"
-      REDIS_PASSWORD: "1234567890" # Any string you want
+    env_file:
+      - .env
     labels:
       caddy: YOUR-DOMAIN-GOES-HERE
       caddy.reverse_proxy: "bot:5000"

enums/cryptocurrency.py

@@ -1,38 +1,37 @@
 from enum import Enum
 
+import config
+
 
 class Cryptocurrency(str, Enum):
+    BNB = "BNB"
     BTC = "BTC"
     LTC = "LTC"
+    ETH = "ETH"
     SOL = "SOL"
-    USDT_TRC20 = "USDT_TRC20"
-    USDT_ERC20 = "USDT_ERC20"
-    USDC_ERC20 = "USDC_ERC20"
 
-    def get_balance_field(self) -> str:
+    def get_divider(self):
         match self:
             case Cryptocurrency.BTC:
-                return "btc_balance"
+                return 8
             case Cryptocurrency.LTC:
-                return "ltc_balance"
+                return 8
+            case Cryptocurrency.ETH:
+                return 18
             case Cryptocurrency.SOL:
-                return "sol_balance"
-            case Cryptocurrency.USDT_TRC20:
-                return "usdt_trc20_balance"
-            case Cryptocurrency.USDT_ERC20:
-                return "usdt_erc20_balance"
-            case Cryptocurrency.USDC_ERC20:
-                return "usdc_erc20_balance"
+                return 9
+            case Cryptocurrency.BNB:
+                return 18
 
-    def get_address_field(self) -> str:
+    def get_coingecko_name(self) -> str:
         match self:
             case Cryptocurrency.BTC:
-                return "btc_address"
+                return "bitcoin"
             case Cryptocurrency.LTC:
-                return "ltc_address"
+                return "litecoin"
+            case Cryptocurrency.ETH:
+                return "ethereum"
+            case Cryptocurrency.BNB:
+                return "binancecoin"
             case Cryptocurrency.SOL:
-                return "sol_address"
-            case Cryptocurrency.USDT_TRC20:
-                return "trx_address"
-            case _:
-                return "eth_address"
+                return "solana"

enums/payment.py

@@ -0,0 +1,6 @@
+from enum import Enum
+
+
+class PaymentType(Enum):
+    DEPOSIT = "DEPOSIT"
+    PAYMENT = "PAYMENT"

enums/withdraw_type.py

@@ -0,0 +1,6 @@
+from enum import Enum
+
+
+class WithdrawType(Enum):
+    ALL = "ALL"
+    SINGLE = "SINGLE"

handlers/admin/constants.py

@@ -35,3 +35,7 @@ class AdminAnnouncementStates(StatesGroup):
 class UserManagementStates(StatesGroup):
     balance_amount = State()
     user_entity = State()
+
+
+class WalletStates(StatesGroup):
+    crypto_address = State()