Upload 2 files

src/main.rs

@@ -123,6 +123,55 @@ async fn media_url_for_track(
     Ok(url)
 }
 
+fn upstream_dzmedia_base() -> Option<String> {
+    let env = std::env::var("DZMEDIA_UPSTREAM").ok().unwrap_or_default();
+    let base = env.trim();
+    if base.is_empty() {
+        return Some("https://lufts-dzmedia.fly.dev".to_string());
+    }
+    if base == "-" {
+        return None;
+    }
+    Some(base.trim_end_matches('/').to_string())
+}
+
+async fn upstream_get_url_text(formats: &Vec<Format>, ids: &Vec<u32>) -> Result<String, String> {
+    let Some(base) = upstream_dzmedia_base() else {
+        return Err("upstream:disabled".to_string());
+    };
+    let url = format!("{base}/get_url");
+
+    let client = reqwest::Client::builder()
+        .timeout(std::time::Duration::from_secs(20))
+        .build()
+        .map_err(|_| "upstream:client".to_string())?;
+
+    let r = client
+        .post(url)
+        .json(&json!({ "formats": formats, "ids": ids }))
+        .send()
+        .await
+        .map_err(|_| "upstream:network".to_string())?;
+
+    let status = r.status();
+    let text = r.text().await.map_err(|_| "upstream:read".to_string())?;
+    if !status.is_success() {
+        let snip: String = text.chars().take(200).collect();
+        return Err(format!("upstream:{}:{}", status.as_u16(), snip));
+    }
+    Ok(text)
+}
+
+async fn upstream_media_url(id: u32, formats: &Vec<Format>) -> Result<String, String> {
+    let text = upstream_get_url_text(formats, &vec![id]).await?;
+    let v: serde_json::Value = serde_json::from_str(&text).map_err(|_| "upstream:parse".to_string())?;
+    let url = extract_media_url(&v);
+    if url.is_empty() {
+        return Err("upstream:no_url".to_string());
+    }
+    Ok(url)
+}
+
 #[derive(Deserialize)]
 struct DeezerTrackList {
     data: Vec<DeezerTrack>
@@ -153,6 +202,13 @@ async fn get_url(State(state): State<AppState>, Json(req): Json<RequestParams>)
         return (StatusCode::BAD_REQUEST, "ID list cannot be empty".to_string());
     }
 
+    let arl = req.arl.clone().unwrap_or_default().trim().to_string();
+    if arl.is_empty() {
+        if let Ok(t) = upstream_get_url_text(&req.formats, &req.ids).await {
+            return (StatusCode::OK, t);
+        }
+    }
+
     let client = api_client_for_arl(&state, req.arl.clone()).await;
     let media_resp = {
         let mut client = client.lock().await;
@@ -277,7 +333,18 @@ async fn stream(State(state): State<AppState>, Query(q): Query<StreamParams>) ->
 
     let mut url: Option<String> = None;
     let mut last_err: Option<String> = None;
+
+    if arl_key.is_none() {
+        match upstream_media_url(q.id, &formats).await {
+            Ok(u) => url = Some(u),
+            Err(e) => last_err = Some(e),
+        }
+    }
+
     for attempt in 0..2 {
+        if url.is_some() {
+            break;
+        }
         let client = api_client_for_arl(&state, arl_key.clone()).await;
         let res = {
             let mut client = client.lock().await;
@@ -304,12 +371,27 @@ async fn stream(State(state): State<AppState>, Query(q): Query<StreamParams>) ->
         }
     }
 
-    let Some(url) = url else {
-        return (
-            StatusCode::SERVICE_UNAVAILABLE,
-            last_err.unwrap_or_else(|| "stream:error".to_string()),
-        )
-            .into_response();
+    let url = match url {
+        Some(u) => u,
+        None => {
+            if arl_key.is_some() {
+                if let Ok(u) = upstream_media_url(q.id, &formats).await {
+                    u
+                } else {
+                    return (
+                        StatusCode::SERVICE_UNAVAILABLE,
+                        last_err.unwrap_or_else(|| "stream:error".to_string()),
+                    )
+                        .into_response();
+                }
+            } else {
+                return (
+                    StatusCode::SERVICE_UNAVAILABLE,
+                    last_err.unwrap_or_else(|| "stream:error".to_string()),
+                )
+                    .into_response();
+            }
+        }
     };
 
     let http = reqwest::Client::builder()
@@ -829,6 +911,7 @@ async fn pair_page(Query(q): Query<PairPageQuery>) -> impl IntoResponse {
         .box{{max-width:420px;margin:0 auto;background:#151515;padding:18px;border-radius:12px}}\
         input{{width:100%;padding:12px 14px;margin:10px 0;border-radius:10px;border:1px solid #333;background:#0f0f0f;color:#fff;font-size:16px}}\
         button{{width:100%;padding:12px 14px;border-radius:10px;border:0;background:#1db954;color:#000;font-weight:700;font-size:16px}}\
+        .secondary{{background:#2b2b2b;color:#fff}}\
         .muted{{color:#aaa;font-size:13px;line-height:1.35}}</style>\
         </head><body><div class=\"box\">\
         <h2 style=\"margin:0 0 6px\">Вход Deezer</h2>\
@@ -840,6 +923,14 @@ async fn pair_page(Query(q): Query<PairPageQuery>) -> impl IntoResponse {
             <button type=\"submit\">Войти</button>\
         </form>\
         <p class=\"muted\" style=\"margin:12px 0 0\">Пароль используется только для по��учения токена доступа и не сохраняется.</p>\
+        <hr style=\"border:0;border-top:1px solid #2a2a2a;margin:16px 0\">\
+        <h3 style=\"margin:0 0 6px;font-size:16px\">ARL (для полного аудио)</h3>\
+        <div class=\"muted\">Если сервер не может автоматически получить ARL (ошибка 503/502), вставь ARL сюда — с телефона это быстрее, чем на ТВ.</div>\
+        <form method=\"POST\" action=\"/pair/arl\">\
+            <input type=\"hidden\" name=\"code\" value=\"{safe_code}\">\
+            <input name=\"arl\" type=\"text\" placeholder=\"arl=...\" autocomplete=\"off\">\
+            <button class=\"secondary\" type=\"submit\">Отправить ARL</button>\
+        </form>\
         </div></body></html>"
     );
     Html(html)
@@ -852,6 +943,12 @@ struct PairCompleteForm {
     password: String,
 }
 
+#[derive(Deserialize)]
+struct PairArlForm {
+    code: String,
+    arl: String,
+}
+
 fn htmlesc(s: &str) -> String {
     s.replace('&', "&amp;")
         .replace('<', "&lt;")
@@ -860,6 +957,27 @@ fn htmlesc(s: &str) -> String {
         .replace('\'', "&#39;")
 }
 
+async fn pair_arl(State(state): State<AppState>, Form(f): Form<PairArlForm>) -> impl IntoResponse {
+    let code = f.code.trim().to_string();
+    let arl = f.arl.trim().to_string();
+    if code.is_empty() {
+        return Html("<h3>Нужен code</h3>".to_string()).into_response();
+    }
+    if arl.len() < 20 {
+        return Html("<h3>ARL выглядит неверно</h3>".to_string()).into_response();
+    }
+
+    let mut map = state.pair.write().await;
+    let Some(sess) = map.get_mut(&code) else {
+        return Html("<h3>Код не найден или истёк</h3>".to_string()).into_response();
+    };
+    sess.arl = Some(arl);
+    sess.error = None;
+
+    Html("<h3>ARL сохранён</h3><p>Вернись на ТВ — треки должны стать полными.</p>".to_string())
+        .into_response()
+}
+
 async fn api_deezer_get(
     client: &reqwest::Client,
     path: &str,
@@ -999,7 +1117,25 @@ async fn pair_complete(State(state): State<AppState>, Form(f): Form<PairComplete
         }
     }
 
-    Html("<h3>Готово</h3><p>Вернись на ТВ — плагин подключится автоматически.</p>".to_string()).into_response()
+    let map = state.pair.read().await;
+    let sess_arl = map.get(&code).and_then(|s| s.arl.clone()).unwrap_or_default();
+    if sess_arl.len() > 20 {
+        return Html("<h3>Готово</h3><p>Вернись на ТВ — плагин подключится автоматически.</p>".to_string())
+            .into_response();
+    }
+
+    let safe_code = htmlesc(&code);
+    Html(format!(
+        "<h3>Вход выполнен</h3>\
+        <p>Токен получен, но ARL сервер автоматически не вытащил (из‑за блокировки/ошибок gw-light на хостинге).</p>\
+        <p>Чтобы включить полные треки, вставь ARL (с телефона):</p>\
+        <form method=\"POST\" action=\"/pair/arl\">\
+            <input type=\"hidden\" name=\"code\" value=\"{safe_code}\">\
+            <input name=\"arl\" type=\"text\" placeholder=\"arl=...\" autocomplete=\"off\" style=\"width:100%;padding:12px 14px;border-radius:10px;border:1px solid #333;background:#0f0f0f;color:#fff;font-size:16px\">\
+            <button type=\"submit\" style=\"margin-top:10px;width:100%;padding:12px 14px;border-radius:10px;border:0;background:#2b2b2b;color:#fff;font-weight:700;font-size:16px\">Отправить ARL</button>\
+        </form>"
+    ))
+    .into_response()
 }
 
 #[derive(Debug, Deserialize)]
@@ -1226,6 +1362,7 @@ async fn main() {
         .route("/pair/status", get(pair_status))
         .route("/pair", get(pair_page))
         .route("/pair/complete", post(pair_complete))
+        .route("/pair/arl", post(pair_arl))
         .route("/oauth/me", post(oauth_me))
         .route("/oauth/playlists", post(oauth_playlists))
         .route("/oauth/playlist_tracks", post(oauth_playlist_tracks))