Update app.py

app.py

@@ -2,26 +2,32 @@ import os
 import time
 import subprocess
 import threading
+import json
+import pickle
 import psutil
+from datetime import datetime, timezone
+from google.oauth2.credentials import Credentials
+from google.auth.transport.requests import Request
+from googleapiclient.discovery import build
 from flask import Flask
 
 # ------------------------------------------------------------
-# الإعدادات
+# الإعدادات الأساسية (تتغير ديناميكياً لتفادي تداخل المشاريع)
 # ------------------------------------------------------------
 TIKTOK_USERNAME = "@a3dam_111"
-CUSTOM_STREAM_KEY = os.getenv("CUSTOM_STREAM_KEY", "p3dq-5cex-bc1k-vet7-bsyx")
-FFMPEG_LOG = "/tmp/ffmpeg.log"
-MONITOR_LOG = "/tmp/monitor.log"
 
-# إعداد Flask لمنع الإغلاق التلقائي
-app = Flask(__name__)
+# استخراج اسم نقي للملفات لمنع التداخل والاصطدام بين السكربتات الشغالة معاً
+USER_TAG = TIKTOK_USERNAME.replace("@", "").strip()
 
-@app.route('/')
-def home():
-    return "البوت يعمل في الخلفية بنجاح!"
+CUSTOM_STREAM_KEY = os.getenv("CUSTOM_STREAM_KEY", "p3dq-5cex-bc1k-vet7-bsyx")
+FFMPEG_LOG = f"/tmp/ffmpeg_{USER_TAG}.log"
+MONITOR_LOG = f"/tmp/monitor_{USER_TAG}.log"
+LOCK_FILE = f"/tmp/app_monitor_core_{USER_TAG}.lock"
+TOKEN_FILE = f"/tmp/yt_token_{USER_TAG}.pkl"
+BROADCAST_FILE = f"/tmp/current_broadcast_{USER_TAG}.json"
 
-def run_web():
-    app.run(host='0.0.0.0', port=7860)
+TIKTOK_COOKIES = os.getenv("TIKTOK_COOKIES", "")
+TIKTOK_COOKIES_ADULT = os.getenv("TIKTOK_COOKIES_ADULT", "")
 
 def log(msg):
     ts = time.strftime("%Y-%m-%d %H:%M:%S")
@@ -31,71 +37,242 @@ def log(msg):
         f.write(line + "\n")
 
 # ------------------------------------------------------------
-# الكوكيز
+# حفظ الكوكيز باسم معزول
 # ------------------------------------------------------------
 def save_cookies():
-    for env_var, file_path in [("TIKTOK_COOKIES", "/tmp/tiktok_cookies.txt"), ("TIKTOK_COOKIES_ADULT", "/tmp/tiktok_cookies_adult.txt")]:
-        cookie_data = os.getenv(env_var, "").strip()
-        if cookie_data:
-            with open(file_path, "w", encoding="utf-8") as f:
-                f.write(cookie_data)
-            log(f"✅ تم حفظ بيانات الكوكيز في {file_path}")
+    if TIKTOK_COOKIES:
+        with open(f"/tmp/tiktok_cookies_{USER_TAG}.txt", "w") as f:
+            f.write(TIKTOK_COOKIES)
+    if TIKTOK_COOKIES_ADULT:
+        with open(f"/tmp/tiktok_cookies_adult_{USER_TAG}.txt", "w") as f:
+            f.write(TIKTOK_COOKIES_ADULT)
+
+save_cookies()
 
 # ------------------------------------------------------------
-# الدوال الأساسية
+# أدوات مساعدة
 # ------------------------------------------------------------
+def kill_process_tree(pid):
+    try:
+        parent = psutil.Process(pid)
+        for child in parent.children(recursive=True):
+            child.kill()
+        parent.kill()
+    except:
+        pass
+
 def get_tiktok_stream_url():
+    cookies_adult = f"/tmp/tiktok_cookies_adult_{USER_TAG}.txt"
+    cookies_normal = f"/tmp/tiktok_cookies_{USER_TAG}.txt"
+    
     try:
         url = f"https://www.tiktok.com/{TIKTOK_USERNAME}/live"
-        cmd = ["streamlink", "--stream-url", url, "best"]
-        if os.path.exists("/tmp/tiktok_cookies.txt"): cmd.extend(["--cookies", "/tmp/tiktok_cookies.txt"])
+        cmd = ["streamlink", "--stream-url", url, "best", "--retry-streams", "5", "--retry-open", "3"]
+        if os.path.exists(cookies_adult): cmd.extend(["--cookies", cookies_adult])
+        elif os.path.exists(cookies_normal): cmd.extend(["--cookies", cookies_normal])
+        result = subprocess.run(cmd, capture_output=True, text=True, timeout=25)
+        if result.returncode == 0 and result.stdout.strip():
+            return result.stdout.strip().split("\n")[0]
+    except: pass
+    
+    try:
+        url = f"https://www.tiktok.com/{TIKTOK_USERNAME}/live"
+        cmd = ["yt-dlp", "--no-warnings", "--no-check-certificates", "--no-live-from-start", "--no-cache-dir", "--rm-cache-dir", "-f", "best", "--get-url", url]
+        if os.path.exists(cookies_adult): cmd.extend(["--cookies", cookies_adult])
+        elif os.path.exists(cookies_normal): cmd.extend(["--cookies", cookies_normal])
         result = subprocess.run(cmd, capture_output=True, text=True, timeout=30)
-        return result.stdout.strip().split("\n")[0] if result.returncode == 0 else None
-    except: return None
+        if result.returncode == 0 and result.stdout.strip():
+            return result.stdout.strip().split("\n")[0]
+    except: pass
+    return None
 
 def start_stream(stream_key, stream_url):
+    if not stream_key or not stream_url: return None
+    log("🎬 بدء البث (تزامن كامل الصوت والصورة + تحويل H.264)...")
     rtmp_url = f"rtmp://a.rtmp.youtube.com/live2/{stream_key}"
-    cmd = ["ffmpeg", "-re", "-i", stream_url, "-c:v", "copy", "-c:a", "aac", "-b:a", "192k", "-f", "flv", rtmp_url]
-    with open(FFMPEG_LOG, "w") as f:
-        return subprocess.Popen(cmd, stderr=f, preexec_fn=os.setsid)
+    
+    # تم حذف سطر killall العام لمنع إغلاق السكربتات الأخرى الشغالة معاً
+    
+    cmd = [
+        "nice", "-n", "19", "ffmpeg", "-hide_banner", "-loglevel", "warning",
+        "-thread_queue_size", "2048",
+        "-user_agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36",
+        "-headers", "Referer: https://www.tiktok.com/\r\n",
+        
+        # --- علاج مشكلة عدم تزامن الصوت والصورة عند تقطيع إنترنت المذيع ---
+        "-use_wallclock_as_timestamps", "1", 
+        "-fflags", "+genpts+igndts+discardcorrupt", 
+        "-flags", "+global_header", "-i", stream_url,
+        
+        # تثبيت الـ Framerate على 30 لسد الفجوات الزمنية عند سقوط الإطارات
+        "-c:v", "libx264", "-preset", "ultrafast", "-b:v", "2500k", "-maxrate", "2500k", "-bufsize", "5000k", "-pix_fmt", "yuv420p", "-g", "60", "-r", "30",
+        "-c:a", "aac", "-b:a", "192k", "-ar", "44100", "-profile:a", "aac_low", "-af", "aresample=async=1000:min_hard_comp=0.100000:first_pts=0,highpass=f=50,lowpass=f=15000,dynaudnorm=f=150:g=15:m=25:p=0.95",
+        "-reconnect", "1", "-reconnect_at_eof", "1", "-reconnect_delay_max", "60", "-rtmp_live", "live", "-max_muxing_queue_size", "2048", "-f", "flv", rtmp_url
+    ]
+    with open(FFMPEG_LOG, "w") as stderr_file:
+        proc = subprocess.Popen(cmd, stdout=subprocess.DEVNULL, stderr=stderr_file, preexec_fn=os.setsid)
+    log(f"✅ ffmpeg PID: {proc.pid} للمستخدم {TIKTOK_USERNAME}")
+    return proc
 
 # ------------------------------------------------------------
-# المحرك الرئيسي
+# YouTube API (الفحص والتعرف الذكي عبر البصمة الوصفية)
+# ------------------------------------------------------------
+def build_youtube_client():
+    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)
+        creds_json = os.getenv("YT_CREDENTIALS_JSON")
+        if not creds_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_new_broadcast():
+    try:
+        yt = build_youtube_client()
+        if not yt: return None, None
+        
+        now = datetime.now(timezone.utc)
+        date_str = now.strftime('%d-%m-%Y %H:%M')
+        title = f"🔥 بث تيك توك أحمد عبدالله العراقي المباشر | {date_str} 🚀"
+        
+        # حقن بصمة الحساب السرية داخل الوصف لمنع السكربتات الأخرى من حذفه
+        description = (
+            "أهلاً بكم في البث المباشر المخصص لـ أحمد عبدالله العراقي.\n"
+            f"معرف الحساب المسؤول: [{TIKTOK_USERNAME}]\n"
+            "استمتعوا معنا بأجواء البث وتفاعلوا في التعليقات!\n\n"
+            "🔔 لا تنسوا الاشتراك في القناة وتفعيل زر التنبيهات ليصلكم كل جديد فور انطلاق البث.\n"
+            "شكراً لدعمكم المتواصل!\n\n"
+            "#أحمد_عبدالله_العراقي #بث_مباشر #تيك_توك #TikTokLive #العراق #بث #مباشر #Ahmd_Abdullah"
+        )
+        
+        insert_body = {
+            "snippet": {"title": title, "description": description, "scheduledStartTime": now.isoformat()},
+            "status": {"privacyStatus": "public"},
+            "contentDetails": {"enableAutoStart": True, "enableAutoStop": False, "latencyPreference": "normal"}
+        }
+        
+        new_broadcast = yt.liveBroadcasts().insert(part="snippet,status,contentDetails", body=insert_body).execute()
+        bid = new_broadcast["id"]
+        log(f"✅ بث جديد تم إنشاؤه للـ ID: {bid}")
+        
+        stream_insert = yt.liveStreams().insert(
+            part="snippet,cdn,contentDetails",
+            body={"snippet": {"title": f"TikTok Stream Key {USER_TAG}"}, "cdn": {"ingestionType": "rtmp", "resolution": "1080p", "frameRate": "30fps"}, "contentDetails": {"isReusable": True}}
+        ).execute()
+        stream_id = stream_insert["id"]
+        stream_key = stream_insert["cdn"]["ingestionInfo"]["streamName"]
+        
+        yt.liveBroadcasts().bind(id=bid, part="id", streamId=stream_id).execute()
+        
+        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_all_broadcasts():
+    try:
+        yt = build_youtube_client()
+        if not yt: return
+        # قراءة الـ snippet لفحص الوصف والتحقق من بصمة المشروع
+        broadcasts = yt.liveBroadcasts().list(mine=True, part="id,status,snippet", maxResults=50).execute()
+        for item in broadcasts.get("items", []):
+            bid = item["id"]
+            status = item.get("status", {}).get("lifeCycleStatus", "")
+            description = item.get("snippet", {}).get("description", "")
+            
+            # نغلق البث فقط وفقط إذا كان يحمل بصمة هذا المستخدم لعدم تدمير البث الآخر
+            if status in ["ready", "created", "live", "testing"] and f"[{TIKTOK_USERNAME}]" in description:
+                try:
+                    yt.liveBroadcasts().transition(broadcastStatus="complete", id=bid, part="status").execute()
+                    log(f"🧹 تم تنظيف وإغلاق بث قديم معلق يخص الحساب {TIKTOK_USERNAME}")
+                except: pass
+    except: pass
+
+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()
+    except: pass
+    if os.path.exists(BROADCAST_FILE): os.remove(BROADCAST_FILE)
+
+# ------------------------------------------------------------
+# المحرك الرئيسي الآمن والمعزول كلياً
 # ------------------------------------------------------------
 def monitor_loop():
+    # حذفنا أمر killall الشامل الذي كان يسبب انهيار السكربت الأول عند نهوض الثاني
+    if os.path.exists(LOCK_FILE):
+        try:
+            with open(LOCK_FILE) as f:
+                os.kill(int(f.read().strip()), 0)
+            log(f"🔒 محرك الحساب {TIKTOK_USERNAME} يعمل بالفعل بشكل مستقر.")
+            return
+        except:
+            os.remove(LOCK_FILE)
+
+    with open(LOCK_FILE, "w") as f: f.write(str(os.getpid()))
+    
+    finalize_all_broadcasts()
+    ffmpeg_proc = None
+    was_streaming = False
+    consecutive_fails = 0
+    current_stream_key = None
+    current_broadcast_id = None
+    
+    log(f"🚀 تم إطلاق محرك المراقبة الخاص بـ {TIKTOK_USERNAME} بأمان وبصمة منفصلة.")
+    
     while True:
         try:
             stream_url = get_tiktok_stream_url()
             if stream_url:
-                log("🟢 البث متاح، بدء التشغيل...")
-                proc = start_stream(CUSTOM_STREAM_KEY, stream_url)
-                while proc.poll() is None:
-                    time.sleep(60)
+                consecutive_fails = 0
+                if was_streaming and ffmpeg_proc and ffmpeg_proc.poll() is None: pass
+                elif was_streaming and ffmpeg_proc and ffmpeg_proc.poll() is not None:
+                    kill_process_tree(ffmpeg_proc.pid)
+                    ffmpeg_proc = start_stream(current_stream_key, stream_url)
+                else:
+                    if not current_broadcast_id:
+                        finalize_all_broadcasts()
+                        stream_key, bid = create_new_broadcast()
+                        if not stream_key: stream_key = CUSTOM_STREAM_KEY
+                        current_broadcast_id = bid
+                        current_stream_key = stream_key
+                    ffmpeg_proc = start_stream(current_stream_key, stream_url)
+                    was_streaming = True
             else:
-                log("💤 بانتظار بدء البث في تيك توك...")
-                time.sleep(60)
-        except Exception as e:
-            log(f"💥 خطأ: {e}")
-            time.sleep(60)
+                if was_streaming:
+                    consecutive_fails += 1
+                    if consecutive_fails >= 6:
+                        if ffmpeg_proc: kill_process_tree(ffmpeg_proc.pid)
+                        ffmpeg_proc = None
+                        was_streaming = False
+                        if current_broadcast_id:
+                            finalize_current_broadcast(current_broadcast_id)
+                            current_broadcast_id = None
+            time.sleep(30)
+        except: time.sleep(30)
+
+app = Flask(__name__)
+@app.route("/")
+def index(): return "OK"
 
 if __name__ == "__main__":
-    log("✅ بدء تشغيل المحرك النهائي...")
-    
-    # 1. تشغيل خادم الويب (Flask) للحفاظ على استقرار السيرفر
-    web_thread = threading.Thread(target=run_web, daemon=True)
-    web_thread.start()
-    
-    # 2. حفظ الكوكيز
-    save_cookies()
-    
-    # 3. تشغيل حلقة البث
     t = threading.Thread(target=monitor_loop, daemon=True)
     t.start()
     
-    # حلقة مراقبة الـ Thread
-    while True:
-        if not t.is_alive():
-            log("🚨 إعادة تشغيل المحرك...")
-            t = threading.Thread(target=monitor_loop, daemon=True)
-            t.start()
-        time.sleep(3600)
+    # تنبيه: إذا قمت بتشغيل السكربت الثاني على نفس السيرفر، يرجى تغيير الـ port هنا إلى 781 مثلاً
+    app.run(host="0.0.0.0", port=7861, debug=False)