Fix: Support TikTok photo posts by resolving redirects and normalizing /photo/ to /video/

app.py

@@ -4,7 +4,7 @@ import time
 from flask import Flask, request, send_file
 
 from config import EXPIRE_SECONDS, API_LIST
-from utils import jsonify, validate_tiktok_url
+from utils import jsonify, validate_tiktok_url, resolve_url
 from cache import video_path, save_meta, load_meta, cleanup_expired
 from extractor import extract_media_info, download_video
 
@@ -23,6 +23,7 @@ def tiktok():
     cleanup_expired()
 
     url = request.args.get("url")
+    debug_mode = request.args.get("debug")
 
     if not url:
         return jsonify({
@@ -31,6 +32,13 @@ def tiktok():
             "example": "/api/tiktok?url=https://www.tiktok.com/@user/video/1234567890"
         }), 400
 
+    # 1. Resolve redirect URL pendek (seperti vt.tiktok.com)
+    url = resolve_url(url)
+
+    # 2. Normalisasi URL photo ke video agar didukung oleh yt-dlp
+    if "/photo/" in url:
+        url = url.replace("/photo/", "/video/")
+
     if not validate_tiktok_url(url):
         return jsonify({
             "status": "error",
@@ -43,19 +51,23 @@ def tiktok():
             ]
         }), 400
 
-    # 1. Ambil metadata
+    # 3. Ambil metadata
     info = extract_media_info(url, platform="tiktok")
     if info.get("error"):
         return jsonify({"status": "error", "message": info["message"]}), 500
 
-    # 2. Download video ke temp file
+    raw = info["raw_data"]
+
+    # Endpoint debug untuk melihat raw data yt-dlp langsung
+    if debug_mode in ["1", "true"]:
+        return jsonify(raw), 200
+
+    # 4. Download video/slideshow mp4 ke temp file
     try:
         file_id, file_path = download_video(url)
     except Exception as e:
         return jsonify({"status": "error", "message": str(e)}), 500
 
-    raw = info["raw_data"]
-
     # Deteksi author & nickname
     author = raw.get("uploader_id")
     if not author:
@@ -83,9 +95,14 @@ def tiktok():
                 audio_url = fmt.get("url")
                 break
 
-    # Deteksi slideshow / photo
-    images = raw.get("images", [])
-    photo = images if images else False
+    # Deteksi slideshow / photo dari thumbnails
+    photo = False
+    thumbnails = raw.get("thumbnails", [])
+    if thumbnails:
+        photo_urls = [t.get("url") for t in thumbnails if t.get("url")]
+        # Jika ada lebih dari 1 thumbnail, biasanya itu adalah slide gambarnya
+        if len(photo_urls) > 1:
+            photo = photo_urls
 
     # Konversi durasi ke format M:SS atau H:MM:SS
     duration_secs = raw.get("duration")
@@ -104,14 +121,14 @@ def tiktok():
         except Exception:
             duration_str = str(duration_secs)
 
-    # 3. Simpan metadata ke disk (filesystem — aman lintas worker)
+    # 5. Simpan metadata ke disk (filesystem — aman lintas worker)
     title = raw.get("title") or raw.get("description") or "video"
     safe_title = re.sub(r'[^\w\s-]', '', title).strip()[:60] or "video"
     filename = f"{safe_title}.mp4"
     expires_at = time.time() + EXPIRE_SECONDS
     save_meta(file_id, filename, expires_at)
 
-    # 4. Susun response
+    # 6. Susun response
     base_url = request.host_url.rstrip("/")
     video_url = f"{base_url}/api/file/{file_id}"
 

utils.py

@@ -1,5 +1,6 @@
 import json
 import re
+import urllib.request
 from flask import make_response
 
 def jsonify(data, status=200):
@@ -9,10 +10,22 @@ def jsonify(data, status=200):
     return response
 
 def validate_tiktok_url(url):
-    """Validasi apakah URL adalah URL TikTok yang valid."""
+    """Validasi apakah URL adalah URL TikTok yang valid (video atau photo)."""
     patterns = [
-        r'https?://(www\.)?tiktok\.com/@[\w.]+/video/\d+/?',
+        r'https?://(www\.)?tiktok\.com/@[\w.]+/(video|photo)/\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 resolve_url(url):
+    """Mengikuti redirect URL untuk mendapatkan URL final (misal dari vt.tiktok ke tiktok.com/@user/video/id)."""
+    try:
+        req = urllib.request.Request(
+            url, 
+            headers={'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36'}
+        )
+        with urllib.request.urlopen(req, timeout=15) as response:
+            return response.geturl()
+    except Exception:
+        return url