Update main.py

main.py

@@ -1,33 +1,93 @@
-# main.py ထဲက catalog နဲ့ meta function များကို ပြင်ရန်
+import os
+from datetime import datetime
+from fastapi import FastAPI, Request
+from fastapi.middleware.cors import CORSMiddleware
+from pymongo import MongoClient, DESCENDING
+from sync import process_sync
+
+app = FastAPI()
+app.add_middleware(CORSMiddleware, allow_origins=["*"])
+
+client = MongoClient(os.environ.get("MONGODB_URI"))
+db = client['myanmarswe_cinema']
+metas_col, streams_col, users_col = db['metas'], db['streams'], db['users']
+
+def is_user_valid(key):
+    user = users_col.find_one({"key": key})
+    if not user or user.get("status") != "active": return False, "Invalid User"
+    expiry_str = user.get("expiry")
+    if expiry_str:
+        try:
+            exp_date = datetime.strptime(expiry_str.split('T')[0], "%Y-%m-%d")
+            if datetime.now() > exp_date: return False, "Expired"
+        except: pass
+    return True, "OK"
+
+@app.get("/{user_key}/manifest.json")
+def manifest(user_key: str):
+    valid, msg = is_user_valid(user_key)
+    return {
+        "id": "com.myanmarswe.pro", "version": "2.5.0",
+        "name": f"MSwe Cinema {'⚠️' if not valid else '💎'}",
+        "resources": ["catalog", "meta", "stream"],
+        "types": ["movie", "series"],
+        "idPrefixes": ["tmdb:"],
+        "catalogs": [
+            {"type": "movie", "id": "mswe_m", "name": "Latest Movies"},
+            {"type": "series", "id": "mswe_s", "name": "Latest TV Shows"},
+            {"type": "series", "id": "mswe_col", "name": "Collections"}
+        ]
+    }
 
 @app.get("/{user_key}/catalog/{m_type}/{cat_id}.json")
 def catalog(user_key: str, m_type: str, cat_id: str):
     if not is_user_valid(user_key)[0]: return {"metas": []}
     
     if cat_id == "mswe_col":
-        # Collection အစစ်အမှန်များကိုသာ ပြရန်
-        cursor = metas_col.find({"id": {"$regex": "^tmdb_col:"}}).sort("updated_at", -1)
+        # Collection IDs သီးသန့်ထုတ်ပေးခြင်း
+        cursor = metas_col.find({"id": {"$regex": "^tmdb_col:"}}).sort("updated_at", DESCENDING)
         return {"metas": [{"id": m["id"], "type": "series", "name": m["name"], "poster": m["poster"]} for m in cursor]}
 
     query = {"type": m_type, "id": {"$regex": "^tmdb:"}}
-    cursor = metas_col.find(query).sort("updated_at", -1).limit(100)
+    cursor = metas_col.find(query).sort("updated_at", DESCENDING).limit(100)
     return {"metas": [{"id": m["id"], "type": m["type"], "name": m["name"], "poster": m["poster"]} for m in cursor]}
 
 @app.get("/{user_key}/meta/{m_type}/{meta_id}.json")
 def meta(user_key: str, m_type: str, meta_id: str):
     if not is_user_valid(user_key)[0]: return {"meta": {}}
-    
     m = metas_col.find_one({"id": meta_id})
     if not m: return {"meta": {}}
+    
+    # Recommendation Logic
+    related = []
+    if m.get("recommendations"):
+        rec_cursor = metas_col.find({"id": {"$in": [f"tmdb:{rid}" for rid in m["recommendations"]]}})
+        related = [{"id": r["id"], "type": r["type"], "name": r["name"], "poster": r["poster"]} for r in rec_cursor]
 
-    # UI formatting
-    res = {
-        "id": m["id"], "type": m["type"] if not meta_id.startswith("tmdb_col:") else "series",
-        "name": m["name"], "poster": m["poster"], "background": m["background"],
-        "logo": m.get("logo"), "description": m["description"],
-        "releaseInfo": m.get("releaseInfo"), "imdbRating": m.get("imdbRating"),
-        "runtime": m.get("runtime"), "genres": m.get("genres", []),
-        "cast": m.get("cast", []), "director": m.get("director", []),
-        "videos": m.get("videos", []), "trailers": m.get("trailers", [])
-    }
-    return {"meta": res}
+    # UI Optimization
+    m.pop("_id", None)
+    m["related"] = related
+    # Collection ဆိုလျှင် type ကို series အဖြစ်ပြမှ videos ပေါ်မည်
+    if meta_id.startswith("tmdb_col:"): m["type"] = "series"
+    
+    return {"meta": m}
+
+@app.get("/{user_key}/stream/{m_type}/{id}.json")
+def stream(user_key: str, m_type: str, id: str):
+    if not is_user_valid(user_key)[0]: return {"streams": []}
+    parts = id.split(":")
+    # id format - tmdb:123 or tmdb:123:1:1
+    tid = parts[1]
+    query = {"tmdb_id": tid}
+    if len(parts) == 4: # Season/Episode ပါလျှင်
+        query["season"], query["episode"] = int(parts[2]), int(parts[3])
+    
+    streams = list(streams_col.find(query))
+    return {"streams": [{"name": "MyanmarSwe 💎", "title": "🚀 High Speed Server", "url": s["url"]} for s in streams]}
+
+@app.post("/sync-data")
+async def sync_route(req: Request):
+    return process_sync(await req.json())
+
+@app.get("/ping")
+def ping(): return {"status": "awake"}