Update app.py

app.py

@@ -1,7 +1,9 @@
 import os
 import threading
 import time
+import hashlib
 import random
+import string
 from datetime import datetime
 
 from flask_srvr import run_flask
@@ -25,7 +27,9 @@ start_flask()
 # =========================
 BOT_TOKEN = os.getenv("tgtkn")
 GROUP_ID = -1003799294466
+PAGE_SIZE = 5
 MAX_STORAGE = 15 * 1024 * 1024 * 1024
+HARDCODE_ADMIN = 83939584939485
 
 # =========================
 # 🧠 STATE
@@ -35,32 +39,53 @@ user_state = {}
 # =========================
 # 🗄 DB
 # =========================
-def get_data(uid):
-    data = mdb.get(f"user_{uid}", "json")
-    if not data:
-        return {
-            "files": [],
-            "index": {},
-            "folders": {
-                "root": {
-                    "files": [],
-                    "folders": []
-                }
-            }
+def default_data():
+    return {
+        "files": [],
+        "index": {},
+        "folders": {
+            "root": {"files": [], "folders": [], "parent": None}
         }
-    return data
+    }
+
+def get_data(uid):
+    return mdb.get(f"user_{uid}", "json") or default_data()
 
 def save_data(uid, data):
     mdb.set(f"user_{uid}", "json", data)
 
+def is_admin(uid):
+    admins = mdb.get("admins", "json") or []
+    return uid == HARDCODE_ADMIN or uid in admins
+
 # =========================
 # 📊 STORAGE BAR
 # =========================
-def storage_bar(used, total):
+def storage_bar(used):
+    total = MAX_STORAGE
     ratio = used / total if total else 0
     filled = int(ratio * 10)
     return "█"*filled + "░"*(10-filled)
 
+# =========================
+# 🧾 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 "📦"
+
+# =========================
+# 🔐 PASSWORD
+# =========================
+def gen_pass():
+    return ''.join(random.choices(string.ascii_letters+string.digits,k=16))
+
+def hash_pass(p):
+    return hashlib.sha256(p.encode()).hexdigest()
+
 # =========================
 # 🏠 MENU
 # =========================
@@ -70,25 +95,20 @@ def main_menu(uid):
 
     return f"""☁️ Cloud Bot
 
-{storage_bar(used, MAX_STORAGE)}
+📊 Storage:
+{storage_bar(used)}
 {round(used/1024/1024,2)} MB / 15360 MB
 """
 
 def main_kb():
     return InlineKeyboardMarkup([
         [InlineKeyboardButton("📤 Upload", callback_data="upload_root")],
-        [InlineKeyboardButton("📂 Files", callback_data="open_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())
-
-# =========================
-# 📁 FOLDER UI
+# 📁 FOLDER VIEW
 # =========================
 def build_folder(data, folder):
     f = data["folders"][folder]
@@ -101,13 +121,27 @@ def build_folder(data, folder):
     for fid in f["files"]:
         file = data["index"][fid]
         kb.append([InlineKeyboardButton(
-            file["original_name"],
+            f"{get_icon(file['original_name'])} {file['original_name']}",
             callback_data=f"file_{fid}"
         )])
 
-    kb.append([InlineKeyboardButton("➕ Folder", callback_data=f"newfolder_{folder}")])
-    kb.append([InlineKeyboardButton("📤 Upload", callback_data=f"upload_{folder}")])
-    kb.append([InlineKeyboardButton("🔙 Back", callback_data="cancel")])
+    kb.append([
+        InlineKeyboardButton("➕ Folder", callback_data=f"newfolder_{folder}"),
+        InlineKeyboardButton("✏️ Rename", callback_data=f"renamefolder_{folder}")
+    ])
+
+    if folder != "root":
+        kb.append([
+            InlineKeyboardButton("🗑 Delete", callback_data=f"delfolder_{folder}")
+        ])
+
+    kb.append([
+        InlineKeyboardButton("📤 Upload Here", callback_data=f"upload_{folder}")
+    ])
+
+    kb.append([
+        InlineKeyboardButton("🔙 Back", callback_data=f"open_{f['parent'] or 'root'}")
+    ])
 
     return f"📁 {folder}", InlineKeyboardMarkup(kb)
 
@@ -122,12 +156,12 @@ async def buttons(update: Update, context: ContextTypes.DEFAULT_TYPE):
         uid = q.from_user.id
         data = get_data(uid)
 
-        # BACK
-        if q.data == "cancel":
-            user_state.pop(uid, None)
-            return await q.message.edit_text(main_menu(uid), reply_markup=main_kb())
+        # SEARCH MODE
+        if q.data == "search":
+            user_state[uid] = {"mode":"search"}
+            return await q.message.edit_text("🔍 Send file name")
 
-        # OPEN FOLDER
+        # OPEN
         if q.data.startswith("open_"):
             folder = q.data.split("_",1)[1]
             text, kb = build_folder(data, folder)
@@ -136,40 +170,95 @@ async def buttons(update: Update, context: ContextTypes.DEFAULT_TYPE):
         # UPLOAD
         if q.data.startswith("upload_"):
             folder = q.data.split("_",1)[1]
-
             user_state[uid] = {"mode":"upload","folder":folder}
             return await q.message.edit_text(f"📤 Upload to {folder}")
 
+        # FILE
+        if q.data.startswith("file_"):
+            fid = q.data.split("_",1)[1]
+            f = data["index"][fid]
+
+            kb = [
+                [InlineKeyboardButton("👁 Preview", callback_data=f"preview_{fid}")],
+                [InlineKeyboardButton("📥 Download", callback_data=f"download_{fid}")],
+                [InlineKeyboardButton("🔐 Share", callback_data=f"share_{fid}")],
+                [InlineKeyboardButton("🔙 Back", callback_data="open_root")]
+            ]
+
+            return await q.message.edit_text(f"📄 {f['original_name']}", reply_markup=kb)
+
+        # PREVIEW
+        if q.data.startswith("preview_"):
+            fid = q.data.split("_",1)[1]
+            f = data["index"][fid]
+            ext = f["original_name"].lower().split('.')[-1]
+
+            if ext in ["jpg","png","jpeg"]:
+                await context.bot.send_photo(uid, f["file_id"])
+            elif ext in ["mp4","mkv"]:
+                await context.bot.send_video(uid, f["file_id"])
+            else:
+                await context.bot.send_document(uid, f["file_id"])
+            return
+
+        # DOWNLOAD
+        if q.data.startswith("download_"):
+            fid = q.data.split("_",1)[1]
+            await context.bot.send_document(uid, data["index"][fid]["file_id"])
+            return
+
+        # SHARE
+        if q.data.startswith("share_"):
+            fid = q.data.split("_",1)[1]
+
+            pwd = gen_pass()
+            data["index"][fid]["share"] = hash_pass(pwd)
+            save_data(uid, data)
+
+            link = f"https://t.me/{context.bot.username}?start=share_{fid}_{uid}"
+
+            return await q.message.edit_text(f"🔗 {link}\n\n🔐 Password:\n{pwd}")
+
         # 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")
 
+        # 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 name")
+
+        # DELETE FOLDER
+        if q.data.startswith("delfolder_"):
+            folder = q.data.split("_",1)[1]
+            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:\n{code}")
+
     except Exception as e:
-        print("ERROR:", e)
+        print("BTN ERROR:", e)
 
 # =========================
-# 📤 FILE UPLOAD
+# 📤 UPLOAD
 # =========================
 async def handle_file(update: Update, context: ContextTypes.DEFAULT_TYPE):
     try:
         uid = update.effective_user.id
 
-        if uid not in user_state:
-            return
-
-        state = user_state[uid]
-        if state["mode"] != "upload":
+        if uid not in user_state or user_state[uid]["mode"] != "upload":
             return
 
-        folder = state["folder"]
+        folder = user_state[uid]["folder"]
         data = get_data(uid)
 
         doc = update.message.document
-        if not doc:
-            return
+        used = sum(f["file_size"] for f in data["files"])
+
+        if not is_admin(uid) and used + doc.file_size > MAX_STORAGE:
+            return await update.message.reply_text("❌ Storage full")
 
         sent = await context.bot.send_document(GROUP_ID, doc.file_id)
 
@@ -188,7 +277,6 @@ async def handle_file(update: Update, context: ContextTypes.DEFAULT_TYPE):
         data["folders"][folder]["files"].append(name)
 
         save_data(uid, data)
-
         user_state.pop(uid)
 
         await update.message.reply_text("✅ Uploaded", reply_markup=main_kb())
@@ -197,7 +285,7 @@ async def handle_file(update: Update, context: ContextTypes.DEFAULT_TYPE):
         print("UPLOAD ERROR:", e)
 
 # =========================
-# 📝 TEXT
+# 📝 TEXT HANDLER
 # =========================
 async def handle_text(update: Update, context: ContextTypes.DEFAULT_TYPE):
     try:
@@ -209,17 +297,65 @@ async def handle_text(update: Update, context: ContextTypes.DEFAULT_TYPE):
         state = user_state[uid]
         data = get_data(uid)
 
+        # SEARCH
+        if state["mode"] == "search":
+            q = update.message.text.lower()
+            res = [f for f in data["files"] if q in f["original_name"].lower()]
+
+            kb = []
+            for f in res[:5]:
+                kb.append([InlineKeyboardButton(
+                    f["original_name"],
+                    callback_data=f"file_{f['file_name']}"
+                )])
+
+            return await update.message.reply_text("🔍 Results", reply_markup=InlineKeyboardMarkup(kb))
+
+        # CREATE FOLDER
         if state["mode"] == "new_folder":
             name = update.message.text
             parent = state["parent"]
 
-            data["folders"][name] = {"files": [], "folders": []}
+            data["folders"][name] = {"files":[],"folders":[],"parent":parent}
             data["folders"][parent]["folders"].append(name)
 
             save_data(uid, data)
+            user_state.pop(uid)
+
+            return await update.message.reply_text("✅ Folder Created", reply_markup=main_kb())
+
+        # RENAME FOLDER
+        if state["mode"] == "rename_folder":
+            new = update.message.text
+            old = state["folder"]
+
+            data["folders"][new] = data["folders"].pop(old)
+
+            parent = data["folders"][new]["parent"]
+            data["folders"][parent]["folders"].remove(old)
+            data["folders"][parent]["folders"].append(new)
 
+            save_data(uid, data)
             user_state.pop(uid)
-            await update.message.reply_text("✅ Folder created", reply_markup=main_kb())
+
+            return await update.message.reply_text("✅ Renamed", reply_markup=main_kb())
+
+        # DELETE FOLDER
+        if state["mode"] == "delete_folder":
+            if update.message.text == state["code"]:
+                folder = state["folder"]
+
+                if data["folders"][folder]["files"] or data["folders"][folder]["folders"]:
+                    return await update.message.reply_text("❌ Not empty")
+
+                parent = data["folders"][folder]["parent"]
+                data["folders"][parent]["folders"].remove(folder)
+                del data["folders"][folder]
+
+                save_data(uid, data)
+                user_state.pop(uid)
+
+                return await update.message.reply_text("🗑 Deleted", reply_markup=main_kb())
 
     except Exception as e:
         print("TEXT ERROR:", e)
@@ -234,5 +370,5 @@ app.add_handler(CallbackQueryHandler(buttons))
 app.add_handler(MessageHandler(filters.Document.ALL, handle_file))
 app.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, handle_text))
 
-print("🔥 BOT RUNNING")
+print("🔥 FULL BOT RUNNING")
 app.run_polling()