Update app.py

app.py

@@ -1,107 +1,238 @@
-from flask import Flask, request, jsonify, redirect, Response
 import os
+import re
+import time
+import json
 import requests
 import subprocess
-import sys
-
-app = Flask(__name__)
-
-watcher_process = None
-
-def start_watcher():
-    global watcher_process
-
-    if watcher_process is None or watcher_process.poll() is not None:
-        watcher_process = subprocess.Popen(
-            [sys.executable, "watcher.py"],
-            stdout=sys.stdout,
-            stderr=sys.stderr,
-            env=os.environ,   # ← 環境変数を引き継ぐ
-        )
-        print("watcher.py を起動しました")
-
-
-#----------
-
-@app.route("/drive.com/files")
-def index():
-    ip = request.remote_addr
-    print(f"アクセスIP: {ip}")
-    return redirect("https://drive.google.com/")
-
-@app.route("/channel-io-managers")
-def get_managers():
-    limit = request.args.get("limit")
-    since = request.args.get("since")
+from datetime import datetime, timezone
+from bs4 import BeautifulSoup
+
+# ===== Channel.io 設定 =====
+GET_URL = "https://desk-api.channel.io/desk/channels/200605/groups/519217/messages"
+POST_URL = GET_URL
+
+PARAMS = {
+    "sortOrder": "desc",
+    "limit": 36,
+    "logFolded": "false",
+}
+
+X_ACCOUNT = os.getenv("channeliotokenbot2")
+if not X_ACCOUNT:
+    raise RuntimeError("環境変数 channeliotokenbot2 が設定されていません")
+
+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,
+}
+
+# ===== ssyoutube =====
+SSYOUTUBE_URL = "https://ssyoutube.online/yt-video-detail/"
+
+# ===== tflink =====
+TFLINK_UPLOAD_URL = os.getenv("TFLINK_UPLOAD_URL")
+if not TFLINK_UPLOAD_URL:
+    raise RuntimeError("環境変数 TFLINK_UPLOAD_URL が設定されていません")
+
+# ===== Utils =====
+def parse_updated_at(value):
+    if isinstance(value, (int, float)):
+        return datetime.fromtimestamp(value / 1000, tz=timezone.utc)
+    elif isinstance(value, str):
+        return datetime.fromisoformat(value.replace("Z", "+00:00"))
+    return None
+
+def extract_youtube_id(text):
+    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
+
+# ===== ssyoutube HTML 解析 =====
+def fetch_download_links(youtube_url):
+    res = requests.post(
+        SSYOUTUBE_URL,
+        data={"videoURL": youtube_url},
+        timeout=30,
+        headers={
+            "User-Agent": "Mozilla/5.0",
+            "Referer": "https://ssyoutube.online/",
+        }
+    )
+    res.raise_for_status()
+
+    soup = BeautifulSoup(res.text, "lxml")
+    buttons = soup.select("button[data-url]")
+
+    results = []
+    for btn in buttons:
+        url = btn.get("data-url")
+        quality = btn.get("data-quality")
+        has_audio = btn.get("data-has-audio")
+
+        if not url:
+            continue
+
+        results.append({
+            "url": url,
+            "quality": quality or "audio",
+            "has_audio": has_audio,
+        })
+
+    return results
+
+# ===== 動画結合 =====
+def merge_video_audio(video_url, audio_url, output_file):
+    # ダウンロード
+    video_path = "video.mp4"
+    audio_path = "audio.m4a"
+    with requests.get(video_url, stream=True) as r:
+        r.raise_for_status()
+        with open(video_path, "wb") as f:
+            for chunk in r.iter_content(chunk_size=8192):
+                f.write(chunk)
+
+    with requests.get(audio_url, stream=True) as r:
+        r.raise_for_status()
+        with open(audio_path, "wb") as f:
+            for chunk in r.iter_content(chunk_size=8192):
+                f.write(chunk)
+
+    # ffmpeg で結合
+    cmd = [
+        "ffmpeg", "-y",
+        "-i", video_path,
+        "-i", audio_path,
+        "-c:v", "copy",
+        "-c:a", "aac",
+        output_file
+    ]
+    subprocess.run(cmd, check=True)
+
+    # 一時ファイル削除
+    os.remove(video_path)
+    os.remove(audio_path)
+
+# ===== tflink アップロード =====
+def upload_to_tflink(file_path):
+    with open(file_path, "rb") as f:
+        res = requests.post(TFLINK_UPLOAD_URL, files={"file": f})
+    res.raise_for_status()
+    return res.json().get("url")
+
+# ===== Channel.io 送信 =====
+def build_links(items):
+    lines = []
+    for item in items:
+        url = item["url"]
+        quality = item["quality"]
+        has_audio = item["has_audio"]
+
+        audio_label = ""
+        if has_audio == "false":
+            audio_label = "（映像のみ）"
+        elif has_audio == "true":
+            audio_label = "（音声付き）"
+
+        line = f'<link type="url" value="{url}"> {quality} {audio_label}</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
+    }
 
-    params = {}
-    if limit:
-        params["limit"] = limit
-    if since:
-        params["since"] = since
+    res = requests.post(POST_URL, headers=HEADERS_POST, data=json.dumps(payload), timeout=30)
+    res.raise_for_status()
 
-    channel_id = request.args.get("channelid", "200605")
-    url = f"https://desk-api.channel.io/desk/channels/{channel_id}/managers"
+# ===== Main =====
+def main():
+    while True:
+        try:
+            res = requests.get(GET_URL, headers=HEADERS_GET, params=PARAMS, timeout=30)
+            res.raise_for_status()
 
-    headers = {
-        "accept": "application/json",
-        "x-account": os.getenv("channeliotokenmain"),
-    }
+            messages = res.json().get("messages", [])
+            latest_msg = None
+            latest_time = None
 
-    res = requests.get(url, headers=headers, params=params)
+            for msg in messages:
+                plain_text = msg.get("plainText")
+                updated_at = msg.get("updatedAt")
 
-    if res.status_code != 200:
-        return jsonify({"error": res.text}), res.status_code
+                if not plain_text or updated_at is None:
+                    continue
 
-    return jsonify(res.json().get("managers", []))
+                t = parse_updated_at(updated_at)
+                if not t:
+                    continue
 
-@app.route("/cors-proxy", methods=["GET"])
-def corsproxy():
-    url = request.args.get("url")
-    if not url:
-        return "url パラメータが必要です", 400
+                if latest_time is None or t > latest_time:
+                    latest_time = t
+                    latest_msg = msg
 
-    if not url.startswith(("http://", "https://")):
-        return "http または https のURLのみ使用できます", 400
+            if not latest_msg:
+                time.sleep(10)
+                continue
 
-    resp = requests.get(url, headers=request.headers, timeout=60)
+            text = latest_msg["plainText"]
+            youtube_id = extract_youtube_id(text)
+            if not youtube_id:
+                time.sleep(10)
+                continue
 
-    response = Response(resp.content, resp.status_code)
-    response.headers["Access-Control-Allow-Origin"] = "*"
-    response.headers["Access-Control-Allow-Headers"] = "*"
-    response.headers["Access-Control-Allow-Methods"] = "GET, POST, PATCH, OPTIONS"
+            youtube_url = f"https://www.youtube.com/watch?v={youtube_id}"
+            items = fetch_download_links(youtube_url)
+            if not items:
+                print("ダウンロードリンクが取得できませんでした")
+                time.sleep(10)
+                continue
 
-    if "Content-Type" in resp.headers:
-        response.headers["Content-Type"] = resp.headers["Content-Type"]
+            # 最も高画質の動画と音声を選択
+            video_item = max((i for i in items if i['has_audio']=='false'), key=lambda x: int(x['quality'].replace('p','') if x['quality']!='audio' else 0))
+            audio_item = next((i for i in items if i['has_audio']=='true' and i['quality']=='audio'), None)
 
-    return response
+            if not video_item or not audio_item:
+                print("動画または音声が不足しています")
+                time.sleep(10)
+                continue
 
-@app.route("/cors-proxy", methods=["POST", "PATCH"])
-def corsproxy_post():
-    url = request.args.get("url")
-    if not url:
-        return "url パラメータが必要です", 400
+            output_file = "merged.mp4"
+            merge_video_audio(video_item['url'], audio_item['url'], output_file)
 
-    if not url.startswith(("http://", "https://")):
-        return "http または https のURLのみ使用できます", 400
+            # tflink アップロード
+            tflink_url = upload_to_tflink(output_file)
+            os.remove(output_file)
 
-    resp = requests.request(
-        method=request.method,
-        url=url,
-        headers=request.headers,
-        data=request.data,
-        timeout=60,
-    )
+            message_text = build_links(items) + f"\n結合ファイル: {tflink_url}"
+            send_to_channel(message_text)
+            print("送信完了")
 
-    headers = {
-        "Access-Control-Allow-Origin": "*",
-        "Access-Control-Allow-Headers": "*",
-        "Access-Control-Allow-Methods": "GET, POST, PATCH, OPTIONS",
-    }
+        except Exception as e:
+            print("エラー:", e)
 
-    return Response(resp.content, resp.status_code, headers=headers)
+        time.sleep(15)
 
 if __name__ == "__main__":
-    if os.environ.get("WERKZEUG_RUN_MAIN") != "true":
-        start_watcher()
-    app.run(debug=True, host="0.0.0.0", port=7860)
+    main()