Update app.py

app.py

@@ -1,107 +1,321 @@
-from flask import Flask, request, jsonify, redirect, Response
+# -*- coding: utf-8 -*-
 import os
+import re
+import time
+import json
 import requests
-import subprocess
-import sys
+from datetime import datetime, timezone
+from bs4 import BeautifulSoup
+from urllib.parse import quote_plus
 
-app = Flask(__name__)
+# ===== Channel.io 設定 =====
+GET_URL = "https://desk-api.channel.io/desk/channels/200605/groups/519217/messages"
+POST_URL = GET_URL
 
-watcher_process = None
+PARAMS = {
+    "sortOrder": "desc",
+    "limit": 36,
+    "logFolded": "false",
+}
 
-def start_watcher():
-    global watcher_process
+X_ACCOUNT = os.getenv("channeliotokenbot2")
+if not X_ACCOUNT:
+    raise RuntimeError("環境変数 channeliotokenbot2 が設定されていません")
 
-    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 を起動しました")
+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,
+}
 
-#----------
+# ===== SaveNow 設定 =====
+# 例示の API キー（実運用では環境変数 SAVENOW_API を設定してください）
+DEFAULT_SAVENOW_API = "dfcb6d76f2f6a9894gjkege8a4ab232222"
+SAVENOW_API = os.getenv("SAVENOW_API", DEFAULT_SAVENOW_API)
+SAVENOW_BASE = "https://p.savenow.to/ajax/download.php"
 
-@app.route("/drive.com/files")
-def index():
-    ip = request.remote_addr
-    print(f"アクセスIP: {ip}")
-    return redirect("https://drive.google.com/")
+# ===== 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
 
-@app.route("/channel-io-managers")
-def get_managers():
-    limit = request.args.get("limit")
-    since = request.args.get("since")
+def extract_youtube_id(text):
+    patterns = [
+        r"(?:[?&](?:v|vi)=)([A-Za-z0-9_-]{11})",
+        r"youtu\.be/([A-Za-z0-9_-]{11})",
+        r"youtube(?:-nocookie)?\.com/(?:embed|v|shorts)/([A-Za-z0-9_-]{11})",
+        r"youtube(?:-nocookie)?\.com/watch\?.*?v=([A-Za-z0-9_-]{11})",
+        r"[#&]v=([A-Za-z0-9_-]{11})"
+    ]
+    for p in patterns:
+        m = re.search(p, text)
+        if m:
+            return m.group(1)
+    return None
 
-    params = {}
-    if limit:
-        params["limit"] = limit
-    if since:
-        params["since"] = since
+def extract_youtube_and_option(text):
+    parts = re.split(r"[ \u3000]+", text.strip())
+    video_id = extract_youtube_id(parts[0])
+    option = None
+    if len(parts) > 1:
+        option = parts[1].lower()
+    return video_id, option
 
-    channel_id = request.args.get("channelid", "200605")
-    url = f"https://desk-api.channel.io/desk/channels/{channel_id}/managers"
+def normalize_quality(option):
+    if option is None:
+        return "720"
+    trans_table = str.maketrans(
+        "０１２３４５６７８９ｋＫｐＰ",
+        "0123456789kKpP"
+    )
+    option = option.translate(trans_table).lower()
+    quality_map = {
+        "144": "144", "240": "240", "360": "360", "480": "480",
+        "720": "720", "1080": "1080", "2k": "1440", "1440": "1440",
+        "4k": "2160", "2160": "2160", "8k": "4320", "4320": "4320",
+        "144p": "144", "240p": "240", "360p": "360", "480p": "480",
+        "720p": "720", "1080p": "1080", "2kp": "1440", "1440p": "1440",
+        "4kp": "2160", "2160p": "2160", "8kp": "4320", "4320p": "4320",
+    }
+    if option in quality_map:
+        return quality_map[option]
+    if option in ["a", "f"]:
+        return option
+    return "720"
 
-    headers = {
-        "accept": "application/json",
-        "x-account": os.getenv("channeliotokenmain"),
+def send_to_channel(text):
+    payload = {
+        "requestId": f"desk-web-{int(time.time() * 1000)}",
+        "blocks": [
+            {"type": "text", "value": text}
+        ],
     }
+    res = requests.post(
+        POST_URL,
+        headers=HEADERS_POST,
+        data=json.dumps(payload),
+        timeout=30
+    )
+    res.raise_for_status()
 
-    res = requests.get(url, headers=headers, params=params)
+# ===== SaveNow 用ユーティリティ =====
+def build_savenow_initial_url(youtube_url, quality_opt, api_key=SAVENOW_API):
+    """
+    quality_opt:
+      - "a" -> mp3 (format=mp3)
+      - "f" -> omit format パラメータ（要求どおり 'f' は削除）
+      - numeric (例 "720") -> format=720
+    """
+    params = []
+    params.append("copyright=0")
+    # f は削除（つまり送らない）
+    if quality_opt == "a":
+        params.append("format=mp3")
+    elif quality_opt == "f":
+        # 指示通り 'f' パラメータは付けない（format自体を送らない）
+        pass
+    else:
+        # 数値系フォーマット
+        params.append(f"format={quote_plus(str(quality_opt))}")
+    params.append(f"url={quote_plus(youtube_url)}")
+    params.append(f"api={quote_plus(api_key)}")
+    return f"{SAVENOW_BASE}?{'&'.join(params)}"
 
-    if res.status_code != 200:
-        return jsonify({"error": res.text}), res.status_code
+def poll_progress_url(progress_url, max_attempts=30, interval=10):
+    """
+    progress_url をポーリングして download_url / alternative_download_urls を取得する。
+    interval は秒。max_attempts 回の試行を行う。
+    成功時は (True, response_json) を返し、失敗（タイムアウト）は (False, last_json) を返す。
+    """
+    attempt = 0
+    last_json = None
+    while attempt < max_attempts:
+        try:
+            r = requests.get(progress_url, timeout=30)
+            r.raise_for_status()
+            data = r.json()
+            last_json = data
+            # 成功条件: download_url または alternative_download_urls が存在すれば成功とみなす
+            if data.get("download_url") or data.get("alternative_download_urls"):
+                return True, data
+            # 進捗表示などを channel に流しても良い
+            progress = data.get("progress")
+            status_text = data.get("text") or data.get("message") or ""
+            # ログとして出力（開発時用）
+            print(f"poll {attempt+1}/{max_attempts} progress={progress} text={status_text}")
+        except Exception as e:
+            print(f"poll error (attempt {attempt+1}): {e}")
+            # 失敗しても次へ（短時間の一時的なエラーを許容）
+        attempt += 1
+        time.sleep(interval)
+    return False, last_json
 
-    return jsonify(res.json().get("managers", []))
+def process_savenow_flow(youtube_url, quality_opt):
+    """
+    SaveNow の初回 GET -> progress_url をポーリング -> 見つかったリンクを channel に送信
+    """
+    initial_url = build_savenow_initial_url(youtube_url, quality_opt)
+    send_to_channel(f"SaveNow にリクエストを送信しています: {initial_url}")
+    try:
+        r = requests.get(initial_url, timeout=30)
+        r.raise_for_status()
+        init_json = r.json()
+    except Exception as e:
+        send_to_channel(f"SaveNow へのリクエストに失敗しました: {e}")
+        return None
 
-@app.route("/cors-proxy", methods=["GET"])
-def corsproxy():
-    url = request.args.get("url")
-    if not url:
-        return "url パラメータが必要です", 400
+    # 初回レスポンスに progress_url がある想定
+    progress_url = init_json.get("progress_url") or init_json.get("progressUrl") or init_json.get("progress")
+    # 一部の API は progress_url をそのまま返し、別のものは result に入れている場合があるため、念のため探索
+    if not progress_url:
+        # もし最初のレスポンスで直接 download_url があるなら即時処理
+        if init_json.get("download_url") or init_json.get("alternative_download_urls"):
+            # 直接リンクが来ている場合
+            urls = []
+            if init_json.get("download_url"):
+                urls.append(init_json["download_url"])
+            for alt in init_json.get("alternative_download_urls", []):
+                if alt.get("url"):
+                    urls.append(alt["url"])
+            # channel に送る
+            for i, u in enumerate(urls, start=1):
+                send_to_channel(f"ダウンロードリンク #{i}: <link type=\"url\" value=\"{u}\">{u}</link>")
+            return urls
+        send_to_channel("progress_url が見つかりませんでした。レスポンスをログに出します。")
+        send_to_channel(f"初回レスポンス: {json.dumps(init_json, ensure_ascii=False)[:1000]}")
+        return None
 
-    if not url.startswith(("http://", "https://")):
-        return "http または https のURLのみ使用できます", 400
+    # progress_url があればポーリング
+    send_to_channel(f"進捗監視を開始します: {progress_url} （最大30回、10秒間隔）")
+    ok, poll_json = poll_progress_url(progress_url, max_attempts=30, interval=10)
+    if not ok:
+        send_to_channel("進捗監視がタイムアウトしました。最後のレスポンスをログに出します。")
+        send_to_channel(f"最後のレスポンス: {json.dumps(poll_json, ensure_ascii=False)[:1000]}")
+        return None
 
-    resp = requests.get(url, headers=request.headers, timeout=60)
+    # 成功した場合、download_url と alternative_download_urls を収集して channel に送信
+    urls = []
+    if poll_json.get("download_url"):
+        urls.append(poll_json["download_url"])
+    for alt in poll_json.get("alternative_download_urls", []):
+        if alt.get("url"):
+            urls.append(alt["url"])
 
-    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"
+    # 重複排除
+    seen = set()
+    unique_urls = []
+    for u in urls:
+        if u and u not in seen:
+            seen.add(u)
+            unique_urls.append(u)
 
-    if "Content-Type" in resp.headers:
-        response.headers["Content-Type"] = resp.headers["Content-Type"]
+    if not unique_urls:
+        send_to_channel("ダウンロードリンクが見つかりませんでした（成功レスポンスでしたが link が空）。")
+        return None
 
-    return response
+    for i, u in enumerate(unique_urls, start=1):
+        send_to_channel(f"ダウンロードリンク #{i}: <link type=\"url\" value=\"{u}\">{u}</link>")
 
-@app.route("/cors-proxy", methods=["POST", "PATCH"])
-def corsproxy_post():
-    url = request.args.get("url")
-    if not url:
-        return "url パラメータが必要です", 400
+    return unique_urls
 
-    if not url.startswith(("http://", "https://")):
-        return "http または https のURLのみ使用できます", 400
+# ===== Main (既存フローの中に組み込む想定) =====
+def main():
+    processed_messages = set()
+    while True:
+        try:
+            res = requests.get(
+                GET_URL,
+                headers=HEADERS_GET,
+                params=PARAMS,
+                timeout=30,
+            )
+            res.raise_for_status()
 
-    resp = requests.request(
-        method=request.method,
-        url=url,
-        headers=request.headers,
-        data=request.data,
-        timeout=60,
-    )
+            messages = res.json().get("messages", [])
+            latest_msg = None
+            latest_time = None
 
-    headers = {
-        "Access-Control-Allow-Origin": "*",
-        "Access-Control-Allow-Headers": "*",
-        "Access-Control-Allow-Methods": "GET, POST, PATCH, OPTIONS",
-    }
+            for msg in messages:
+                msg_id = msg.get("id")
+                plain_text = msg.get("plainText")
+                updated_at = msg.get("updatedAt")
+
+                if not plain_text or updated_at is None:
+                    continue
+                if msg_id in processed_messages:
+                    continue
+
+                t = parse_updated_at(updated_at)
+                if not t:
+                    continue
+
+                if latest_time is None or t > latest_time:
+                    latest_time = t
+                    latest_msg = msg
+
+            if not latest_msg:
+                time.sleep(10)
+                continue
+
+            video_id, option = extract_youtube_and_option(latest_msg["plainText"])
+            if not video_id:
+                # YouTube ID が取れないメッセージはスキップ
+                processed_messages.add(latest_msg["id"])
+                time.sleep(10)
+                continue
+
+            valid_options = ["144", "240", "360", "480", "720", "1080", "1440", "2160", "4320", "4k", "8k", "2k", "a", "f"]
+
+            if option is None:
+                send_to_channel(
+                    "画質が指定されていないため、デフォルトの720pを使用します。\n"
+                    "指定する場合は、動画URLの後にスペースを入れて、"
+                    "144/240/360/480/720/1080/2k(または1440)/4k(または2160)/8k(または4320)/a(音声/mp3)/f(最大画質・formatパラメータ未送信) のいずれかを入力してください。"
+                )
+            else:
+                opt_lower = option.lower()
+                if opt_lower not in valid_options:
+                    send_to_channel(
+                        "不明な画質、または予期しない形式でした。\n"
+                        "対応画質: 144, 240, 360, 480, 720, 1080, "
+                        "2k(1440), 4k(2160), 8k(4320), a(音声/mp3), f(最大画質)\n"
+                        "動画のURLの後にスペースを入れて指定してください。\n"
+                        "例: https://youtu.be/abcdEFGHijk 1080"
+                    )
+                    processed_messages.add(latest_msg["id"])
+                    continue
+
+            quality_opt = normalize_quality(option)
+            youtube_url = f"https://www.youtube.com/watch?v={video_id}"
+            send_to_channel(f"{video_id} のダウンロード（SaveNow）を開始します。")
+
+            # SaveNow フローを使う（要求にあった処理）
+            urls = process_savenow_flow(youtube_url, quality_opt)
+
+            if urls:
+                send_to_channel("リンクを取得しました。必要ならファイルをアップロードする処理を追加してください。")
+            else:
+                send_to_channel("リンクが取得できませんでした。別の方法を試すか、ログを確認してください。")
+
+            processed_messages.add(latest_msg["id"])
+            print(f"処理完了: {video_id}")
+
+        except Exception as e:
+            print("エラー:", e)
+            import traceback
+            traceback.print_exc()
 
-    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()