d4rktg 1.2.4__py3-none-any.whl → 1.2.6__py3-none-any.whl

d4rk/Web/web_server.py

@@ -0,0 +1,52 @@
+# src/Web/web_server.py
+
+from typing import Literal
+
+from pyrogram import Client
+
+from aiohttp import web
+from aiohttp.web_runner import AppRunner
+
+from d4rk.Logs import setup_logger
+from d4rk.Utils import get_public_ip, check_public_ip_reachable
+
+from d4rk.Web.web import _web_server
+WEB_APP = 'http://localhost'
+
+logger = setup_logger("WebServerManager")
+
+class WebServerManager:
+    def __init__(self,bot:Client=None) -> None:
+        self._web_runner = None
+        self._tcp_site = None
+        self._web_port = None
+        self._bot = bot
+
+    async def setup_web_server(self, preferred_port=8443) -> AppRunner | Literal[False]:
+        try:
+            self._web_port = preferred_port
+            logger.info(f'Starting API server on port {preferred_port}...')
+            
+            self._web_runner = web.AppRunner(await _web_server(self._bot))
+            await self._web_runner.setup()
+            self._tcp_site = web.TCPSite(self._web_runner, "0.0.0.0", preferred_port)
+            await self._tcp_site.start()
+            if WEB_APP:
+                if 'localhost' in WEB_APP:logger.info(f"Web app is running on http://localhost:{preferred_port}")
+                else:logger.info(f"Web app is running on {WEB_APP}")
+            else:
+                myIP = get_public_ip()
+                if await check_public_ip_reachable(myIP):logger.info(f"Web app running on http://{myIP}:{preferred_port}")
+                else:logger.info(f"Web app running on http://localhost:{preferred_port}")
+            return self._web_runner
+        except Exception as e:
+            logger.error(f"Failed to setup web server: {e}")
+            return False
+
+    async def cleanup(self) -> None:
+        if self._web_runner:
+            await self._web_runner.cleanup()
+            logger.info("Web server cleaned up")
+
+    
+

d4rk/__init__.py

@@ -1,6 +1,19 @@
 from .Database import db
-from .Handlers import BotManager, scheduler , FontMessageMixin
+from .Handlers import BotManager , FontMessageMixin
 from .Logs import setup_logger , get_timezone_offset
 from .Utils import *
+from ._base import TGBase
+from ._bot_manager import D4RK_BotManager
+from . import errors
 
-__version__ = "0.9.7"
+__version__ = "0.9.7"
+__all__ = [
+    "TGBase",
+    "D4RK_BotManager",
+    "db",
+    "errors",
+    "FontMessageMixin",
+    "setup_logger",
+    "get_timezone_offset",
+    "BotManager",
+]

d4rk/_base.py

@@ -0,0 +1,115 @@
+import sys
+import signal
+import asyncio
+
+from pyrogram import Client
+
+from d4rk.Logs import setup_logger
+from d4rk.Handlers import BotManager, FontMessageMixin , LoggerBotUtil
+
+logger = setup_logger("TGBase")
+
+class TGBase(FontMessageMixin, BotManager, Client):
+    def __init__(self, 
+                api_id: int = None,
+                api_hash: str = None, 
+                token: str = None,
+                app_name: str = None,
+                plugins: dict = None,
+                database_url: str = None,
+                log_chat_id: int = None,
+                owner_id: int = None,
+                web_app_url: str = None,
+                web_server_manager = None,
+                rename: bool = False,
+                logger_bot_util: LoggerBotUtil = None) -> None:
+        
+        self.api_id = api_id
+        self.api_hash = api_hash
+        self.token = token
+        self.app_name = app_name
+        self.plugins = plugins
+        self.database_url = database_url
+        self.log_chat_id = log_chat_id
+        self.owner_id = owner_id
+        self.web_app_url = web_app_url
+        self.web_server_manager = web_server_manager
+        self._rename = rename
+        self.logger_bot_util = logger_bot_util
+        self.start_message = ""
+        self.start_message_id = None
+        self.start_message_chat_id = None
+        self._is_connected = False
+        
+        super().__init__(
+            name=self.app_name,
+            api_id=self.api_id,
+            api_hash=self.api_hash,
+            bot_token=self.token,
+            plugins=self.plugins,
+            in_memory=True
+        )
+
+    def run(self):
+        """Run single bot - simple blocking call"""
+        signal.signal(signal.SIGINT, self._signal_handler)
+        signal.signal(signal.SIGTERM, self._signal_handler)
+
+        try:
+            loop = asyncio.get_event_loop()
+        except RuntimeError:
+            loop = asyncio.new_event_loop()
+            asyncio.set_event_loop(loop)
+        
+        try:
+            loop.run_until_complete(self._run_async())
+        except KeyboardInterrupt:
+            logger.info("Received interrupt signal, stopping bot...")
+            loop.run_until_complete(self._stop_single_bot())
+
+    def _signal_handler(self, signum, frame):
+        """Handle shutdown signals"""
+        logger.info(f"Received signal {signum}, stopping bot...")
+        sys.exit(0)
+
+    async def _run_async(self):
+        """Internal async method to run single bot"""
+        await self._send_startup_message()
+        await self._start_single_bot()
+        try:
+            while True:
+                await asyncio.sleep(5)
+        except Exception as e:
+            logger.error(f"Error in bot operation: {e}")
+        finally:
+            await self._stop_single_bot()
+
+    async def _send_startup_message(self):
+        """Send startup message using logger bot if available"""
+        if self.logger_bot_util and self.log_chat_id:
+            try:
+                logger.info("Sending startup message via logger bot...")
+                await self.logger_bot_util.send_log_message(
+                    chat_id=self.log_chat_id,
+                    message=f"🚀 Starting bot {self.app_name}..."
+                )
+            except Exception as e:
+                logger.error(f"Failed to send startup message: {e}")
+
+    async def _start_single_bot(self):
+        """Start this single bot instance"""
+        await self.powerup(self.app_name)
+
+    async def _stop_single_bot(self):
+        """Stop this single bot instance"""
+        try:
+            if hasattr(self, '_is_connected') and self._is_connected:
+                await self.powerdown()
+                await asyncio.sleep(2)
+                logger.info(f"Bot instance {self.app_name} stopped successfully")
+            else:
+                logger.info(f"Bot instance {self.app_name} was not running, skipping stop")
+        except Exception as e:
+            logger.error(f"Error stopping bot instance {self.app_name}: {e}")
+
+__all__ = ['TGBase']