Add offline mode: Allow website to work without Telegram clients

main.py

@@ -25,7 +25,13 @@ async def lifespan(app: FastAPI):
     reset_cache_dir()
 
     # Initialize the clients
-    await initialize_clients()
+    clients_initialized = await initialize_clients()
+    
+    # If clients failed to initialize, initialize DRIVE_DATA in offline mode
+    if not clients_initialized:
+        from utils.directoryHandler import initDriveDataWithoutClients
+        await initDriveDataWithoutClients()
+        logger.warning("Website is running in OFFLINE MODE - Telegram features are disabled")
 
     # Start the website auto ping task
     asyncio.create_task(auto_ping_website())
@@ -60,7 +66,11 @@ async def static_files(file_path):
 @app.get("/file")
 async def dl_file(request: Request):
     from utils.directoryHandler import DRIVE_DATA
+    from utils.clients import has_clients
 
+    if not has_clients():
+        raise HTTPException(status_code=503, detail="Telegram clients not available - Service running in offline mode")
+    
     path = request.query_params["path"]
     file = DRIVE_DATA.get_file(path)
     return await media_streamer(STORAGE_CHANNEL, file.file_id, file.name, request)
@@ -156,10 +166,14 @@ async def upload_file(
     total_size: str = Form(...),
 ):
     global SAVE_PROGRESS
+    from utils.clients import has_clients
 
     if password != ADMIN_PASSWORD:
         return JSONResponse({"status": "Invalid password"})
 
+    if not has_clients():
+        return JSONResponse({"status": "Telegram clients not available - Service running in offline mode"})
+
     total_size = int(total_size)
     SAVE_PROGRESS[id] = ("running", 0, total_size)
 
@@ -304,11 +318,16 @@ async def getFileInfoFromUrl(request: Request):
 
 @app.post("/api/startFileDownloadFromUrl")
 async def startFileDownloadFromUrl(request: Request):
+    from utils.clients import has_clients
+    
     data = await request.json()
 
     if data["password"] != ADMIN_PASSWORD:
         return JSONResponse({"status": "Invalid password"})
 
+    if not has_clients():
+        return JSONResponse({"status": "Telegram clients not available - Service running in offline mode"})
+
     logger.info(f"startFileDownloadFromUrl {data}")
     try:
         id = getRandomID()

utils/clients.py

@@ -84,9 +84,12 @@ async def initialize_clients():
             ]
         )
     )
-    if len(multi_clients) == 0:
-        logger.error("No Clients Were Initialized")
-        raise RuntimeError("No Clients Were Initialized - Check BOT_TOKENS configuration")
+    
+    clients_initialized = len(multi_clients) > 0
+    
+    if not clients_initialized:
+        logger.warning("No Clients Were Initialized - Website will run in offline mode")
+        return False
 
     if len(premium_clients) == 0:
         logger.info("No Premium Clients Were Initialized")
@@ -98,6 +101,14 @@ async def initialize_clients():
 
     # Start the backup drive data task
     asyncio.create_task(backup_drive_data())
+    
+    return True
+
+
+def has_clients():
+    """Check if any clients are available"""
+    global multi_clients
+    return len(multi_clients) > 0
 
 
 def get_client(premium_required=False) -> Client:

utils/directoryHandler.py

@@ -345,23 +345,19 @@ async def loadDriveData():
     logger.info("Loading drive data.")
     from utils.clients import get_client
 
-    client = get_client()
     try:
+        client = get_client()
         try:
             msg: Message = await client.get_messages(
                 config.STORAGE_CHANNEL, config.DATABASE_BACKUP_MSG_ID
             )
         except Exception as e:
             logger.error(f"Error fetching backup message: {e}")
-
-            # Forcefully terminates the program immediately
-            os.kill(os.getpid(), signal.SIGKILL)
+            raise
 
         if not msg.document:
-            logger.error(f"Error fetching backup message: {e}")
-
-            # Forcefully terminates the program immediately
-            os.kill(os.getpid(), signal.SIGKILL)
+            logger.error("Backup message does not contain a document")
+            raise Exception("Backup message does not contain a document")
 
         if msg.document.file_name == "drive.data":
             dl_path = await msg.download()
@@ -385,3 +381,29 @@ async def loadDriveData():
         BOT_MODE = NewBotMode(DRIVE_DATA)
         await start_bot_mode(DRIVE_DATA, BOT_MODE)
         logger.info("Bot mode started.")
+
+
+async def initDriveDataWithoutClients():
+    """Initialize DRIVE_DATA without requiring Telegram clients (offline mode)"""
+    global DRIVE_DATA, BOT_MODE
+    
+    logger.info("Initializing drive data in offline mode (without Telegram clients).")
+    
+    # Try to load from local cache if it exists
+    if drive_cache_path.exists():
+        try:
+            with open(drive_cache_path, "rb") as f:
+                DRIVE_DATA = dill.load(f)
+            logger.info("Drive data loaded from local cache.")
+        except Exception as e:
+            logger.warning(f"Failed to load from local cache: {e}")
+            logger.info("Creating new drive.data file.")
+            DRIVE_DATA = NewDriveData({"/": Folder("/", "/")}, [])
+            DRIVE_DATA.save()
+    else:
+        logger.info("No local cache found. Creating new drive.data file.")
+        DRIVE_DATA = NewDriveData({"/": Folder("/", "/")}, [])
+        DRIVE_DATA.save()
+    
+    await init_drive_data()
+    logger.info("Drive data initialized in offline mode.")