feat(api): Add TMDB API calls

Cargo.toml

@@ -10,7 +10,10 @@ clap = { version = "4.4.7", features = ["derive"] }
 infer = "0.15.0"
 inquire = "0.6.2"
 log = "0.4.20"
+reqwest = { version = "0.11.22", features = ["json", "blocking"] }
 serde = { version = "1.0.190", features = ["derive"] }
 serde_json = "1.0.108"
 stderrlog = "0.5.4"
+tmdb_client = "1.6.0"
+urlencoding = "2.1.3"
 walkdir = "2.4.0"

src/directory.rs

@@ -1,6 +1,6 @@
 use std::{path::PathBuf, fs::{self, DirEntry}, error::Error};
 
-use crate::media::handle_media;
+use crate::{media::handle_media, config::Config};
 
 /*fn is_not_hidden(entry: &DirEntry) -> bool {
     entry
@@ -20,7 +20,7 @@ pub fn walk_path(path: PathBuf) -> Vec<PathBuf> {
     entries
 }*/
 
-pub fn search_path(path: PathBuf) -> Result<(), Box<dyn Error>> {
+pub fn search_path(path: PathBuf, cfg: Config) -> Result<(), Box<dyn Error>> {
     let entries = fs::read_dir(path)?;
     let mut files: Vec<DirEntry> = Vec::new();
     let mut dirs: Vec<DirEntry> = Vec::new();
@@ -41,7 +41,7 @@ pub fn search_path(path: PathBuf) -> Result<(), Box<dyn Error>> {
     if dirs.len() == 0 {
         // No folders present, assuming there are only distinct media files
         for file in files {
-            handle_media(file);
+            handle_media(file, cfg.clone());
         }
     }
 

src/main.rs

@@ -64,7 +64,7 @@ fn main() {
     };
 
     //let files = directory::walk_path(search_path);
-    directory::search_path(search_path).unwrap();
+    directory::search_path(search_path, cfg).unwrap();
 
     /*for file in files.clone() {
         info!("Found: {}", file.to_str().unwrap());

src/media.rs

@@ -1,11 +1,45 @@
-use std::{path::PathBuf, error::Error, io::Read, fs::{File, DirEntry}, cmp};
+use std::{path::PathBuf, error::Error, io::Read, fs::{File, DirEntry}, cmp, fmt, ops::Deref};
 use infer;
+use inquire::Select;
 use log::{info, warn, error, trace, debug};
+use serde::Deserialize;
+use urlencoding::encode;
 
-#[derive(Debug)]
-struct MediaName {
-    name: String,
-    year: String
+use crate::config::Config;
+
+#[derive(Deserialize, Debug)]
+struct TMDBResponse {
+    page: i32,
+    results: Vec<TMDBEntry>,
+    total_pages: i32,
+    total_results: i32
+}
+
+#[derive(Deserialize, Debug)]
+struct TMDBEntry {
+    id: i32,
+    #[serde(alias = "name")]
+    title: String,
+    original_language: Option<String>,
+    #[serde(alias = "original_name")]
+    original_title: String,
+    overview: Option<String>,
+    media_type: String,
+    popularity: f32,
+    #[serde(alias = "first_air_date")]
+    release_date: Option<String>,
+}
+
+impl fmt::Display for TMDBEntry {
+    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+        if self.media_type == "movie" {
+            write!(f, "[MOVIE] {} ({}, {}) (ID: {})", self.title, self.release_date.clone().unwrap_or("unknown".to_string()), self.original_language.as_ref().unwrap(), self.id)
+        } else if self.media_type == "tv" {
+            write!(f, "[SHOW] {} ({}, {}) (ID: {})", self.title, self.release_date.clone().unwrap_or("unknown".to_string()), self.original_language.as_ref().unwrap(), self.id)
+        } else {
+            write!(f, "[{}] {} (ID: {})", self.media_type, self.title, self.id)
+        }
+    }
 }
 
 fn get_file_header(path: PathBuf) -> Result<Vec<u8>, Box<dyn Error>> {
@@ -20,22 +54,13 @@ fn get_file_header(path: PathBuf) -> Result<Vec<u8>, Box<dyn Error>> {
     Ok(bytes)
 }
 
-fn token_year_likely(t: &&str) -> bool {
-    if t.len() == 6 &&
-        (t.starts_with('[') && t.ends_with(']')) ||
-        (t.starts_with('(') && t.ends_with(')')) ||
-        (t.starts_with('{') && t.ends_with('}'))   
-    {
-        return true;    
-    }
-    return false
-}
-
 fn token_valid(t: &&str) -> bool {
     if
         t.eq_ignore_ascii_case("dvd") ||
         t.eq_ignore_ascii_case("bluray") ||
         t.eq_ignore_ascii_case("webrip") ||
+        t.eq_ignore_ascii_case("youtube") ||
+        t.eq_ignore_ascii_case("download") ||
         t.eq_ignore_ascii_case("web") ||
         t.eq_ignore_ascii_case("uhd") ||
         t.eq_ignore_ascii_case("hd") ||
@@ -44,54 +69,102 @@ fn token_valid(t: &&str) -> bool {
         t.eq_ignore_ascii_case("1080p") ||
         t.eq_ignore_ascii_case("1080i") ||
         t.eq_ignore_ascii_case("2160p") ||
-        (t.len() != 6 && t.starts_with('[') && t.ends_with(']')) ||
-        (t.len() != 6 && t.starts_with('(') && t.ends_with(')')) ||
-        (t.len() != 6 && t.starts_with('{') && t.ends_with('}'))
+        t.eq_ignore_ascii_case("x264") ||
+        t.eq_ignore_ascii_case("x265") ||
+        t.eq_ignore_ascii_case("h265") ||
+        t.eq_ignore_ascii_case("dts") ||
+        t.eq_ignore_ascii_case("hevc") ||
+        t.eq_ignore_ascii_case("10bit") ||
+        t.eq_ignore_ascii_case("12bit") ||
+        t.eq_ignore_ascii_case("hdr") ||
+        t.eq_ignore_ascii_case("xvid") ||
+        t.eq_ignore_ascii_case("AAC5") ||
+        t.eq_ignore_ascii_case("AAC") ||
+        t.eq_ignore_ascii_case("AC3") ||
+        t.eq_ignore_ascii_case("sample") ||             // This just removes the word sample, maybe we want to ban files with the word sample all together
+        (t.starts_with('[') || t.ends_with(']')) ||
+        (t.starts_with('(') || t.ends_with(')')) ||
+        (t.starts_with('{') || t.ends_with('}'))
     {
         return false;
     }
     true
 }
 
-fn find_media_name(file_name: String) -> MediaName {
-    let mut tokens: Vec<&str> = file_name.split(&['-', ' ', ':', '@', '.'][..]).filter(|t| token_valid(t)).collect();
+fn tokenize_media_name(file_name: String) -> Vec<String> {
+    let mut tokens: Vec<String> = file_name.split(&['-', ' ', ':', '@', '.'][..]).filter(|t| token_valid(t)).map(String::from).collect();
     trace!("Tokens are: {:#?}", tokens);
 
     // Remove last token (file ext)
     _ = tokens.pop();
+    tokens
+}
 
-    let mut year = String::new();
-    let mut name = String::new();
+fn lookup_media(file_name: PathBuf, mut name_tokens: Vec<String>, cfg: Config) -> Option<TMDBEntry> {
+    let mut h = reqwest::header::HeaderMap::new();
+    h.insert("Accept", reqwest::header::HeaderValue::from_static("application/json"));
+    h.insert("Authorization", reqwest::header::HeaderValue::from_str(format!("Bearer {}", cfg.tmdb_key).as_str()).unwrap());
     
-    for token in tokens {
-        if token_year_likely(&token)  {
-            year = token.strip_prefix(['(', '[', '{']).unwrap().strip_suffix([')', ']', '}']).unwrap().to_string();
-        } else if token.len() != 0 {
-            name.push_str(token);
-            name.push(' ');
+    let client = reqwest::blocking::Client::builder()
+        .default_headers(h)
+        .build().unwrap();
+
+    let mut response: TMDBResponse;
+    loop {
+        if name_tokens.len() == 0 {
+            error!("Could not find title on TMDB!");
+            return None;
+        }
+
+        let name = name_tokens.join(" ");
+        trace!("Searching on TMDB for {:#?}", name);
+
+        let http_response = client
+            .get(format!("https://api.themoviedb.org/3/search/multi?query={}&include_adult=false&language=en-US&page=1", encode(name.as_str()).into_owned()))
+            .send().unwrap();
+
+        response = http_response.json::<TMDBResponse>().unwrap();
+        trace!("TMDB Reponse: {:#?}", response);
+
+        if response.total_results == 0 {
+            name_tokens.pop();
+        } else {
+            break;
         }
     }
 
-    // Remove last added space
-    name.pop();
+    let options = response.results;
 
-    let media_name = MediaName { name: name, year: year };
-    debug!("Name is now: {:#?}", media_name);
-    media_name
+    let ans = Select::new(format!("Select movie or show that matches the file \x1b[93m{}\x1b[0m:", file_name.display()).as_str(), options).prompt();
+    match ans {
+        Ok(choice) => {
+            debug!("Selected: {:#?}", choice);
+            return Some(choice);
+        },
+        Err(e) => {
+            error!("Error while selecting content: {:#?}", e);
+            return None;
+        },
+    }
 }
 
-fn video_file_handler(entry: DirEntry) {
+fn video_file_handler(entry: DirEntry, cfg: Config) {
     let path = entry.path();
     info!("Found video file: {:#?}", path);
 
     let file_name = path.file_name().unwrap_or_default();
     trace!("File name is: {:#?}", file_name);
 
-    let name = find_media_name(file_name.to_str().unwrap_or_default().to_string());
-    todo!("Do TMDB API calls");
+    let name_tokens = tokenize_media_name(file_name.to_str().unwrap_or_default().to_string());
+    
+    match lookup_media(entry.path(), name_tokens, cfg) {
+        Some(entry) => todo!("Save media info in some struct to move media afterwards, or move directly"),
+        None => {}, 
+    }
+
 }
 
-pub fn handle_media(entry: DirEntry) {
+pub fn handle_media(entry: DirEntry, cfg: Config) {
     if entry.file_type().is_ok_and(|t| t.is_dir()) {
         warn!("Directory passed to handle_media, {:#?} will be skipped", entry);
         return
@@ -101,7 +174,7 @@ pub fn handle_media(entry: DirEntry) {
         Ok(header) => {
             // Handle video files
             if infer::is_video(&header) {
-                video_file_handler(entry);
+                video_file_handler(entry, cfg.clone());
             }
         },
         Err(error) => error!("Can not get file header for {:#?}, Error: {:#?}", entry, error),