Ultradesu
80 lines
2.1 KiB
Rust
80 lines
2.1 KiB
Rust
pub mod artist_image_backfill;
|
|
pub mod artist_track_image_backfill;
|
|
pub mod cover_backfill;
|
|
pub mod cover_variant_backfill;
|
|
pub mod inbox_discover;
|
|
pub mod inbox_process;
|
|
pub mod lastfm_popularity;
|
|
pub mod lastfm_scrobble;
|
|
pub mod metadata_backfill;
|
|
|
|
use std::path::{Component, Path, PathBuf};
|
|
|
|
#[derive(Debug, Clone)]
|
|
pub struct UploaderAttribution {
|
|
pub user_id: Option<i64>,
|
|
pub name: String,
|
|
}
|
|
|
|
impl UploaderAttribution {
|
|
pub fn unknown() -> Self {
|
|
Self {
|
|
user_id: None,
|
|
name: "UFO".to_string(),
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn strip_user_upload_prefix(relative_path: &Path) -> PathBuf {
|
|
let components: Vec<_> = relative_path.components().collect();
|
|
if components.len() >= 3
|
|
&& matches!(components[0], Component::Normal(value) if value == "user_uploads")
|
|
{
|
|
components[2..].iter().collect()
|
|
} else {
|
|
relative_path.to_path_buf()
|
|
}
|
|
}
|
|
|
|
pub async fn uploader_from_relative_path(
|
|
pool: &sqlx::PgPool,
|
|
relative_path: &Path,
|
|
) -> UploaderAttribution {
|
|
let components: Vec<_> = relative_path.components().collect();
|
|
let Some(Component::Normal(root)) = components.first() else {
|
|
return UploaderAttribution::unknown();
|
|
};
|
|
if *root != "user_uploads" {
|
|
return UploaderAttribution::unknown();
|
|
}
|
|
|
|
let Some(Component::Normal(user_id_os)) = components.get(1) else {
|
|
return UploaderAttribution::unknown();
|
|
};
|
|
let Some(user_id_str) = user_id_os.to_str() else {
|
|
return UploaderAttribution::unknown();
|
|
};
|
|
let Ok(user_id) = user_id_str.parse::<i64>() else {
|
|
return UploaderAttribution::unknown();
|
|
};
|
|
|
|
let name: Option<String> = sqlx::query_scalar(
|
|
r#"SELECT COALESCE(NULLIF(display_name, ''), username)::text
|
|
FROM furumusic__user
|
|
WHERE id = $1 AND is_active = true"#,
|
|
)
|
|
.bind(user_id)
|
|
.fetch_optional(pool)
|
|
.await
|
|
.ok()
|
|
.flatten();
|
|
|
|
match name {
|
|
Some(name) if !name.trim().is_empty() => UploaderAttribution {
|
|
user_id: Some(user_id),
|
|
name,
|
|
},
|
|
_ => UploaderAttribution::unknown(),
|
|
}
|
|
}
|