Upload soundcloud.rs

src/soundcloud.rs

@@ -134,10 +134,49 @@ pub async fn download(Query(params): Query<DownloadParams>) -> impl IntoResponse
     let quality = params.quality.as_deref().unwrap_or("best");
     
     // Формируем аргументы для yt-dlp
+    // Приоритизируем M4A/AAC (256kbps для Go+), затем MP3
     let format_arg = match quality {
-        "256" => "bestaudio[abr<=256]",
-        "128" => "bestaudio[abr<=128]",
-        _ => "bestaudio",
+        "256" => "bestaudio[ext=m4a]/bestaudio[ext=aac]/bestaudio[abr<=256]/bestaudio",
+        "128" => "bestaudio[abr<=128]/bestaudio",
+        _ => "bestaudio[ext=m4a]/bestaudio[ext=aac]/bestaudio", // Приоритет M4A/AAC
+    };
+    
+    tracing::info!("Using format selector: {}", format_arg);
+
+    // Сначала получаем информацию о доступных форматах для отладки
+    if let Ok(info_output) = Command::new("yt-dlp")
+        .args([
+            "--quiet",
+            "-F", // Показать все доступные форматы
+            &params.url,
+        ])
+        .output()
+        .await
+    {
+        let formats_info = String::from_utf8_lossy(&info_output.stdout);
+        tracing::info!("Available formats:\n{}", formats_info);
+    }
+
+    // Получаем URL thumbnail'а
+    let thumbnail_url = if let Ok(thumb_output) = Command::new("yt-dlp")
+        .args([
+            "--quiet",
+            "--no-warnings",
+            "--print", "%(thumbnail)s",
+            &params.url,
+        ])
+        .output()
+        .await
+    {
+        let thumb = String::from_utf8_lossy(&thumb_output.stdout).trim().to_string();
+        if !thumb.is_empty() && thumb.starts_with("http") {
+            tracing::info!("Found thumbnail: {}", thumb);
+            Some(thumb)
+        } else {
+            None
+        }
+    } else {
+        None
     };
 
     // Скачиваем во временный файл БЕЗ конвертации - оригинальный формат
@@ -146,15 +185,19 @@ pub async fn download(Query(params): Query<DownloadParams>) -> impl IntoResponse
         .unwrap()
         .as_nanos());
 
+    // Добавляем расширение .%(ext)s чтобы yt-dlp сам определил правильное расширение
+    let output_template = format!("{}.%(ext)s", temp_file);
+
     let output = match Command::new("yt-dlp")
         .args([
             "--quiet",
             "--no-warnings",
             "-f", format_arg,
-            "-o", &temp_file,
+            "-o", &output_template,
+            "--no-post-overwrites", // Не перезаписываем после обработки
             &params.url,
         ])
-        .stdout(Stdio::null())
+        .stdout(Stdio::piped())
         .stderr(Stdio::piped())
         .output()
         .await
@@ -172,15 +215,22 @@ pub async fn download(Query(params): Query<DownloadParams>) -> impl IntoResponse
         return (StatusCode::INTERNAL_SERVER_ERROR, "Download failed").into_response();
     }
 
-    // Находим скачанный файл (yt-dlp добавит расширение автоматически)
+    // Логируем stdout чтобы увидеть что скачал yt-dlp
+    let stdout = String::from_utf8_lossy(&output.stdout);
+    if !stdout.is_empty() {
+        tracing::info!("yt-dlp stdout: {}", stdout);
+    }
+
+    // Находим скачанный файл (yt-dlp добавил расширение через .%(ext)s)
     let actual_file = if tokio::fs::metadata(&temp_file).await.is_ok() {
         temp_file.clone()
     } else {
-        // Пробуем найти с расширением
+        // Пробуем найти с расширением - ВАЖНО: сначала ищем M4A/AAC, потом остальные
         let mut found = None;
-        for ext in &["m4a", "opus", "webm", "mp3", "aac"] {
+        for ext in &["m4a", "aac", "opus", "webm", "mp3"] {
             let path = format!("{}.{}", temp_file, ext);
             if tokio::fs::metadata(&path).await.is_ok() {
+                tracing::info!("Found downloaded file with extension: {}", ext);
                 found = Some(path);
                 break;
             }
@@ -188,7 +238,7 @@ pub async fn download(Query(params): Query<DownloadParams>) -> impl IntoResponse
         match found {
             Some(p) => p,
             None => {
-                tracing::error!("Could not find downloaded file");
+                tracing::error!("Could not find downloaded file with any extension");
                 return (StatusCode::INTERNAL_SERVER_ERROR, "File not found").into_response();
             }
         }
@@ -293,14 +343,21 @@ pub async fn download(Query(params): Query<DownloadParams>) -> impl IntoResponse
 
     tracing::info!("Sending SoundCloud file: {} ({} bytes, {})", filename_safe, final_data.len(), final_content_type);
 
-    axum::response::Response::builder()
+    let mut response = axum::response::Response::builder()
         .status(200)
         .header("Content-Type", final_content_type)
         .header("Content-Length", final_data.len().to_string())
         .header("Content-Disposition", format!("inline; filename=\"{}\"", filename_safe))
         .header("Accept-Ranges", "bytes")
         .header("Access-Control-Allow-Origin", "*")
-        .header("Cache-Control", "public, max-age=31536000")
+        .header("Cache-Control", "public, max-age=31536000");
+    
+    // Добавляем URL thumbnail'а в заголовок если есть
+    if let Some(thumb_url) = thumbnail_url {
+        response = response.header("X-Thumbnail-Url", thumb_url);
+    }
+    
+    response
         .body(axum::body::Body::from(final_data))
         .unwrap()
         .into_response()