Update main.py

main.py

@@ -1,60 +1,100 @@
 import os
-from fastapi import FastAPI
+from datetime import datetime
+from fastapi import FastAPI, Request
 from fastapi.middleware.cors import CORSMiddleware
-from pymongo import MongoClient
+from pymongo import MongoClient, DESCENDING
 
 app = FastAPI()
 app.add_middleware(CORSMiddleware, allow_origins=["*"])
 
 client = MongoClient(os.environ.get("MONGODB_URI"))
 db = client['myanmarswe_cinema']
-metas_col, streams_col = db['metas'], db['streams']
+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"
+    return True, "OK"
 
 @app.get("/{user_key}/manifest.json")
 def manifest(user_key: str):
+    valid, _ = is_user_valid(user_key)
     return {
-        "id": "com.myanmarswe.pro",
-        "version": "2.6.0",
-        "name": "MSwe Cinema 💎",
-        "resources": [
-            {"name": "catalog", "types": ["movie", "series"], "idPrefixes": ["mswe_"]},
-            {"name": "meta", "types": ["movie", "series"], "idPrefixes": ["tmdb:", "tmdb_col:"]},
-            {"name": "stream", "types": ["movie", "series"], "idPrefixes": ["tmdb:", "tmdb_col:"]}
-        ],
+        "id": "com.myanmarswe.pro", "version": "2.6.0",
+        "name": f"MSwe Cinema {'💎' if valid else '⚠️'}",
+        "resources": ["catalog", "meta", "stream"],
         "types": ["movie", "series"],
+        "idPrefixes": ["tmdb:", "tmdb_col:"],
         "catalogs": [
-            {"type": "movie", "id": "mswe_m", "name": "Latest Movies", "extra": [{"name": "search", "isRequired": False}]},
-            {"type": "series", "id": "mswe_s", "name": "Latest TV Shows"},
+            {
+                "type": "movie", "id": "mswe_m", "name": "Latest Movies",
+                "extra": [{"name": "search", "isRequired": False}] # Search Support
+            },
+            {
+                "type": "series", "id": "mswe_s", "name": "Latest TV Shows",
+                "extra": [{"name": "search", "isRequired": False}]
+            },
             {"type": "series", "id": "mswe_col", "name": "Collections"}
         ]
     }
 
+@app.get("/{user_key}/catalog/{m_type}/{cat_id}/{extra}.json")
+@app.get("/{user_key}/catalog/{m_type}/{cat_id}.json")
+def catalog(user_key: str, m_type: str, cat_id: str, extra: str = None):
+    if not is_user_valid(user_key)[0]: return {"metas": []}
+    
+    query = {}
+    # Search Logic
+    if extra:
+        try:
+            from urllib.parse import parse_qs
+            params = parse_qs(extra)
+            if "search" in params:
+                query["name"] = {"$regex": params["search"][0], "$options": "i"}
+        except: pass
+
+    if cat_id == "mswe_col":
+        query["id"] = {"$regex": "^tmdb_col:"}
+    else:
+        query["type"] = m_type
+        if "id" not in query: query["id"] = {"$regex": "^tmdb:"}
+
+    cursor = metas_col.find(query).sort("updated_at", DESCENDING).limit(100)
+    return {"metas": [{"id": m["id"], "type": m.get("type", m_type), "name": m.get("name"), "poster": m.get("poster")} for m in cursor]}
+
 @app.get("/{user_key}/meta/{m_type}/{meta_id}.json")
-def get_meta(user_key: str, m_type: str, meta_id: str):
+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": {}}
-    m.pop("_id")
-    # Cast မပေါ်တဲ့ ပြဿနာအတွက် Response မှာ Cast ပါတာ သေချာအောင်လုပ်ထားပါတယ်
+    
+    m.pop("_id", None)
+    # Stremio UI က Cast ကို Array(String) ပဲလက်ခံတာ သေချာအောင်လုပ်ခြင်း
+    if "cast" in m:
+        m["cast"] = [str(name) for name in m["cast"] if name]
+        
     return {"meta": m}
 
 @app.get("/{user_key}/stream/{m_type}/{id}.json")
-def get_stream(user_key: str, m_type: str, id: str):
+def stream(user_key: str, m_type: str, id: str):
+    if not is_user_valid(user_key)[0]: return {"streams": []}
+    
     # Collection ထဲက Movie ကို မူရင်း Stream နဲ့ ချိတ်ပေးခြင်း
     if id.startswith("tmdb_col:"):
         parts = id.split(":")
         col_meta = metas_col.find_one({"id": f"tmdb_col:{parts[1]}"})
-        if col_meta:
-            for v in col_meta.get("videos", []):
-                if v["season"] == int(parts[2]) and v["episode"] == int(parts[3]):
+        if col_meta and "videos" in col_meta:
+            for v in col_meta["videos"]:
+                if v.get("season") == int(parts[2]) and v.get("episode") == int(parts[3]):
                     real_id = v.get("movie_id")
-                    streams = list(streams_col.find({"tmdb_id": real_id}))
-                    return {"streams": [{"name": "Server 1 🚀", "url": s["url"]} for s in streams]}
-    
-    # Normal Stream
+                    streams = list(streams_col.find({"tmdb_id": str(real_id)}))
+                    return {"streams": [{"name": "MyanmarSwe 💎", "url": s["url"]} for s in streams]}
+
     parts = id.split(":")
-    query = {"tmdb_id": parts[1]}
+    tid = parts[1]
+    query = {"tmdb_id": tid}
     if len(parts) == 4:
         query["season"], query["episode"] = int(parts[2]), int(parts[3])
     
     streams = list(streams_col.find(query))
-    return {"streams": [{"name": "Server 1 🚀", "url": s["url"]} for s in streams]}
+    return {"streams": [{"name": "MyanmarSwe 💎", "url": s["url"]} for s in streams]}