Update app.py

app.py

@@ -181,24 +181,77 @@ def get_stream_url(driver):
     try:
         log("🔍 Đang tìm stream URL...")
         
-        # Đợi thêm để video load
-        time.sleep(8)
+        # Đợi video load
+        time.sleep(10)
+        
+        # Phương pháp 1: Lấy trực tiếp từ video element
+        log("🔍 Kiểm tra video element...")
+        try:
+            wait = WebDriverWait(driver, 15)
+            video_element = wait.until(
+                EC.presence_of_element_located((By.TAG_NAME, "video"))
+            )
+            
+            # Lấy tất cả thuộc tính của video
+            video_info = driver.execute_script("""
+                var video = arguments[0];
+                return {
+                    src: video.src || '',
+                    currentSrc: video.currentSrc || '',
+                    networkState: video.networkState,
+                    readyState: video.readyState,
+                    duration: video.duration,
+                    paused: video.paused
+                };
+            """, video_element)
+            
+            log(f"📊 Video info: readyState={video_info['readyState']}, paused={video_info['paused']}")
+            
+            # Kiểm tra src
+            for src_key in ['currentSrc', 'src']:
+                url = video_info.get(src_key, '')
+                if url and not url.startswith('blob:') and url.startswith('http'):
+                    log(f"✅ Stream từ video.{src_key}")
+                    log(f"📺 {url}")
+                    return url
+            
+            # Nếu là blob URL, cần tìm URL gốc
+            if video_info.get('currentSrc', '').startswith('blob:'):
+                log("⚠️ Video dùng blob URL - cần tìm stream gốc")
+                
+        except Exception as e:
+            log(f"⚠️ Video element: {str(e)[:100]}")
         
-        # Phương pháp 1: Tìm trong network requests (ưu tiên)
-        log("🔍 Phân tích network requests...")
+        # Phương pháp 2: Tìm HLS/DASH manifest trong network
+        log("🔍 Tìm manifest trong network...")
         
         try:
             logs = driver.get_log('performance')
-            stream_candidates = {
+            
+            # Danh sách pattern tìm kiếm
+            video_patterns = [
+                '.m3u8',      # HLS
+                '.mpd',       # DASH
+                '/hls/',
+                '/dash/',
+                'manifest',
+                'playlist',
+                'master.m3u8',
+                'index.m3u8',
+                '/live/',
+                '/stream/'
+            ]
+            
+            # Extensions không hợp lệ
+            invalid_exts = ['.js', '.css', '.json', '.html', '.woff', '.ttf', '.svg', 
+                           '.png', '.jpg', '.jpeg', '.gif', '.ico', '.xml', '.txt', '.webp']
+            
+            stream_urls = {
                 'hls': [],
                 'dash': [],
-                'mp4': [],
-                'other': []
+                'live': []
             }
             
-            # Danh sách extension không phải video
-            invalid_extensions = ['.js', '.css', '.json', '.html', '.woff', '.ttf', '.svg', '.png', '.jpg', '.gif', '.ico', '.xml']
-            
             for entry in logs:
                 import json
                 try:
@@ -207,129 +260,60 @@ def get_stream_url(driver):
                     if log_entry['method'] == 'Network.requestWillBeSent':
                         url = log_entry['params']['request']['url']
                         
-                        # Bỏ qua các file không phải video
-                        if any(url.endswith(ext) for ext in invalid_extensions):
+                        # Skip invalid
+                        if not url.startswith('http'):
                             continue
-                        
-                        # Bỏ qua các domain không liên quan
-                        if any(x in url.lower() for x in ['google', 'analytics', 'facebook', 'twitter', 'ads', 'tracking']):
+                        if any(url.endswith(ext) for ext in invalid_exts):
+                            continue
+                        if 'google' in url or 'facebook' in url or 'analytics' in url:
                             continue
                         
-                        # Phân loại stream
+                        # Tìm HLS
                         if '.m3u8' in url:
-                            stream_candidates['hls'].append(url)
-                            log(f"🎯 HLS: {url[:80]}...")
+                            stream_urls['hls'].append(url)
+                        # Tìm DASH
                         elif '.mpd' in url:
-                            stream_candidates['dash'].append(url)
-                            log(f"🎯 DASH: {url[:80]}...")
-                        elif '.mp4' in url or '.flv' in url or '.webm' in url:
-                            stream_candidates['mp4'].append(url)
-                            log(f"🎯 MP4: {url[:80]}...")
-                        elif any(keyword in url.lower() for keyword in ['stream', 'video', 'live', 'hls', 'manifest', 'playlist']):
-                            # Chỉ thêm nếu không phải file tĩnh
-                            if not any(url.endswith(ext) for ext in invalid_extensions):
-                                stream_candidates['other'].append(url)
-                                log(f"🎯 Stream: {url[:80]}...")
+                            stream_urls['dash'].append(url)
+                        # Tìm live stream URLs
+                        elif any(pattern in url.lower() for pattern in ['/hls/', '/dash/', '/live/', '/stream/', 'manifest', 'playlist']):
+                            if not any(url.endswith(ext) for ext in invalid_exts):
+                                stream_urls['live'].append(url)
                 except:
                     continue
             
-            # Chọn stream tốt nhất theo thứ tự ưu tiên
-            if stream_candidates['hls']:
-                url = stream_candidates['hls'][0]
+            # Log kết quả
+            if stream_urls['hls']:
+                log(f"🎯 Tìm thấy {len(stream_urls['hls'])} HLS stream(s)")
+                url = stream_urls['hls'][0]
                 log(f"✅ Chọn HLS stream")
                 log(f"📺 {url}")
                 return url
             
-            if stream_candidates['dash']:
-                url = stream_candidates['dash'][0]
+            if stream_urls['dash']:
+                log(f"🎯 Tìm thấy {len(stream_urls['dash'])} DASH stream(s)")
+                url = stream_urls['dash'][0]
                 log(f"✅ Chọn DASH stream")
                 log(f"📺 {url}")
                 return url
             
-            if stream_candidates['mp4']:
-                url = stream_candidates['mp4'][0]
-                log(f"✅ Chọn MP4 stream")
-                log(f"📺 {url}")
-                return url
-            
-            if stream_candidates['other']:
-                url = stream_candidates['other'][0]
-                log(f"✅ Chọn stream")
-                log(f"📺 {url}")
-                return url
+            if stream_urls['live']:
+                log(f"🎯 Tìm thấy {len(stream_urls['live'])} live stream(s)")
+                for url in stream_urls['live']:
+                    log(f"   - {url[:100]}...")
+                # Chọn URL có vẻ đúng nhất
+                for url in stream_urls['live']:
+                    if any(x in url.lower() for x in ['m3u8', 'mpd', 'manifest', 'playlist', 'master']):
+                        log(f"✅ Chọn live stream")
+                        log(f"📺 {url}")
+                        return url
                 
         except Exception as e:
-            log(f"⚠️ Lỗi network analysis: {e}")
+            log(f"⚠️ Network analysis: {str(e)[:100]}")
         
-        # Phương pháp 2: Lấy từ video element
-        log("🔍 Kiểm tra video element...")
-        try:
-            wait = WebDriverWait(driver, 10)
-            video_element = wait.until(
-                EC.presence_of_element_located((By.TAG_NAME, "video"))
-            )
-            
-            # Thử nhiều cách lấy URL
-            for method in ['src', 'currentSrc']:
-                try:
-                    stream_url = driver.execute_script(f"return arguments[0].{method}", video_element)
-                    if stream_url and stream_url.startswith('http') and not any(stream_url.endswith(ext) for ext in ['.js', '.css']):
-                        log(f"✅ Stream từ video.{method}")
-                        log(f"📺 {stream_url}")
-                        return stream_url
-                except:
-                    pass
-            
-            # Lấy từ source elements
-            try:
-                sources = driver.execute_script("""
-                    var video = arguments[0];
-                    var sources = video.querySelectorAll('source');
-                    return Array.from(sources).map(s => s.src);
-                """, video_element)
-                
-                for src in sources:
-                    if src and src.startswith('http') and not any(src.endswith(ext) for ext in ['.js', '.css']):
-                        log(f"✅ Stream từ source element")
-                        log(f"📺 {src}")
-                        return src
-            except:
-                pass
-                
-        except Exception as e:
-            log(f"⚠️ Video element: {e}")
-        
-        # Phương pháp 3: Tìm trong iframe
-        log("🔍 Kiểm tra iframe...")
-        try:
-            iframes = driver.find_elements(By.TAG_NAME, "iframe")
-            log(f"Tìm thấy {len(iframes)} iframe(s)")
-            
-            for i, iframe in enumerate(iframes):
-                try:
-                    driver.switch_to.frame(iframe)
-                    
-                    # Tìm video trong iframe
-                    videos = driver.find_elements(By.TAG_NAME, "video")
-                    if videos:
-                        for vid in videos:
-                            src = driver.execute_script("return arguments[0].src || arguments[0].currentSrc", vid)
-                            if src and src.startswith('http'):
-                                driver.switch_to.default_content()
-                                log(f"✅ Stream từ iframe {i}")
-                                log(f"📺 {src}")
-                                return src
-                    
-                    driver.switch_to.default_content()
-                except:
-                    driver.switch_to.default_content()
-                    continue
-        except Exception as e:
-            log(f"⚠️ Iframe check: {e}")
-        
-        log("❌ Không tìm thấy stream URL hợp lệ")
-        log("💡 Gợi ý: Thử URL khác hoặc kiểm tra trang có phát video không")
-        return None
+        # Phương pháp 3: Screenshot + screen recording
+        log("💡 Không tìm thấy stream URL trực tiếp")
+        log("🎥 Sẽ thử ghi màn hình video player...")
+        return "SCREEN_RECORD"
         
     except Exception as e:
         log(f"❌ Lỗi: {e}")