Update watcher.py

watcher.py

@@ -1,9 +1,14 @@
 import os
+import re
 import time
+import json
 import requests
 from datetime import datetime, timezone
 
-URL = "https://desk-api.channel.io/desk/channels/200605/groups/519217/messages"
+# ===== Channel.io 設定 =====
+GET_URL = "https://desk-api.channel.io/desk/channels/200605/groups/519217/messages"
+POST_URL = GET_URL
+
 PARAMS = {
     "sortOrder": "desc",
     "limit": 36,
@@ -14,12 +19,23 @@ X_ACCOUNT = os.getenv("channeliotokenbot2")
 if not X_ACCOUNT:
     raise RuntimeError("環境変数 channeliotokenbot2 が設定されていません")
 
-HEADERS = {
+HEADERS_GET = {
+    "accept": "application/json",
+    "accept-language": "ja",
+    "x-account": X_ACCOUNT,
+}
+
+HEADERS_POST = {
     "accept": "application/json",
     "accept-language": "ja",
+    "content-type": "application/json",
     "x-account": X_ACCOUNT,
 }
 
+# ===== Vidfly API =====
+VIDFLY_API = "https://api.vidfly.ai/api/media/youtube/download?url="
+
+# ===== Utils =====
 def parse_updated_at(value):
     if isinstance(value, (int, float)):
         return datetime.fromtimestamp(value / 1000, tz=timezone.utc)
@@ -27,19 +43,78 @@ def parse_updated_at(value):
         return datetime.fromisoformat(value.replace("Z", "+00:00"))
     return None
 
+def extract_youtube_id(text):
+    """
+    対応例:
+    https://www.youtube.com/watch?v=ID
+    https://youtu.be/ID
+    """
+    patterns = [
+        r"v=([A-Za-z0-9_-]{11})",
+        r"youtu\.be/([A-Za-z0-9_-]{11})",
+    ]
+    for p in patterns:
+        m = re.search(p, text)
+        if m:
+            return m.group(1)
+    return None
+
+def build_links(items):
+    lines = []
+    for item in items:
+        url = item.get("url")
+        ext = item.get("ext")
+        height = item.get("height")
+        media_type = item.get("type")
+
+        if not url or not ext or not height:
+            continue
+
+        label_type = "動画" if ext in ("mp4", "webm") else "音声"
+        audio_suffix = " (音声付き)" if media_type == "video_with_audio" else ""
+
+        line = f'<link type="url" value="{url}">{label_type}{audio_suffix} {height}p</link>'
+        lines.append(line)
+
+    return "\n".join(lines)
+
+def send_to_channel(text):
+    payload = {
+        "requestId": f"desk-web-{int(time.time() * 1000)}",
+        "blocks": [
+            {
+                "type": "text",
+                "value": text
+            }
+        ],
+        "buttons": None,
+        "form": None,
+        "webPage": None,
+        "files": None,
+        "customPayload": None
+    }
+
+    res = requests.post(
+        POST_URL,
+        headers=HEADERS_POST,
+        data=json.dumps(payload),
+        timeout=30
+    )
+    res.raise_for_status()
+
+# ===== Main =====
 def main():
     while True:
         try:
-            response = requests.get(
-                URL,
-                headers=HEADERS,
+            res = requests.get(
+                GET_URL,
+                headers=HEADERS_GET,
                 params=PARAMS,
                 timeout=30,
             )
-            response.raise_for_status()
+            res.raise_for_status()
 
-            data = response.json()
-            messages = data.get("messages", [])
+            messages = res.json().get("messages", [])
 
             latest_msg = None
             latest_time = None
@@ -51,18 +126,40 @@ def main():
                 if not plain_text or updated_at is None:
                     continue
 
-                updated_time = parse_updated_at(updated_at)
-                if not updated_time:
+                t = parse_updated_at(updated_at)
+                if not t:
                     continue
 
-                if latest_time is None or updated_time > latest_time:
-                    latest_time = updated_time
+                if latest_time is None or t > latest_time:
+                    latest_time = t
                     latest_msg = msg
 
-            if latest_msg:
-                print(latest_msg["plainText"])
-            else:
+            if not latest_msg:
                 print("条件に合う message が見つかりませんでした")
+                time.sleep(10)
+                continue
+
+            text = latest_msg["plainText"]
+            youtube_id = extract_youtube_id(text)
+
+            if not youtube_id:
+                print("YouTube ID が見つかりません")
+                time.sleep(10)
+                continue
+
+            youtube_url = f"https://www.youtube.com/watch?v={youtube_id}"
+            vidfly_res = requests.get(VIDFLY_API + youtube_url, timeout=30)
+            vidfly_res.raise_for_status()
+
+            items = vidfly_res.json().get("data", {}).get("items", [])
+            if not items:
+                print("Vidfly の items が空です")
+                time.sleep(10)
+                continue
+
+            message_text = build_links(items)
+            send_to_channel(message_text)
+            print("送信完了")
 
         except Exception as e:
             print("エラー:", e)