Refactor: Split app.py into modular files

app.py

@@ -1,215 +1,14 @@
-import json
 import os
 import re
-import subprocess
 import time
-import uuid
-from flask import Flask, request, make_response, send_file
-
-app = Flask(__name__)
-
-# ========================================
-# Konfigurasi temp file
-# ========================================
-TEMP_DIR = "/tmp/ytdlp_cache"
-EXPIRE_SECONDS = 6 * 3600  # 6 jam
-os.makedirs(TEMP_DIR, exist_ok=True)
-
-
-def jsonify(data, status=200):
-    """Custom jsonify dengan indent 2 dan ensure_ascii False."""
-    response = make_response(json.dumps(data, indent=2, ensure_ascii=False), status)
-    response.headers["Content-Type"] = "application/json; charset=utf-8"
-    return response
+from flask import Flask, request, send_file
 
+from config import EXPIRE_SECONDS, API_LIST
+from utils import jsonify, validate_tiktok_url
+from cache import video_path, save_meta, load_meta, cleanup_expired
+from extractor import extract_media_info, download_video
 
-# ========================================
-# Filesystem cache — aman lintas worker gunicorn
-# ========================================
-
-def meta_path(file_id):
-    return os.path.join(TEMP_DIR, f"{file_id}.json")
-
-def video_path(file_id):
-    return os.path.join(TEMP_DIR, f"{file_id}.mp4")
-
-def save_meta(file_id, filename, expires_at):
-    """Simpan metadata file ke disk."""
-    with open(meta_path(file_id), "w") as f:
-        json.dump({"filename": filename, "expires_at": expires_at}, f)
-
-def load_meta(file_id):
-    """Baca metadata file dari disk. Return None jika tidak ada."""
-    p = meta_path(file_id)
-    if not os.path.exists(p):
-        return None
-    try:
-        with open(p) as f:
-            return json.load(f)
-    except Exception:
-        return None
-
-def cleanup_expired():
-    """Hapus file temp (video + meta) yang sudah expired."""
-    now = time.time()
-    for fname in os.listdir(TEMP_DIR):
-        if not fname.endswith(".json"):
-            continue
-        fid = fname[:-5]
-        meta = load_meta(fid)
-        if meta and meta["expires_at"] < now:
-            try:
-                vp = video_path(fid)
-                if os.path.exists(vp):
-                    os.remove(vp)
-                os.remove(meta_path(fid))
-            except Exception:
-                pass
-
-
-# ========================================
-# Daftar API yang tersedia
-# ========================================
-API_LIST = {
-    "name": "yt-dlp API",
-    "version": "1.0.0",
-    "description": "API untuk mengunduh dan mengekstrak informasi media dari berbagai platform menggunakan yt-dlp.",
-    "endpoints": {
-        "/api/tiktok": {
-            "method": "GET",
-            "description": (
-                "Mengekstrak info video TikTok sekaligus mendownload videonya ke server. "
-                "Mengembalikan download_url yang siap pakai selama 6 jam."
-            ),
-            "parameters": {
-                "url": {
-                    "type": "string",
-                    "required": True,
-                    "description": "URL video TikTok."
-                }
-            },
-            "example": "/api/tiktok?url=https://www.tiktok.com/@user/video/1234567890"
-        },
-        "/api/file/<file_id>": {
-            "method": "GET",
-            "description": "Akses file video yang sudah didownload. Berlaku selama 6 jam.",
-            "parameters": {
-                "file_id": {
-                    "type": "string",
-                    "required": True,
-                    "description": "ID file dari response /api/tiktok."
-                }
-            },
-            "example": "/api/file/uuid-xxxxx"
-        }
-    }
-}
-
-
-def validate_tiktok_url(url):
-    """Validasi apakah URL adalah URL TikTok yang valid."""
-    patterns = [
-        r'https?://(www\.)?tiktok\.com/@[\w.]+/video/\d+/?',
-        r'https?://(vm|vt)\.tiktok\.com/[\w]+/?',
-        r'https?://(www\.)?tiktok\.com/t/[\w]+/?',
-    ]
-    return any(re.match(pattern, url) for pattern in patterns)
-
-
-def extract_media_info(url, platform="tiktok"):
-    """
-    Menggunakan yt-dlp --dump-json untuk mengekstrak metadata video.
-    """
-    try:
-        cmd = [
-            "yt-dlp",
-            "--dump-json",
-            "--no-warnings",
-            "--no-playlist",
-            "--skip-download",
-            url
-        ]
-
-        result = subprocess.run(cmd, capture_output=True, text=True, timeout=60)
-
-        if result.returncode != 0:
-            error_msg = result.stderr.strip() if result.stderr else "Unknown error"
-            return {"error": True, "message": f"yt-dlp error: {error_msg}"}
-
-        data = json.loads(result.stdout)
-
-        formats = data.get("formats", [])
-        best_video = None
-        best_audio = None
-
-        for fmt in formats:
-            if fmt.get("vcodec", "none") != "none" and fmt.get("acodec", "none") != "none":
-                if best_video is None or (fmt.get("height", 0) or 0) > (best_video.get("height", 0) or 0):
-                    best_video = fmt
-            elif fmt.get("acodec", "none") != "none" and fmt.get("vcodec", "none") == "none":
-                if best_audio is None:
-                    best_audio = fmt
-
-        return {
-            "status": "success",
-            "platform": platform,
-            "data": {
-                "id": data.get("id"),
-                "title": data.get("title", "Tidak ada judul"),
-                "description": data.get("description", ""),
-                "duration": data.get("duration"),
-                "uploader": data.get("uploader", data.get("creator", "Unknown")),
-                "uploader_url": data.get("uploader_url", ""),
-                "thumbnail": data.get("thumbnail", ""),
-                "view_count": data.get("view_count"),
-                "like_count": data.get("like_count"),
-                "comment_count": data.get("comment_count"),
-                "upload_date": data.get("upload_date"),
-                "resolution": f'{best_video.get("width", "?")}x{best_video.get("height", "?")}' if best_video else None,
-            }
-        }
-
-    except subprocess.TimeoutExpired:
-        return {"error": True, "message": "Request timeout: yt-dlp membutuhkan waktu terlalu lama."}
-    except json.JSONDecodeError:
-        return {"error": True, "message": "Gagal memparse response dari yt-dlp."}
-    except Exception as e:
-        return {"error": True, "message": f"Internal error: {str(e)}"}
-
-
-def download_video(url):
-    """
-    Download video TikTok ke temp file.
-    Return: (file_id, file_path, filename) atau raise Exception.
-    """
-    file_id = str(uuid.uuid4())
-    output_path = os.path.join(TEMP_DIR, f"{file_id}.mp4")
-
-    cmd = [
-        "yt-dlp",
-        "--no-warnings",
-        "--no-playlist",
-        "-f", "bestvideo+bestaudio/best",
-        "--merge-output-format", "mp4",
-        "-o", output_path,
-        url
-    ]
-
-    result = subprocess.run(cmd, capture_output=True, timeout=180)
-
-    if result.returncode != 0:
-        error_msg = result.stderr.decode(errors="ignore").strip()
-        raise Exception(f"yt-dlp error: {error_msg}")
-
-    if not os.path.exists(output_path):
-        raise Exception("File tidak berhasil dibuat oleh yt-dlp.")
-
-    return file_id, output_path
-
-
-# ========================================
-# Routes
-# ========================================
+app = Flask(__name__)
 
 @app.route("/")
 def index():

cache.py

@@ -0,0 +1,43 @@
+import os
+import json
+import time
+from config import TEMP_DIR
+
+def meta_path(file_id):
+    return os.path.join(TEMP_DIR, f"{file_id}.json")
+
+def video_path(file_id):
+    return os.path.join(TEMP_DIR, f"{file_id}.mp4")
+
+def save_meta(file_id, filename, expires_at):
+    """Simpan metadata file ke disk."""
+    with open(meta_path(file_id), "w") as f:
+        json.dump({"filename": filename, "expires_at": expires_at}, f)
+
+def load_meta(file_id):
+    """Baca metadata file dari disk. Return None jika tidak ada."""
+    p = meta_path(file_id)
+    if not os.path.exists(p):
+        return None
+    try:
+        with open(p) as f:
+            return json.load(f)
+    except Exception:
+        return None
+
+def cleanup_expired():
+    """Hapus file temp (video + meta) yang sudah expired."""
+    now = time.time()
+    for fname in os.listdir(TEMP_DIR):
+        if not fname.endswith(".json"):
+            continue
+        fid = fname[:-5]
+        meta = load_meta(fid)
+        if meta and meta["expires_at"] < now:
+            try:
+                vp = video_path(fid)
+                if os.path.exists(vp):
+                    os.remove(vp)
+                os.remove(meta_path(fid))
+            except Exception:
+                pass

config.py

@@ -0,0 +1,40 @@
+import os
+
+TEMP_DIR = "/tmp/ytdlp_cache"
+EXPIRE_SECONDS = 6 * 3600  # 6 jam
+os.makedirs(TEMP_DIR, exist_ok=True)
+
+API_LIST = {
+    "name": "yt-dlp API",
+    "version": "1.0.0",
+    "description": "API untuk mengunduh dan mengekstrak informasi media dari berbagai platform menggunakan yt-dlp.",
+    "endpoints": {
+        "/api/tiktok": {
+            "method": "GET",
+            "description": (
+                "Mengekstrak info video TikTok sekaligus mendownload videonya ke server. "
+                "Mengembalikan download_url yang siap pakai selama 6 jam."
+            ),
+            "parameters": {
+                "url": {
+                    "type": "string",
+                    "required": True,
+                    "description": "URL video TikTok."
+                }
+            },
+            "example": "/api/tiktok?url=https://www.tiktok.com/@user/video/1234567890"
+        },
+        "/api/file/<file_id>": {
+            "method": "GET",
+            "description": "Akses file video yang sudah didownload. Berlaku selama 6 jam.",
+            "parameters": {
+                "file_id": {
+                    "type": "string",
+                    "required": True,
+                    "description": "ID file dari response /api/tiktok."
+                }
+            },
+            "example": "/api/file/uuid-xxxxx"
+        }
+    }
+}

extractor.py

@@ -0,0 +1,94 @@
+import json
+import os
+import subprocess
+import uuid
+from config import TEMP_DIR
+
+def extract_media_info(url, platform="tiktok"):
+    """
+    Menggunakan yt-dlp --dump-json untuk mengekstrak metadata video.
+    """
+    try:
+        cmd = [
+            "yt-dlp",
+            "--dump-json",
+            "--no-warnings",
+            "--no-playlist",
+            "--skip-download",
+            url
+        ]
+
+        result = subprocess.run(cmd, capture_output=True, text=True, timeout=60)
+
+        if result.returncode != 0:
+            error_msg = result.stderr.strip() if result.stderr else "Unknown error"
+            return {"error": True, "message": f"yt-dlp error: {error_msg}"}
+
+        data = json.loads(result.stdout)
+
+        formats = data.get("formats", [])
+        best_video = None
+        best_audio = None
+
+        for fmt in formats:
+            if fmt.get("vcodec", "none") != "none" and fmt.get("acodec", "none") != "none":
+                if best_video is None or (fmt.get("height", 0) or 0) > (best_video.get("height", 0) or 0):
+                    best_video = fmt
+            elif fmt.get("acodec", "none") != "none" and fmt.get("vcodec", "none") == "none":
+                if best_audio is None:
+                    best_audio = fmt
+
+        return {
+            "status": "success",
+            "platform": platform,
+            "data": {
+                "id": data.get("id"),
+                "title": data.get("title", "Tidak ada judul"),
+                "description": data.get("description", ""),
+                "duration": data.get("duration"),
+                "uploader": data.get("uploader", data.get("creator", "Unknown")),
+                "uploader_url": data.get("uploader_url", ""),
+                "thumbnail": data.get("thumbnail", ""),
+                "view_count": data.get("view_count"),
+                "like_count": data.get("like_count"),
+                "comment_count": data.get("comment_count"),
+                "upload_date": data.get("upload_date"),
+                "resolution": f'{best_video.get("width", "?")}x{best_video.get("height", "?")}' if best_video else None,
+            }
+        }
+
+    except subprocess.TimeoutExpired:
+        return {"error": True, "message": "Request timeout: yt-dlp membutuhkan waktu terlalu lama."}
+    except json.JSONDecodeError:
+        return {"error": True, "message": "Gagal memparse response dari yt-dlp."}
+    except Exception as e:
+        return {"error": True, "message": f"Internal error: {str(e)}"}
+
+def download_video(url):
+    """
+    Download video TikTok ke temp file.
+    Return: (file_id, output_path) atau raise Exception.
+    """
+    file_id = str(uuid.uuid4())
+    output_path = os.path.join(TEMP_DIR, f"{file_id}.mp4")
+
+    cmd = [
+        "yt-dlp",
+        "--no-warnings",
+        "--no-playlist",
+        "-f", "bestvideo+bestaudio/best",
+        "--merge-output-format", "mp4",
+        "-o", output_path,
+        url
+    ]
+
+    result = subprocess.run(cmd, capture_output=True, timeout=180)
+
+    if result.returncode != 0:
+        error_msg = result.stderr.decode(errors="ignore").strip()
+        raise Exception(f"yt-dlp error: {error_msg}")
+
+    if not os.path.exists(output_path):
+        raise Exception("File tidak berhasil dibuat oleh yt-dlp.")
+
+    return file_id, output_path

utils.py

@@ -0,0 +1,18 @@
+import json
+import re
+from flask import make_response
+
+def jsonify(data, status=200):
+    """Custom jsonify dengan indent 2 dan ensure_ascii False."""
+    response = make_response(json.dumps(data, indent=2, ensure_ascii=False), status)
+    response.headers["Content-Type"] = "application/json; charset=utf-8"
+    return response
+
+def validate_tiktok_url(url):
+    """Validasi apakah URL adalah URL TikTok yang valid."""
+    patterns = [
+        r'https?://(www\.)?tiktok\.com/@[\w.]+/video/\d+/?',
+        r'https?://(vm|vt)\.tiktok\.com/[\w]+/?',
+        r'https?://(www\.)?tiktok\.com/t/[\w]+/?',
+    ]
+    return any(re.match(pattern, url) for pattern in patterns)