working on new sources

GoodDay360 · GoodDay360 · commit 6359d0afa39e · 2025-10-06T23:04:05.000+07:00
diff --git a/src-tauri/Cargo.lock b/src-tauri/Cargo.lock
diff --git a/src-tauri/Cargo.toml b/src-tauri/Cargo.toml
@@ -58,6 +58,7 @@ lazy_static = "1.5.0"
 tauri-plugin-android-fs = "16.1.2"
 tauri-plugin-dialog = "2"
 dunce = "1.0.5"
+html-escape = "0.2.13"
 
 [target.'cfg(target_os = "android")'.dependencies]
 tauri-plugin-android-package-install = "2.0.2"
diff --git a/src-tauri/src/commands/get_content.rs b/src-tauri/src/commands/get_content.rs
@@ -6,7 +6,7 @@ use tracing::{error, warn};
 
 use chlaty_core::request_plugin::get_episode_list::DataResult;
 
-use crate::anime;
+use crate::sources::anime;
 use crate::commands::favorite::{get_recent_from_favorite, get_tag_from_favorite};
 use crate::commands::local_manifest::{get_local_manifest, save_local_manifest};
 use crate::commands::request_plugin::get_episode_list::get_episode_list;
diff --git a/src-tauri/src/lib.rs b/src-tauri/src/lib.rs
@@ -7,7 +7,7 @@ use tauri::async_runtime;
 
 use chlaty_core;
 
-pub mod anime;
+pub mod sources;
 pub mod models;
 pub mod utils;
 
diff --git a/src-tauri/src/sources/anime/home.rs b/src-tauri/src/sources/anime/home.rs
@@ -3,7 +3,7 @@ use std::time::Duration;
 use tokio;
 use tracing::error;
 
-use crate::anime::models::ApiResponse;
+use crate::sources::anime::models::ApiResponse;
 use crate::models::home::{Content, ContentData, HomeData, RelevantContent, Trailer};
 use crate::utils::get_calendar;
 
diff --git a/src-tauri/src/sources/anime/mod.rs b/src-tauri/src/sources/anime/mod.rs
diff --git a/src-tauri/src/sources/anime/models.rs b/src-tauri/src/sources/anime/models.rs
diff --git a/src-tauri/src/sources/anime/search.rs b/src-tauri/src/sources/anime/search.rs
@@ -2,7 +2,7 @@ use reqwest::Client;
 use std::time::Duration;
 use urlencoding::encode;
 
-use crate::anime::models::ApiResponse;
+use crate::sources::anime::models::ApiResponse;
 use crate::models::search::{Content, SearchData};
 
 const LIMIT: usize = 20;
diff --git a/src-tauri/src/sources/anime/view.rs b/src-tauri/src/sources/anime/view.rs
@@ -4,7 +4,7 @@ use std::time::Duration;
 use std::vec;
 use urlencoding::encode;
 
-use crate::anime::models::Data;
+use crate::sources::anime::models::Data;
 use crate::models::view::{ManifestData, Trailer};
 
 async fn get_content(id: &str) -> Result<ManifestData, String> {
diff --git a/src-tauri/src/sources/mod.rs b/src-tauri/src/sources/mod.rs
@@ -0,0 +1,2 @@
+pub mod anime;
+// pub mod movie;
diff --git a/src-tauri/src/sources/movie/home.rs b/src-tauri/src/sources/movie/home.rs
@@ -0,0 +1,61 @@
+use reqwest::Client;
+use visdom::Vis;
+use visdom::types::Elements;
+use html_escape::decode_html_entities;
+
+
+use crate::models::home::{HomeData, RelevantContent, Content, ContentData};
+
+pub async fn get_relevant_content(vis_root: Elements<'static>) -> Result<(), String> {
+    
+    let swiper_wrapper = vis_root.find(".swiper-wrapper");
+    let swiper_slide = swiper_wrapper.find(".swiper-slide");
+    for slide_dom in swiper_slide {
+        let slide_node = Vis::dom(&slide_dom);
+        let figure = slide_node.find("figure");
+
+        let poster_div = figure.find(".ipc-poster__poster-image");
+        let poster_img = poster_div.find("img");
+
+        let poster_img_src_set = poster_img.attr("srcset").ok_or("no src")?.to_string();
+        let poster_img_src_set_split = poster_img_src_set.split(", ").collect::<Vec<&str>>();
+        let last_poster_img_src_set = poster_img_src_set_split.last().ok_or("no src")?.to_string();
+
+        let poster: String = last_poster_img_src_set.split(" ").collect::<Vec<&str>>().first().ok_or("no src")?.to_string();
+
+        
+        let slate_div = figure.find(".ipc-slate__slate-image");
+        let slate_img = slate_div.find("img");
+
+        let slate_img_src_set = slate_img.attr("srcset").ok_or("no src")?.to_string();
+        let slate_img_src_set_split = slate_img_src_set.split(", ").collect::<Vec<&str>>();
+        let last_slate_img_src_set = slate_img_src_set_split.last().ok_or("no src")?.to_string();
+
+        let banner: String = last_slate_img_src_set.split(" ").collect::<Vec<&str>>().first().ok_or("no src")?.to_string();
+
+        let title = decode_html_entities(&slate_img.attr("alt").ok_or("no title")?.to_string()).to_string();
+        println!("banner: {}", banner);
+        println!("poster: {}", poster);
+        println!("title: {}", title);
+
+
+    };
+    return Ok(());
+}
+
+
+pub async fn new() -> Result<HomeData, String> {
+    let client = Client::new();
+
+    let res = client.get("https://www.imdb.com/").send().await.map_err(|e| e.to_string())?;
+
+    if res.status().is_success() {
+        let html = res.text().await.map_err(|e| e.to_string())?;
+        let vis_root = Vis::load(html).map_err(|e| e.to_string())?;
+        get_relevant_content(vis_root).await?;
+    }else{
+        return Err("[home]: Request failed.".to_string());
+    }
+
+    return Err("Unkown Source".to_string());
+}
diff --git a/src-tauri/src/sources/movie/mod.rs b/src-tauri/src/sources/movie/mod.rs
@@ -0,0 +1 @@
+pub mod home;
diff --git a/src-tauri/src/test.rs b/src-tauri/src/test.rs
@@ -1,37 +1,40 @@
 #[cfg(test)]
 mod tests {
     use crate::commands;
+    use crate::sources;
     use crate::utils;
     use std::env;
     use tauri::async_runtime;
 
-    #[tokio::test]
-    async fn init() {
-        env::set_var(
-            "HYPERIONBOX_APPDATA",
-            env::var("APPDATA").unwrap() + "/io.github.goodday360.hyperionbox",
-        );
-        utils::configs::init().map_err(|e| e.to_string()).unwrap();
-    }
+    // #[tokio::test]
+    // async fn init() {
+    //     env::set_var(
+    //         "HYPERIONBOX_APPDATA",
+    //         env::var("APPDATA").unwrap() + "/io.github.goodday360.hyperionbox",
+    //     );
+    //     utils::configs::init().map_err(|e| e.to_string()).unwrap();
+    // }
 
     #[tokio::test]
     async fn test() {
         env::set_var(
             "HYPERIONBOX_APPDATA",
             env::var("APPDATA").unwrap() + "/io.github.goodday360.hyperionbox",
         );
-        match commands::download::get_download().await {
+        match sources::movie::home::new().await {
             Ok(d) => {
                 println!("Data: {:?}", d)
             }
-            Err(_) => assert!(false),
+            Err(e) => {
+                println!("Error: {}", e);
+            }
         }
     }
 
-    use crate::utils::configs;
+    // use crate::utils::configs;
 
-    #[tokio::test]
-    async fn test_setting() {
-        let _ = configs::get();
-    }
+    // #[tokio::test]
+    // async fn test_setting() {
+    //     let _ = configs::get();
+    // }
 }
diff --git a/src-tauri/src/worker/download.rs b/src-tauri/src/worker/download.rs
@@ -473,7 +473,7 @@ async fn download_episode(
 
                 let segment_path = segments_dir.join(format!("segment-{}", index));
 
-                download_file::new(&_url, &segment_path, headers.clone(), 60, |_, _| {})
+                download_file::new(&_url, &segment_path, headers.clone(), 300, |_, _| {})
                     .await
                     .map_err(|e| e.to_string())?;
 
@@ -576,7 +576,7 @@ async fn download_episode(
                     }
                 );
                 let caption_path = captions_dir.join(&caption_file_name);
-                match download_file::new(&caption.file, &caption_path, headers.clone(), 60, |_, _| {}).await {
+                match download_file::new(&caption.file, &caption_path, headers.clone(), 300, |_, _| {}).await {
                     Ok(_) => {
                         let new_caption_file = PathBuf::from(source)
                             .join(id)

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,2 @@`
	`1`	`+pub mod anime;`
	`2`	`+// pub mod movie;`