Merge pull request #25 from brqnko/main

feat: search suggestion

examples/suggestion.rs

@@ -0,0 +1,12 @@
+use rusty_ytdl::search::YouTube;
+
+#[tokio::main]
+async fn main() {
+    use rusty_ytdl::search::YouTube;
+
+    let youtube = YouTube::new().unwrap();
+
+    let res = youtube.suggestion("i know ").await;
+
+    println!("{res:#?}");
+}

src/search/youtube.rs

@@ -201,6 +201,43 @@ impl YouTube {
         Ok(res.first().cloned())
     }
 
+    /// Fetch search suggestion with specific `query`
+    /// # Example
+    /// ```ignore
+    /// let youtube = YouTube::new().unwrap();
+    /// 
+    /// let res = youtube.suggestion("i know ").await;
+    /// 
+    /// println!("{res:#?}");
+    /// ```
+    pub async fn suggestion(
+        &self,
+        query: impl Into<String>,
+    ) -> Result<Vec<String>, VideoError> {
+        let query: String = query.into();
+        let url = format!(
+            "https://suggestqueries-clients6.youtube.com/complete/search?client=android&q={query}",
+            query = encode(query.trim())
+        );
+
+        let body = get_html(&self.client, url, None).await?;
+
+        let serde_value = serde_json::from_str::<serde_json::Value>(&body).unwrap();
+
+        let suggestion = serde_value
+            .as_array()
+            .unwrap()
+            .get(1)
+            .unwrap()
+            .as_array()
+            .unwrap()
+            .iter()
+            .map(|x| x.to_string())
+            .collect();
+
+        Ok(suggestion)
+    }
+
     async fn innertube_key(&self) -> String {
         {
             let innertube_cache = self.innertube_cache.read().unwrap();

tests/suggestion.rs

@@ -0,0 +1,10 @@
+#[tokio::test]
+async fn suggestion() {
+    use rusty_ytdl::search::YouTube;
+
+    let youtube = YouTube::new().unwrap();
+
+    let res = youtube.suggestion("i know ").await;
+
+    println!("{res:#?}");
+}