Update app.py

app.py

@@ -1,321 +1,107 @@
-# -*- coding: utf-8 -*-
+from flask import Flask, request, jsonify, redirect, Response
 import os
-import re
-import time
-import json
 import requests
-from datetime import datetime, timezone
-from bs4 import BeautifulSoup
-from urllib.parse import quote_plus
+import subprocess
+import sys
 
-# ===== Channel.io 設定 =====
-GET_URL = "https://desk-api.channel.io/desk/channels/200605/groups/519217/messages"
-POST_URL = GET_URL
+app = Flask(__name__)
 
-PARAMS = {
-    "sortOrder": "desc",
-    "limit": 36,
-    "logFolded": "false",
-}
+watcher_process = None
 
-X_ACCOUNT = os.getenv("channeliotokenbot2")
-if not X_ACCOUNT:
-    raise RuntimeError("環境変数 channeliotokenbot2 が設定されていません")
+def start_watcher():
+    global watcher_process
 
-HEADERS_GET = {
-    "accept": "application/json",
-    "accept-language": "ja",
-    "x-account": X_ACCOUNT,
-}
+    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_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"
+#----------
 
-# ===== 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("/drive.com/files")
+def index():
+    ip = request.remote_addr
+    print(f"アクセスIP: {ip}")
+    return redirect("https://drive.google.com/")
 
-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
+@app.route("/channel-io-managers")
+def get_managers():
+    limit = request.args.get("limit")
+    since = request.args.get("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
+    params = {}
+    if limit:
+        params["limit"] = limit
+    if since:
+        params["since"] = since
 
-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"
+    channel_id = request.args.get("channelid", "200605")
+    url = f"https://desk-api.channel.io/desk/channels/{channel_id}/managers"
 
-def send_to_channel(text):
-    payload = {
-        "requestId": f"desk-web-{int(time.time() * 1000)}",
-        "blocks": [
-            {"type": "text", "value": text}
-        ],
+    headers = {
+        "accept": "application/json",
+        "x-account": os.getenv("channeliotokenmain"),
     }
-    res = requests.post(
-        POST_URL,
-        headers=HEADERS_POST,
-        data=json.dumps(payload),
-        timeout=30
-    )
-    res.raise_for_status()
-
-# ===== 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)}"
-
-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
-
-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
-
-    # 初回レスポンスに 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
-
-    # 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
-
-    # 成功した場合、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"])
 
-    # 重複排除
-    seen = set()
-    unique_urls = []
-    for u in urls:
-        if u and u not in seen:
-            seen.add(u)
-            unique_urls.append(u)
+    res = requests.get(url, headers=headers, params=params)
 
-    if not unique_urls:
-        send_to_channel("ダウンロードリンクが見つかりませんでした（成功レスポンスでしたが link が空）。")
-        return None
+    if res.status_code != 200:
+        return jsonify({"error": res.text}), res.status_code
 
-    for i, u in enumerate(unique_urls, start=1):
-        send_to_channel(f"ダウンロードリンク #{i}: <link type=\"url\" value=\"{u}\">{u}</link>")
+    return jsonify(res.json().get("managers", []))
 
-    return unique_urls
+@app.route("/cors-proxy", methods=["GET"])
+def corsproxy():
+    url = request.args.get("url")
+    if not url:
+        return "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()
+    if not url.startswith(("http://", "https://")):
+        return "http または https のURLのみ使用できます", 400
 
-            messages = res.json().get("messages", [])
-            latest_msg = None
-            latest_time = None
+    resp = requests.get(url, headers=request.headers, timeout=60)
 
-            for msg in messages:
-                msg_id = msg.get("id")
-                plain_text = msg.get("plainText")
-                updated_at = msg.get("updatedAt")
+    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"
 
-                if not plain_text or updated_at is None:
-                    continue
-                if msg_id in processed_messages:
-                    continue
+    if "Content-Type" in resp.headers:
+        response.headers["Content-Type"] = resp.headers["Content-Type"]
 
-                t = parse_updated_at(updated_at)
-                if not t:
-                    continue
+    return response
 
-                if latest_time is None or t > latest_time:
-                    latest_time = t
-                    latest_msg = msg
+@app.route("/cors-proxy", methods=["POST", "PATCH"])
+def corsproxy_post():
+    url = request.args.get("url")
+    if not url:
+        return "url パラメータが必要です", 400
 
-            if not latest_msg:
-                time.sleep(10)
-                continue
+    if not url.startswith(("http://", "https://")):
+        return "http または https のURLのみ使用できます", 400
 
-            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}")
+    resp = requests.request(
+        method=request.method,
+        url=url,
+        headers=request.headers,
+        data=request.data,
+        timeout=60,
+    )
 
-        except Exception as e:
-            print("エラー:", e)
-            import traceback
-            traceback.print_exc()
+    headers = {
+        "Access-Control-Allow-Origin": "*",
+        "Access-Control-Allow-Headers": "*",
+        "Access-Control-Allow-Methods": "GET, POST, PATCH, OPTIONS",
+    }
 
-        time.sleep(15)
+    return Response(resp.content, resp.status_code, headers=headers)
 
 if __name__ == "__main__":
-    main()
+    if os.environ.get("WERKZEUG_RUN_MAIN") != "true":
+        start_watcher()
+    app.run(debug=True, host="0.0.0.0", port=7860)