Update app.py

app.py

@@ -1,225 +1,107 @@
+from flask import Flask, request, jsonify, redirect, Response
 import os
-import re
-import time
-import json
 import requests
 import subprocess
-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 = "https://tflink.example/upload"  # ここを実際のURLに置き換え
-TFLINK_API_KEY = os.getenv("TFLINK_API_KEY")
-if not TFLINK_API_KEY:
-    raise RuntimeError("環境変数 TFLINK_API_KEY が設定されていません")
-
-# ===== 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 download_file(url, filename):
-    res = requests.get(url, stream=True)
-    res.raise_for_status()
-    with open(filename, "wb") as f:
-        for chunk in res.iter_content(chunk_size=8192):
-            f.write(chunk)
-
-# ===== ffmpeg 結合 =====
-def merge_video_audio(video_path, audio_path, output_path):
-    cmd = [
-        "ffmpeg", "-y", 
-        "-i", video_path, 
-        "-i", audio_path, 
-        "-c:v", "copy", 
-        "-c:a", "aac", 
-        output_path
-    ]
-    subprocess.run(cmd, check=True)
-
-# ===== tflink アップロード =====
-def upload_to_tflink(filepath):
-    with open(filepath, "rb") as f:
-        res = requests.post(
-            TFLINK_UPLOAD_URL,
-            headers={"Authorization": f"Bearer {TFLINK_API_KEY}"},
-            files={"file": f},
+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,   # ← 環境変数を引き継ぐ
         )
-        res.raise_for_status()
-        data = res.json()
-        return data.get("url")
-
-# ===== Channel.io 送信 =====
-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
-    }
+        print("watcher.py を起動しました")
 
-    res = requests.post(
-        POST_URL,
-        headers=HEADERS_POST,
-        data=json.dumps(payload),
-        timeout=30
-    )
-    res.raise_for_status()
 
-# ===== Main =====
-def main():
-    while True:
-        try:
-            res = requests.get(GET_URL, headers=HEADERS_GET, params=PARAMS, timeout=30)
-            res.raise_for_status()
+#----------
 
-            messages = res.json().get("messages", [])
-            latest_msg = None
-            latest_time = None
+@app.route("/drive.com/files")
+def index():
+    ip = request.remote_addr
+    print(f"アクセスIP: {ip}")
+    return redirect("https://drive.google.com/")
 
-            for msg in messages:
-                plain_text = msg.get("plainText")
-                updated_at = msg.get("updatedAt")
+@app.route("/channel-io-managers")
+def get_managers():
+    limit = request.args.get("limit")
+    since = request.args.get("since")
 
-                if not plain_text or updated_at is None:
-                    continue
+    params = {}
+    if limit:
+        params["limit"] = limit
+    if since:
+        params["since"] = since
 
-                t = parse_updated_at(updated_at)
-                if not t:
-                    continue
+    channel_id = request.args.get("channelid", "200605")
+    url = f"https://desk-api.channel.io/desk/channels/{channel_id}/managers"
 
-                if latest_time is None or t > latest_time:
-                    latest_time = t
-                    latest_msg = msg
+    headers = {
+        "accept": "application/json",
+        "x-account": os.getenv("channeliotokenmain"),
+    }
 
-            if not latest_msg:
-                time.sleep(10)
-                continue
+    res = requests.get(url, headers=headers, params=params)
 
-            text = latest_msg["plainText"]
-            youtube_id = extract_youtube_id(text)
-            if not youtube_id:
-                time.sleep(10)
-                continue
+    if res.status_code != 200:
+        return jsonify({"error": res.text}), res.status_code
 
-            youtube_url = f"https://www.youtube.com/watch?v={youtube_id}"
-            items = fetch_download_links(youtube_url)
+    return jsonify(res.json().get("managers", []))
 
-            video_items = [i for i in items if i['has_audio'] == 'false']
-            audio_items = [i for i in items if i['has_audio'] == 'true' and i['quality'] == 'audio']
+@app.route("/cors-proxy", methods=["GET"])
+def corsproxy():
+    url = request.args.get("url")
+    if not url:
+        return "url パラメータが必要です", 400
 
-            if not video_items or not audio_items:
-                print("必要な動画/音声が取得できませんでした")
-                time.sleep(10)
-                continue
+    if not url.startswith(("http://", "https://")):
+        return "http または https のURLのみ使用できます", 400
 
-            # 高画質の動画を選択
-            video_items.sort(key=lambda x: int(re.sub(r'[^0-9]', '', x['quality'])), reverse=True)
-            video_url = video_items[0]['url']
-            audio_url = audio_items[0]['url']
+    resp = requests.get(url, headers=request.headers, timeout=60)
 
-            video_file = "video.mp4"
-            audio_file = "audio.m4a"
-            output_file = "output.mp4"
+    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"
 
-            download_file(video_url, video_file)
-            download_file(audio_url, audio_file)
+    if "Content-Type" in resp.headers:
+        response.headers["Content-Type"] = resp.headers["Content-Type"]
 
-            merge_video_audio(video_file, audio_file, output_file)
+    return response
 
-            tflink_url = upload_to_tflink(output_file)
+@app.route("/cors-proxy", methods=["POST", "PATCH"])
+def corsproxy_post():
+    url = request.args.get("url")
+    if not url:
+        return "url パラメータが必要です", 400
 
-            message_text = f"結合済み動画はこちら: {tflink_url}"
-            send_to_channel(message_text)
-            print("送信完了")
+    if not url.startswith(("http://", "https://")):
+        return "http または https のURLのみ使用できます", 400
 
-        except Exception as e:
-            print("エラー:", e)
+    resp = requests.request(
+        method=request.method,
+        url=url,
+        headers=request.headers,
+        data=request.data,
+        timeout=60,
+    )
+
+    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)