Update app.py

app.py

@@ -1,283 +1,286 @@
 import os
 import threading
 import time
+import hashlib
+import random
+import string
+from datetime import datetime
+
 from flask_srvr import run_flask
+from telegram import *
+from telegram.ext import *
 
+import mdb
+
+# =========================
+# 🚀 FLASK
+# =========================
 def start_flask():
-    flask_thread = threading.Thread(target=run_flask)
-    flask_thread.daemon = True  # background me chalega
-    flask_thread.start()
+    t = threading.Thread(target=run_flask)
+    t.daemon = True
+    t.start()
 
-# Flask start karo
 start_flask()
 
-print("rinning")
+# =========================
+# ⚙️ CONFIG
+# =========================
 BOT_TOKEN = os.getenv("tgtkn")
 GROUP_ID = -1003799294466
-
-import time
-from datetime import datetime
-from telegram import *
-from telegram.ext import *
-import mdb
-
-waiting_for_file = set()
-PAGE_SIZE = 5
+MAX_STORAGE = 15 * 1024 * 1024 * 1024
+HARDCODE_ADMIN = 83939584939485
 
 # =========================
-# 🧠 DATABASE
+# 🧠 STATE
 # =========================
+user_state = {}
 
-def get_data(user_id):
-    key = f"user_{user_id}"
-    data = mdb.get(key, "json")
+# =========================
+# 🗄 DB
+# =========================
+def get_data(uid):
+    data = mdb.get(f"user_{uid}", "json")
     if not data:
-        return {"files": [], "index": {}}
+        return {
+            "files": [],
+            "index": {},
+            "folders": {
+                "root": {
+                    "name": "root",
+                    "files": [],
+                    "folders": []
+                }
+            }
+        }
     return data
 
+def save_data(uid, data):
+    mdb.set(f"user_{uid}", "json", data)
 
-def save_data(user_id, data):
-    key = f"user_{user_id}"
-    mdb.set(key, "json", data)
+def get_admins():
+    return mdb.get("admins", "json") or []
 
+def is_admin(uid):
+    return uid == HARDCODE_ADMIN or uid in get_admins()
 
 # =========================
-# 🚀 START
+# 📊 STORAGE BAR
 # =========================
-
-async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    kb = [
-        [InlineKeyboardButton("📤 Upload", callback_data="upload")],
-        [InlineKeyboardButton("📂 My Files", callback_data="files_0")],
-        [InlineKeyboardButton("📊 Stats", callback_data="stats")]
-    ]
-    await update.message.reply_text(
-        "☁️ Advanced Cloud Bot",
-        reply_markup=InlineKeyboardMarkup(kb)
-    )
-
+def storage_bar(used, total, length=10):
+    ratio = used / total if total else 0
+    filled = int(ratio * length)
+    return "█" * filled + "░" * (length - filled)
 
 # =========================
-# 📂 PAGINATION
+# 🧾 ICON
 # =========================
+def get_icon(name):
+    ext = name.lower().split('.')[-1]
+    if ext == "pdf": return "📄"
+    if ext in ["mp4","mkv"]: return "🎥"
+    if ext in ["jpg","png","jpeg"]: return "🖼"
+    return "📦"
 
-def build_file_page(data, page):
-    files = data["files"]
-    start = page * PAGE_SIZE
-    end = start + PAGE_SIZE
-
-    chunk = files[start:end]
-
-    text = "📂 Your Files:\n\n"
-    kb = []
-
-    for f in chunk:
-        fname = f["file_name"]
-        text += f"• {fname}\n"
+# =========================
+# 🏠 MENU
+# =========================
+def main_menu(uid):
+    data = get_data(uid)
+    used = sum(f["file_size"] for f in data["files"])
 
-        kb.append([
-            InlineKeyboardButton("ℹ️", callback_data=f"info_{fname}"),
-            InlineKeyboardButton("🗑", callback_data=f"del_{fname}")
-        ])
+    bar = storage_bar(used, MAX_STORAGE)
 
-    nav = []
-    if page > 0:
-        nav.append(InlineKeyboardButton("⬅️", callback_data=f"files_{page-1}"))
-    if end < len(files):
-        nav.append(InlineKeyboardButton("➡️", callback_data=f"files_{page+1}"))
+    return f"""☁️ Cloud Bot
 
-    if nav:
-        kb.append(nav)
+📊 Storage:
+{bar}
+{round(used/1024/1024,2)} MB / 15360 MB
+"""
 
-    return text, InlineKeyboardMarkup(kb)
+def main_kb():
+    return InlineKeyboardMarkup([
+        [InlineKeyboardButton("📤 Upload", callback_data="upload_root")],
+        [InlineKeyboardButton("📂 My Files", callback_data="open_root")],
+        [InlineKeyboardButton("🔍 Search", callback_data="search")]
+    ])
 
+# =========================
+# 🚀 START
+# =========================
+async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    uid = update.effective_user.id
+    await update.message.reply_text(main_menu(uid), reply_markup=main_kb())
 
 # =========================
-# 🔘 BUTTONS
+# 📁 FOLDER VIEW
 # =========================
+def build_folder(data, folder):
+    f = data["folders"][folder]
 
-async def buttons(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    q = update.callback_query
-    await q.answer()
-    user_id = q.from_user.id
-    data = get_data(user_id)
+    kb = []
 
-    # 📤 upload
-    if q.data == "upload":
-        waiting_for_file.add(user_id)
-        await q.message.reply_text("📁 Send file")
+    for sub in f["folders"]:
+        kb.append([InlineKeyboardButton(f"📁 {sub}", callback_data=f"open_{sub}")])
 
-    # 📂 files pagination
-    elif q.data.startswith("files_"):
-        page = int(q.data.split("_")[1])
-        text, kb = build_file_page(data, page)
-        await q.message.reply_text(text, reply_markup=kb)
+    for fid in f["files"]:
+        file = data["index"][fid]
+        kb.append([InlineKeyboardButton(
+            f"{get_icon(file['original_name'])} {file['original_name']}",
+            callback_data=f"file_{fid}"
+        )])
 
-    # 📊 stats
-    elif q.data == "stats":
-        await q.message.reply_text(f"📊 Files: {len(data['files'])}")
+    kb.append([
+        InlineKeyboardButton("➕ Folder", callback_data=f"newfolder_{folder}"),
+        InlineKeyboardButton("✏️ Rename", callback_data=f"renamefolder_{folder}")
+    ])
 
-    # 🗑 delete confirm
-    elif q.data.startswith("del_"):
-        fname = q.data.split("_", 1)[1]
-        kb = [
-            [InlineKeyboardButton("✅ Confirm", callback_data=f"confirmdel_{fname}")],
-            [InlineKeyboardButton("❌ Cancel", callback_data="files_0")]
-        ]
-        await q.message.reply_text(f"Delete {fname}?", reply_markup=InlineKeyboardMarkup(kb))
+    kb.append([
+        InlineKeyboardButton("🗑 Delete", callback_data=f"delfolder_{folder}")
+    ])
 
-    elif q.data.startswith("confirmdel_"):
-        fname = q.data.split("_", 1)[1]
+    kb.append([
+        InlineKeyboardButton("📤 Upload", callback_data=f"upload_{folder}")
+    ])
 
-        if fname in data["index"]:
-            data["files"] = [f for f in data["files"] if f["file_name"] != fname]
-            del data["index"][fname]
-            save_data(user_id, data)
+    kb.append([
+        InlineKeyboardButton("🔙 Back", callback_data="cancel")
+    ])
 
-            await q.message.reply_text("🗑 Deleted")
+    return f"📁 {folder}", InlineKeyboardMarkup(kb)
 
-    # ℹ️ info
-    elif q.data.startswith("info_"):
-        fname = q.data.split("_", 1)[1]
+# =========================
+# 🔘 BUTTONS
+# =========================
+async def buttons(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    q = update.callback_query
+    await q.answer()
 
-        if fname in data["index"]:
-            f = data["index"][fname]
+    uid = q.from_user.id
+    data = get_data(uid)
 
-            kb = [
-                [InlineKeyboardButton("🔗 Share", callback_data=f"share_{fname}")],
-                [InlineKeyboardButton("❌ Close", callback_data="files_0")]
-            ]
+    # BACK
+    if q.data == "cancel":
+        user_state.pop(uid, None)
+        return await q.message.edit_text(main_menu(uid), reply_markup=main_kb())
 
-            text = (
-                f"📄 File Info\n\n"
-                f"Name: {f['original_name']}\n"
-                f"Saved As: {f['file_name']}\n"
-                f"Size: {f['file_size']} bytes\n"
-                f"Uploaded: {f['upload_date']}\n"
-                f"Shared With: {len(f.get('shared_with', []))} users"
-            )
+    # OPEN FOLDER
+    if q.data.startswith("open_"):
+        folder = q.data.split("_",1)[1]
+        text, kb = build_folder(data, folder)
+        return await q.message.edit_text(text, reply_markup=kb)
 
-            await q.message.reply_text(text, reply_markup=InlineKeyboardMarkup(kb))
+    # NEW FOLDER
+    if q.data.startswith("newfolder_"):
+        parent = q.data.split("_",1)[1]
+        user_state[uid] = {"mode":"new_folder","parent":parent}
+        return await q.message.edit_text("📁 Send folder name")
 
-    # 🔗 share
-    elif q.data.startswith("share_"):
-        fname = q.data.split("_", 1)[1]
+    # RENAME FOLDER
+    if q.data.startswith("renamefolder_"):
+        folder = q.data.split("_",1)[1]
+        user_state[uid] = {"mode":"rename_folder","folder":folder}
+        return await q.message.edit_text("✏️ Send new folder name")
 
-        if fname in data["index"]:
-            f = data["index"][fname]
-            link = f"https://t.me/{context.bot.username}?start=share_{fname}_{user_id}"
+    # DELETE FOLDER
+    if q.data.startswith("delfolder_"):
+        folder = q.data.split("_",1)[1]
 
-            await q.message.reply_text(f"🔗 Share Link:\n{link}")
+        code = str(random.randint(100000,999999))
+        user_state[uid] = {"mode":"delete_folder","folder":folder,"code":code}
 
+        return await q.message.edit_text(f"Type code to delete:\n{code}")
 
 # =========================
-# 📤 FILE UPLOAD
+# 📤 UPLOAD
 # =========================
-
 async def handle_file(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    user_id = update.effective_user.id
+    uid = update.effective_user.id
 
-    if user_id not in waiting_for_file:
+    if uid not in user_state:
         return
 
-    doc = update.message.document
-    if not doc:
+    state = user_state[uid]
+    if state["mode"] != "upload":
         return
 
-    data = get_data(user_id)
+    folder = state.get("folder","root")
+    data = get_data(uid)
+
+    doc = update.message.document
+    used = sum(f["file_size"] for f in data["files"])
 
-    original = doc.file_name
-    ext = original.split('.')[-1] if '.' in original else "dat"
+    if not is_admin(uid) and used + doc.file_size > MAX_STORAGE:
+        return await update.message.reply_text("❌ Storage Full")
 
-    ts = int(time.time())
-    new_name = f"{user_id}_{ts}.{ext}"
+    sent = await context.bot.send_document(GROUP_ID, doc.file_id)
 
-    sent = await context.bot.send_document(
-        chat_id=GROUP_ID,
-        document=doc.file_id,
-        filename=new_name
-    )
+    name = f"{uid}_{int(time.time())}"
 
     entry = {
-        "file_name": new_name,
-        "original_name": original,
+        "file_name": name,
+        "original_name": doc.file_name,
         "file_id": sent.document.file_id,
         "file_size": doc.file_size,
-        "upload_date": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
-        "shared_with": []
+        "upload_date": str(datetime.now())
     }
 
     data["files"].append(entry)
-    data["index"][new_name] = entry
-
-    save_data(user_id, data)
+    data["index"][name] = entry
+    data["folders"][folder]["files"].append(name)
 
-    waiting_for_file.discard(user_id)
-    await update.message.reply_text("✅ Uploaded")
+    save_data(uid, data)
 
+    user_state.pop(uid)
+    await update.message.reply_text("✅ Uploaded", reply_markup=main_kb())
 
 # =========================
-# 🔍 SEARCH
+# 📝 TEXT HANDLER
 # =========================
+async def handle_text(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    uid = update.effective_user.id
 
-async def search(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    user_id = update.effective_user.id
-    data = get_data(user_id)
-
-    if not context.args:
-        return await update.message.reply_text("Usage: /search name")
-
-    query = context.args[0].lower()
-
-    results = [
-        f for f in data["files"]
-        if query in f["file_name"].lower() or query in f["original_name"].lower()
-    ]
+    if uid not in user_state:
+        return
 
-    if not results:
-        return await update.message.reply_text("❌ No results")
+    state = user_state[uid]
+    data = get_data(uid)
 
-    msg = "🔍 Results:\n\n"
-    for f in results[:10]:
-        msg += f"• {f['file_name']}\n"
+    # CREATE FOLDER
+    if state["mode"] == "new_folder":
+        name = update.message.text
+        parent = state["parent"]
 
-    await update.message.reply_text(msg)
+        data["folders"][name] = {"name":name,"files":[],"folders":[]}
+        data["folders"][parent]["folders"].append(name)
 
+        save_data(uid, data)
+        user_state.pop(uid)
 
-# =========================
-# 📥 GET
-# =========================
+        return await update.message.reply_text("✅ Folder Created", reply_markup=main_kb())
 
-async def get_file(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    user_id = update.effective_user.id
-    data = get_data(user_id)
+    # RENAME FOLDER
+    if state["mode"] == "rename_folder":
+        new = update.message.text
+        old = state["folder"]
 
-    if not context.args:
-        return
+        data["folders"][new] = data["folders"].pop(old)
 
-    fname = context.args[0]
+        save_data(uid, data)
+        user_state.pop(uid)
 
-    if fname in data["index"]:
-        await context.bot.send_document(
-            chat_id=update.effective_chat.id,
-            document=data["index"][fname]["file_id"]
-        )
-    else:
-        await update.message.reply_text("❌ Not found")
+        return await update.message.reply_text("✅ Renamed", reply_markup=main_kb())
 
+    # DELETE FOLDER CONFIRM
+    if state["mode"] == "delete_folder":
+        if update.message.text == state["code"]:
+            folder = state["folder"]
 
-# =========================
-# 🚀 RUN
-# =========================
-
-app = ApplicationBuilder().token(BOT_TOKEN).build()
+            if data["folders"][folder]["files"] or data["folders"][folder]["folders"]:
+                return await update.message.reply_text("❌ Folder not empty")
 
-app.add_handler(CommandHandler("start", start))
-app.add_handler(CommandHandler("search", search))
-app.add_handler(CommandHandler("get", get_file))
+            del data["folders"][folder]
 
-app.add_handler(CallbackQueryHandler(buttons))
-app.add_handler(MessageHandler(filters.Document.ALL, handle_file))
+            save_data(uid, data)
+            user_state.pop(uid)
 
-print("🔥 ADVANCED BOT RUNNING...")
-app.run_polling()
+            return await update.message.reply_text("🗑 Deleted", reply_markup=main_kb())