Add rust implementation test

2 files changed, 99 insertions, 0 deletions

diff --git a/src/lyrics_engine/genius.rs b/src/lyrics_engine/genius.rs
new file mode 100644
index 0000000..e14589d
--- /dev/null
+++ b/src/lyrics_engine/genius.rs
@@ -0,0 +1,73 @@
+use std::io::{Result, Error, ErrorKind::Other};
+use serde::Deserialize;
+use reqwest::get;
+use scraper::{ElementRef, Html, Selector, node::Node::{Text, Element}};
+
+#[derive(Deserialize, Debug)]
+struct GeniusApiResponse {
+    response: GeniusResponseBlock,
+}
+
+#[derive(Deserialize, Debug)]
+struct GeniusResponseBlock {
+    hits: Vec<GeniusHit>,
+}
+
+#[derive(Deserialize, Debug)]
+struct GeniusHit {
+    #[serde(rename = "type")]
+    hit_type: String,
+    result: GeniusResult,
+}
+
+#[derive(Deserialize, Debug)]
+struct GeniusResult {
+    url: String,
+}
+
+pub async fn search(url: &str) -> Result<String> {
+    let response = get(url).await
+        .map_err(|_| Error::new(Other, "Could not perform lyrics search in engine."))?
+        .json::<GeniusApiResponse>().await
+        .map_err(|_| Error::new(Other, "Lyrics engine returned invalid response from search."))?;
+    let url = response.response.hits.into_iter()
+        .find(|hit| hit.hit_type == "song")
+        .map(|hit| hit.result.url)
+        .ok_or_else(|| Error::new(Other, "Could not find a matching track in lyrics engine."))?;
+
+    Ok(url)
+}
+
+pub async fn get_lyrics(url: &str) -> Result<String> {
+    let song_html = get(url).await
+        .map_err(|_| Error::new(Other, "Could not fetch lyrics from engine."))?
+        .text().await
+        .map_err(|_| Error::new(Other, "Lyrics engine returned invalid response."))?;
+
+    let document = Html::parse_document(&song_html);
+    let selector = Selector::parse(r#"div[data-lyrics-container="true"]"#).unwrap();
+    let lyrics_div = document.select(&selector).next()
+        .ok_or_else(|| Error::new(Other, "Could not find lyrics in response."))?;
+
+    let mut lyrics = String::new();
+    for node in lyrics_div.children() {
+        match node.value() {
+            Element(element) => {
+                if element.name() == "br" {
+                    lyrics.push('\n');
+                } else {
+                    if let Some(element_ref) = ElementRef::wrap(node) {
+                        let text = element_ref.text().collect::<Vec<_>>().join("");
+                        lyrics.push_str(&text);
+                    }
+                }
+            },
+            Text(text) => {
+                lyrics.push_str(text);
+            },
+            _ => {}
+        }
+    }
+
+    Ok(lyrics)
+}
diff --git a/src/lyrics_engine/mod.rs b/src/lyrics_engine/mod.rs
new file mode 100644
index 0000000..f83d219
--- /dev/null
+++ b/src/lyrics_engine/mod.rs
@@ -0,0 +1,26 @@
+use std::io::Result;
+use percent_encoding::{utf8_percent_encode, NON_ALPHANUMERIC};
+
+mod genius;
+
+use crate::Track;
+use genius::{get_lyrics, search};
+
+const GENIUS_CLIENT_TOKEN: &str = env!("LYRICLI_GENIUS_TOKEN");
+const GENIUS_API_URL: &str = "https://api.genius.com/search";
+
+pub async fn print_lyrics(track: Track, show_title: bool) -> Result<()> {
+    if show_title {
+        println!("{} - {}", track.artist, track.name);
+    }
+
+    let artist = utf8_percent_encode(&track.artist, NON_ALPHANUMERIC).to_string();
+    let name = utf8_percent_encode(&track.name, NON_ALPHANUMERIC).to_string();
+    let url = format!("{}?access_token={}&q={}%20{}", GENIUS_API_URL, GENIUS_CLIENT_TOKEN, artist, name);
+
+    let song_url = search(&url).await?;
+    let lyrics = get_lyrics(&song_url).await?;
+
+    println!("{}", lyrics);
+    Ok(())
+}