Update watcher.py

watcher.py

@@ -2,9 +2,6 @@ import os
 import re
 import time
 import json
-import shutil
-import tempfile
-import subprocess
 import requests
 from datetime import datetime, timezone
 from bs4 import BeautifulSoup
@@ -58,6 +55,18 @@ def extract_youtube_id(text):
             return m.group(1)
     return None
 
+def get_video_title(youtube_url):
+    """YouTube動画のタイトルを取得する簡易的な関数"""
+    try:
+        res = requests.get(youtube_url, timeout=10)
+        soup = BeautifulSoup(res.text, "lxml")
+        title_tag = soup.find("meta", property="og:title")
+        if title_tag:
+            return title_tag.get("content", "").replace(" - YouTube", "")
+    except:
+        pass
+    return "YouTube動画"
+
 # ===== ssyoutube HTML 解析 =====
 def fetch_download_links(youtube_url):
     res = requests.post(
@@ -98,21 +107,28 @@ def fetch_download_links(youtube_url):
 # ===== 最高画質映像 + 音声を選択 =====
 def select_best_video_and_audio(items):
     video_items = []
-    audio_item = None
+    audio_items = []
 
     for item in items:
         if item["quality"] == "audio":
-            audio_item = item
+            audio_items.append(item)
         elif item["has_audio"] == "false":
             m = re.match(r"(\d+)p", item["quality"])
             if m:
                 video_items.append((int(m.group(1)), item))
 
-    if not video_items or not audio_item:
+    if not video_items or not audio_items:
         return None, None
 
+    # 最高画質の動画を選択
     video_items.sort(key=lambda x: x[0], reverse=True)
-    return video_items[0][1], audio_item
+    best_video = video_items[0][1]
+    
+    # 最高音質の音声を選択（例: 128kbps > 64kbps）
+    # 実際の音声品質情報に基づいて選択する場合は要修正
+    best_audio = audio_items[0] if audio_items else None
+    
+    return best_video, best_audio
 
 # ===== ssyoutube サーバー側結合 =====
 def merge_video_on_server(video_url, audio_url, video_id, quality, video_title, nonce):
@@ -229,50 +245,6 @@ def send_to_channel(text):
     )
     res.raise_for_status()
 
-# ===== ダウンロードリンクを整理して表示 =====
-def format_download_links(items):
-    """
-    ダウンロードリンクを整理してフォーマットする
-    """
-    # 画質順にソート
-    video_items = []
-    audio_items = []
-    
-    for item in items:
-        if item["quality"] == "audio":
-            audio_items.append(item)
-        else:
-            m = re.match(r"(\d+)p", item["quality"])
-            if m:
-                video_items.append((int(m.group(1)), item))
-    
-    # 画質の高い順にソート
-    video_items.sort(key=lambda x: x[0], reverse=True)
-    
-    # メッセージ作成
-    lines = ["📥 利用可能なダウンロードリンク:"]
-    
-    # 映像リンク（音声付き）
-    video_with_audio = [item for item in items if item.get("has_audio") == "true" and item["quality"] != "audio"]
-    if video_with_audio:
-        lines.append("\n🎬 映像+音声（単一ファイル）:")
-        for item in video_with_audio:
-            lines.append(f"  • {item['quality']}: {item['url']}")
-    
-    # 映像のみリンク
-    if video_items:
-        lines.append("\n📹 映像のみ（高画質）:")
-        for quality, item in video_items:
-            lines.append(f"  • {item['quality']}: {item['url']}")
-    
-    # 音声のみリンク
-    if audio_items:
-        lines.append("\n🎵 音声のみ:")
-        for item in audio_items:
-            lines.append(f"  • {item['quality']}: {item['url']}")
-    
-    return "\n".join(lines)
-
 # ===== Main =====
 def main():
     processed_messages = set()  # 処理済みメッセージを追跡
@@ -321,60 +293,82 @@ def main():
                 continue
 
             youtube_url = f"https://www.youtube.com/watch?v={youtube_id}"
+            
+            # 動画タイトルを取得
+            video_title = get_video_title(youtube_url)
+            
             items, nonce = fetch_download_links(youtube_url)
             
-            if not items:
-                print("ダウンロードリンクが見つかりません")
+            if not nonce:
+                print("nonceの取得に失敗しました")
                 time.sleep(10)
                 continue
 
-            # 1. すべてのダウンロードリンクを送信
-            download_links_message = format_download_links(items)
-            send_to_channel(download_links_message)
+            # すべてのダウンロードリンクを収集
+            video_items = []
+            audio_items = []
+            all_items = []
+            
+            for item in items:
+                all_items.append(item)
+                if item["quality"] == "audio":
+                    audio_items.append(item)
+                elif item["has_audio"] == "false":
+                    video_items.append(item)
+            
+            # 最高画質の動画と音声を選択
+            best_video, best_audio = select_best_video_and_audio(items)
+            
+            # メッセージを構築
+            message_lines = []
+            message_lines.append(f"**{video_title}**\n")
+            message_lines.append("**ダウンロードリンク:**")
             
-            # 2. 最高画質の映像と音声を結合（オプション）
-            if nonce:
-                video_item, audio_item = select_best_video_and_audio(items)
+            # すべての動画リンクを追加
+            for item in all_items:
+                quality = item["quality"]
+                if item["quality"] == "audio":
+                    quality_display = "音声"
+                elif item["has_audio"] == "true":
+                    quality_display = f"{quality} (音声付き)"
+                else:
+                    quality_display = f"{quality} (映像のみ)"
                 
-                if video_item and audio_item:
-                    # 画質情報を取得
-                    quality_match = re.match(r"(\d+)p", video_item["quality"])
-                    if quality_match:
-                        quality = quality_match.group(1)
-                        
-                        # 動画タイトル（簡易版）
-                        video_title = f"youtube_video_{youtube_id}"
-                        
-                        # 結合処理開始メッセージを送信
-                        send_to_channel(f"🔄 最高画質({quality}p)の動画と音声を結合中... しばらくお待ちください。")
-                        
-                        try:
-                            # サーバー側で結合
-                            print(f"サーバー側での結合を開始します: {youtube_id} {quality}p")
-                            merged_url = merge_video_on_server(
-                                video_item["url"],
-                                audio_item["url"],
-                                youtube_id,
-                                quality,
-                                video_title,
-                                nonce
-                            )
-                            
-                            # 結合済み動画のリンクを送信
-                            merged_message = f"✅ 結合完了！最高画質({quality}p)の動画:\n{merged_url}"
-                            send_to_channel(merged_message)
-                            
-                            print("結合完了:", merged_url)
-                            
-                        except Exception as merge_error:
-                            error_message = f"⚠️ 結合処理中にエラーが発生しました: {str(merge_error)}"
-                            send_to_channel(error_message)
-                            print("結合エラー:", merge_error)
+                message_lines.append(f"• {quality_display}: {item['url']}")
+            
+            # 最高画質の結合動画を作成
+            if best_video and best_audio and nonce:
+                message_lines.append("\n **最高画質結合動画 (処理中):**")
+                
+                # 画質情報を取得
+                quality_match = re.match(r"(\d+)p", best_video["quality"])
+                if quality_match:
+                    quality = quality_match.group(1)
+                    
+                    # サーバー側で結合
+                    print(f"サーバー側での結合を開始します: {youtube_id} {quality}p")
+                    try:
+                        merged_url = merge_video_on_server(
+                            best_video["url"],
+                            best_audio["url"],
+                            youtube_id,
+                            quality,
+                            video_title,
+                            nonce
+                        )
+                        message_lines.append(f"{quality}p 結合済み動画: {merged_url}")
+                    except Exception as e:
+                        print(f"結合処理エラー: {e}")
+                        message_lines.append(f"結合処理に失敗: {str(e)}")
+            
+            # メッセージを送信
+            message = "\n".join(message_lines)
+            send_to_channel(message)
             
             # 処理済みとしてマーク
             processed_messages.add(latest_msg.get("id"))
             
-            print(f"処理完了: {youtube_id}")
+            print(f"送信完了: {youtube_id}")
 
         except Exception as e:
             print("エラー:", e)