Update main.py

main.py

@@ -3,6 +3,7 @@ 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 # sync.py ထဲက function ကို ခေါ်သုံးမယ်
 
 app = FastAPI()
 app.add_middleware(CORSMiddleware, allow_origins=["*"])
@@ -13,15 +14,20 @@ 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
-    return True
+    return user and user.get("status") == "active"
+
+# --- Google Sheet ကနေ Data လက်ခံမည့် Endpoint ---
+@app.post("/sync-data")
+async def sync_route(req: Request):
+    data = await req.json()
+    return process_sync(data)
 
 @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.5",
+        "version": "2.7.0",
         "name": f"MSwe Cinema {'💎' if valid else '⚠️'}",
         "resources": ["catalog", "meta", "stream"],
         "types": ["movie", "series"],
@@ -33,18 +39,14 @@ def manifest(user_key: str):
         ]
     }
 
-# Catalog Route ကို တစ်ခုတည်းနဲ့ အကုန်မိအောင် ပြင်လိုက်ပါတယ်
 @app.get("/{user_key}/catalog/{m_type}/{cat_id}.json")
 @app.get("/{user_key}/catalog/{m_type}/{cat_id}/{extra}.json")
 def catalog(user_key: str, m_type: str, cat_id: str, extra: str = None):
     if not is_user_valid(user_key): return {"metas": []}
-    
     query = {}
-    # Search Logic: Stremio က search=query ဆိုပြီး ပို့လာရင် ဖတ်ဖို့
     if extra and "search=" in extra:
-        search_query = extra.replace("search=", "")
-        query["name"] = {"$regex": search_query, "$options": "i"}
-
+        query["name"] = {"$regex": extra.replace("search=", ""), "$options": "i"}
+    
     if cat_id == "mswe_col":
         query["id"] = {"$regex": "^tmdb_col:"}
     else:
@@ -52,56 +54,33 @@ def catalog(user_key: str, m_type: str, cat_id: str, extra: str = None):
         if "id" not in query: query["id"] = {"$regex": "^tmdb:"}
 
     cursor = metas_col.find(query).sort("updated_at", DESCENDING).limit(100)
-    metas = []
-    for m in cursor:
-        metas.append({
-            "id": m["id"],
-            "type": m.get("type", m_type),
-            "name": m.get("name"),
-            "poster": m.get("poster")
-        })
-    return {"metas": metas}
+    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 meta(user_key: str, m_type: str, meta_id: str):
     if not is_user_valid(user_key): return {"meta": {}}
     m = metas_col.find_one({"id": meta_id})
     if not m: return {"meta": {}}
-    
     m.pop("_id", None)
-    # Cast အားလုံးကို String ဖြစ်အောင် သေချာပြောင်းပေးခြင်း (Detail Screen မှာ Cast ပေါ်စေရန်)
-    if "cast" in m and isinstance(m["cast"], list):
-        m["cast"] = [str(person) for person in m["cast"] if person]
-    
-    # TV Show ဖြစ်ရင် Episode တစ်ခုချင်းစီရဲ့ Thumbnail (Poster) ပါဝင်မှု စစ်ဆေးခြင်း
-    if "videos" in m:
-        for v in m["videos"]:
-            # အကယ်၍ Episode မှာ Thumbnail မပါရင် Series Poster ကိုပဲ သုံးဖို့
-            if not v.get("thumbnail"):
-                v["thumbnail"] = m.get("poster")
-
+    if "cast" in m: m["cast"] = [str(p) for p in m["cast"] if p]
     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): return {"streams": []}
     
-    # Collection IDs: tmdb_col:123:1:1
+    # Collection Stream Link Logic
     if id.startswith("tmdb_col:"):
         parts = id.split(":")
-        col_id, s_num, e_num = parts[1], int(parts[2]), int(parts[3])
-        col_meta = metas_col.find_one({"id": f"tmdb_col:{col_id}"})
+        col_meta = metas_col.find_one({"id": f"tmdb_col:{parts[1]}"})
         if col_meta and "videos" in col_meta:
-            target = next((v for v in col_meta["videos"] if v["season"] == s_num and v["episode"] == e_num), None)
+            target = next((v for v in col_meta["videos"] if v["season"] == int(parts[2]) and v["episode"] == int(parts[3])), None)
             if target:
                 streams = list(streams_col.find({"tmdb_id": str(target["movie_id"])}))
-                return {"streams": [{"name": "MyanmarSwe 💎", "title": "🚀 High Speed", "url": s["url"]} for s in streams]}
+                return {"streams": [{"name": "MyanmarSwe 💎", "url": s["url"]} for s in streams]}
 
-    # ပ���ံမှန် Movie/Series
     parts = id.split(":")
     query = {"tmdb_id": parts[1]}
-    if len(parts) == 4:
-        query["season"], query["episode"] = int(parts[2]), int(parts[3])
-    
+    if len(parts) == 4: query["season"], query["episode"] = int(parts[2]), int(parts[3])
     streams = list(streams_col.find(query))
-    return {"streams": [{"name": "MyanmarSwe 💎", "title": "🚀 High Speed", "url": s["url"]} for s in streams]}
+    return {"streams": [{"name": "MyanmarSwe 💎", "url": s["url"]} for s in streams]}