Update app.py

app.py

@@ -2,14 +2,26 @@ import os
 import time
 import subprocess
 import threading
+import json
+import pickle
 import psutil
+from google.oauth2.credentials import Credentials
+from google.auth.transport.requests import Request
+from googleapiclient.discovery import build
+from googleapiclient.errors import HttpError
 
+# ------------------------------------------------------------
+# الإعدادات الأساسية
+# ------------------------------------------------------------
 TIKTOK_USERNAME = "@osei_iraq"
-STREAM_KEY = "p3dq-5cex-bc1k-vet7-bsyx"
+CUSTOM_STREAM_KEY = os.getenv("CUSTOM_STREAM_KEY", "p3dq-5cex-bc1k-vet7-bsyx")  # مفتاح احتياطي
 FFMPEG_LOG = "/tmp/ffmpeg.log"
 MONITOR_LOG = "/tmp/monitor.log"
 LOCK_FILE = "/tmp/app_monitor_core.lock"
+BROADCAST_FILE = "/tmp/current_broadcast.json"
+TOKEN_FILE = "/tmp/yt_token.pkl"
 
+# الكوكيز من المتغيرات البيئية
 TIKTOK_COOKIES = os.getenv("TIKTOK_COOKIES", "")
 TIKTOK_COOKIES_ADULT = os.getenv("TIKTOK_COOKIES_ADULT", "")
 
@@ -20,6 +32,9 @@ def log(msg):
     with open(MONITOR_LOG, "a", encoding="utf-8") as f:
         f.write(line + "\n")
 
+# ------------------------------------------------------------
+# إعداد الكوكيز
+# ------------------------------------------------------------
 def save_cookies():
     if TIKTOK_COOKIES:
         with open("/tmp/tiktok_cookies.txt", "w") as f:
@@ -30,6 +45,9 @@ def save_cookies():
 
 save_cookies()
 
+# ------------------------------------------------------------
+# أدوات مساعدة
+# ------------------------------------------------------------
 def kill_process_tree(pid):
     try:
         parent = psutil.Process(pid)
@@ -76,7 +94,7 @@ def get_tiktok_stream_url():
 def start_stream(stream_key, stream_url):
     if not stream_key or not stream_url:
         return None
-    log("🎬 بدء البث (فيديو نسخ + صوت احترافي مع إصلاح التزامن)...")
+    log("🎬 بدء البث (فيديو نسخ + صوت احترافي)...")
     rtmp_url = f"rtmp://a.rtmp.youtube.com/live2/{stream_key}"
 
     try:
@@ -92,9 +110,7 @@ def start_stream(stream_key, stream_url):
         "-fflags", "+genpts",
         "-flags", "+global_header",
         "-i", stream_url,
-
         "-c:v", "copy",
-
         "-c:a", "aac",
         "-b:a", "192k",
         "-ar", "44100",
@@ -102,11 +118,9 @@ def start_stream(stream_key, stream_url):
         "-cutoff", "18000",
         "-af", "highpass=f=60,lowpass=f=17000,aresample=async=1:first_pts=0",
         "-async", "1",
-
         "-start_at_zero",
         "-copyts",
         "-vsync", "2",
-
         "-bsf:v", "h264_mp4toannexb",
         "-reconnect", "1",
         "-reconnect_at_eof", "1",
@@ -122,6 +136,150 @@ def start_stream(stream_key, stream_url):
     log(f"✅ ffmpeg PID: {proc.pid}")
     return proc
 
+# ------------------------------------------------------------
+# YouTube API دوال (مأخوذة من الكود المرجعي ومعدلة)
+# ------------------------------------------------------------
+def build_youtube_client():
+    """إنشاء عميل YouTube API من التوكن المحفوظ أو من JSON"""
+    try:
+        # محاولة تحميل توكن موجود
+        if os.path.exists(TOKEN_FILE):
+            with open(TOKEN_FILE, "rb") as f:
+                creds = pickle.load(f)
+            if creds and creds.valid:
+                return build('youtube', 'v3', credentials=creds)
+            if creds and creds.expired and creds.refresh_token:
+                creds.refresh(Request())
+                with open(TOKEN_FILE, "wb") as f:
+                    pickle.dump(creds, f)
+                return build('youtube', 'v3', credentials=creds)
+
+        # إنشاء من JSON الموجود في المتغير البيئي
+        creds_json = os.getenv("YT_CREDENTIALS_JSON")
+        if not creds_json:
+            log("❌ YT_CREDENTIALS_JSON غير موجود")
+            return None
+
+        data = json.loads(creds_json)
+        creds = Credentials(
+            token=data.get("token"),
+            refresh_token=data.get("refresh_token"),
+            token_uri=data.get("token_uri"),
+            client_id=data.get("client_id"),
+            client_secret=data.get("client_secret"),
+            scopes=data.get("scopes", ["https://www.googleapis.com/auth/youtube.force-ssl"])
+        )
+        if creds.expired and creds.refresh_token:
+            creds.refresh(Request())
+        with open(TOKEN_FILE, "wb") as f:
+            pickle.dump(creds, f)
+        return build('youtube', 'v3', credentials=creds)
+    except Exception as e:
+        log(f"❌ فشل بناء عميل YouTube: {e}")
+        return None
+
+def create_live_stream(yt):
+    """إنشاء liveStream وإرجاع stream_id و stream_key"""
+    try:
+        stream_insert = yt.liveStreams().insert(
+            part="snippet,cdn,contentDetails",
+            body={
+                "snippet": {"title": "TikTok Stream Key"},
+                "cdn": {
+                    "ingestionType": "rtmp",
+                    "resolution": "1080p",
+                    "frameRate": "30fps"
+                },
+                "contentDetails": {"isReusable": True}
+            }
+        ).execute()
+        stream_id = stream_insert["id"]
+        stream_key = stream_insert["cdn"]["ingestionInfo"]["streamName"]
+        log(f"✅ liveStream جاهز: {stream_id} | key: ...{stream_key[-4:]}")
+        return stream_id, stream_key
+    except Exception as e:
+        log(f"❌ فشل إنشاء liveStream: {e}")
+        return None, None
+
+def bind_stream_to_broadcast(yt, broadcast_id, stream_id):
+    try:
+        yt.liveBroadcasts().bind(id=broadcast_id, part="id", streamId=stream_id).execute()
+        log(f"✅ تم ربط البث {broadcast_id} مع {stream_id}")
+        return True
+    except Exception as e:
+        log(f"❌ فشل ربط البث: {e}")
+        return False
+
+def finalize_all_broadcasts():
+    """إنهاء جميع البثوث النشطة لبداية نظيفة"""
+    try:
+        yt = build_youtube_client()
+        if not yt:
+            return
+        broadcasts = yt.liveBroadcasts().list(mine=True, part="id,status").execute()
+        for item in broadcasts.get("items", []):
+            bid = item["id"]
+            status = item.get("status", {}).get("lifeCycleStatus", "")
+            if status in ["ready", "created", "live", "testing"]:
+                try:
+                    yt.liveBroadcasts().transition(broadcastStatus="complete", id=bid, part="status").execute()
+                    log(f"🧹 أنهيت البث: {bid} (كان {status})")
+                except Exception as e:
+                    log(f"⚠️ تعذر إنهاء البث {bid}: {e}")
+    except Exception as e:
+        log(f"❌ خطأ في التنظيف: {e}")
+
+def create_new_broadcast():
+    """إنشاء بث جديد كامل وإرجاع (stream_key, broadcast_id)"""
+    try:
+        yt = build_youtube_client()
+        if not yt:
+            return None, None
+
+        now = time.strftime("%Y-%m-%d %H:%M:%S")
+        title = f"بث تيك توك مباشر | {now}"
+        insert_body = {
+            "snippet": {"title": title},
+            "status": {"privacyStatus": "public"},
+            "contentDetails": {"enableAutoStart": True, "enableAutoStop": True, "latencyPreference": "normal"}
+        }
+        new_broadcast = yt.liveBroadcasts().insert(part="snippet,status,contentDetails", body=insert_body).execute()
+        bid = new_broadcast["id"]
+        log(f"✅ بث جديد: {bid} | {title}")
+
+        stream_id, stream_key = create_live_stream(yt)
+        if not stream_key:
+            return None, None
+        if not bind_stream_to_broadcast(yt, bid, stream_id):
+            return None, None
+
+        # حفظ معلومات البث
+        with open(BROADCAST_FILE, "w") as f:
+            json.dump({"broadcast_id": bid, "stream_key": stream_key}, f)
+
+        return stream_key, bid
+    except Exception as e:
+        log(f"❌ فشل إنشاء بث جديد: {e}")
+        return None, None
+
+def finalize_current_broadcast(broadcast_id):
+    """إنهاء بث محدد"""
+    if not broadcast_id:
+        return
+    try:
+        yt = build_youtube_client()
+        if not yt:
+            return
+        yt.liveBroadcasts().transition(broadcastStatus="complete", id=broadcast_id, part="status").execute()
+        log(f"✅ تم إنهاء البث {broadcast_id}")
+    except Exception as e:
+        log(f"⚠️ خطأ في إنهاء البث {broadcast_id}: {e}")
+    if os.path.exists(BROADCAST_FILE):
+        os.remove(BROADCAST_FILE)
+
+# ------------------------------------------------------------
+# المحرك الرئيسي (معدل لاستخدام API مع احتياطي المفتاح الثابت)
+# ------------------------------------------------------------
 def monitor_loop():
     try:
         subprocess.run(["killall", "-9", "ffmpeg", "yt-dlp", "streamlink"], stderr=subprocess.DEVNULL)
@@ -141,13 +299,19 @@ def monitor_loop():
     with open(LOCK_FILE, "w") as f:
         f.write(str(os.getpid()))
 
+    # تنظيف البثوث القديمة عبر API
+    log("🧹 تنظيف البثوث السابقة...")
+    finalize_all_broadcasts()
+
     is_streaming = False
     ffmpeg_proc = None
     was_streaming = False
     consecutive_fails = 0
     ffmpeg_restart_count = 0
+    current_stream_key = None
+    current_broadcast_id = None
 
-    log("🚀 محرك النقل الشفاف (بدون واجهة) بدأ")
+    log("🚀 محرك النقل الشفاف (YouTube API + احتياطي) بدأ")
 
     while True:
         try:
@@ -163,25 +327,51 @@ def monitor_loop():
                                 log("📋 آخر سجلات ffmpeg:\n" + "".join(f.readlines()[-10:]))
                         ffmpeg_restart_count += 1
                         if ffmpeg_restart_count >= 10:
-                            log("💀 10 محاولات فاشلة – توقف لمدة 5 دقائق")
+                            log("�� 10 محاولات فاشلة – توقف 5 دقائق")
                             kill_process_tree(ffmpeg_proc.pid)
                             ffmpeg_proc = None
                             is_streaming = False
                             was_streaming = False
+                            if current_broadcast_id:
+                                finalize_current_broadcast(current_broadcast_id)
+                                current_broadcast_id = None
+                                current_stream_key = None
                             time.sleep(300)
                             ffmpeg_restart_count = 0
                             continue
                         time.sleep(10)
 
+                    # الحصول على مفتاح بث جديد (API أولاً، ثم احتياطي)
+                    stream_key = None
+                    bid = None
+                    # محاولة إنشاء بث عبر API
+                    stream_key, bid = create_new_broadcast()
+                    if not stream_key:
+                        log("⚠️ فشل إنشاء بث بالـAPI، استخدام المفتاح الاحتياطي...")
+                        stream_key = CUSTOM_STREAM_KEY
+                        if not stream_key:
+                            log("❌ لا يوجد مفتاح احتياطي")
+                            time.sleep(30)
+                            continue
+                    else:
+                        current_broadcast_id = bid
+
                     log("🟢 بدء بث جديد...")
                     if ffmpeg_proc:
                         kill_process_tree(ffmpeg_proc.pid)
-                    ffmpeg_proc = start_stream(STREAM_KEY, stream_url)
+                    ffmpeg_proc = start_stream(stream_key, stream_url)
                     if ffmpeg_proc:
                         is_streaming = True
                         was_streaming = True
                         consecutive_fails = 0
                         ffmpeg_restart_count = 0
+                        current_stream_key = stream_key
+                        log(f"🔑 المفتاح الحالي: ...{stream_key[-4:]}")
+                    else:
+                        if current_broadcast_id:
+                            finalize_current_broadcast(current_broadcast_id)
+                            current_broadcast_id = None
+                            current_stream_key = None
             else:
                 consecutive_fails += 1
                 if was_streaming and consecutive_fails > 5:
@@ -191,6 +381,10 @@ def monitor_loop():
                         ffmpeg_proc = None
                     is_streaming = False
                     was_streaming = False
+                    if current_broadcast_id:
+                        finalize_current_broadcast(current_broadcast_id)
+                        current_broadcast_id = None
+                        current_stream_key = None
                     consecutive_fails = 0
 
             time.sleep(30)
@@ -200,7 +394,7 @@ def monitor_loop():
             time.sleep(30)
 
 if __name__ == "__main__":
-    log("✅ بدء تشغيل المحرك بدون واجهة")
+    log("✅ بدء تشغيل المحرك مع YouTube API")
     t = threading.Thread(target=monitor_loop, daemon=True)
     t.start()
     while True: