From 2c6f753797b8e223c5982c4790f1be8b08fe63d6 Mon Sep 17 00:00:00 2001
From: metamuffin <metamuffin@disroot.org>
Date: Fri, 25 Apr 2025 21:34:59 +0200
Subject: images from wikidata works

---
 import/src/musicbrainz.rs | 107 +++++++++++++++++++++++++++++++++++++++++++---
 1 file changed, 102 insertions(+), 5 deletions(-)

(limited to 'import/src/musicbrainz.rs')

diff --git a/import/src/musicbrainz.rs b/import/src/musicbrainz.rs
index 0fe839b..704807b 100644
--- a/import/src/musicbrainz.rs
+++ b/import/src/musicbrainz.rs
@@ -53,6 +53,24 @@ pub struct MbRecordingRel {
     pub artist_credit: Vec<MbArtistCredit>,
 }
 
+#[derive(Debug, Deserialize, Encode, Decode)]
+#[serde(rename_all = "kebab-case")]
+pub struct MbArtistRel {
+    pub id: String,
+    pub isnis: Vec<String>,
+    pub ipis: Vec<String>,
+    pub name: String,
+    pub disambiguation: String,
+    pub country: Option<String>,
+    pub sort_name: String,
+    pub gender_id: Option<String>,
+    pub area: Option<MbArea>,
+    pub begin_area: Option<MbArea>,
+    pub end_area: Option<MbArea>,
+    pub life_span: MbTimespan,
+    pub relations: Vec<MbRelation>,
+}
+
 #[derive(Debug, Deserialize, Encode, Decode)]
 #[serde(rename_all = "kebab-case")]
 pub struct MbArtistCredit {
@@ -81,6 +99,7 @@ pub struct MbRelation {
     pub url: Option<MbUrl>,
     pub recording: Option<MbRecording>,
     pub series: Option<MbSeries>,
+    pub event: Option<MbEvent>,
 }
 
 #[derive(Debug, Deserialize, Encode, Decode)]
@@ -98,10 +117,12 @@ pub struct MbSeries {
 pub struct MbRecording {
     pub id: String,
     pub title: String,
+    #[serde(default)]
     pub isrcs: Vec<String>,
     pub video: bool,
     pub disambiguation: String,
-    pub length: u32,
+    pub length: Option<u32>,
+    #[serde(default)]
     pub artist_credit: Vec<MbArtistCredit>,
 }
 
@@ -109,8 +130,8 @@ pub struct MbRecording {
 #[serde(rename_all = "kebab-case")]
 pub struct MbWork {
     pub id: String,
-    pub r#type: String,
-    pub type_id: String,
+    pub r#type: Option<String>,
+    pub type_id: Option<String>,
     pub languages: Vec<String>,
     pub iswcs: Vec<String>,
     pub language: Option<String>,
@@ -119,6 +140,19 @@ pub struct MbWork {
     pub disambiguation: String,
 }
 
+#[derive(Debug, Deserialize, Encode, Decode)]
+#[serde(rename_all = "kebab-case")]
+pub struct MbEvent {
+    pub id: String,
+    pub r#type: String,
+    pub type_id: String,
+    pub name: String,
+    pub time: String,
+    pub cancelled: bool,
+    pub setlist: String,
+    pub life_span: MbTimespan,
+}
+
 #[derive(Debug, Deserialize, Encode, Decode)]
 #[serde(rename_all = "kebab-case")]
 pub struct MbArtist {
@@ -131,11 +165,30 @@ pub struct MbArtist {
     pub sort_name: String,
 }
 
+#[derive(Debug, Deserialize, Encode, Decode)]
+#[serde(rename_all = "kebab-case")]
+pub struct MbTimespan {
+    pub begin: Option<String>,
+    pub end: Option<String>,
+    pub ended: bool,
+}
+
+#[derive(Debug, Deserialize, Encode, Decode)]
+#[serde(rename_all = "kebab-case")]
+pub struct MbArea {
+    pub name: String,
+    pub sort_name: String,
+    #[serde(default)]
+    pub iso_3166_1_codes: Vec<String>,
+    pub id: String,
+    pub disambiguation: String,
+}
+
 #[derive(Debug, Deserialize, Encode, Decode)]
 #[serde(rename_all = "kebab-case")]
 pub struct MbUrl {
-    id: String,
-    resource: String,
+    pub id: String,
+    pub resource: String,
 }
 
 impl MusicBrainz {
@@ -206,4 +259,48 @@ impl MusicBrainz {
         })
         .await
     }
+
+    pub async fn lookup_artist(&self, id: String) -> Result<Arc<MbArtistRel>> {
+        async_cache_memory("api-musicbrainz-artist", id.clone(), || async move {
+            let _permit = self.rate_limit.clone().acquire_owned().await?;
+            let permit_drop_ts = Instant::now() + Duration::from_secs(10);
+            info!("artist lookup: {id}");
+
+            let inc = [
+                "area-rels",
+                "artist-rels",
+                "event-rels",
+                "genre-rels",
+                "instrument-rels",
+                "label-rels",
+                "place-rels",
+                "recording-rels",
+                "release-rels",
+                "release-group-rels",
+                "series-rels",
+                "url-rels",
+                "work-rels",
+            ]
+            .join("+");
+
+            let resp = self
+                .client
+                .get(format!(
+                    "https://musicbrainz.org/ws/2/artist/{id}?inc={inc}"
+                ))
+                .send()
+                .await?
+                .error_for_status()?
+                .json::<MbArtistRel>()
+                .await?;
+
+            tokio::task::spawn(async move {
+                sleep_until(permit_drop_ts).await;
+                drop(_permit);
+            });
+
+            Ok(resp)
+        })
+        .await
+    }
 }
-- 
cgit v1.2.3-70-g09d2