Compare commits
1 Commits
d09557d953
...
d358b79a6b
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
d358b79a6b |
20
src/main.rs
20
src/main.rs
@@ -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,
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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
|
||||||
|
|||||||
@@ -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(
|
||||||
.await
|
Some(db.conn()),
|
||||||
.unwrap();
|
"test:artist:query",
|
||||||
|
"musicbrainz",
|
||||||
|
&results,
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
.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
|
||||||
|
|||||||
Reference in New Issue
Block a user