Compare commits

...

1 Commits

Author SHA1 Message Date
Connor Johnstone
d358b79a6b Formatting 2026-03-18 15:37:21 -04:00
3 changed files with 59 additions and 29 deletions

View File

@@ -82,7 +82,10 @@ async fn main() -> anyhow::Result<()> {
} else if results.is_empty() { } else if results.is_empty() {
println!("No artists found."); println!("No artists found.");
} else { } else {
println!("{:<40} {:<8} {:<5} {}", "NAME", "COUNTRY", "SCORE", "DISAMBIGUATION"); println!(
"{:<40} {:<8} {:<5} DISAMBIGUATION",
"NAME", "COUNTRY", "SCORE"
);
for r in &results { for r in &results {
println!( println!(
"{:<40} {:<8} {:<5} {}", "{:<40} {:<8} {:<5} {}",
@@ -104,7 +107,10 @@ async fn main() -> anyhow::Result<()> {
} else if results.is_empty() { } else if results.is_empty() {
println!("No albums found."); println!("No albums found.");
} else { } else {
println!("{:<35} {:<25} {:<6} {:<5}", "TITLE", "ARTIST", "YEAR", "SCORE"); println!(
"{:<35} {:<25} {:<6} {:<5}",
"TITLE", "ARTIST", "YEAR", "SCORE"
);
for r in &results { for r in &results {
println!( println!(
"{:<35} {:<25} {:<6} {:<5}", "{:<35} {:<25} {:<6} {:<5}",
@@ -126,13 +132,19 @@ async fn main() -> anyhow::Result<()> {
} else if results.is_empty() { } else if results.is_empty() {
println!("No tracks found."); println!("No tracks found.");
} else { } else {
println!("{:<35} {:<25} {:<25} {:<5}", "TITLE", "ARTIST", "ALBUM", "SCORE"); println!(
"{:<35} {:<25} {:<25} {:<5}",
"TITLE", "ARTIST", "ALBUM", "SCORE"
);
for r in &results { for r in &results {
println!( println!(
"{:<35} {:<25} {:<25} {:<5}", "{:<35} {:<25} {:<25} {:<5}",
truncate(&r.title, 35), truncate(&r.title, 35),
truncate(&r.artist, 25), truncate(&r.artist, 25),
r.album.as_deref().map(|a| truncate(a, 25)).unwrap_or_default(), r.album
.as_deref()
.map(|a| truncate(a, 25))
.unwrap_or_default(),
r.score, r.score,
); );
} }

View File

@@ -1,5 +1,5 @@
use shanty_tag::provider::MetadataProvider;
use shanty_tag::MusicBrainzClient; use shanty_tag::MusicBrainzClient;
use shanty_tag::provider::MetadataProvider;
use crate::error::SearchResult; use crate::error::SearchResult;
use crate::provider::{ use crate::provider::{
@@ -20,11 +20,7 @@ impl MusicBrainzSearch {
} }
impl SearchProvider for MusicBrainzSearch { impl SearchProvider for MusicBrainzSearch {
async fn search_artist( async fn search_artist(&self, query: &str, limit: u32) -> SearchResult<Vec<ArtistResult>> {
&self,
query: &str,
limit: u32,
) -> SearchResult<Vec<ArtistResult>> {
let results = self.client.search_artist(query, limit).await?; let results = self.client.search_artist(query, limit).await?;
Ok(results Ok(results
.into_iter() .into_iter()
@@ -55,7 +51,11 @@ impl SearchProvider for MusicBrainzSearch {
title: r.title, title: r.title,
artist: r.artist, artist: r.artist,
artist_id: r.artist_mbid, artist_id: r.artist_mbid,
year: r.date.as_deref().and_then(|d| d.split('-').next()).map(String::from), year: r
.date
.as_deref()
.and_then(|d| d.split('-').next())
.map(String::from),
track_count: r.track_count, track_count: r.track_count,
score: r.score, score: r.score,
}) })
@@ -85,10 +85,7 @@ impl SearchProvider for MusicBrainzSearch {
.collect()) .collect())
} }
async fn get_discography( async fn get_discography(&self, artist_id: &str) -> SearchResult<Discography> {
&self,
artist_id: &str,
) -> SearchResult<Discography> {
let releases = self.client.get_artist_releases(artist_id, 100).await?; let releases = self.client.get_artist_releases(artist_id, 100).await?;
// Try to get the artist name from the first release, or use the MBID // Try to get the artist name from the first release, or use the MBID

View File

@@ -87,7 +87,10 @@ impl SearchProvider for MockSearch {
}) })
} }
async fn get_release_groups(&self, _artist_id: &str) -> SearchResult<Vec<shanty_search::ReleaseGroupResult>> { async fn get_release_groups(
&self,
_artist_id: &str,
) -> SearchResult<Vec<shanty_search::ReleaseGroupResult>> {
Ok(vec![]) Ok(vec![])
} }
} }
@@ -104,14 +107,20 @@ async fn test_search_artist() {
#[tokio::test] #[tokio::test]
async fn test_search_artist_no_results() { async fn test_search_artist_no_results() {
let provider = MockSearch; let provider = MockSearch;
let results = provider.search_artist("Nonexistent Band", 10).await.unwrap(); let results = provider
.search_artist("Nonexistent Band", 10)
.await
.unwrap();
assert!(results.is_empty()); assert!(results.is_empty());
} }
#[tokio::test] #[tokio::test]
async fn test_search_album() { async fn test_search_album() {
let provider = MockSearch; let provider = MockSearch;
let results = provider.search_album("Dark Side", Some("Pink Floyd"), 10).await.unwrap(); let results = provider
.search_album("Dark Side", Some("Pink Floyd"), 10)
.await
.unwrap();
assert_eq!(results.len(), 1); assert_eq!(results.len(), 1);
assert_eq!(results[0].title, "The Dark Side of the Moon"); assert_eq!(results[0].title, "The Dark Side of the Moon");
} }
@@ -119,10 +128,16 @@ async fn test_search_album() {
#[tokio::test] #[tokio::test]
async fn test_search_track() { async fn test_search_track() {
let provider = MockSearch; let provider = MockSearch;
let results = provider.search_track("Time", Some("Pink Floyd"), 10).await.unwrap(); let results = provider
.search_track("Time", Some("Pink Floyd"), 10)
.await
.unwrap();
assert_eq!(results.len(), 1); assert_eq!(results.len(), 1);
assert_eq!(results[0].title, "Time"); assert_eq!(results[0].title, "Time");
assert_eq!(results[0].album.as_deref(), Some("The Dark Side of the Moon")); assert_eq!(
results[0].album.as_deref(),
Some("The Dark Side of the Moon")
);
} }
#[tokio::test] #[tokio::test]
@@ -162,18 +177,25 @@ async fn test_cache_roundtrip() {
}]; }];
// Cache miss // Cache miss
let cached: Option<Vec<ArtistResult>> = let cached: Option<Vec<ArtistResult>> = cache::get_cached(Some(db.conn()), "test:artist:query")
cache::get_cached(Some(db.conn()), "test:artist:query").await.unwrap(); .await
.unwrap();
assert!(cached.is_none()); assert!(cached.is_none());
// Store // Store
cache::set_cached(Some(db.conn()), "test:artist:query", "musicbrainz", &results) cache::set_cached(
Some(db.conn()),
"test:artist:query",
"musicbrainz",
&results,
)
.await .await
.unwrap(); .unwrap();
// Cache hit // Cache hit
let cached: Option<Vec<ArtistResult>> = let cached: Option<Vec<ArtistResult>> = cache::get_cached(Some(db.conn()), "test:artist:query")
cache::get_cached(Some(db.conn()), "test:artist:query").await.unwrap(); .await
.unwrap();
assert!(cached.is_some()); assert!(cached.is_some());
assert_eq!(cached.unwrap()[0].name, "Cached Artist"); assert_eq!(cached.unwrap()[0].name, "Cached Artist");
} }
@@ -181,8 +203,7 @@ async fn test_cache_roundtrip() {
#[tokio::test] #[tokio::test]
async fn test_cache_none_conn() { async fn test_cache_none_conn() {
// With no DB connection, caching is a no-op // With no DB connection, caching is a no-op
let cached: Option<Vec<ArtistResult>> = let cached: Option<Vec<ArtistResult>> = cache::get_cached(None, "anything").await.unwrap();
cache::get_cached(None, "anything").await.unwrap();
assert!(cached.is_none()); assert!(cached.is_none());
// set_cached with None conn should not error // set_cached with None conn should not error