Compare commits
13
Commits
4efcf1f0ea
..
main
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
7d68ff5620 | ||
|
|
8b9bdeb56a | ||
|
|
029837d0f1 | ||
|
|
8a55fd91a4 | ||
|
|
5cec284cd6 | ||
|
|
adfbbfcea7 | ||
|
|
09c706cc61 | ||
|
|
93828c7253 | ||
|
|
5470d74921 | ||
|
|
bbb8bed389 | ||
|
|
e6f543078b | ||
|
|
2769b11918 | ||
|
|
8d74d04b43 |
No files matched your search
@@ -15,7 +15,7 @@
|
||||
};
|
||||
|
||||
scraper = pkgs.stdenv.mkDerivation {
|
||||
name = "scraper-0.1.0";
|
||||
name = "scraper-0.1.1";
|
||||
src = ./.;
|
||||
|
||||
nativeBuildInputs = with pkgs; [
|
||||
|
||||
@@ -42,6 +42,13 @@ pub async fn download_youtube(url: &str, quality: &str) -> Result<DownloadResult
|
||||
DownloaderRepository::download_youtube(url, quality).await
|
||||
}
|
||||
|
||||
pub async fn download_youtube_merge(
|
||||
url: &str,
|
||||
quality: &str,
|
||||
) -> Result<DownloadResult, ScrapingError> {
|
||||
DownloaderRepository::fetch_youtube_merge(url, quality).await
|
||||
}
|
||||
|
||||
pub async fn download_youtube_mp3(url: &str) -> Result<DownloadResult, ScrapingError> {
|
||||
DownloaderRepository::download_youtube_mp3(url).await
|
||||
}
|
||||
|
||||
@@ -8,4 +8,3 @@ pub mod proxy;
|
||||
pub mod search;
|
||||
pub mod stalk;
|
||||
pub mod tools;
|
||||
pub mod weebs;
|
||||
@@ -17,3 +17,10 @@ pub async fn youtube(username: &str) -> Result<Value, ScrapingError> {
|
||||
.await
|
||||
.map_err(ScrapingError::Http)
|
||||
}
|
||||
|
||||
/// Twitter/X user stalk.
|
||||
pub async fn twitter(username: &str) -> Result<Value, ScrapingError> {
|
||||
Repo::fetch_twitter_stalk(username)
|
||||
.await
|
||||
.map_err(ScrapingError::Http)
|
||||
}
|
||||
@@ -1,35 +0,0 @@
|
||||
//! Application use-cases for weebs (anime/manga info, waifu, whatanime).
|
||||
|
||||
use crate::domain::error::ScrapingError;
|
||||
use crate::infrastructure::repository::weebs as Repo;
|
||||
use serde_json::Value;
|
||||
|
||||
pub async fn anime_info(query: &str) -> Result<Value, ScrapingError> {
|
||||
Repo::fetch_anime_info(query)
|
||||
.await
|
||||
.map_err(ScrapingError::Http)
|
||||
}
|
||||
|
||||
pub async fn manga_info(query: &str) -> Result<Value, ScrapingError> {
|
||||
Repo::fetch_manga_info(query)
|
||||
.await
|
||||
.map_err(ScrapingError::Http)
|
||||
}
|
||||
|
||||
pub async fn whatanime(url: &str) -> Result<Value, ScrapingError> {
|
||||
Repo::fetch_whatanime(url)
|
||||
.await
|
||||
.map_err(ScrapingError::Http)
|
||||
}
|
||||
|
||||
pub async fn sfw_waifu(tag: &str) -> Result<Value, ScrapingError> {
|
||||
Repo::fetch_sfw_waifu(tag)
|
||||
.await
|
||||
.map_err(ScrapingError::Http)
|
||||
}
|
||||
|
||||
pub async fn nsfw_waifu(tag: &str) -> Result<Value, ScrapingError> {
|
||||
Repo::fetch_nsfw_waifu(tag)
|
||||
.await
|
||||
.map_err(ScrapingError::Http)
|
||||
}
|
||||
@@ -5,6 +5,8 @@
|
||||
|
||||
use std::borrow::Cow;
|
||||
use std::collections::HashMap;
|
||||
use std::path::{Path, PathBuf};
|
||||
use std::process::{Command, Stdio};
|
||||
use std::time::Duration;
|
||||
|
||||
use aes::cipher::{BlockDecrypt, KeyInit};
|
||||
@@ -511,6 +513,14 @@ impl DownloaderRepository {
|
||||
fetch_youtube_mp4(url, quality).await
|
||||
}
|
||||
|
||||
/// YouTube video + audio merged server-side into a single MP4.
|
||||
pub async fn fetch_youtube_merge(
|
||||
url: &str,
|
||||
quality: &str,
|
||||
) -> Result<DownloadResult, ScrapingError> {
|
||||
fetch_youtube_merge_free(url, quality).await
|
||||
}
|
||||
|
||||
/// YouTube to MP3 via ydlp.yard.id.
|
||||
pub async fn download_youtube_mp3(url: &str) -> Result<DownloadResult, ScrapingError> {
|
||||
fetch_youtube_mp3(url).await
|
||||
@@ -844,12 +854,26 @@ async fn detect_media_type(url: &str) -> MediaType {
|
||||
/// SnapSave parser — extracts Instagram/Facebook media via snapsave.app.
|
||||
/// Uses downr.org as fallback for robustness, then Playwright browser scraping.
|
||||
pub(crate) async fn fetch_snapsave(url: &str) -> Result<DownloadResult, ScrapingError> {
|
||||
// Validate Instagram/Facebook URL
|
||||
let valid_fb = regex::Regex::new(r"https?://(web\.|www\.|m\.)(facebook|fb)\.(com|watch)\S+")
|
||||
// Validate Instagram/Facebook URL.
|
||||
//
|
||||
// Facebook link shapes (all must pass):
|
||||
// https://www.facebook.com/watch/?v=123
|
||||
// https://fb.watch/abc123/
|
||||
// https://m.facebook.com/video.php?v=123
|
||||
// https://www.facebook.com/100012345678901/videos/1234567890
|
||||
// https://www.facebook.com/reel/123
|
||||
// https://www.facebook.com/story.php?story_fbid=123
|
||||
// https://web.facebook.com/...
|
||||
// NOTE: the old regex required a `www.|m.|web.` subdomain AND ended the
|
||||
// host capture with `(facebook|fb)\.(com|watch)` — that rejected
|
||||
// `facebook.com/watch...` (no subdomain) and `fb.watch/...` entirely.
|
||||
let fb_host = r"(?:[a-z0-9-]*\.)?(?:facebook|fb)\.(?:com|watch)";
|
||||
let valid_fb = regex::Regex::new(&format!(r"https?://{}(?:/|$)", fb_host))
|
||||
.unwrap()
|
||||
.is_match(url);
|
||||
let valid_ig =
|
||||
regex::Regex::new(r"https?://(www\.)?instagram\.com/(p|reel|reels|tv|stories)/\S+")
|
||||
let valid_ig = url.contains("instagram.com")
|
||||
|| url.contains("threads.net")
|
||||
|| regex::Regex::new(r"https?://(www\.)?instagram\.com/[^\s]+")
|
||||
.unwrap()
|
||||
.is_match(url);
|
||||
|
||||
@@ -859,7 +883,24 @@ pub(crate) async fn fetch_snapsave(url: &str) -> Result<DownloadResult, Scraping
|
||||
));
|
||||
}
|
||||
|
||||
// Try downr.org first
|
||||
// Try well-known scrapers in order — Playwright is a LAST resort (heavy,
|
||||
// slow, and its naive response-harvesting returns many broken/404 links
|
||||
// mixed with the one good URL, which is exactly the "link exists but can't
|
||||
// download" bug we've seen).
|
||||
//
|
||||
// 1) yt-dlp facebook extractor — authoritative, returns direct fbcdn
|
||||
// progressive URLs with REAL title/format metadata and only the working
|
||||
// scales (sd + hd, no junk). Also handles /reel, /watch, /videos,
|
||||
// story.php, video.php, fb.watch URL shapes.
|
||||
if let Ok(yt) = run_ytdlp_json(url, &["-f", "bestvideo+bestaudio/best"]).await {
|
||||
if let Some(res) = ytdlp_to_download_result(&yt) {
|
||||
if !res.media.is_empty() {
|
||||
return Ok(res);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 2) downr.org all-in-one (same provider as Shirokami's /fbdl)
|
||||
match fetch_all_in_one(url).await {
|
||||
Ok(result) if !result.media.is_empty() => return Ok(result),
|
||||
Ok(_) => {}
|
||||
@@ -871,7 +912,7 @@ pub(crate) async fn fetch_snapsave(url: &str) -> Result<DownloadResult, Scraping
|
||||
}
|
||||
}
|
||||
|
||||
// Fallback: use Playwright browser scraping to extract video URLs
|
||||
// 3) Playwright browser scraping to extract video URLs
|
||||
let platform = if valid_ig { "instagram" } else { "facebook" };
|
||||
let data = run_playwright_scraper(url, platform).await?;
|
||||
|
||||
@@ -889,17 +930,25 @@ pub(crate) async fn fetch_snapsave(url: &str) -> Result<DownloadResult, Scraping
|
||||
let media_arr = data.get("media").and_then(|v| v.as_array());
|
||||
if let Some(medias) = media_arr {
|
||||
for m in medias {
|
||||
let u = m
|
||||
.get("url")
|
||||
.and_then(|v| v.as_str())
|
||||
.unwrap_or("")
|
||||
.to_string();
|
||||
// Drop Playwright-harvested links that are actually broken
|
||||
// byte-range chunks (fbcdn URLs with bytestart=…byteend=… that
|
||||
// return tiny 200 responses — they 404 or return an HTML error
|
||||
// page when fetched as a whole), and the empty-string url.
|
||||
if u.is_empty() || u.contains("bytestart=") {
|
||||
continue;
|
||||
}
|
||||
let item = MediaItem {
|
||||
url: m
|
||||
.get("url")
|
||||
.and_then(|v| v.as_str())
|
||||
.unwrap_or("")
|
||||
.to_string(),
|
||||
url: u,
|
||||
quality: None,
|
||||
file_type: m.get("ext").and_then(|v| v.as_str()).and_then(|e| match e {
|
||||
"mp4" | "m3u8" => Some(MediaType::Video),
|
||||
"mp3" | "m4a" => Some(MediaType::Audio),
|
||||
_ => Some(MediaType::Video),
|
||||
file_type: m.get("ext").and_then(|v| v.as_str()).map(|e| match e {
|
||||
"mp4" | "m3u8" => MediaType::Video,
|
||||
"mp3" | "m4a" => MediaType::Audio,
|
||||
_ => MediaType::Video,
|
||||
}),
|
||||
extension: m.get("ext").and_then(|v| v.as_str()).map(|s| s.to_string()),
|
||||
thumbnail: None,
|
||||
@@ -913,6 +962,13 @@ pub(crate) async fn fetch_snapsave(url: &str) -> Result<DownloadResult, Scraping
|
||||
}
|
||||
}
|
||||
|
||||
// De-duplicate by URL (Playwright harvests the same CDN file at many
|
||||
// byte-ranges, and the primary URL appears multiple times).
|
||||
{
|
||||
let mut seen = std::collections::HashSet::new();
|
||||
result.media.retain(|m| seen.insert(m.url.clone()));
|
||||
}
|
||||
|
||||
if result.media.is_empty() {
|
||||
return Ok(DownloadResult::error(format!(
|
||||
"Failed to extract media from {} — server IP may be blocked by anti-bot protection",
|
||||
@@ -922,6 +978,122 @@ pub(crate) async fn fetch_snapsave(url: &str) -> Result<DownloadResult, Scraping
|
||||
|
||||
Ok(result)
|
||||
}
|
||||
|
||||
/// Convert a yt-dlp --dump-single-json result into a DownloadResult with the
|
||||
/// direct progressive/adaptive media URLs (only entries that carry a URL).
|
||||
/// Used by the Facebook downloader as the primary provider.
|
||||
fn ytdlp_to_download_result(data: &serde_json::Value) -> Option<DownloadResult> {
|
||||
let title = data
|
||||
.get("title")
|
||||
.and_then(|v| v.as_str())
|
||||
.map(|s| s.to_string());
|
||||
let mut result = DownloadResult::success(title);
|
||||
result.provider = Some("yt-dlp".to_string());
|
||||
result.thumbnail = data
|
||||
.get("thumbnail")
|
||||
.and_then(|v| v.as_str())
|
||||
.map(|s| s.to_string());
|
||||
result.duration = data
|
||||
.get("duration")
|
||||
.and_then(|v| v.as_u64())
|
||||
.map(|d| format!("{}s", d));
|
||||
|
||||
let formats = data.get("formats").and_then(|v| v.as_array());
|
||||
let requested_formats = data.get("requested_formats").and_then(|v| v.as_array());
|
||||
let mut seen = std::collections::HashSet::new();
|
||||
|
||||
// `-f bestvideo+bestaudio/best` puts the merged URLs in requested_formats.
|
||||
// `best` single format lands in top-level url + formats[0].
|
||||
for list in [requested_formats, formats].into_iter().flatten() {
|
||||
for f in list {
|
||||
let Some(furl) = f.get("url").and_then(|v| v.as_str()) else {
|
||||
continue;
|
||||
};
|
||||
if !furl.starts_with("http") || !seen.insert(furl.to_string()) {
|
||||
continue;
|
||||
}
|
||||
let proto = f.get("protocol").and_then(|v| v.as_str()).unwrap_or("");
|
||||
if proto.contains("m3u8")
|
||||
|| f.get("vcodec").and_then(|v| v.as_str()) == Some("none")
|
||||
&& f.get("acodec").and_then(|v| v.as_str()) != Some("none")
|
||||
{
|
||||
// Skip HLS + audio-only; Facebook's progressive mp4s are direct.
|
||||
continue;
|
||||
}
|
||||
let ext: String = f
|
||||
.get("ext")
|
||||
.and_then(|v| v.as_str())
|
||||
.unwrap_or("mp4")
|
||||
.to_string();
|
||||
result.media.push(MediaItem {
|
||||
url: furl.to_string(),
|
||||
quality: f
|
||||
.get("height")
|
||||
.and_then(|v| v.as_u64())
|
||||
.map(|h| format!("{}p", h))
|
||||
.or_else(|| {
|
||||
f.get("format_note")
|
||||
.and_then(|v| v.as_str())
|
||||
.map(|s| s.to_string())
|
||||
}),
|
||||
file_type: Some(if ext == "mp4" || ext == "webm" || ext == "mkv" {
|
||||
MediaType::Video
|
||||
} else {
|
||||
MediaType::File
|
||||
}),
|
||||
extension: Some(ext),
|
||||
thumbnail: result.thumbnail.clone(),
|
||||
file_size: f
|
||||
.get("filesize")
|
||||
.and_then(|v| v.as_u64())
|
||||
.map(format_filesize),
|
||||
size_bytes: f.get("filesize").and_then(|v| v.as_u64()),
|
||||
frame_width: f
|
||||
.get("width")
|
||||
.and_then(|v| v.as_u64())
|
||||
.map(|w| w.to_string()),
|
||||
frame_height: f
|
||||
.get("height")
|
||||
.and_then(|v| v.as_u64())
|
||||
.map(|h| h.to_string()),
|
||||
note: None,
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
// Fall back to the merged top-level URL (bestvideo+bestaudio sets it).
|
||||
if result.media.is_empty() {
|
||||
if let Some(dl_url) = data.get("url").and_then(|v| v.as_str()) {
|
||||
if dl_url.starts_with("http") {
|
||||
result.media.push(MediaItem {
|
||||
url: dl_url.to_string(),
|
||||
quality: None,
|
||||
file_type: Some(MediaType::Video),
|
||||
extension: data
|
||||
.get("ext")
|
||||
.and_then(|v| v.as_str())
|
||||
.map(|s| s.to_string())
|
||||
.or_else(|| Some("mp4".to_string())),
|
||||
thumbnail: None,
|
||||
file_size: data
|
||||
.get("filesize")
|
||||
.and_then(|v| v.as_u64())
|
||||
.map(format_filesize),
|
||||
size_bytes: data.get("filesize").and_then(|v| v.as_u64()),
|
||||
frame_width: None,
|
||||
frame_height: None,
|
||||
note: None,
|
||||
});
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if result.media.is_empty() {
|
||||
None
|
||||
} else {
|
||||
Some(result)
|
||||
}
|
||||
}
|
||||
/// TikTok via embed-page scraping (primary method).
|
||||
/// Scrapes `https://www.tiktok.com/embed/v2/{video_id}` HTML and extracts the
|
||||
/// direct `v16m.tiktokcdn.com` MP4 URL from the `<video data-testid="play-video">` tag.
|
||||
@@ -1402,7 +1574,11 @@ pub async fn fetch_youtube_mp3(url: &str) -> Result<DownloadResult, ScrapingErro
|
||||
data.get("abr").and_then(|v| v.as_u64()).unwrap_or(128)
|
||||
)),
|
||||
file_type: Some(MediaType::Audio),
|
||||
extension: Some("mp3".to_string()),
|
||||
extension: data
|
||||
.get("ext")
|
||||
.and_then(|v| v.as_str())
|
||||
.map(|s| s.to_string())
|
||||
.or_else(|| Some("webm".to_string())),
|
||||
thumbnail: data
|
||||
.get("thumbnail")
|
||||
.and_then(|v| v.as_str())
|
||||
@@ -1425,11 +1601,116 @@ pub async fn fetch_youtube_mp4(url: &str, quality: &str) -> Result<DownloadResul
|
||||
extract_youtube_id(url)
|
||||
.ok_or_else(|| ScrapingError::Http("Invalid YouTube URL".to_string()))?;
|
||||
|
||||
let q = quality.trim_end_matches('p');
|
||||
|
||||
// ---------------------------------------------------------------
|
||||
// 1) PREFER a PROGRESSIVE format (video + audio in ONE file).
|
||||
// The `android` player client exposes progressive itags (18/22) that
|
||||
// are NOT rate-limited from datacenter IPs (verified 2026-09-05:
|
||||
// itag 18 downloads at full speed, while audio-only streams crawl).
|
||||
// Fall back to adaptive (bestvideo+bestaudio) below when no
|
||||
// progressive format exists at the requested height.
|
||||
// ---------------------------------------------------------------
|
||||
let prog_fmt = format!("best[height<={}][ext=mp4]/best[height<={}]", q, q);
|
||||
if let Ok(prog_data) = run_ytdlp_json(
|
||||
url,
|
||||
&[
|
||||
"--extractor-args",
|
||||
"youtube:player_client=android",
|
||||
"-f",
|
||||
&prog_fmt,
|
||||
],
|
||||
)
|
||||
.await
|
||||
{
|
||||
let prog_url = prog_data
|
||||
.get("url")
|
||||
.and_then(|v| v.as_str())
|
||||
.map(|s| s.to_string());
|
||||
let prog_vcodec = prog_data
|
||||
.get("vcodec")
|
||||
.and_then(|v| v.as_str())
|
||||
.unwrap_or("none");
|
||||
let prog_acodec = prog_data
|
||||
.get("acodec")
|
||||
.and_then(|v| v.as_str())
|
||||
.unwrap_or("none");
|
||||
let has_audio = prog_acodec != "none" && !prog_acodec.is_empty();
|
||||
let has_video = prog_vcodec != "none" && !prog_vcodec.is_empty();
|
||||
if let Some(purl) = prog_url {
|
||||
if has_audio && has_video && purl.starts_with("http") {
|
||||
let height = prog_data
|
||||
.get("height")
|
||||
.and_then(|v| v.as_u64())
|
||||
.unwrap_or(360);
|
||||
// Only use the progressive format if it actually meets the
|
||||
// requested quality. `best[height<=Q][ext=mp4]` can return a
|
||||
// lower-height progressive (e.g. 360p when 720p was requested)
|
||||
// because no 720p progressive exists; in that case fall through
|
||||
// to the adaptive (bestvideo+bestaudio) path so the caller
|
||||
// still gets the requested resolution.
|
||||
let q_num: u64 = q.parse().unwrap_or(0);
|
||||
if q_num != 0 && height < q_num {
|
||||
// fall through to adaptive below
|
||||
} else {
|
||||
let ext = prog_data
|
||||
.get("ext")
|
||||
.and_then(|v| v.as_str())
|
||||
.unwrap_or("mp4");
|
||||
let mut result = DownloadResult::success(
|
||||
prog_data
|
||||
.get("title")
|
||||
.and_then(|v| v.as_str())
|
||||
.map(|s| s.to_string()),
|
||||
);
|
||||
result.author = prog_data
|
||||
.get("uploader")
|
||||
.and_then(|v| v.as_str())
|
||||
.map(|s| s.to_string());
|
||||
result.thumbnail = prog_data
|
||||
.get("thumbnail")
|
||||
.and_then(|v| v.as_str())
|
||||
.map(|s| s.to_string());
|
||||
result.duration = prog_data
|
||||
.get("duration")
|
||||
.and_then(|v| v.as_u64())
|
||||
.map(|d| format!("{}s", d));
|
||||
result.provider = Some("yt-dlp".to_string());
|
||||
result.media.push(MediaItem {
|
||||
url: purl,
|
||||
quality: Some(format!("{}p", height)),
|
||||
file_type: Some(MediaType::Video),
|
||||
extension: Some(ext.to_string()),
|
||||
thumbnail: prog_data
|
||||
.get("thumbnail")
|
||||
.and_then(|v| v.as_str())
|
||||
.map(|s| s.to_string()),
|
||||
file_size: prog_data
|
||||
.get("filesize")
|
||||
.and_then(|v| v.as_u64())
|
||||
.map(format_filesize),
|
||||
size_bytes: prog_data.get("filesize").and_then(|v| v.as_u64()),
|
||||
frame_width: prog_data
|
||||
.get("width")
|
||||
.and_then(|v| v.as_u64())
|
||||
.map(|w| w.to_string()),
|
||||
frame_height: prog_data
|
||||
.get("height")
|
||||
.and_then(|v| v.as_u64())
|
||||
.map(|h| h.to_string()),
|
||||
note: Some("Progressive MP4 — video and audio in one file".to_string()),
|
||||
});
|
||||
return Ok(result);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
let fmt = format!(
|
||||
"bestvideo[height<={}]/best+bestaudio/best",
|
||||
quality.trim_end_matches('p')
|
||||
"bestvideo[height<={}]+bestaudio/best[height<={}]/best",
|
||||
q, q
|
||||
);
|
||||
let data = run_ytdlp_json(url, &["-f", &fmt]).await?;
|
||||
let data = run_ytdlp_json(url, &["-f", &fmt, "--merge-output-format", "mp4"]).await?;
|
||||
|
||||
let title = data
|
||||
.get("title")
|
||||
@@ -1455,23 +1736,101 @@ pub async fn fetch_youtube_mp4(url: &str, quality: &str) -> Result<DownloadResul
|
||||
result.provider = Some("yt-dlp".to_string());
|
||||
|
||||
let formats = data.get("formats").and_then(|v| v.as_array());
|
||||
let requested_formats = data.get("requested_formats").and_then(|v| v.as_array());
|
||||
let mut seen = std::collections::HashSet::new();
|
||||
|
||||
if let Some(fmts) = formats {
|
||||
// When yt-dlp uses a merged format (bestvideo+bestaudio), the actual
|
||||
// download URLs live in `requested_formats` (video + audio separate URLs).
|
||||
// The top-level `url` is empty and `formats[]` is the FULL list of all
|
||||
// individual format IDs, not just the selected ones.
|
||||
if let Some(rf) = requested_formats {
|
||||
for f in rf {
|
||||
if let (Some(furl), Some(ext_val)) = (
|
||||
f.get("url").and_then(|v| v.as_str()),
|
||||
f.get("ext").and_then(|v| v.as_str()),
|
||||
) {
|
||||
let proto = f.get("protocol").and_then(|v| v.as_str()).unwrap_or("");
|
||||
// Skip mhtml/storyboard and m3u8/hls streams
|
||||
if ext_val == "mhtml" || proto.contains("m3u8") {
|
||||
continue;
|
||||
}
|
||||
let url_string = furl.to_string();
|
||||
if !url_string.starts_with("http") {
|
||||
continue;
|
||||
}
|
||||
if seen.insert(url_string.clone()) {
|
||||
let is_video = f
|
||||
.get("vcodec")
|
||||
.and_then(|v| v.as_str())
|
||||
.map(|c| c != "none")
|
||||
.unwrap_or(false);
|
||||
let is_audio_only = f
|
||||
.get("acodec")
|
||||
.and_then(|v| v.as_str())
|
||||
.map(|c| c != "none")
|
||||
.unwrap_or(false);
|
||||
let fmt_type = if is_video && is_audio_only {
|
||||
MediaType::Video
|
||||
} else if is_audio_only {
|
||||
MediaType::Audio
|
||||
} else if is_video {
|
||||
MediaType::Video
|
||||
} else {
|
||||
MediaType::File
|
||||
};
|
||||
let q = format!("{}p", f.get("height").and_then(|v| v.as_u64()).unwrap_or(0));
|
||||
result.media.push(MediaItem {
|
||||
url: url_string,
|
||||
quality: Some(q),
|
||||
file_type: Some(fmt_type),
|
||||
extension: Some(ext_val.to_string()),
|
||||
thumbnail: data
|
||||
.get("thumbnail")
|
||||
.and_then(|v| v.as_str())
|
||||
.map(|s| s.to_string()),
|
||||
file_size: f
|
||||
.get("filesize")
|
||||
.and_then(|v| v.as_u64())
|
||||
.map(format_filesize),
|
||||
size_bytes: f.get("filesize").and_then(|v| v.as_u64()),
|
||||
frame_width: f
|
||||
.get("width")
|
||||
.and_then(|v| v.as_u64())
|
||||
.map(|w| w.to_string()),
|
||||
frame_height: f
|
||||
.get("height")
|
||||
.and_then(|v| v.as_u64())
|
||||
.map(|h| h.to_string()),
|
||||
note: None,
|
||||
});
|
||||
}
|
||||
}
|
||||
}
|
||||
} else if let Some(fmts) = formats {
|
||||
for f in fmts {
|
||||
if let (Some(furl), Some(ext_val)) = (
|
||||
f.get("url").and_then(|v| v.as_str()),
|
||||
f.get("ext").and_then(|v| v.as_str()),
|
||||
) {
|
||||
if ext_val == "mhtml" {
|
||||
// Skip storyboard, mhtml, m3u8/hls, and audio-only formats.
|
||||
// We only want formats that have both video and audio.
|
||||
let proto = f.get("protocol").and_then(|v| v.as_str()).unwrap_or("");
|
||||
if ext_val == "mhtml" || proto.contains("m3u8") {
|
||||
continue;
|
||||
}
|
||||
// Skip audio-only formats (vcodec == "none", acodec != "none")
|
||||
let vcodec = f.get("vcodec").and_then(|v| v.as_str()).unwrap_or("none");
|
||||
let acodec = f.get("acodec").and_then(|v| v.as_str()).unwrap_or("none");
|
||||
if vcodec == "none" && acodec != "none" {
|
||||
continue; // audio-only
|
||||
}
|
||||
let url_string = furl.to_string();
|
||||
if !url_string.starts_with("http") {
|
||||
continue;
|
||||
}
|
||||
if seen.insert(url_string.clone()) {
|
||||
let fmt_type = if ext_val == "mp4" || ext_val == "webm" || ext_val == "mkv" {
|
||||
MediaType::Video
|
||||
} else if ext_val == "mp3" || ext_val == "m4a" || ext_val == "webm" {
|
||||
MediaType::Audio
|
||||
} else {
|
||||
MediaType::File
|
||||
};
|
||||
@@ -1536,6 +1895,252 @@ pub async fn fetch_youtube_mp4(url: &str, quality: &str) -> Result<DownloadResul
|
||||
Ok(result)
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// YouTube video + audio SERVER-SIDE MERGE (produces a single MP4 with audio).
|
||||
//
|
||||
// Modern YouTube videos (AV1/VP9, 720p+) only expose separate video-only and
|
||||
// audio-only streams. `fetch_youtube_mp4` returns those two URLs — the caller
|
||||
// would have to merge them manually. This endpoint downloads both streams and
|
||||
// remuxes them into ONE MP4 with `ffmpeg`, stored under the uploads dir and
|
||||
// served via GET /file/yt_merge/{name}. Files older than 2h are cleaned up on
|
||||
// each new merge request to bound disk usage.
|
||||
// ============================================================================
|
||||
pub async fn fetch_youtube_merge_free(
|
||||
url: &str,
|
||||
quality: &str,
|
||||
) -> Result<DownloadResult, ScrapingError> {
|
||||
let video_id = extract_youtube_id(url)
|
||||
.ok_or_else(|| ScrapingError::Http("Invalid YouTube URL".to_string()))?;
|
||||
|
||||
let q = quality.trim_end_matches('p');
|
||||
let fmt = format!(
|
||||
"bestvideo[height<={}]+bestaudio/best[height<={}]/best",
|
||||
q, q
|
||||
);
|
||||
let data = run_ytdlp_json(url, &["-f", &fmt, "--merge-output-format", "mp4"]).await?;
|
||||
|
||||
let title = data
|
||||
.get("title")
|
||||
.and_then(|v| v.as_str())
|
||||
.map(|s| s.to_string());
|
||||
let author = data
|
||||
.get("uploader")
|
||||
.and_then(|v| v.as_str())
|
||||
.map(|s| s.to_string());
|
||||
let thumbnail = data
|
||||
.get("thumbnail")
|
||||
.and_then(|v| v.as_str())
|
||||
.map(|s| s.to_string());
|
||||
let duration = data
|
||||
.get("duration")
|
||||
.and_then(|v| v.as_u64())
|
||||
.map(|d| format!("{}s", d));
|
||||
|
||||
// Locate the video-only and audio-only URLs from requested_formats.
|
||||
let requested_formats = data
|
||||
.get("requested_formats")
|
||||
.and_then(|v| v.as_array())
|
||||
.ok_or_else(|| ScrapingError::Http("No requested_formats from yt-dlp".to_string()))?;
|
||||
|
||||
let mut video_url: Option<String> = None;
|
||||
let mut audio_url: Option<String> = None;
|
||||
let mut video_ext = "mp4";
|
||||
let mut audio_ext = "webm";
|
||||
for f in requested_formats {
|
||||
let vcodec = f.get("vcodec").and_then(|v| v.as_str()).unwrap_or("none");
|
||||
let acodec = f.get("acodec").and_then(|v| v.as_str()).unwrap_or("none");
|
||||
let f_url = f.get("url").and_then(|v| v.as_str());
|
||||
let proto = f.get("protocol").and_then(|v| v.as_str()).unwrap_or("");
|
||||
if proto.contains("m3u8") {
|
||||
continue; // HLS streams are not directly downloadable via reqwest
|
||||
}
|
||||
if vcodec != "none" && acodec == "none" {
|
||||
if let Some(u) = f_url {
|
||||
video_url = Some(u.to_string());
|
||||
video_ext = f.get("ext").and_then(|v| v.as_str()).unwrap_or("mp4");
|
||||
}
|
||||
} else if vcodec == "none" && acodec != "none" {
|
||||
if let Some(u) = f_url {
|
||||
audio_url = Some(u.to_string());
|
||||
audio_ext = f.get("ext").and_then(|v| v.as_str()).unwrap_or("webm");
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
let video_url = video_url
|
||||
.ok_or_else(|| ScrapingError::Http("No video stream URL from yt-dlp".to_string()))?;
|
||||
let audio_url = audio_url
|
||||
.ok_or_else(|| ScrapingError::Http("No audio stream URL from yt-dlp".to_string()))?;
|
||||
|
||||
// --- download both streams to the merge temp dir ---
|
||||
let merge_dir = PathBuf::from("/var/lib/scraper/uploads/yt_merge");
|
||||
tokio::fs::create_dir_all(&merge_dir)
|
||||
.await
|
||||
.map_err(|e| ScrapingError::Http(format!("create merge dir failed: {e}")))?;
|
||||
|
||||
// Cleanup stale merged files (> 2h) before writing anything new.
|
||||
cleanup_stale_merges(&merge_dir).await;
|
||||
|
||||
let job_id = uuid::Uuid::new_v4().simple().to_string();
|
||||
let vpath = merge_dir.join(format!("{video_id}_{job_id}_v.{video_ext}"));
|
||||
let apath = merge_dir.join(format!("{video_id}_{job_id}_a.{audio_ext}"));
|
||||
let outpath = merge_dir.join(format!("{video_id}_{job_id}_merged.mp4"));
|
||||
|
||||
download_to_file(&video_url, &vpath).await?;
|
||||
download_to_file(&audio_url, &apath).await?;
|
||||
|
||||
// --- merge with ffmpeg ---
|
||||
let merge_result = tokio::task::spawn_blocking({
|
||||
let vpath = vpath.clone();
|
||||
let apath = apath.clone();
|
||||
let outpath = outpath.clone();
|
||||
move || merge_with_ffmpeg(&vpath, &apath, &outpath)
|
||||
})
|
||||
.await
|
||||
.map_err(|e| ScrapingError::Http(format!("ffmpeg join failed: {e}")))?;
|
||||
|
||||
match merge_result {
|
||||
Ok(()) => {}
|
||||
Err(e) => {
|
||||
let _ = tokio::fs::remove_file(&vpath).await;
|
||||
let _ = tokio::fs::remove_file(&apath).await;
|
||||
return Err(ScrapingError::Http(format!("ffmpeg merge failed: {e}")));
|
||||
}
|
||||
}
|
||||
|
||||
// Clean up the transient inputs now that the merged file exists.
|
||||
let _ = tokio::fs::remove_file(&vpath).await;
|
||||
let _ = tokio::fs::remove_file(&apath).await;
|
||||
|
||||
let file_size = tokio::fs::metadata(&outpath).await.ok().map(|m| m.len());
|
||||
|
||||
let file_name = outpath
|
||||
.file_name()
|
||||
.map(|s| s.to_string_lossy().to_string())
|
||||
.unwrap_or_default();
|
||||
let public_url = format!("/file/yt_merge/{file_name}");
|
||||
|
||||
let mut result = DownloadResult::success(title);
|
||||
result.author = author;
|
||||
result.thumbnail = thumbnail.clone();
|
||||
result.duration = duration;
|
||||
result.provider = Some("yt-dlp+ffmpeg".to_string());
|
||||
result.media.push(MediaItem {
|
||||
url: public_url,
|
||||
quality: Some(format!("{}p merged", q)),
|
||||
file_type: Some(MediaType::Video),
|
||||
extension: Some("mp4".to_string()),
|
||||
thumbnail: thumbnail.clone(),
|
||||
file_size: file_size.map(format_filesize),
|
||||
size_bytes: file_size,
|
||||
frame_width: None,
|
||||
frame_height: None,
|
||||
note: Some(
|
||||
"Merged server-side: single MP4 file containing both video and audio".to_string(),
|
||||
),
|
||||
});
|
||||
|
||||
Ok(result)
|
||||
}
|
||||
|
||||
/// Download `url` to `path` with a generous timeout. The yt CDN stream URLs are
|
||||
/// signed and short-lived, so we must download promptly and cannot retry after
|
||||
/// the URL expires.
|
||||
async fn download_to_file(url: &str, path: &Path) -> Result<(), ScrapingError> {
|
||||
// Dedicated client with a longer timeout than the shared 30s one: merged
|
||||
// files are often tens of MB and the yt CDN can be slow from a datacenter.
|
||||
// .no_gzip()/.no_brotli()/.no_deflate() disable reqwest's auto-decompress
|
||||
// so we can write raw bytes — some CDN responses are mis-labelled and fail
|
||||
// the auto-decode step.
|
||||
let client = reqwest::Client::builder()
|
||||
.timeout(Duration::from_secs(300))
|
||||
.connect_timeout(Duration::from_secs(20))
|
||||
.no_gzip()
|
||||
.no_brotli()
|
||||
.no_deflate()
|
||||
.build()
|
||||
.map_err(|e| ScrapingError::Http(format!("stream client build failed: {e}")))?;
|
||||
let resp = client
|
||||
.get(url)
|
||||
.header(USER_AGENT, "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36")
|
||||
.header("Accept-Encoding", "identity")
|
||||
.send()
|
||||
.await
|
||||
.map_err(|e| ScrapingError::Http(format!("stream download failed: {e}")))?;
|
||||
|
||||
let status = resp.status();
|
||||
if !status.is_success() {
|
||||
return Err(ScrapingError::Http(format!(
|
||||
"stream download HTTP {status}"
|
||||
)));
|
||||
}
|
||||
|
||||
let bytes = resp
|
||||
.bytes()
|
||||
.await
|
||||
.map_err(|e| ScrapingError::Http(format!("stream body read failed: {e}")))?;
|
||||
|
||||
tokio::fs::write(path, bytes)
|
||||
.await
|
||||
.map_err(|e| ScrapingError::Http(format!("stream write failed: {e}")))?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Run ffmpeg to remux the video and audio streams into a single MP4.
|
||||
/// `-c:v copy` avoids a full re-encode (fast); audio is transcoded to AAC so
|
||||
/// the output plays everywhere. `+faststart` moves the moov atom to the front
|
||||
/// so the file streams immediately.
|
||||
fn merge_with_ffmpeg(video: &Path, audio: &Path, out: &Path) -> Result<(), String> {
|
||||
let status = Command::new("/bin/ffmpeg")
|
||||
.args([
|
||||
"-y",
|
||||
"-v",
|
||||
"error",
|
||||
"-i",
|
||||
video.to_str().unwrap_or_default(),
|
||||
"-i",
|
||||
audio.to_str().unwrap_or_default(),
|
||||
"-c:v",
|
||||
"copy",
|
||||
"-c:a",
|
||||
"aac",
|
||||
"-movflags",
|
||||
"+faststart",
|
||||
"-shortest",
|
||||
out.to_str().unwrap_or_default(),
|
||||
])
|
||||
.stdout(Stdio::null())
|
||||
.stderr(Stdio::piped())
|
||||
.status()
|
||||
.map_err(|e| format!("spawn ffmpeg failed: {e}"))?;
|
||||
|
||||
if status.success() {
|
||||
Ok(())
|
||||
} else {
|
||||
Err(format!("ffmpeg exited with {status:?}"))
|
||||
}
|
||||
}
|
||||
|
||||
/// Remove merged output + transient files older than 2 hours. Bounded, best
|
||||
/// effort — never fails the request.
|
||||
async fn cleanup_stale_merges(dir: &Path) {
|
||||
let deadline = std::time::SystemTime::now() - std::time::Duration::from_secs(2 * 60 * 60);
|
||||
let Ok(mut entries) = tokio::fs::read_dir(dir).await else {
|
||||
return;
|
||||
};
|
||||
while let Ok(Some(entry)) = entries.next_entry().await {
|
||||
let Ok(meta) = entry.metadata().await else {
|
||||
continue;
|
||||
};
|
||||
let Ok(modified) = meta.modified() else {
|
||||
continue;
|
||||
};
|
||||
if modified < deadline {
|
||||
let _ = tokio::fs::remove_file(entry.path()).await;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Spotify downloaders
|
||||
// ============================================================================
|
||||
|
||||
@@ -2804,55 +3409,130 @@ pub async fn fetch_terabox(url: &str) -> Result<DownloadResult, ScrapingError> {
|
||||
|
||||
let surl = surl.ok_or_else(|| ScrapingError::Http("SURL not found.".to_string()))?;
|
||||
|
||||
// Call local TeraBox resolver (Playwright + residential proxy + cookies)
|
||||
let resolver_url = std::env::var("TERABOX_RESOLVER_URL")
|
||||
.unwrap_or_else(|_| "http://127.0.0.1:4092".to_string());
|
||||
let resolve_url = format!("{}/resolve?surl={}", resolver_url, surl);
|
||||
|
||||
let client = http_client();
|
||||
let resp = client
|
||||
.client()
|
||||
.get("https://tera2.sylyt93.workers.dev/info")
|
||||
.query(&[("s", &surl)])
|
||||
.header("origin", "https://www.kauruka.com")
|
||||
.header("referer", "https://www.kauruka.com/")
|
||||
.get(&resolve_url)
|
||||
.header(USER_AGENT, "Mozilla/5.0")
|
||||
.timeout(Duration::from_secs(15))
|
||||
.timeout(Duration::from_secs(60))
|
||||
.send()
|
||||
.await
|
||||
.map_err(|e| ScrapingError::Http(format!("TeraBox info fetch failed: {}", e)))?
|
||||
.json::<serde_json::Value>()
|
||||
.map_err(|e| ScrapingError::Http(format!("TeraBox resolver fetch failed: {}", e)))?;
|
||||
|
||||
if !resp.status().is_success() {
|
||||
let status = resp.status();
|
||||
let body = resp.text().await.unwrap_or_default();
|
||||
return Ok(DownloadResult::error(format!(
|
||||
"TeraBox resolver returned HTTP {}: {}",
|
||||
status, body
|
||||
)));
|
||||
}
|
||||
|
||||
let data: serde_json::Value = resp
|
||||
.json()
|
||||
.await
|
||||
.map_err(|e| ScrapingError::Http(format!("TeraBox JSON parse failed: {}", e)))?;
|
||||
.map_err(|e| ScrapingError::Http(format!("TeraBox resolver JSON parse failed: {}", e)))?;
|
||||
|
||||
if let Some(download_url) = resp.get("url").and_then(|v| v.as_str()) {
|
||||
let mut result = DownloadResult::success(
|
||||
resp.get("name")
|
||||
.and_then(|v| v.as_str())
|
||||
.map(|s| s.to_string()),
|
||||
);
|
||||
result.provider = Some("terabox".to_string());
|
||||
result.thumbnail = resp
|
||||
.get("img")
|
||||
.and_then(|v| v.as_str())
|
||||
.map(|s| s.to_string());
|
||||
if let Some(error) = data.get("error").and_then(|v| v.as_str()) {
|
||||
return Ok(DownloadResult::error(error));
|
||||
}
|
||||
|
||||
let file_name = data
|
||||
.get("file_name")
|
||||
.and_then(|v| v.as_str())
|
||||
.map(|s| s.to_string());
|
||||
|
||||
let mut result = DownloadResult::success(file_name.clone());
|
||||
result.provider = Some("terabox".to_string());
|
||||
result.thumbnail = data
|
||||
.get("thumbnail")
|
||||
.and_then(|v| v.as_str())
|
||||
.map(|s| s.to_string());
|
||||
|
||||
// Primary download link.
|
||||
//
|
||||
// IMPORTANT: the resolver's download_link is a raw terabox dlink that
|
||||
// 403s with {"error_code":31045,"user not exists"} when fetched WITHOUT
|
||||
// cookies — even from residential IPs. So we rewrite the public URL to
|
||||
// OUR OWN /proxy/terabox streaming endpoint (which forwards the resolver's
|
||||
// cookie-authenticated stream through the public API). This is the only
|
||||
// way users get a link that actually downloads.
|
||||
// The public API base that users can reach. `CONFIG.urls.site_url` is the
|
||||
// main site (hub → 4003) — NOT the scraper API domain. Use an explicit
|
||||
// env override (TERABOX_PUBLIC_BASE_URL), falling back to the scraper's
|
||||
// own public domain via site_url only when it's clearly the API host.
|
||||
let public_base = std::env::var("TERABOX_PUBLIC_BASE_URL").unwrap_or_else(|_| {
|
||||
let site = crate::config::CONFIG.urls.site_url.clone();
|
||||
if site.contains("scraper") || site.contains("api") {
|
||||
site
|
||||
} else {
|
||||
// default to the scraper API subdomain (this API's public host)
|
||||
"https://api.asepharyana.my.id".to_string()
|
||||
}
|
||||
});
|
||||
let proxy_dl = format!("{}/proxy/terabox?surl={}", public_base, surl);
|
||||
if let Some(download_url) = data.get("download_link").and_then(|v| v.as_str()) {
|
||||
let size_bytes = data.get("file_size").and_then(|v| v.as_u64()).unwrap_or(0);
|
||||
|
||||
// PRIMARY = the cookie-authenticated public proxy. The raw terabox
|
||||
// dlink is only a secondary entry — it 403s without cookies, so it
|
||||
// must NOT be the first link users see.
|
||||
result.media.push(MediaItem {
|
||||
url: proxy_dl.clone(),
|
||||
quality: None,
|
||||
file_type: Some(MediaType::File),
|
||||
extension: None,
|
||||
thumbnail: None,
|
||||
file_size: Some(format_filesize(size_bytes)),
|
||||
size_bytes: Some(size_bytes),
|
||||
frame_width: None,
|
||||
frame_height: None,
|
||||
note: Some("Direct download via cookie-authenticated proxy (works).".into()),
|
||||
});
|
||||
result.media.push(MediaItem {
|
||||
url: download_url.to_string(),
|
||||
quality: None,
|
||||
file_type: Some(MediaType::File),
|
||||
extension: None,
|
||||
thumbnail: None,
|
||||
file_size: None,
|
||||
size_bytes: None,
|
||||
file_size: Some(format_filesize(size_bytes)),
|
||||
size_bytes: Some(size_bytes),
|
||||
frame_width: None,
|
||||
frame_height: None,
|
||||
note: None,
|
||||
note: Some("Raw CDN link (may require a TeraBox session cookie).".into()),
|
||||
});
|
||||
|
||||
Ok(result)
|
||||
} else {
|
||||
Ok(DownloadResult::error(
|
||||
resp.get("error")
|
||||
.and_then(|v| v.as_str())
|
||||
.unwrap_or("Failed to fetch TeraBox info"),
|
||||
))
|
||||
// Directory: add child file links
|
||||
if let Some(files) = data.get("files").and_then(|v| v.as_array()) {
|
||||
for f in files {
|
||||
if let Some(furl) = f.get("download_link").and_then(|v| v.as_str()) {
|
||||
let fsize = f.get("file_size").and_then(|v| v.as_u64()).unwrap_or(0);
|
||||
result.media.push(MediaItem {
|
||||
url: furl.to_string(),
|
||||
quality: None,
|
||||
file_type: Some(MediaType::File),
|
||||
extension: None,
|
||||
thumbnail: None,
|
||||
file_size: Some(format_filesize(fsize)),
|
||||
size_bytes: Some(fsize),
|
||||
frame_width: None,
|
||||
frame_height: None,
|
||||
note: f
|
||||
.get("file_name")
|
||||
.and_then(|v| v.as_str())
|
||||
.map(|s| s.to_string()),
|
||||
});
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Ok(result)
|
||||
}
|
||||
|
||||
pub async fn fetch_videy(url: &str) -> Result<DownloadResult, ScrapingError> {
|
||||
|
||||
@@ -9,7 +9,6 @@ pub mod proxy;
|
||||
pub mod search;
|
||||
pub mod stalk;
|
||||
pub mod tools;
|
||||
pub mod weebs;
|
||||
|
||||
pub use alqanime::AlqanimeRepository;
|
||||
pub use downloader::DownloaderRepository;
|
||||
|
||||
@@ -1,4 +1,4 @@
|
||||
//! Infrastructure — Stalk utilities (github, youtube, instagram, tiktok, twitter).
|
||||
//! Infrastructure — Stalk utilities (github, youtube, twitter).
|
||||
//!
|
||||
//! Ported from Shirokami-API `scraper/stalk/*.js`.
|
||||
|
||||
@@ -179,7 +179,7 @@ pub async fn fetch_youtube_stalk(username: &str) -> Result<Value, String> {
|
||||
.and_then(|t| t.as_array())
|
||||
{
|
||||
if let Some(contents) = tabs
|
||||
.get(0)
|
||||
.first()
|
||||
.and_then(|t| t.pointer("/tabRenderer/content/sectionListRenderer/contents"))
|
||||
.and_then(|c| c.as_array())
|
||||
{
|
||||
@@ -223,8 +223,8 @@ fn extract_balanced_json(body: &str, open_brace: usize) -> Result<String, String
|
||||
let mut in_str = false;
|
||||
let mut escaped = false;
|
||||
let mut close = None;
|
||||
for i in open_brace..bytes.len() {
|
||||
let c = bytes[i] as char;
|
||||
for (i, b) in bytes.iter().enumerate().skip(open_brace) {
|
||||
let c = *b as char;
|
||||
if in_str {
|
||||
if escaped {
|
||||
escaped = false;
|
||||
@@ -341,7 +341,7 @@ fn video_from_lockup(lv: &Value) -> Value {
|
||||
.pointer("/metadata/lockupMetadataViewModel/metadata/contentMetadataViewModel/metadataRows/0/metadataParts")
|
||||
.and_then(|p| p.as_array())
|
||||
{
|
||||
if let Some(p0) = parts.get(0) {
|
||||
if let Some(p0) = parts.first() {
|
||||
view_count = p0
|
||||
.pointer("/text/content")
|
||||
.cloned()
|
||||
@@ -416,3 +416,43 @@ fn video_from_grid_video(gv: &Value) -> Value {
|
||||
}
|
||||
v
|
||||
}
|
||||
|
||||
// ---------------------------------------------------------------------------
|
||||
// Twitter — fxtwitter API
|
||||
// ---------------------------------------------------------------------------
|
||||
|
||||
pub async fn fetch_twitter_stalk(username: &str) -> Result<Value, String> {
|
||||
let url = format!("https://api.fxtwitter.com/{}", username);
|
||||
let resp = http_client()
|
||||
.client()
|
||||
.get(&url)
|
||||
.header(
|
||||
USER_AGENT,
|
||||
"Mozilla/5.0 (Windows NT 10.0; Win64; x64) Chrome/130.0 Safari/537.36",
|
||||
)
|
||||
.send()
|
||||
.await
|
||||
.map_err(|e| format!("HTTP: {}", e))?;
|
||||
let data: Value = resp.json().await.map_err(|e| format!("JSON: {}", e))?;
|
||||
|
||||
// Transform to the source's shape
|
||||
let u = &data["user"];
|
||||
Ok(json!({
|
||||
"message": data.get("message").cloned().unwrap_or(Value::Null),
|
||||
"user": {
|
||||
"id": u.get("id").cloned().unwrap_or(Value::Null),
|
||||
"url": u.get("url").cloned().unwrap_or(Value::Null),
|
||||
"screen_name": u.get("screen_name").cloned().unwrap_or(Value::Null),
|
||||
"name": u.get("name").cloned().unwrap_or(Value::Null),
|
||||
"location": u.get("location").cloned().unwrap_or(Value::Null),
|
||||
"description": u.get("description").cloned().unwrap_or(Value::Null),
|
||||
"followers": u.get("followers").cloned().unwrap_or(Value::Null),
|
||||
"following": u.get("following").cloned().unwrap_or(Value::Null),
|
||||
"likes": u.get("likes").cloned().unwrap_or(Value::Null),
|
||||
"banner_url": u.get("banner_url").cloned().unwrap_or(Value::Null),
|
||||
"avatar_url": u.get("avatar_url").cloned().unwrap_or(Value::Null),
|
||||
"joined_at": u.get("joined").cloned().unwrap_or(Value::Null),
|
||||
"website": u.get("website").cloned().unwrap_or(Value::Null),
|
||||
}
|
||||
}))
|
||||
}
|
||||
@@ -1,196 +0,0 @@
|
||||
//! Infrastructure — Weebs endpoints (anime/manga info, waifu pics, whatanime).
|
||||
//!
|
||||
//! Ported from Shirokami-API `scraper/weebs/*.js`. These are public JSON
|
||||
//! APIs that we proxy/forward (jikan.moe, waifu.pics, trace.moe).
|
||||
|
||||
use crate::infrastructure::utils::http_client::http_client;
|
||||
use reqwest::header::USER_AGENT;
|
||||
use serde_json::{json, Value};
|
||||
|
||||
/// Anime info from jikan.moe, ported from anime-search's `getAnimeInfo`.
|
||||
pub async fn fetch_anime_info(query: &str) -> Result<Value, String> {
|
||||
let url = format!("https://api.jikan.moe/v4/anime?q={}", urlencode(query));
|
||||
let data = get_json(&url).await?;
|
||||
|
||||
let arr = data
|
||||
.get("data")
|
||||
.and_then(|d| d.as_array())
|
||||
.cloned()
|
||||
.unwrap_or_default();
|
||||
if arr.is_empty() {
|
||||
return Ok(json!({ "error": "Anime tidak ditemukan" }));
|
||||
}
|
||||
let anime = &arr[0];
|
||||
|
||||
let judul = anime
|
||||
.get("titles")
|
||||
.and_then(|t| t.as_array())
|
||||
.and_then(|t| t.first())
|
||||
.and_then(|t| t.get("title"))
|
||||
.and_then(|t| t.as_str())
|
||||
.unwrap_or("Title not available");
|
||||
let genrenya = join_names(anime.get("genres"));
|
||||
|
||||
Ok(json!({
|
||||
"title": judul,
|
||||
"url": anime.get("url").and_then(|v| v.as_str()).unwrap_or(""),
|
||||
"type": anime.get("type").and_then(|v| v.as_str()).unwrap_or(""),
|
||||
"score": anime.get("score").cloned().unwrap_or(Value::Null),
|
||||
"members": anime.get("members").cloned().unwrap_or(Value::Null),
|
||||
"status": anime.get("status").and_then(|v| v.as_str()).unwrap_or(""),
|
||||
"synopsis": anime.get("synopsis").and_then(|v| v.as_str()).unwrap_or(""),
|
||||
"favorites": anime.get("favorites").cloned().unwrap_or(Value::Null),
|
||||
"images": anime.get("images").cloned().unwrap_or(Value::Null),
|
||||
"genres": genrenya,
|
||||
}))
|
||||
}
|
||||
|
||||
/// Manga info from jikan.moe, ported from anime-search's `getMangaInfo`.
|
||||
pub async fn fetch_manga_info(query: &str) -> Result<Value, String> {
|
||||
let url = format!("https://api.jikan.moe/v4/manga?q={}", urlencode(query));
|
||||
let data = get_json(&url).await?;
|
||||
|
||||
let arr = data
|
||||
.get("data")
|
||||
.and_then(|d| d.as_array())
|
||||
.cloned()
|
||||
.unwrap_or_default();
|
||||
if arr.is_empty() {
|
||||
return Ok(json!({ "error": "Manga tidak ditemukan" }));
|
||||
}
|
||||
let manga = &arr[0];
|
||||
|
||||
let judul = manga
|
||||
.get("titles")
|
||||
.and_then(|t| t.as_array())
|
||||
.map(|tt| {
|
||||
tt.iter()
|
||||
.filter_map(|j| {
|
||||
let title = j.get("title").and_then(|v| v.as_str()).unwrap_or("");
|
||||
let ty = j.get("type").and_then(|v| v.as_str()).unwrap_or("");
|
||||
if title.is_empty() {
|
||||
None
|
||||
} else {
|
||||
Some(format!("{} [{}]", title, ty))
|
||||
}
|
||||
})
|
||||
.collect::<Vec<_>>()
|
||||
.join(", ")
|
||||
})
|
||||
.unwrap_or_default();
|
||||
let genrenya = join_names(manga.get("genres"));
|
||||
|
||||
Ok(json!({
|
||||
"title": judul,
|
||||
"chapters": manga.get("chapters").cloned().unwrap_or(Value::Null),
|
||||
"type": manga.get("type").and_then(|v| v.as_str()).unwrap_or(""),
|
||||
"status": manga.get("status").and_then(|v| v.as_str()).unwrap_or(""),
|
||||
"genre": genrenya,
|
||||
"volumes": manga.get("volumes").cloned().unwrap_or(Value::Null),
|
||||
"favorites": manga.get("favorites").cloned().unwrap_or(Value::Null),
|
||||
"score": manga.get("score").cloned().unwrap_or(Value::Null),
|
||||
"scored": manga.get("scored").cloned().unwrap_or(Value::Null),
|
||||
"scored_by": manga.get("scored_by").cloned().unwrap_or(Value::Null),
|
||||
"rank": manga.get("rank").cloned().unwrap_or(Value::Null),
|
||||
"popularity": manga.get("popularity").cloned().unwrap_or(Value::Null),
|
||||
"members": manga.get("members").cloned().unwrap_or(Value::Null),
|
||||
"url": manga.get("url").and_then(|v| v.as_str()).unwrap_or(""),
|
||||
"background": manga.get("background").and_then(|v| v.as_str()).unwrap_or(""),
|
||||
"synopsis": manga.get("synopsis").and_then(|v| v.as_str()).unwrap_or(""),
|
||||
}))
|
||||
}
|
||||
|
||||
/// What anime is in this screenshot? trace.moe reverse-image search.
|
||||
pub async fn fetch_whatanime(url: &str) -> Result<Value, String> {
|
||||
let api = format!(
|
||||
"https://api.trace.moe/search?cutBorders&url={}",
|
||||
urlencode(url)
|
||||
);
|
||||
let data = get_json(&api).await?;
|
||||
|
||||
let result = data
|
||||
.get("result")
|
||||
.and_then(|r| r.as_array())
|
||||
.and_then(|r| r.first())
|
||||
.cloned();
|
||||
let Some(r) = result else {
|
||||
return Ok(json!({ "error": "Anime tidak ditemukan" }));
|
||||
};
|
||||
|
||||
let similarity = r
|
||||
.get("similarity")
|
||||
.and_then(|s| s.as_f64())
|
||||
.map(|s| (s * 100.0).round() as i64)
|
||||
.unwrap_or(0);
|
||||
|
||||
Ok(json!({
|
||||
"judul": r.get("filename").and_then(|v| v.as_str()).unwrap_or(""),
|
||||
"episode": r.get("episode").and_then(|v| v.as_str()).unwrap_or(""),
|
||||
"similarity": similarity,
|
||||
"videoURL": r.get("video").and_then(|v| v.as_str()).unwrap_or(""),
|
||||
"videoIMG": r.get("image").and_then(|v| v.as_str()).unwrap_or(""),
|
||||
}))
|
||||
}
|
||||
|
||||
/// SFW waifu image from waifu.pics.
|
||||
pub async fn fetch_sfw_waifu(tag: &str) -> Result<Value, String> {
|
||||
let url = format!("https://api.waifu.pics/sfw/{}", urlencode(tag));
|
||||
get_json(&url).await
|
||||
}
|
||||
|
||||
/// NSFW waifu image from waifu.pics.
|
||||
pub async fn fetch_nsfw_waifu(tag: &str) -> Result<Value, String> {
|
||||
let url = format!("https://api.waifu.pics/nsfw/{}", urlencode(tag));
|
||||
get_json(&url).await
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// Helpers
|
||||
// ============================================================================
|
||||
|
||||
async fn get_json(url: &str) -> Result<Value, String> {
|
||||
let resp = http_client()
|
||||
.client()
|
||||
.get(url)
|
||||
.header(USER_AGENT, "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0 Safari/537.36")
|
||||
.send()
|
||||
.await
|
||||
.map_err(|e| format!("HTTP error: {}", e))?;
|
||||
if !resp.status().is_success() {
|
||||
return Err(format!("HTTP {} from upstream", resp.status()));
|
||||
}
|
||||
resp.json::<Value>()
|
||||
.await
|
||||
.map_err(|e| format!("JSON parse: {}", e))
|
||||
}
|
||||
|
||||
/// Join an array of { name: "X" } objects into a comma-separated string.
|
||||
fn join_names(v: Option<&Value>) -> String {
|
||||
v.and_then(|g| g.as_array())
|
||||
.map(|arr| {
|
||||
arr.iter()
|
||||
.filter_map(|g| {
|
||||
g.get("name")
|
||||
.and_then(|n| n.as_str())
|
||||
.map(|s| s.to_string())
|
||||
})
|
||||
.collect::<Vec<_>>()
|
||||
.join(", ")
|
||||
})
|
||||
.unwrap_or_default()
|
||||
}
|
||||
|
||||
/// Minimal URL-encode (space -> %20, keep alphanumerics and safe chars).
|
||||
fn urlencode(s: &str) -> String {
|
||||
let mut out = String::new();
|
||||
for b in s.bytes() {
|
||||
match b {
|
||||
b'A'..=b'Z' | b'a'..=b'z' | b'0'..=b'9' | b'-' | b'_' | b'.' | b'~' => {
|
||||
out.push(b as char)
|
||||
}
|
||||
b' ' => out.push_str("%20"),
|
||||
_ => out.push_str(&format!("%{:02X}", b)),
|
||||
}
|
||||
}
|
||||
out
|
||||
}
|
||||
@@ -4,7 +4,7 @@ use utoipa::OpenApi;
|
||||
#[derive(OpenApi)]
|
||||
#[openapi(
|
||||
paths(
|
||||
// Anime module handlers
|
||||
// ── Anime module ──────────────────────────────────────────
|
||||
crate::presentation::handler::anime::anime_index,
|
||||
crate::presentation::handler::anime::genres,
|
||||
crate::presentation::handler::anime::detail_slug,
|
||||
@@ -16,7 +16,7 @@ use utoipa::OpenApi;
|
||||
crate::presentation::handler::anime::search_slug_page,
|
||||
crate::presentation::handler::anime::genre_slug_index,
|
||||
crate::presentation::handler::anime::genre_slug_page,
|
||||
// Anime2 module handlers
|
||||
// ── Anime2 module ─────────────────────────────────────────
|
||||
crate::presentation::handler::anime2::index,
|
||||
crate::presentation::handler::anime2::genre_list,
|
||||
crate::presentation::handler::anime2::filter,
|
||||
@@ -28,7 +28,7 @@ use utoipa::OpenApi;
|
||||
crate::presentation::handler::anime2::latest_slug,
|
||||
crate::presentation::handler::anime2::ongoing_anime_slug,
|
||||
crate::presentation::handler::anime2::complete_anime_slug,
|
||||
// Komik module handlers
|
||||
// ── Komik module ──────────────────────────────────────────
|
||||
crate::presentation::handler::komik::genre_list,
|
||||
crate::presentation::handler::komik::chapter_slug,
|
||||
crate::presentation::handler::komik::detail_slug,
|
||||
@@ -40,19 +40,75 @@ use utoipa::OpenApi;
|
||||
crate::presentation::handler::komik::popular_slug,
|
||||
crate::presentation::handler::komik::search_slug,
|
||||
crate::presentation::handler::komik::search_slug_page,
|
||||
// Proxy module handlers
|
||||
// ── Download module ────────────────────────────────────────
|
||||
crate::presentation::handler::downloader::download,
|
||||
crate::presentation::handler::downloader::detect_platform_handler,
|
||||
crate::presentation::handler::downloader::download_instagram,
|
||||
crate::presentation::handler::downloader::download_facebook,
|
||||
crate::presentation::handler::downloader::download_tiktok,
|
||||
crate::presentation::handler::downloader::download_youtube,
|
||||
crate::presentation::handler::downloader::download_youtube_mp3,
|
||||
crate::presentation::handler::downloader::download_spotify,
|
||||
crate::presentation::handler::downloader::download_twitter,
|
||||
crate::presentation::handler::downloader::download_pinterest,
|
||||
crate::presentation::handler::downloader::download_mega,
|
||||
crate::presentation::handler::downloader::download_terabox,
|
||||
crate::presentation::handler::downloader::download_gdrive,
|
||||
crate::presentation::handler::downloader::download_mediafire,
|
||||
crate::presentation::handler::downloader::download_pixeldrain,
|
||||
crate::presentation::handler::downloader::download_threads,
|
||||
crate::presentation::handler::downloader::download_doodstream,
|
||||
crate::presentation::handler::downloader::download_krakenfiles,
|
||||
crate::presentation::handler::downloader::download_danbooru,
|
||||
crate::presentation::handler::downloader::download_soundcloud,
|
||||
crate::presentation::handler::downloader::download_dailymotion,
|
||||
crate::presentation::handler::downloader::download_reddit,
|
||||
crate::presentation::handler::downloader::download_streamable,
|
||||
crate::presentation::handler::downloader::download_videy,
|
||||
crate::presentation::handler::downloader::download_bilibili,
|
||||
crate::presentation::handler::downloader::serve_merged_file,
|
||||
// ── Misc module ────────────────────────────────────────────
|
||||
crate::presentation::handler::misc::currency_converter_handler,
|
||||
crate::presentation::handler::misc::harga_emas_handler,
|
||||
crate::presentation::handler::misc::kurs_bca_handler,
|
||||
crate::presentation::handler::misc::server_info_handler,
|
||||
// ── Stalk module ───────────────────────────────────────────
|
||||
crate::presentation::handler::stalk::github_handler,
|
||||
crate::presentation::handler::stalk::youtube_handler,
|
||||
crate::presentation::handler::stalk::twitter_handler,
|
||||
// ── Search module ──────────────────────────────────────────
|
||||
crate::presentation::handler::search::bmkg_handler,
|
||||
crate::presentation::handler::search::jadwal_sholat_handler,
|
||||
crate::presentation::handler::search::weather_handler,
|
||||
crate::presentation::handler::search::google_handler,
|
||||
crate::presentation::handler::search::yt_handler,
|
||||
// ── Tool module ────────────────────────────────────────────
|
||||
crate::presentation::handler::tools::whois_handler,
|
||||
crate::presentation::handler::tools::ip_location_handler,
|
||||
crate::presentation::handler::tools::tinyurl_handler,
|
||||
crate::presentation::handler::tools::check_hosting_handler,
|
||||
crate::presentation::handler::tools::hargapangan_handler,
|
||||
crate::presentation::handler::tools::cek_resi_handler,
|
||||
// ── Image module ───────────────────────────────────────────
|
||||
crate::presentation::handler::image::brat_handler,
|
||||
crate::presentation::handler::image::brat_animated_handler,
|
||||
),
|
||||
components(
|
||||
schemas(
|
||||
// Application response wrapper
|
||||
crate::presentation::dto::common::ApiResponse<String>,
|
||||
crate::presentation::dto::downloader::DownloadResponse,
|
||||
)
|
||||
),
|
||||
tags(
|
||||
(name = "anime", description = "Anime endpoints"),
|
||||
(name = "anime2", description = "Anime2 endpoints"),
|
||||
(name = "komik", description = "Komik endpoints"),
|
||||
(name = "proxy", description = "Proxy endpoints"),
|
||||
(name = "download", description = "Media download endpoints (23 platforms)"),
|
||||
(name = "misc", description = "Miscellaneous utilities (currency, gold, server)"),
|
||||
(name = "stalk", description = "User profile stalking (GitHub, YT, Twitter)"),
|
||||
(name = "search", description = "Search endpoints (BMKG, prayer times, weather, YouTube, Google)"),
|
||||
(name = "tool", description = "Utility tools (WHOIS, IP lookup, URL shortener, tracking, hosting)"),
|
||||
(name = "image", description = "Image generation (brat static & animated)"),
|
||||
)
|
||||
)]
|
||||
pub struct ModuleApiDoc;
|
||||
@@ -7,40 +7,85 @@
|
||||
use axum::extract::Query;
|
||||
use axum::Json;
|
||||
use serde::Deserialize;
|
||||
use serde_json::Value;
|
||||
use utoipa::IntoParams;
|
||||
|
||||
use crate::application::downloader as use_cases;
|
||||
use crate::presentation::dto::downloader::DownloadResponse;
|
||||
use crate::presentation::error::AppError;
|
||||
|
||||
// ============================================================================
|
||||
// DownloadParams + response
|
||||
// ============================================================================
|
||||
|
||||
/// Request params for downloader endpoints.
|
||||
#[derive(Debug, Deserialize)]
|
||||
#[derive(Debug, Deserialize, IntoParams)]
|
||||
pub struct DownloadParams {
|
||||
pub url: String,
|
||||
pub cookies: Option<String>,
|
||||
pub api_key: Option<String>,
|
||||
pub quality: Option<String>,
|
||||
/// When `true`, YouTube downloads are merged server-side (video + audio
|
||||
/// into a single MP4) instead of returning separate video/audio URLs.
|
||||
pub merge: Option<String>,
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// ========================================================================
|
||||
// Handlers
|
||||
// ============================================================================
|
||||
// ========================================================================
|
||||
|
||||
/// All-in-one downloader with auto-detection.
|
||||
/// All-in-one auto-detect downloader.
|
||||
/// `/download?url=...`
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/download",
|
||||
tag = "download",
|
||||
operation_id = "dl_download",
|
||||
params(DownloadParams),
|
||||
responses(
|
||||
(status = 200, description = "Download result", body = DownloadResponse),
|
||||
(status = 400, description = "Bad URL or platform"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn download(
|
||||
Query(params): Query<DownloadParams>,
|
||||
) -> Result<Json<DownloadResponse>, AppError> {
|
||||
let result = use_cases::download_all_in_one(¶ms.url, params.cookies.as_deref()).await?;
|
||||
let merge = params
|
||||
.merge
|
||||
.as_deref()
|
||||
.map(|v| v == "true" || v == "1")
|
||||
.unwrap_or(false);
|
||||
let result = if merge {
|
||||
// All-in-one with merge=true → force merge for YouTube URLs.
|
||||
if crate::application::downloader::detect_platform(¶ms.url) == "youtube" {
|
||||
use_cases::download_youtube_merge(
|
||||
¶ms.url,
|
||||
params.quality.as_deref().unwrap_or("720"),
|
||||
)
|
||||
.await?
|
||||
} else {
|
||||
use_cases::download_all_in_one(¶ms.url, params.cookies.as_deref()).await?
|
||||
}
|
||||
} else {
|
||||
use_cases::download_all_in_one(¶ms.url, params.cookies.as_deref()).await?
|
||||
};
|
||||
Ok(Json(DownloadResponse::ok(result)))
|
||||
}
|
||||
|
||||
/// Platform detection endpoint.
|
||||
/// Detect platform from URL.
|
||||
/// `/detect-platform?url=...`
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/detect-platform",
|
||||
tag = "download",
|
||||
operation_id = "dl_detect_platform_handler",
|
||||
params(DownloadParams),
|
||||
responses(
|
||||
(status = 200, description = "Download result", body = DownloadResponse),
|
||||
(status = 400, description = "Bad URL or platform"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn detect_platform_handler(
|
||||
Query(params): Query<DownloadParams>,
|
||||
) -> Result<Json<serde_json::Value>, AppError> {
|
||||
) -> Result<Json<Value>, AppError> {
|
||||
let platform = use_cases::detect_platform(¶ms.url);
|
||||
Ok(Json(serde_json::json!({
|
||||
"platform": platform,
|
||||
@@ -48,8 +93,20 @@ pub async fn detect_platform_handler(
|
||||
})))
|
||||
}
|
||||
|
||||
/// Generate handler for each downloader platform.
|
||||
/// Handler for _instagram.
|
||||
/// Download Instagram media.
|
||||
/// `/download/instagram?url=...`
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/download/instagram",
|
||||
tag = "download",
|
||||
operation_id = "dl_download_instagram",
|
||||
params(DownloadParams),
|
||||
responses(
|
||||
(status = 200, description = "Download result", body = DownloadResponse),
|
||||
(status = 400, description = "Bad URL or platform"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn download_instagram(
|
||||
Query(params): Query<DownloadParams>,
|
||||
) -> Result<Json<DownloadResponse>, AppError> {
|
||||
@@ -57,7 +114,20 @@ pub async fn download_instagram(
|
||||
Ok(Json(DownloadResponse::ok(result)))
|
||||
}
|
||||
|
||||
/// Handler for _facebook.
|
||||
/// Download Facebook video.
|
||||
/// `/download/facebook?url=...`
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/download/facebook",
|
||||
tag = "download",
|
||||
operation_id = "dl_download_facebook",
|
||||
params(DownloadParams),
|
||||
responses(
|
||||
(status = 200, description = "Download result", body = DownloadResponse),
|
||||
(status = 400, description = "Bad URL or platform"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn download_facebook(
|
||||
Query(params): Query<DownloadParams>,
|
||||
) -> Result<Json<DownloadResponse>, AppError> {
|
||||
@@ -65,7 +135,20 @@ pub async fn download_facebook(
|
||||
Ok(Json(DownloadResponse::ok(result)))
|
||||
}
|
||||
|
||||
/// Handler for _tiktok.
|
||||
/// Download TikTok video.
|
||||
/// `/download/tiktok?url=...`
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/download/tiktok",
|
||||
tag = "download",
|
||||
operation_id = "dl_download_tiktok",
|
||||
params(DownloadParams),
|
||||
responses(
|
||||
(status = 200, description = "Download result", body = DownloadResponse),
|
||||
(status = 400, description = "Bad URL or platform"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn download_tiktok(
|
||||
Query(params): Query<DownloadParams>,
|
||||
) -> Result<Json<DownloadResponse>, AppError> {
|
||||
@@ -73,17 +156,51 @@ pub async fn download_tiktok(
|
||||
Ok(Json(DownloadResponse::ok(result)))
|
||||
}
|
||||
|
||||
/// Handler for _youtube.
|
||||
/// Download YouTube video.
|
||||
/// `/download/youtube?url=...`
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/download/youtube",
|
||||
tag = "download",
|
||||
operation_id = "dl_download_youtube",
|
||||
params(DownloadParams),
|
||||
responses(
|
||||
(status = 200, description = "Download result", body = DownloadResponse),
|
||||
(status = 400, description = "Bad URL or platform"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn download_youtube(
|
||||
Query(params): Query<DownloadParams>,
|
||||
) -> Result<Json<DownloadResponse>, AppError> {
|
||||
let result =
|
||||
use_cases::download_youtube(¶ms.url, params.quality.as_deref().unwrap_or("720"))
|
||||
.await?;
|
||||
let quality = params.quality.as_deref().unwrap_or("720");
|
||||
let merge = params
|
||||
.merge
|
||||
.as_deref()
|
||||
.map(|v| v == "true" || v == "1")
|
||||
.unwrap_or(false);
|
||||
let result = if merge {
|
||||
use_cases::download_youtube_merge(¶ms.url, quality).await?
|
||||
} else {
|
||||
use_cases::download_youtube(¶ms.url, quality).await?
|
||||
};
|
||||
Ok(Json(DownloadResponse::ok(result)))
|
||||
}
|
||||
|
||||
/// Handler for _youtube_mp3.
|
||||
/// Convert YouTube to MP3.
|
||||
/// `/download/youtube/mp3?url=...`
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/download/youtube/mp3",
|
||||
tag = "download",
|
||||
operation_id = "dl_download_youtube_mp3",
|
||||
params(DownloadParams),
|
||||
responses(
|
||||
(status = 200, description = "Download result", body = DownloadResponse),
|
||||
(status = 400, description = "Bad URL or platform"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn download_youtube_mp3(
|
||||
Query(params): Query<DownloadParams>,
|
||||
) -> Result<Json<DownloadResponse>, AppError> {
|
||||
@@ -91,7 +208,20 @@ pub async fn download_youtube_mp3(
|
||||
Ok(Json(DownloadResponse::ok(result)))
|
||||
}
|
||||
|
||||
/// Handler for _spotify.
|
||||
/// Download Spotify track.
|
||||
/// `/download/spotify?url=...`
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/download/spotify",
|
||||
tag = "download",
|
||||
operation_id = "dl_download_spotify",
|
||||
params(DownloadParams),
|
||||
responses(
|
||||
(status = 200, description = "Download result", body = DownloadResponse),
|
||||
(status = 400, description = "Bad URL or platform"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn download_spotify(
|
||||
Query(params): Query<DownloadParams>,
|
||||
) -> Result<Json<DownloadResponse>, AppError> {
|
||||
@@ -99,7 +229,20 @@ pub async fn download_spotify(
|
||||
Ok(Json(DownloadResponse::ok(result)))
|
||||
}
|
||||
|
||||
/// Handler for _twitter.
|
||||
/// Download Twitter/X media.
|
||||
/// `/download/twitter?url=...`
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/download/twitter",
|
||||
tag = "download",
|
||||
operation_id = "dl_download_twitter",
|
||||
params(DownloadParams),
|
||||
responses(
|
||||
(status = 200, description = "Download result", body = DownloadResponse),
|
||||
(status = 400, description = "Bad URL or platform"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn download_twitter(
|
||||
Query(params): Query<DownloadParams>,
|
||||
) -> Result<Json<DownloadResponse>, AppError> {
|
||||
@@ -107,7 +250,20 @@ pub async fn download_twitter(
|
||||
Ok(Json(DownloadResponse::ok(result)))
|
||||
}
|
||||
|
||||
/// Handler for _pinterest.
|
||||
/// Download Pinterest pin.
|
||||
/// `/download/pinterest?url=...`
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/download/pinterest",
|
||||
tag = "download",
|
||||
operation_id = "dl_download_pinterest",
|
||||
params(DownloadParams),
|
||||
responses(
|
||||
(status = 200, description = "Download result", body = DownloadResponse),
|
||||
(status = 400, description = "Bad URL or platform"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn download_pinterest(
|
||||
Query(params): Query<DownloadParams>,
|
||||
) -> Result<Json<DownloadResponse>, AppError> {
|
||||
@@ -115,7 +271,20 @@ pub async fn download_pinterest(
|
||||
Ok(Json(DownloadResponse::ok(result)))
|
||||
}
|
||||
|
||||
/// Handler for _mega.
|
||||
/// Download from Mega.
|
||||
/// `/download/mega?url=...`
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/download/mega",
|
||||
tag = "download",
|
||||
operation_id = "dl_download_mega",
|
||||
params(DownloadParams),
|
||||
responses(
|
||||
(status = 200, description = "Download result", body = DownloadResponse),
|
||||
(status = 400, description = "Bad URL or platform"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn download_mega(
|
||||
Query(params): Query<DownloadParams>,
|
||||
) -> Result<Json<DownloadResponse>, AppError> {
|
||||
@@ -123,7 +292,20 @@ pub async fn download_mega(
|
||||
Ok(Json(DownloadResponse::ok(result)))
|
||||
}
|
||||
|
||||
/// Handler for _terabox.
|
||||
/// Download from Terabox.
|
||||
/// `/download/terabox?url=...`
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/download/terabox",
|
||||
tag = "download",
|
||||
operation_id = "dl_download_terabox",
|
||||
params(DownloadParams),
|
||||
responses(
|
||||
(status = 200, description = "Download result", body = DownloadResponse),
|
||||
(status = 400, description = "Bad URL or platform"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn download_terabox(
|
||||
Query(params): Query<DownloadParams>,
|
||||
) -> Result<Json<DownloadResponse>, AppError> {
|
||||
@@ -131,7 +313,115 @@ pub async fn download_terabox(
|
||||
Ok(Json(DownloadResponse::ok(result)))
|
||||
}
|
||||
|
||||
/// Handler for _gdrive.
|
||||
/// Stream a TeraBox file through this API.
|
||||
/// `/proxy/terabox?surl=...`
|
||||
///
|
||||
/// The resolver's `/download` endpoint replays the full 302 chain WITH
|
||||
/// cookies and streams real bytes — this handler forwards those bytes to the
|
||||
/// caller so the returned URL is a public, downloadable link (not a 403
|
||||
/// dlink, not a private 127.0.0.1 URL).
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/proxy/terabox",
|
||||
tag = "download",
|
||||
operation_id = "dl_proxy_terabox",
|
||||
params(
|
||||
("surl" = String, Query, description = "TeraBox share short URL ID"),
|
||||
("filename" = Option<String>, Query, description = "Download file name hint"),
|
||||
),
|
||||
responses(
|
||||
(status = 200, description = "File stream"),
|
||||
(status = 400, description = "Missing surl"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn proxy_terabox(
|
||||
Query(params): Query<TeraboxProxyParams>,
|
||||
) -> Result<axum::response::Response, AppError> {
|
||||
let surl = params.surl;
|
||||
if surl.is_empty() {
|
||||
return Err(AppError::BadRequest("missing surl parameter".into()));
|
||||
}
|
||||
|
||||
let resolver_url = std::env::var("TERABOX_RESOLVER_URL")
|
||||
.unwrap_or_else(|_| "http://127.0.0.1:4092".to_string());
|
||||
let fetch_url = format!("{}/download?surl={}", resolver_url, surl);
|
||||
|
||||
// Ask the resolver for the redirect chain (it follows 302s internally with
|
||||
// cookies) — but we only want the final CDN stream URL. The resolver
|
||||
// already streams; simplest reliable path is to just proxy the resolver's
|
||||
// response body as-is.
|
||||
let client = crate::infrastructure::utils::http_client::http_client();
|
||||
|
||||
let resp = client
|
||||
.client()
|
||||
.get(&fetch_url)
|
||||
.header("User-Agent", "Mozilla/5.0")
|
||||
.timeout(std::time::Duration::from_secs(120))
|
||||
.send()
|
||||
.await
|
||||
.map_err(|e| AppError::ScraperError(format!("terabox proxy fetch failed: {e}")))?;
|
||||
|
||||
if !resp.status().is_success() {
|
||||
let status = resp.status();
|
||||
let body = resp.text().await.unwrap_or_default();
|
||||
return Err(AppError::ScraperError(format!(
|
||||
"terabox proxy returned HTTP {status}: {body}"
|
||||
)));
|
||||
}
|
||||
|
||||
let content_type = resp
|
||||
.headers()
|
||||
.get(reqwest::header::CONTENT_TYPE)
|
||||
.and_then(|v| v.to_str().ok())
|
||||
.unwrap_or("application/octet-stream")
|
||||
.to_string();
|
||||
|
||||
let content_disposition = resp
|
||||
.headers()
|
||||
.get(reqwest::header::CONTENT_DISPOSITION)
|
||||
.and_then(|v| v.to_str().ok())
|
||||
.map(|s| s.to_string())
|
||||
.unwrap_or_else(|| {
|
||||
let fname = params
|
||||
.filename
|
||||
.filter(|s| !s.is_empty())
|
||||
.unwrap_or_else(|| "terabox-download.bin".to_string());
|
||||
format!("attachment; filename=\"{fname}\"")
|
||||
});
|
||||
|
||||
let stream = resp.bytes_stream();
|
||||
|
||||
Ok(axum::response::Response::builder()
|
||||
.header("Content-Type", content_type)
|
||||
.header("Content-Disposition", content_disposition)
|
||||
.header("X-Accel-Buffering", "no")
|
||||
.header("Cache-Control", "no-store")
|
||||
.body(axum::body::Body::from_stream(stream))
|
||||
.map_err(|e| AppError::Internal(format!("build stream response: {e}")))?)
|
||||
}
|
||||
|
||||
/// Params for the TeraBox streaming proxy.
|
||||
#[derive(Debug, Deserialize, IntoParams)]
|
||||
pub struct TeraboxProxyParams {
|
||||
pub surl: String,
|
||||
pub filename: Option<String>,
|
||||
}
|
||||
|
||||
/// Download from Google Drive.
|
||||
/// `/download/gdrive?url=...`
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/download/gdrive",
|
||||
tag = "download",
|
||||
operation_id = "dl_download_gdrive",
|
||||
params(DownloadParams),
|
||||
responses(
|
||||
(status = 200, description = "Download result", body = DownloadResponse),
|
||||
(status = 400, description = "Bad URL or platform"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn download_gdrive(
|
||||
Query(params): Query<DownloadParams>,
|
||||
) -> Result<Json<DownloadResponse>, AppError> {
|
||||
@@ -139,7 +429,20 @@ pub async fn download_gdrive(
|
||||
Ok(Json(DownloadResponse::ok(result)))
|
||||
}
|
||||
|
||||
/// Handler for _mediafire.
|
||||
/// Download from MediaFire.
|
||||
/// `/download/mediafire?url=...`
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/download/mediafire",
|
||||
tag = "download",
|
||||
operation_id = "dl_download_mediafire",
|
||||
params(DownloadParams),
|
||||
responses(
|
||||
(status = 200, description = "Download result", body = DownloadResponse),
|
||||
(status = 400, description = "Bad URL or platform"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn download_mediafire(
|
||||
Query(params): Query<DownloadParams>,
|
||||
) -> Result<Json<DownloadResponse>, AppError> {
|
||||
@@ -147,7 +450,20 @@ pub async fn download_mediafire(
|
||||
Ok(Json(DownloadResponse::ok(result)))
|
||||
}
|
||||
|
||||
/// Handler for _pixeldrain.
|
||||
/// Download from Pixeldrain.
|
||||
/// `/download/pixeldrain?url=...`
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/download/pixeldrain",
|
||||
tag = "download",
|
||||
operation_id = "dl_download_pixeldrain",
|
||||
params(DownloadParams),
|
||||
responses(
|
||||
(status = 200, description = "Download result", body = DownloadResponse),
|
||||
(status = 400, description = "Bad URL or platform"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn download_pixeldrain(
|
||||
Query(params): Query<DownloadParams>,
|
||||
) -> Result<Json<DownloadResponse>, AppError> {
|
||||
@@ -155,7 +471,20 @@ pub async fn download_pixeldrain(
|
||||
Ok(Json(DownloadResponse::ok(result)))
|
||||
}
|
||||
|
||||
/// Handler for _threads.
|
||||
/// Download Threads media.
|
||||
/// `/download/threads?url=...`
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/download/threads",
|
||||
tag = "download",
|
||||
operation_id = "dl_download_threads",
|
||||
params(DownloadParams),
|
||||
responses(
|
||||
(status = 200, description = "Download result", body = DownloadResponse),
|
||||
(status = 400, description = "Bad URL or platform"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn download_threads(
|
||||
Query(params): Query<DownloadParams>,
|
||||
) -> Result<Json<DownloadResponse>, AppError> {
|
||||
@@ -163,7 +492,20 @@ pub async fn download_threads(
|
||||
Ok(Json(DownloadResponse::ok(result)))
|
||||
}
|
||||
|
||||
/// Handler for _doodstream.
|
||||
/// Download from DoodStream.
|
||||
/// `/download/doodstream?url=...`
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/download/doodstream",
|
||||
tag = "download",
|
||||
operation_id = "dl_download_doodstream",
|
||||
params(DownloadParams),
|
||||
responses(
|
||||
(status = 200, description = "Download result", body = DownloadResponse),
|
||||
(status = 400, description = "Bad URL or platform"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn download_doodstream(
|
||||
Query(params): Query<DownloadParams>,
|
||||
) -> Result<Json<DownloadResponse>, AppError> {
|
||||
@@ -171,7 +513,20 @@ pub async fn download_doodstream(
|
||||
Ok(Json(DownloadResponse::ok(result)))
|
||||
}
|
||||
|
||||
/// Handler for _krakenfiles.
|
||||
/// Download from KrakenFiles.
|
||||
/// `/download/krakenfiles?url=...`
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/download/krakenfiles",
|
||||
tag = "download",
|
||||
operation_id = "dl_download_krakenfiles",
|
||||
params(DownloadParams),
|
||||
responses(
|
||||
(status = 200, description = "Download result", body = DownloadResponse),
|
||||
(status = 400, description = "Bad URL or platform"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn download_krakenfiles(
|
||||
Query(params): Query<DownloadParams>,
|
||||
) -> Result<Json<DownloadResponse>, AppError> {
|
||||
@@ -179,7 +534,20 @@ pub async fn download_krakenfiles(
|
||||
Ok(Json(DownloadResponse::ok(result)))
|
||||
}
|
||||
|
||||
/// Handler for _danbooru.
|
||||
/// Download from Danbooru.
|
||||
/// `/download/danbooru?url=...`
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/download/danbooru",
|
||||
tag = "download",
|
||||
operation_id = "dl_download_danbooru",
|
||||
params(DownloadParams),
|
||||
responses(
|
||||
(status = 200, description = "Download result", body = DownloadResponse),
|
||||
(status = 400, description = "Bad URL or platform"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn download_danbooru(
|
||||
Query(params): Query<DownloadParams>,
|
||||
) -> Result<Json<DownloadResponse>, AppError> {
|
||||
@@ -187,7 +555,20 @@ pub async fn download_danbooru(
|
||||
Ok(Json(DownloadResponse::ok(result)))
|
||||
}
|
||||
|
||||
/// Handler for _soundcloud.
|
||||
/// Download SoundCloud track.
|
||||
/// `/download/soundcloud?url=...`
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/download/soundcloud",
|
||||
tag = "download",
|
||||
operation_id = "dl_download_soundcloud",
|
||||
params(DownloadParams),
|
||||
responses(
|
||||
(status = 200, description = "Download result", body = DownloadResponse),
|
||||
(status = 400, description = "Bad URL or platform"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn download_soundcloud(
|
||||
Query(params): Query<DownloadParams>,
|
||||
) -> Result<Json<DownloadResponse>, AppError> {
|
||||
@@ -195,7 +576,20 @@ pub async fn download_soundcloud(
|
||||
Ok(Json(DownloadResponse::ok(result)))
|
||||
}
|
||||
|
||||
/// Handler for _dailymotion.
|
||||
/// Download Dailymotion video.
|
||||
/// `/download/dailymotion?url=...`
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/download/dailymotion",
|
||||
tag = "download",
|
||||
operation_id = "dl_download_dailymotion",
|
||||
params(DownloadParams),
|
||||
responses(
|
||||
(status = 200, description = "Download result", body = DownloadResponse),
|
||||
(status = 400, description = "Bad URL or platform"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn download_dailymotion(
|
||||
Query(params): Query<DownloadParams>,
|
||||
) -> Result<Json<DownloadResponse>, AppError> {
|
||||
@@ -203,7 +597,20 @@ pub async fn download_dailymotion(
|
||||
Ok(Json(DownloadResponse::ok(result)))
|
||||
}
|
||||
|
||||
/// Handler for _reddit.
|
||||
/// Download Reddit media.
|
||||
/// `/download/reddit?url=...`
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/download/reddit",
|
||||
tag = "download",
|
||||
operation_id = "dl_download_reddit",
|
||||
params(DownloadParams),
|
||||
responses(
|
||||
(status = 200, description = "Download result", body = DownloadResponse),
|
||||
(status = 400, description = "Bad URL or platform"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn download_reddit(
|
||||
Query(params): Query<DownloadParams>,
|
||||
) -> Result<Json<DownloadResponse>, AppError> {
|
||||
@@ -211,7 +618,20 @@ pub async fn download_reddit(
|
||||
Ok(Json(DownloadResponse::ok(result)))
|
||||
}
|
||||
|
||||
/// Handler for _streamable.
|
||||
/// Download Streamable video.
|
||||
/// `/download/streamable?url=...`
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/download/streamable",
|
||||
tag = "download",
|
||||
operation_id = "dl_download_streamable",
|
||||
params(DownloadParams),
|
||||
responses(
|
||||
(status = 200, description = "Download result", body = DownloadResponse),
|
||||
(status = 400, description = "Bad URL or platform"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn download_streamable(
|
||||
Query(params): Query<DownloadParams>,
|
||||
) -> Result<Json<DownloadResponse>, AppError> {
|
||||
@@ -219,7 +639,20 @@ pub async fn download_streamable(
|
||||
Ok(Json(DownloadResponse::ok(result)))
|
||||
}
|
||||
|
||||
/// Handler for _videy.
|
||||
/// Download Videy video.
|
||||
/// `/download/videy?url=...`
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/download/videy",
|
||||
tag = "download",
|
||||
operation_id = "dl_download_videy",
|
||||
params(DownloadParams),
|
||||
responses(
|
||||
(status = 200, description = "Download result", body = DownloadResponse),
|
||||
(status = 400, description = "Bad URL or platform"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn download_videy(
|
||||
Query(params): Query<DownloadParams>,
|
||||
) -> Result<Json<DownloadResponse>, AppError> {
|
||||
@@ -227,10 +660,82 @@ pub async fn download_videy(
|
||||
Ok(Json(DownloadResponse::ok(result)))
|
||||
}
|
||||
|
||||
/// Handler for _bilibili.
|
||||
/// Download Bilibili video.
|
||||
/// `/download/bilibili?url=...`
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/download/bilibili",
|
||||
tag = "download",
|
||||
operation_id = "dl_download_bilibili",
|
||||
params(DownloadParams),
|
||||
responses(
|
||||
(status = 200, description = "Download result", body = DownloadResponse),
|
||||
(status = 400, description = "Bad URL or platform"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn download_bilibili(
|
||||
Query(params): Query<DownloadParams>,
|
||||
) -> Result<Json<DownloadResponse>, AppError> {
|
||||
let result = use_cases::download_bilibili(¶ms.url).await?;
|
||||
Ok(Json(DownloadResponse::ok(result)))
|
||||
}
|
||||
|
||||
// ========================================================================
|
||||
// Merged-file static serving
|
||||
// ========================================================================
|
||||
|
||||
/// Serve a merged YouTube MP4 produced by `/download/youtube?merge=true`.
|
||||
/// Path is sanitised: only a plain filename (no `/`, `..`) inside the
|
||||
/// `yt_merge` uploads dir is allowed.
|
||||
///
|
||||
/// `/file/yt_merge/{filename}`
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/file/yt_merge/{filename}",
|
||||
tag = "download",
|
||||
operation_id = "dl_serve_merged_file",
|
||||
params(
|
||||
("filename" = String, Path, description = "Merged MP4 file name")
|
||||
),
|
||||
responses(
|
||||
(status = 200, description = "Merged MP4 file", content_type = "video/mp4"),
|
||||
(status = 404, description = "File not found or invalid name")
|
||||
)
|
||||
)]
|
||||
pub async fn serve_merged_file(
|
||||
axum::extract::Path(filename): axum::extract::Path<String>,
|
||||
) -> Result<axum::response::Response, AppError> {
|
||||
// Reject any path traversal or nested segments.
|
||||
if filename.contains('/')
|
||||
|| filename.contains('\\')
|
||||
|| filename.contains("..")
|
||||
|| filename.is_empty()
|
||||
{
|
||||
return Err(AppError::NotFound(format!("invalid file name: {filename}")));
|
||||
}
|
||||
|
||||
let dir = std::path::PathBuf::from("/var/lib/scraper/uploads/yt_merge");
|
||||
let path = dir.join(&filename);
|
||||
let bytes = tokio::fs::read(&path)
|
||||
.await
|
||||
.map_err(|_| AppError::NotFound(format!("file not found: {filename}")))?;
|
||||
|
||||
let content_type = if filename.ends_with(".mp4") {
|
||||
"video/mp4"
|
||||
} else if filename.ends_with(".webm") {
|
||||
"video/webm"
|
||||
} else {
|
||||
"application/octet-stream"
|
||||
};
|
||||
|
||||
Ok(axum::response::Response::builder()
|
||||
.header("Content-Type", content_type)
|
||||
.header("Content-Length", bytes.len().to_string())
|
||||
.header(
|
||||
"Content-Disposition",
|
||||
format!("attachment; filename=\"{filename}\""),
|
||||
)
|
||||
.body(axum::body::Body::from(bytes))
|
||||
.map_err(|e| AppError::Internal(format!("build response: {e}")))?)
|
||||
}
|
||||
@@ -7,35 +7,62 @@ use axum::extract::Query;
|
||||
use axum::http::{header, StatusCode};
|
||||
use axum::response::Response;
|
||||
use serde::Deserialize;
|
||||
use utoipa::IntoParams;
|
||||
|
||||
use crate::application::image as use_cases;
|
||||
use crate::presentation::error::AppError;
|
||||
|
||||
#[derive(Deserialize)]
|
||||
#[derive(Deserialize, IntoParams)]
|
||||
pub struct BratParams {
|
||||
pub text: Option<String>,
|
||||
}
|
||||
|
||||
/// GET /image/brat?text=Hello
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/image/brat",
|
||||
tag = "image",
|
||||
operation_id = "image_brat",
|
||||
params(BratParams),
|
||||
responses(
|
||||
(status = 200, description = "Static brat image (PNG)", body = Vec<u8>),
|
||||
(status = 400, description = "Missing text"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn brat_handler(Query(params): Query<BratParams>) -> Result<Response, AppError> {
|
||||
let text = params
|
||||
.text
|
||||
.ok_or_else(|| AppError::BadRequest("text parameter is required".into()))?;
|
||||
let bytes = use_cases::brat(&text).await?;
|
||||
Ok(Response::builder()
|
||||
Response::builder()
|
||||
.status(StatusCode::OK)
|
||||
.header(header::CONTENT_TYPE, "image/png")
|
||||
.body(Body::from(bytes))
|
||||
.expect("valid response"))
|
||||
.map_err(|e| AppError::Internal(format!("build response: {e}")))
|
||||
}
|
||||
|
||||
/// GET /image/brat/animated?text=Hello
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/image/brat/animated",
|
||||
tag = "image",
|
||||
operation_id = "image_brat_animated",
|
||||
params(BratParams),
|
||||
responses(
|
||||
(status = 200, description = "Animated brat image (GIF)", body = Vec<u8>),
|
||||
(status = 400, description = "Missing text"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn brat_animated_handler(Query(params): Query<BratParams>) -> Result<Response, AppError> {
|
||||
let text = params
|
||||
.text
|
||||
.ok_or_else(|| AppError::BadRequest("text parameter is required".into()))?;
|
||||
let bytes = use_cases::brat_animated(&text).await?;
|
||||
Ok(Response::builder()
|
||||
Response::builder()
|
||||
.status(StatusCode::OK)
|
||||
.header(header::CONTENT_TYPE, "image/gif")
|
||||
.body(Body::from(bytes))
|
||||
.expect("valid response"))
|
||||
.map_err(|e| AppError::Internal(format!("build response: {e}")))
|
||||
}
|
||||
@@ -7,12 +7,13 @@ use axum::http::StatusCode;
|
||||
use axum::Json;
|
||||
use serde::Deserialize;
|
||||
use serde_json::Value;
|
||||
use utoipa::IntoParams;
|
||||
|
||||
use crate::application::misc as use_cases;
|
||||
use crate::presentation::error::AppError;
|
||||
|
||||
/// Query params for currency converter.
|
||||
#[derive(Deserialize)]
|
||||
#[derive(Deserialize, IntoParams)]
|
||||
pub struct CurrencyParams {
|
||||
pub amount: Option<f64>,
|
||||
pub from: Option<String>,
|
||||
@@ -20,6 +21,17 @@ pub struct CurrencyParams {
|
||||
}
|
||||
|
||||
/// GET /misc/currency-converter?amount=100&from=USD&to=IDR
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/misc/currency-converter",
|
||||
tag = "misc",
|
||||
operation_id = "currency_converter",
|
||||
params(CurrencyParams),
|
||||
responses(
|
||||
(status = 200, description = "Currency conversion result", body = Value),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn currency_converter_handler(
|
||||
Query(params): Query<CurrencyParams>,
|
||||
) -> Result<(StatusCode, Json<Value>), AppError> {
|
||||
@@ -32,18 +44,48 @@ pub async fn currency_converter_handler(
|
||||
}
|
||||
|
||||
/// GET /misc/harga-emas
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/misc/harga-emas",
|
||||
tag = "misc",
|
||||
operation_id = "harga_emas",
|
||||
responses(
|
||||
(status = 200, description = "Gold prices (Antam)", body = Value),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn harga_emas_handler() -> Result<(StatusCode, Json<Value>), AppError> {
|
||||
let result = use_cases::harga_emas().await?;
|
||||
Ok((StatusCode::OK, Json(result)))
|
||||
}
|
||||
|
||||
/// GET /misc/kurs-bca
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/misc/kurs-bca",
|
||||
tag = "misc",
|
||||
operation_id = "kurs_bca",
|
||||
responses(
|
||||
(status = 200, description = "BCA exchange rates", body = Value),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn kurs_bca_handler() -> Result<(StatusCode, Json<Value>), AppError> {
|
||||
let result = use_cases::kurs_bca().await?;
|
||||
Ok((StatusCode::OK, Json(result)))
|
||||
}
|
||||
|
||||
/// GET /misc/server-info
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/misc/server-info",
|
||||
tag = "misc",
|
||||
operation_id = "server_info",
|
||||
responses(
|
||||
(status = 200, description = "Server info", body = Value),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn server_info_handler() -> Result<(StatusCode, Json<Value>), AppError> {
|
||||
let result = use_cases::server_info().await?;
|
||||
Ok((StatusCode::OK, Json(result)))
|
||||
|
||||
@@ -9,4 +9,3 @@ pub mod proxy;
|
||||
pub mod search;
|
||||
pub mod stalk;
|
||||
pub mod tools;
|
||||
pub mod weebs;
|
||||
@@ -19,17 +19,41 @@ pub struct SearchParams {
|
||||
pub city: Option<String>,
|
||||
}
|
||||
|
||||
fn require<'a>(v: Option<&'a String>, name: &str) -> Result<String, AppError> {
|
||||
fn require(v: Option<&String>, name: &str) -> Result<String, AppError> {
|
||||
v.cloned()
|
||||
.ok_or_else(|| AppError::BadRequest(format!("Missing '{}' parameter", name)))
|
||||
}
|
||||
|
||||
/// GET /search/bmkg
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/search/bmkg",
|
||||
tag = "search",
|
||||
operation_id = "search_bmkg_handler",
|
||||
responses(
|
||||
(status = 200, description = "OK", body = serde_json::Value),
|
||||
(status = 400, description = "Bad Request"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
|
||||
pub async fn bmkg_handler() -> Result<(StatusCode, Json<Value>), AppError> {
|
||||
Ok((StatusCode::OK, Json(use_cases::bmkg().await?)))
|
||||
}
|
||||
|
||||
/// GET /search/jadwal-sholat?kota=jakarta
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/search/jadwal-sholat",
|
||||
tag = "search",
|
||||
operation_id = "search_jadwal_sholat_handler",
|
||||
responses(
|
||||
(status = 200, description = "OK", body = serde_json::Value),
|
||||
(status = 400, description = "Bad Request"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
|
||||
pub async fn jadwal_sholat_handler(
|
||||
Query(p): Query<SearchParams>,
|
||||
) -> Result<(StatusCode, Json<Value>), AppError> {
|
||||
@@ -38,6 +62,18 @@ pub async fn jadwal_sholat_handler(
|
||||
}
|
||||
|
||||
/// GET /search/weather?city=jakarta
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/search/weather",
|
||||
tag = "search",
|
||||
operation_id = "search_weather_handler",
|
||||
responses(
|
||||
(status = 200, description = "OK", body = serde_json::Value),
|
||||
(status = 400, description = "Bad Request"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
|
||||
pub async fn weather_handler(
|
||||
Query(p): Query<SearchParams>,
|
||||
) -> Result<(StatusCode, Json<Value>), AppError> {
|
||||
@@ -46,6 +82,18 @@ pub async fn weather_handler(
|
||||
}
|
||||
|
||||
/// GET /search/google?query=rust
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/search/google",
|
||||
tag = "search",
|
||||
operation_id = "search_google_handler",
|
||||
responses(
|
||||
(status = 200, description = "OK", body = serde_json::Value),
|
||||
(status = 400, description = "Bad Request"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
|
||||
pub async fn google_handler(
|
||||
Query(p): Query<SearchParams>,
|
||||
) -> Result<(StatusCode, Json<Value>), AppError> {
|
||||
@@ -54,6 +102,18 @@ pub async fn google_handler(
|
||||
}
|
||||
|
||||
/// GET /search/yt?query=rickroll
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/search/yt",
|
||||
tag = "search",
|
||||
operation_id = "search_yt_handler",
|
||||
responses(
|
||||
(status = 200, description = "OK", body = serde_json::Value),
|
||||
(status = 400, description = "Bad Request"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
|
||||
pub async fn yt_handler(
|
||||
Query(p): Query<SearchParams>,
|
||||
) -> Result<(StatusCode, Json<Value>), AppError> {
|
||||
|
||||
@@ -7,12 +7,13 @@ use axum::http::StatusCode;
|
||||
use axum::Json;
|
||||
use serde::Deserialize;
|
||||
use serde_json::Value;
|
||||
use utoipa::IntoParams;
|
||||
|
||||
use crate::application::stalk as use_cases;
|
||||
use crate::presentation::error::AppError;
|
||||
|
||||
/// Query params for stalk endpoints.
|
||||
#[derive(Deserialize)]
|
||||
#[derive(Deserialize, IntoParams)]
|
||||
pub struct StalkParams {
|
||||
pub username: Option<String>,
|
||||
pub user_id: Option<String>,
|
||||
@@ -22,6 +23,18 @@ pub struct StalkParams {
|
||||
}
|
||||
|
||||
/// GET /stalk/github?username=asepharyana
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/stalk/github",
|
||||
tag = "stalk",
|
||||
operation_id = "stalk_github",
|
||||
params(StalkParams),
|
||||
responses(
|
||||
(status = 200, description = "GitHub user profile", body = Value),
|
||||
(status = 400, description = "Missing username"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn github_handler(
|
||||
Query(params): Query<StalkParams>,
|
||||
) -> Result<(StatusCode, Json<Value>), AppError> {
|
||||
@@ -33,6 +46,18 @@ pub async fn github_handler(
|
||||
}
|
||||
|
||||
/// GET /stalk/youtube?username=<handle>
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/stalk/youtube",
|
||||
tag = "stalk",
|
||||
operation_id = "stalk_youtube",
|
||||
params(StalkParams),
|
||||
responses(
|
||||
(status = 200, description = "YouTube channel profile", body = Value),
|
||||
(status = 400, description = "Missing username"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn youtube_handler(
|
||||
Query(params): Query<StalkParams>,
|
||||
) -> Result<(StatusCode, Json<Value>), AppError> {
|
||||
@@ -42,3 +67,26 @@ pub async fn youtube_handler(
|
||||
let result = use_cases::youtube(&username).await?;
|
||||
Ok((StatusCode::OK, Json(result)))
|
||||
}
|
||||
|
||||
/// GET /stalk/twitter?username=...
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/stalk/twitter",
|
||||
tag = "stalk",
|
||||
operation_id = "stalk_twitter",
|
||||
params(StalkParams),
|
||||
responses(
|
||||
(status = 200, description = "Twitter/X user profile", body = Value),
|
||||
(status = 400, description = "Missing username"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn twitter_handler(
|
||||
Query(params): Query<StalkParams>,
|
||||
) -> Result<(StatusCode, Json<Value>), AppError> {
|
||||
let username = params
|
||||
.username
|
||||
.ok_or_else(|| AppError::BadRequest("Missing 'username' parameter".to_string()))?;
|
||||
let result = use_cases::twitter(&username).await?;
|
||||
Ok((StatusCode::OK, Json(result)))
|
||||
}
|
||||
@@ -1,35 +1,49 @@
|
||||
//! Axum handlers for tools.
|
||||
//! Axum handlers for tool endpoints.
|
||||
|
||||
use axum::extract::Query;
|
||||
use axum::http::StatusCode;
|
||||
use axum::Json;
|
||||
use serde::Deserialize;
|
||||
use serde_json::Value;
|
||||
use utoipa::IntoParams;
|
||||
|
||||
use crate::application::tools as use_cases;
|
||||
use crate::presentation::error::AppError;
|
||||
|
||||
#[derive(Deserialize)]
|
||||
#[derive(Deserialize, IntoParams)]
|
||||
pub struct DomainParams {
|
||||
pub domain: Option<String>,
|
||||
}
|
||||
|
||||
#[derive(Deserialize)]
|
||||
#[derive(Deserialize, IntoParams)]
|
||||
pub struct IpParams {
|
||||
pub ip: Option<String>,
|
||||
}
|
||||
|
||||
#[derive(Deserialize)]
|
||||
#[derive(Deserialize, IntoParams)]
|
||||
pub struct UrlParams {
|
||||
pub url: Option<String>,
|
||||
}
|
||||
|
||||
#[derive(Deserialize)]
|
||||
#[derive(Deserialize, IntoParams)]
|
||||
pub struct ResiParams {
|
||||
pub resi: Option<String>,
|
||||
pub ekspedisi: Option<String>,
|
||||
}
|
||||
|
||||
/// GET /tool/whois?domain=example.com
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/tool/whois",
|
||||
tag = "tool",
|
||||
operation_id = "tool_whois",
|
||||
params(DomainParams),
|
||||
responses(
|
||||
(status = 200, description = "WHOIS info", body = Value),
|
||||
(status = 400, description = "Missing domain"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn whois_handler(
|
||||
Query(params): Query<DomainParams>,
|
||||
) -> Result<(StatusCode, Json<Value>), AppError> {
|
||||
@@ -39,6 +53,19 @@ pub async fn whois_handler(
|
||||
Ok((StatusCode::OK, Json(use_cases::whois(&domain).await?)))
|
||||
}
|
||||
|
||||
/// GET /tool/ip-location?ip=1.1.1.1
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/tool/ip-location",
|
||||
tag = "tool",
|
||||
operation_id = "tool_ip_location",
|
||||
params(IpParams),
|
||||
responses(
|
||||
(status = 200, description = "IP geolocation", body = Value),
|
||||
(status = 400, description = "Missing IP"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn ip_location_handler(
|
||||
Query(params): Query<IpParams>,
|
||||
) -> Result<(StatusCode, Json<Value>), AppError> {
|
||||
@@ -48,6 +75,19 @@ pub async fn ip_location_handler(
|
||||
Ok((StatusCode::OK, Json(use_cases::ip_location(&ip).await?)))
|
||||
}
|
||||
|
||||
/// GET /tool/tinyurl?url=https://example.com
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/tool/tinyurl",
|
||||
tag = "tool",
|
||||
operation_id = "tool_tinyurl",
|
||||
params(UrlParams),
|
||||
responses(
|
||||
(status = 200, description = "Shortened URL", body = Value),
|
||||
(status = 400, description = "Missing URL"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn tinyurl_handler(
|
||||
Query(params): Query<UrlParams>,
|
||||
) -> Result<(StatusCode, Json<Value>), AppError> {
|
||||
@@ -57,6 +97,19 @@ pub async fn tinyurl_handler(
|
||||
Ok((StatusCode::OK, Json(use_cases::tinyurl(&url).await?)))
|
||||
}
|
||||
|
||||
/// GET /tool/check-hosting?domain=example.com
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/tool/check-hosting",
|
||||
tag = "tool",
|
||||
operation_id = "tool_check_hosting",
|
||||
params(DomainParams),
|
||||
responses(
|
||||
(status = 200, description = "Hosting info", body = Value),
|
||||
(status = 400, description = "Missing domain"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn check_hosting_handler(
|
||||
Query(params): Query<DomainParams>,
|
||||
) -> Result<(StatusCode, Json<Value>), AppError> {
|
||||
@@ -69,10 +122,34 @@ pub async fn check_hosting_handler(
|
||||
))
|
||||
}
|
||||
|
||||
/// GET /tool/hargapangan
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/tool/hargapangan",
|
||||
tag = "tool",
|
||||
operation_id = "tool_hargapangan",
|
||||
responses(
|
||||
(status = 200, description = "Food prices", body = Value),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn hargapangan_handler() -> Result<(StatusCode, Json<Value>), AppError> {
|
||||
Ok((StatusCode::OK, Json(use_cases::hargapangan().await?)))
|
||||
}
|
||||
|
||||
/// GET /tool/cek-resi?resi=...&ekspedisi=...
|
||||
#[utoipa::path(
|
||||
get,
|
||||
path = "/tool/cek-resi",
|
||||
tag = "tool",
|
||||
operation_id = "tool_cek_resi",
|
||||
params(ResiParams),
|
||||
responses(
|
||||
(status = 200, description = "Shipment tracking", body = Value),
|
||||
(status = 400, description = "Missing resi"),
|
||||
(status = 502, description = "Upstream error"),
|
||||
)
|
||||
)]
|
||||
pub async fn cek_resi_handler(
|
||||
Query(params): Query<ResiParams>,
|
||||
) -> Result<(StatusCode, Json<Value>), AppError> {
|
||||
|
||||
@@ -1,65 +0,0 @@
|
||||
//! Axum handlers for weebs endpoints.
|
||||
//!
|
||||
//! Ported from Shirokami-API `scraper/weebs/*.js`.
|
||||
|
||||
use axum::extract::Query;
|
||||
use axum::http::StatusCode;
|
||||
use axum::Json;
|
||||
use serde::Deserialize;
|
||||
use serde_json::Value;
|
||||
|
||||
use crate::application::weebs as use_cases;
|
||||
use crate::presentation::error::AppError;
|
||||
|
||||
#[derive(Deserialize)]
|
||||
pub struct WeebsParams {
|
||||
pub query: Option<String>,
|
||||
pub q: Option<String>,
|
||||
pub url: Option<String>,
|
||||
pub tag: Option<String>,
|
||||
}
|
||||
|
||||
fn require<'a>(v: Option<&'a String>, name: &str) -> Result<String, AppError> {
|
||||
v.cloned()
|
||||
.ok_or_else(|| AppError::BadRequest(format!("Missing '{}' parameter", name)))
|
||||
}
|
||||
|
||||
/// GET /weebs/anime-info?query=naruto
|
||||
pub async fn anime_info_handler(
|
||||
Query(p): Query<WeebsParams>,
|
||||
) -> Result<(StatusCode, Json<Value>), AppError> {
|
||||
let query = require(p.query.as_ref().or(p.q.as_ref()), "query")?;
|
||||
Ok((StatusCode::OK, Json(use_cases::anime_info(&query).await?)))
|
||||
}
|
||||
|
||||
/// GET /weebs/manga-info?query=one%20piece
|
||||
pub async fn manga_info_handler(
|
||||
Query(p): Query<WeebsParams>,
|
||||
) -> Result<(StatusCode, Json<Value>), AppError> {
|
||||
let query = require(p.query.as_ref().or(p.q.as_ref()), "query")?;
|
||||
Ok((StatusCode::OK, Json(use_cases::manga_info(&query).await?)))
|
||||
}
|
||||
|
||||
/// GET /weebs/whatanime?url=<image-url>
|
||||
pub async fn whatanime_handler(
|
||||
Query(p): Query<WeebsParams>,
|
||||
) -> Result<(StatusCode, Json<Value>), AppError> {
|
||||
let url = require(p.url.as_ref(), "url")?;
|
||||
Ok((StatusCode::OK, Json(use_cases::whatanime(&url).await?)))
|
||||
}
|
||||
|
||||
/// GET /weebs/sfw-waifu?tag=waifu
|
||||
pub async fn sfw_waifu_handler(
|
||||
Query(p): Query<WeebsParams>,
|
||||
) -> Result<(StatusCode, Json<Value>), AppError> {
|
||||
let tag = require(p.tag.as_ref(), "tag")?;
|
||||
Ok((StatusCode::OK, Json(use_cases::sfw_waifu(&tag).await?)))
|
||||
}
|
||||
|
||||
/// GET /weebs/nsfw-waifu?tag=waifu
|
||||
pub async fn nsfw_waifu_handler(
|
||||
Query(p): Query<WeebsParams>,
|
||||
) -> Result<(StatusCode, Json<Value>), AppError> {
|
||||
let tag = require(p.tag.as_ref(), "tag")?;
|
||||
Ok((StatusCode::OK, Json(use_cases::nsfw_waifu(&tag).await?)))
|
||||
}
|
||||
+10
-19
@@ -184,6 +184,10 @@ pub fn build_router(app_state: Arc<AppState>) -> anyhow::Result<Router> {
|
||||
"/download/youtube/mp3",
|
||||
axum::routing::get(crate::presentation::handler::downloader::download_youtube_mp3),
|
||||
)
|
||||
.route(
|
||||
"/file/yt_merge/{filename}",
|
||||
axum::routing::get(crate::presentation::handler::downloader::serve_merged_file),
|
||||
)
|
||||
.route(
|
||||
"/download/spotify",
|
||||
axum::routing::get(crate::presentation::handler::downloader::download_spotify),
|
||||
@@ -204,6 +208,10 @@ pub fn build_router(app_state: Arc<AppState>) -> anyhow::Result<Router> {
|
||||
"/download/terabox",
|
||||
axum::routing::get(crate::presentation::handler::downloader::download_terabox),
|
||||
)
|
||||
.route(
|
||||
"/proxy/terabox",
|
||||
axum::routing::get(crate::presentation::handler::downloader::proxy_terabox),
|
||||
)
|
||||
.route(
|
||||
"/download/gdrive",
|
||||
axum::routing::get(crate::presentation::handler::downloader::download_gdrive),
|
||||
@@ -282,26 +290,9 @@ pub fn build_router(app_state: Arc<AppState>) -> anyhow::Result<Router> {
|
||||
"/stalk/youtube",
|
||||
axum::routing::get(crate::presentation::handler::stalk::youtube_handler),
|
||||
)
|
||||
// Weebs routes
|
||||
.route(
|
||||
"/weebs/anime-info",
|
||||
axum::routing::get(crate::presentation::handler::weebs::anime_info_handler),
|
||||
)
|
||||
.route(
|
||||
"/weebs/manga-info",
|
||||
axum::routing::get(crate::presentation::handler::weebs::manga_info_handler),
|
||||
)
|
||||
.route(
|
||||
"/weebs/whatanime",
|
||||
axum::routing::get(crate::presentation::handler::weebs::whatanime_handler),
|
||||
)
|
||||
.route(
|
||||
"/weebs/sfw-waifu",
|
||||
axum::routing::get(crate::presentation::handler::weebs::sfw_waifu_handler),
|
||||
)
|
||||
.route(
|
||||
"/weebs/nsfw-waifu",
|
||||
axum::routing::get(crate::presentation::handler::weebs::nsfw_waifu_handler),
|
||||
"/stalk/twitter",
|
||||
axum::routing::get(crate::presentation::handler::stalk::twitter_handler),
|
||||
)
|
||||
// Search routes
|
||||
.route(
|
||||
|
||||
@@ -0,0 +1,64 @@
|
||||
"""Extract TikTok user-detail JSON (__UNIVERSAL_DATA_FOR_REHYDRATION__) via Playwright.
|
||||
|
||||
The direct HTTP path gets a SlardarWAF challenge page; a real browser with JS
|
||||
executes the challenge and exposes the __UNIVERSAL_DATA_FOR_REHYDRATION__ script.
|
||||
Usage: python3 stalk_tiktok.py <username>
|
||||
"""
|
||||
import sys
|
||||
import asyncio
|
||||
import json
|
||||
import os
|
||||
|
||||
os.environ.setdefault("PLAYWRIGHT_BROWSERS_PATH", "/usr/local/share/ms-playwright")
|
||||
|
||||
from playwright.async_api import async_playwright
|
||||
|
||||
|
||||
async def main(username):
|
||||
url = f"https://www.tiktok.com/@{username}?_t=ZS-8tHANz7ieoS&_r=1"
|
||||
async with async_playwright() as p:
|
||||
browser = await p.chromium.launch(
|
||||
headless=True,
|
||||
args=["--no-sandbox", "--disable-bypass", "--disable-blink-features=AutomationControlled"],
|
||||
)
|
||||
context = await browser.new_context(
|
||||
user_agent="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36",
|
||||
viewport={"width": 1366, "height": 768},
|
||||
java_script_enabled=True,
|
||||
locale="en-US",
|
||||
)
|
||||
page = await context.new_page()
|
||||
try:
|
||||
await page.goto(url, timeout=60000, wait_until="domcontentloaded")
|
||||
except Exception as e:
|
||||
await browser.close()
|
||||
print(json.dumps({"success": False, "error": f"Navigation: {e}"}))
|
||||
return
|
||||
|
||||
# Wait for the rehydration script to appear (WAF challenge may take a few seconds)
|
||||
found = None
|
||||
for _ in range(30):
|
||||
try:
|
||||
found = await page.evaluate(
|
||||
"""() => {
|
||||
const s = document.querySelector('script#__UNIVERSAL_DATA_FOR_REHYDRATION__');
|
||||
return s ? s.textContent : null;
|
||||
}"""
|
||||
)
|
||||
if found:
|
||||
break
|
||||
except Exception:
|
||||
pass
|
||||
await page.wait_for_timeout(1000)
|
||||
|
||||
await browser.close()
|
||||
|
||||
if not found:
|
||||
print(json.dumps({"success": False, "error": "Rehydration data not found (WAF challenge)"}))
|
||||
return
|
||||
|
||||
print(json.dumps({"success": True, "data": found}))
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
asyncio.run(main(sys.argv[1]))
|
||||
Reference in new issue
Block a user