Update main.py

main.py

@@ -1,94 +1,127 @@
-from fastapi import FastAPI, HTTPException
+import os, json
+from datetime import datetime
+from fastapi import FastAPI, HTTPException, Request
 from fastapi.middleware.cors import CORSMiddleware
-from sheets import is_user_valid, get_data
-from tmdb import get_meta
+from pymongo import MongoClient, DESCENDING
+from sync import process_sync # အပေါ်က sync logic ကို ခေါ်သုံးခြင်း
 
 app = FastAPI()
 app.add_middleware(CORSMiddleware, allow_origins=["*"])
 
-@app.get("/ping")
-@app.get("/ping/")
-def ping():
-    return {"status": "awake"}
+# MongoDB Setup
+MONGO_URI = os.environ.get("MONGODB_URI")
+client = MongoClient(MONGO_URI)
+db = client['myanmarswe_cinema']
+metas_col = db['metas']
+streams_col = db['streams']
+users_col = db['users']
+
+# --- Helper Functions ---
+
+def is_user_valid(user_key: str):
+    user = users_col.find_one({"key": user_key})
+    if not user or user.get("status") != "active":
+        return False, "Invalid or Inactive User"
+    
+    # Expiry Date Check
+    expiry_str = user.get("expiry")
+    if expiry_str:
+        try:
+            expiry_date = datetime.strptime(expiry_str, "%Y-%m-%d")
+            if datetime.now() > expiry_date:
+                return False, "Account Expired"
+        except: pass
+    return True, "OK"
+
+# --- API Endpoints ---
+
+@app.post("/sync-data")
+async def sync_data(request: Request):
+    # Google Sheet မှ လာသော data ကို လက်ခံခြင်း
+    data = await request.json()
+    result = process_sync(data)
+    return result
 
 @app.get("/{user_key}/manifest.json")
 def manifest(user_key: str):
-    if not is_user_valid(user_key): 
-        raise HTTPException(status_code=401)
+    valid, msg = is_user_valid(user_key)
     return {
-        "id": "com.myanmarswe.private", 
-        "version": "1.2.0", # Version မြှင့်လိုက်မှ Details အသစ်တွေပေါ်မှာပါ
-        "name": "MyanmarSwe Cinema",
-        "resources": ["catalog", "meta", "stream"], 
+        "id": "com.myanmarswe.pro",
+        "version": "2.0.0",
+        "name": f"MSwe Cinema {'⚠️' if not valid else '💎'}",
+        "description": msg if not valid else "Ultra-Detail Private Cinema",
+        "resources": ["catalog", "meta", "stream"],
         "types": ["movie", "series"],
         "idPrefixes": ["tmdb:"],
         "catalogs": [
-            {"type": "movie", "id": "mswe_m", "name": "Movies"}, 
-            {"type": "series", "id": "mswe_s", "name": "Series"}
+            {"type": "movie", "id": "mswe_m", "name": "Latest Movies"},
+            {"type": "series", "id": "mswe_s", "name": "Latest Series"}
         ]
     }
 
 @app.get("/{user_key}/catalog/{type}/{id}.json")
 def catalog(user_key: str, type: str):
-    if not is_user_valid(user_key): return {"metas": []}
-    tab = "movies" if type == "movie" else "tv_shows"
-    items = get_data(tab)
+    valid, _ = is_user_valid(user_key)
+    if not valid: return {"metas": []}
     
-    metas = []
-    # 重复 (Duplicate) ဖြစ်နေတဲ့ ID တွေကို ဖယ်ထုတ်ရန် set သုံးပါမည်
-    seen_ids = set()
+    # နောက်ဆုံးထည့်ထားတာကို ရှေ့ဆုံးမှာပြဖို့ Sort လုပ်ခြင်း (DESCENDING)
+    cursor = metas_col.find({"type": type}).sort("updated_at", DESCENDING).limit(100)
     
-    for i in items:
-        tid = str(i['TMDB_ID'])
-        if tid not in seen_ids:
-            # Catalog အတွက် ပုံရုံပဲယူပါမည် (မြန်ဆန်စေရန်)
-            m = get_meta(tid, type == 'movie')
-            metas.append({
-                "id": f"tmdb:{tid}", 
-                "type": type, 
-                "name": i.get("Title") or m.get("name"), 
-                "poster": m.get("poster")
-            })
-            seen_ids.add(tid)
+    metas = []
+    for m in cursor:
+        metas.append({
+            "id": m["id"],
+            "type": m["type"],
+            "name": m["name"],
+            "poster": m["poster"]
+        })
     return {"metas": metas}
 
 @app.get("/{user_key}/meta/{type}/{id}.json")
 def meta(user_key: str, type: str, id: str):
-    if not is_user_valid(user_key): raise HTTPException(status_code=401)
-    # id format: tmdb:12345
-    tmdb_id = id.split(":")[1]
-    return {"meta": get_meta(tmdb_id, type == 'movie')}
+    valid, _ = is_user_valid(user_key)
+    if not valid: raise HTTPException(status_code=401)
+    
+    meta_data = metas_col.find_one({"id": id})
+    if not meta_data: return {"meta": {}}
+    
+    # Recommended Movies Logic
+    # (Metas ထဲမှာရှိနေတဲ့ ကားတွေကိုပဲ Recommendation အဖြစ် ပြန်ပြပေးခြင်း)
+    rec_ids = meta_data.get("recommendations", [])
+    related = []
+    if rec_ids:
+        # DB ထဲမှာ တကယ်ရှိတဲ့ ကားတွေကိုပဲ ရှာယ���မယ်
+        rec_cursor = metas_col.find({"id": {"$in": [f"tmdb:{rid}" for rid in rec_ids]}})
+        for r in rec_cursor:
+            related.append({
+                "id": r["id"],
+                "type": r["type"],
+                "name": r["name"],
+                "poster": r["poster"]
+            })
+    
+    meta_data["related"] = related # Stremio UI တွင် ပြသရန်
+    return {"meta": meta_data}
 
 @app.get("/{user_key}/stream/{type}/{id}.json")
 def stream(user_key: str, type: str, id: str):
-    if not is_user_valid(user_key): return {"streams": []}
+    valid, _ = is_user_valid(user_key)
+    if not valid: return {"streams": []}
     
-    # ID ခွဲထုတ်ခြင်း (Movie: tmdb:123, Series: tmdb:123:1:5)
     parts = id.split(":")
-    tmdb_id = parts[1]
+    tid = parts[1]
     
-    tab = "movies" if type == "movie" else "tv_shows"
-    items = get_data(tab)
+    query = {"tmdb_id": tid}
+    if type == "series" and len(parts) == 4:
+        query["season"] = int(parts[2])
+        query["episode"] = int(parts[3])
     
-    streams = []
-    for i in items:
-        if str(i.get("TMDB_ID")) == tmdb_id:
-            # Movie ဖြစ်လျှင်
-            if type == "movie":
-                streams.append({
-                    "name": "MyanmarSwe 💎",
-                    "title": "🚀 High Speed Server",
-                    "url": i['Stream_URL']
-                })
-            # Series ဖြစ်လျှင် Season နှင့် Episode တိုက်စစ်မည်
-            elif len(parts) == 4:
-                season = parts[2]
-                episode = parts[3]
-                if str(i.get("Season")) == season and str(i.get("Episode")) == episode:
-                    streams.append({
-                        "name": "MyanmarSwe 💎",
-                        "title": f"🚀 S{season} E{episode} - Premium Link",
-                        "url": i['Stream_URL']
-                    })
-                    
-    return {"streams": streams}
+    streams = streams_col.find(query)
+    results = []
+    for s in streams:
+        results.append({
+            "name": "MyanmarSwe 💎",
+            "title": "🚀 High Speed Server",
+            "url": s["url"]
+        })
+    return {"streams": results}