Update app.py

app.py

@@ -1,12 +1,270 @@
+import os
+BOT_TOKEN = os.getenv("tgtkn")
+GROUP_ID = -1003799294466
+
+import time
+from datetime import datetime
+from telegram import *
+from telegram.ext import *
 from mdb import *
 
-set("test", "string", "hello")
-print(get("test"))
+waiting_for_file = set()
+PAGE_SIZE = 5
+
+# =========================
+# 🧠 DATABASE
+# =========================
+
+def get_data(user_id):
+    key = f"user_{user_id}"
+    data = get(key, "json")
+    if not data:
+        return {"files": [], "index": {}}
+    return data
+
+
+def save_data(user_id, data):
+    key = f"user_{user_id}"
+    set(key, "json", data)
+
+
+# =========================
+# 🚀 START
+# =========================
+
+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)
+    )
+
+
+# =========================
+# 📂 PAGINATION
+# =========================
+
+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"
+
+        kb.append([
+            InlineKeyboardButton("ℹ️", callback_data=f"info_{fname}"),
+            InlineKeyboardButton("🗑", callback_data=f"del_{fname}")
+        ])
+
+    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}"))
+
+    if nav:
+        kb.append(nav)
+
+    return text, InlineKeyboardMarkup(kb)
+
+
+# =========================
+# 🔘 BUTTONS
+# =========================
+
+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)
+
+    # 📤 upload
+    if q.data == "upload":
+        waiting_for_file.add(user_id)
+        await q.message.reply_text("📁 Send file")
+
+    # 📂 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)
+
+    # 📊 stats
+    elif q.data == "stats":
+        await q.message.reply_text(f"📊 Files: {len(data['files'])}")
+
+    # 🗑 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))
+
+    elif q.data.startswith("confirmdel_"):
+        fname = q.data.split("_", 1)[1]
+
+        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)
+
+            await q.message.reply_text("🗑 Deleted")
+
+    # ℹ️ info
+    elif q.data.startswith("info_"):
+        fname = q.data.split("_", 1)[1]
+
+        if fname in data["index"]:
+            f = data["index"][fname]
+
+            kb = [
+                [InlineKeyboardButton("🔗 Share", callback_data=f"share_{fname}")],
+                [InlineKeyboardButton("❌ Close", callback_data="files_0")]
+            ]
+
+            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"
+            )
+
+            await q.message.reply_text(text, reply_markup=InlineKeyboardMarkup(kb))
+
+    # 🔗 share
+    elif q.data.startswith("share_"):
+        fname = q.data.split("_", 1)[1]
+
+        if fname in data["index"]:
+            f = data["index"][fname]
+            link = f"https://t.me/{context.bot.username}?start=share_{fname}_{user_id}"
+
+            await q.message.reply_text(f"🔗 Share Link:\n{link}")
+
+
+# =========================
+# 📤 FILE UPLOAD
+# =========================
+
+async def handle_file(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    user_id = update.effective_user.id
+
+    if user_id not in waiting_for_file:
+        return
+
+    doc = update.message.document
+    if not doc:
+        return
+
+    data = get_data(user_id)
+
+    original = doc.file_name
+    ext = original.split('.')[-1] if '.' in original else "dat"
+
+    ts = int(time.time())
+    new_name = f"{user_id}_{ts}.{ext}"
+
+    sent = await context.bot.send_document(
+        chat_id=GROUP_ID,
+        document=doc.file_id,
+        filename=new_name
+    )
+
+    entry = {
+        "file_name": new_name,
+        "original_name": original,
+        "file_id": sent.document.file_id,
+        "file_size": doc.file_size,
+        "upload_date": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
+        "shared_with": []
+    }
+
+    data["files"].append(entry)
+    data["index"][new_name] = entry
+
+    save_data(user_id, data)
+
+    waiting_for_file.discard(user_id)
+    await update.message.reply_text("✅ Uploaded")
+
+
+# =========================
+# 🔍 SEARCH
+# =========================
+
+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 not results:
+        return await update.message.reply_text("❌ No results")
+
+    msg = "🔍 Results:\n\n"
+    for f in results[:10]:
+        msg += f"• {f['file_name']}\n"
+
+    await update.message.reply_text(msg)
+
+
+# =========================
+# 📥 GET
+# =========================
+
+async def get_file(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    user_id = update.effective_user.id
+    data = get_data(user_id)
+
+    if not context.args:
+        return
+
+    fname = context.args[0]
+
+    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")
+
+
+# =========================
+# 🚀 RUN
+# =========================
+
+app = ApplicationBuilder().token(BOT_TOKEN).build()
 
-set("test", "json", {"a": 1})
-print(get("test", "json"))
+app.add_handler(CommandHandler("start", start))
+app.add_handler(CommandHandler("search", search))
+app.add_handler(CommandHandler("get", get_file))
 
-update("test", "string", "updated")
-print(get("test"))
+app.add_handler(CallbackQueryHandler(buttons))
+app.add_handler(MessageHandler(filters.Document.ALL, handle_file))
 
-#delete("test")
+print("🔥 ADVANCED BOT RUNNING...")
+app.run_polling()