fix(downloads): correct extension metadata and list UI

This commit is contained in:
NimBold
2026-07-03 03:55:59 +03:30
parent 63abf23622
commit 4026e5bc86
4 changed files with 172 additions and 67 deletions
+141 -36
View File
@@ -18,6 +18,113 @@ fn get_metadata_cache() -> &'static std::sync::Mutex<HashMap<String, String>> {
CACHE.get_or_init(|| std::sync::Mutex::new(HashMap::new()))
}
fn sanitize_metadata_filename(filename: &str) -> Option<String> {
let normalized = filename.trim().replace('\\', "/");
let basename = std::path::Path::new(&normalized)
.file_name()?
.to_str()?
.trim()
.trim_end_matches(|c| c == '.' || c == ' ');
if basename.is_empty() || basename == "." || basename == ".." || basename.len() > 255 {
return None;
}
Some(basename.to_string())
}
fn percent_decode_metadata_value(value: &str) -> Option<String> {
let bytes = value.as_bytes();
let mut decoded = Vec::with_capacity(bytes.len());
let mut index = 0;
while index < bytes.len() {
if bytes[index] == b'%' && index + 2 < bytes.len() {
let hex = std::str::from_utf8(&bytes[index + 1..index + 3]).ok()?;
if let Ok(byte) = u8::from_str_radix(hex, 16) {
decoded.push(byte);
index += 3;
continue;
}
}
decoded.push(bytes[index]);
index += 1;
}
String::from_utf8(decoded).ok()
}
fn filename_from_content_disposition(disposition: &str) -> Option<String> {
let mut fallback = None;
for part in disposition.split(';').map(str::trim) {
let Some((name, value)) = part.split_once('=') else {
continue;
};
let normalized_name = name.trim().to_ascii_lowercase();
let raw_value = value.trim().trim_matches('"').trim_matches('\'');
if normalized_name == "filename*" {
let encoded = raw_value
.split_once("''")
.map(|(_, value)| value)
.unwrap_or(raw_value);
if let Some(filename) = percent_decode_metadata_value(encoded)
.and_then(|value| sanitize_metadata_filename(&value))
{
return Some(filename);
}
} else if normalized_name == "filename" {
fallback = sanitize_metadata_filename(raw_value);
}
}
fallback
}
fn filename_from_url_disposition_query(raw_url: &str) -> Option<String> {
let parsed = reqwest::Url::parse(raw_url).ok()?;
for (name, value) in parsed.query_pairs() {
if name.eq_ignore_ascii_case("response-content-disposition")
|| name.eq_ignore_ascii_case("rscd")
{
if let Some(filename) = filename_from_content_disposition(&value) {
return Some(filename);
}
}
}
None
}
fn filename_from_url_path(raw_url: &str) -> Option<String> {
let parsed = reqwest::Url::parse(raw_url).ok()?;
let last = parsed.path_segments()?.next_back()?;
sanitize_metadata_filename(last)
}
fn metadata_filename_from_response(
response: &reqwest::Response,
current_url: &str,
original_url: &str,
) -> String {
if let Some(filename) = response
.headers()
.get(reqwest::header::CONTENT_DISPOSITION)
.and_then(|value| value.to_str().ok())
.and_then(filename_from_content_disposition)
{
return filename;
}
filename_from_url_disposition_query(current_url)
.or_else(|| filename_from_url_path(original_url))
.or_else(|| filename_from_url_path(current_url))
.unwrap_or_else(|| "download".to_string())
}
#[derive(Serialize, TS)]
#[ts(export, export_to = "../../src/bindings/")]
pub struct MetadataResponse {
@@ -1002,39 +1109,7 @@ async fn fetch_metadata(
break;
}
let mut filename = String::new();
if let Some(cd) = res.headers().get(reqwest::header::CONTENT_DISPOSITION) {
if let Ok(cd_str) = cd.to_str() {
if let Some(idx) = cd_str.find("filename=") {
let rest = &cd_str[idx + 9..];
let raw_filename = rest.trim_matches(|c| c == '"' || c == '\'');
let normalized = raw_filename.replace('\\', "/");
filename = std::path::Path::new(&normalized)
.file_name()
.and_then(|n| n.to_str())
.unwrap_or("download")
.to_string();
}
}
}
if filename.is_empty() {
if let Ok(parsed) = reqwest::Url::parse(&current_url) {
if let Some(mut segments) = parsed.path_segments() {
if let Some(last) = segments.next_back() {
let normalized = last.replace('\\', "/");
filename = std::path::Path::new(&normalized)
.file_name()
.and_then(|n| n.to_str())
.unwrap_or("download")
.to_string();
}
}
}
}
if filename.is_empty() {
filename = "download".to_string();
}
let filename = metadata_filename_from_response(&res, &current_url, &url);
let mut size_str = "Unknown".to_string();
let mut size_bytes = 0;
@@ -4225,9 +4300,11 @@ fn set_extension_frontend_ready(state: tauri::State<'_, AppState>, ready: bool)
mod tests {
use super::{
aggregate_media_fraction, build_media_format_options, collect_download_uris,
is_excluded_yt_dlp_format, json_lower, media_output_template, media_progress_speed,
normalize_speed_limit_for_aria2, parse_firelink_deep_link, parse_media_progress_line,
redact_log_line, FirelinkDeepLink, MediaProgress, MEDIA_PROGRESS_PREFIX,
filename_from_content_disposition, filename_from_url_disposition_query,
filename_from_url_path, is_excluded_yt_dlp_format, json_lower, media_output_template,
media_progress_speed, normalize_speed_limit_for_aria2, parse_firelink_deep_link,
parse_media_progress_line, redact_log_line, FirelinkDeepLink, MediaProgress,
MEDIA_PROGRESS_PREFIX,
};
use serde_json::json;
use std::time::{Duration, Instant};
@@ -4249,6 +4326,34 @@ mod tests {
assert_eq!(template, destination.join("clip.mp4"));
}
#[test]
fn metadata_filename_prefers_content_disposition_filename() {
assert_eq!(
filename_from_content_disposition(
"attachment; filename*=UTF-8''OnionHop-3.5-macOS-arm64.dmg; filename=ignored.bin"
),
Some("OnionHop-3.5-macOS-arm64.dmg".to_string())
);
assert_eq!(
filename_from_content_disposition("attachment; filename=OnionHop-3.5-macOS-arm64.dmg"),
Some("OnionHop-3.5-macOS-arm64.dmg".to_string())
);
}
#[test]
fn metadata_filename_reads_redirect_disposition_query_before_opaque_path() {
let redirected = "https://release-assets.githubusercontent.com/github-production-release-asset/1117828249/7aae36e6-00ec-4e7d-8dec-f14ace170bdb?rscd=attachment%3B+filename%3DOnionHop-3.5-macOS-arm64.dmg";
assert_eq!(
filename_from_url_disposition_query(redirected),
Some("OnionHop-3.5-macOS-arm64.dmg".to_string())
);
assert_eq!(
filename_from_url_path(redirected),
Some("7aae36e6-00ec-4e7d-8dec-f14ace170bdb".to_string())
);
}
#[test]
fn normalizes_bare_global_speed_limits_as_kib_per_second() {
assert_eq!(
+1 -1
View File
@@ -451,7 +451,7 @@ export const DownloadTable: React.FC<DownloadTableProps> = ({ filter }) => {
className={`${index === 5 ? 'download-cell-right' : ''} ${filter.startsWith('queue:') ? 'opacity-70 cursor-default' : 'cursor-pointer hover:text-text-primary transition-colors'} flex items-center justify-between`}
onClick={() => handleSort(label)}
>
<div className={`flex items-center gap-1 w-full h-full select-none ${index > 0 ? 'justify-center' : ''}`}>
<div className="flex items-center gap-1 w-full h-full select-none">
<span>{label}</span>
{sortConfig?.column === label && (
sortConfig.direction === 'asc' ? <ChevronUp size={14} /> : <ChevronDown size={14} />
+4 -23
View File
@@ -60,11 +60,6 @@ const browserCookieSourceOptions = [
{ value: 'whale', label: 'Whale' },
] as const;
const normalizeBrowserCookieSource = (source: string) => {
const trimmed = source.trim();
return trimmed.length > 0 ? trimmed : 'none';
};
type EngineCheck = typeof engineChecks[number];
const engineStatusCache = new Map<string, EngineStatusItem>();
@@ -1121,32 +1116,18 @@ className="app-button px-3 py-1.5 text-[12px] flex items-center gap-1.5 disabled
</div>
))}
<div className="grid grid-cols-[180px_1fr] items-start gap-4 text-[13px] border-t border-border-modal/50 pt-3 mt-2">
<label className="text-text-secondary font-semibold pt-1.5">Browser Cookies Source:</label>
<div className="space-y-2">
<div className="grid grid-cols-[180px_1fr] items-center gap-4 text-[13px] border-t border-border-modal/50 pt-3 mt-2">
<label className="text-text-secondary font-semibold pt-1.5">Browser Cookies Source:</label>
<select
value={browserCookieSourceOptions.some(option => option.value === settings.mediaCookieSource) ? settings.mediaCookieSource : 'custom'}
onChange={(e) => {
if (e.target.value !== 'custom') {
settings.setMediaCookieSource(e.target.value);
}
}}
value={browserCookieSourceOptions.some(option => option.value === settings.mediaCookieSource) ? settings.mediaCookieSource : 'none'}
onChange={(e) => settings.setMediaCookieSource(e.target.value)}
className="bg-bg-input border border-border-modal rounded-lg p-1.5 text-[13px] text-text-primary focus:outline-none focus:border-accent w-full"
>
{browserCookieSourceOptions.map(option => (
<option key={option.value} value={option.value}>{option.label}</option>
))}
<option value="custom">Custom yt-dlp source</option>
</select>
<input
value={settings.mediaCookieSource === 'none' ? '' : settings.mediaCookieSource}
onChange={(e) => settings.setMediaCookieSource(normalizeBrowserCookieSource(e.target.value))}
placeholder="firefox:/path/to/profile or chrome:Profile 1"
className="app-control w-full px-3 py-1.5 bg-surface-overlay/50 border-border-color/50 focus:border-accent-color focus:bg-surface-overlay text-[13px]"
aria-label="Custom yt-dlp browser cookie source"
/>
</div>
</div>
<p className="text-text-muted text-xs mt-1">yt-dlp reads browser cookies to bypass video download limits or access restricted media. Firelink does not save browser cookies.</p>
</div>
</div>
+26 -7
View File
@@ -1667,6 +1667,7 @@ html[data-list-density="relaxed"] {
box-sizing: border-box;
width: 100%;
min-width: 0;
--download-column-padding-x: 14px;
}
.download-table-header > div {
@@ -1676,6 +1677,15 @@ html[data-list-density="relaxed"] {
align-items: center;
min-width: 0;
overflow: hidden;
padding-inline: var(--download-column-padding-x);
}
.download-table-header > div:first-child {
padding-left: 0;
}
.download-table-header > div:last-child {
padding-right: 0;
}
.download-table-header > div > span {
@@ -1686,7 +1696,7 @@ html[data-list-density="relaxed"] {
}
.download-table-header > div:not(:first-child) {
padding-left: 0;
padding-left: var(--download-column-padding-x);
}
.column-resize-handle {
@@ -1766,6 +1776,15 @@ html[data-list-density="relaxed"] .download-ghost-row {
.download-row > div {
min-width: 0;
overflow: hidden;
padding-inline: var(--download-column-padding-x);
}
.download-row > div:first-child {
padding-left: 0;
}
.download-row > div:last-child {
padding-right: 0;
}
@@ -1805,7 +1824,7 @@ html[data-list-density="relaxed"] .download-ghost-row {
.download-cell-truncate {
display: flex;
align-items: center;
justify-content: center;
justify-content: flex-start;
min-width: 0;
overflow: hidden;
}
@@ -1813,7 +1832,7 @@ html[data-list-density="relaxed"] .download-ghost-row {
.download-cell-truncate > span,
.download-cell-right > span {
display: block;
text-align: center;
text-align: left;
max-width: 100%;
min-width: 0;
overflow: hidden;
@@ -1826,9 +1845,9 @@ html[data-list-density="relaxed"] .download-ghost-row {
}
.download-cell-right {
text-align: center;
text-align: left;
display: flex;
justify-content: center;
justify-content: flex-start;
align-items: center;
min-width: 0;
overflow: hidden;
@@ -1858,7 +1877,7 @@ html[data-list-density="relaxed"] .download-ghost-row {
.download-status-cell {
display: flex;
align-items: center;
justify-content: center;
justify-content: flex-start;
gap: var(--download-status-gap);
min-width: 0;
overflow: hidden;
@@ -1868,7 +1887,7 @@ html[data-list-density="relaxed"] .download-ghost-row {
min-width: 0;
max-width: 100%;
flex: 0 0 auto;
text-align: center;
text-align: left;
overflow: hidden;
text-overflow: ellipsis;
white-space: nowrap;