chore: update file name (#54)

* chore: update file name

Signed-off-by: kerthcet <kerthcet@gmail.com>

* fix

Signed-off-by: kerthcet <kerthcet@gmail.com>

* update

Signed-off-by: kerthcet <kerthcet@gmail.com>

* update

Signed-off-by: kerthcet <kerthcet@gmail.com>

---------

Signed-off-by: kerthcet <kerthcet@gmail.com>

Author: kerthcet

src/cli/commands.rs

@@ -2,8 +2,8 @@ use clap::{Parser, Subcommand};
 use prettytable::{format, row, Table};
 
 use crate::cli::{inspect, ls, rm};
-use crate::downloader::downloader::Downloader;
 use crate::downloader::huggingface::HuggingFaceDownloader;
+use crate::downloader::Downloader;
 use crate::registry::model_registry::ModelRegistry;
 use crate::system::system_info::SystemInfo;
 use crate::utils::format::{format_size_decimal, format_time_ago};

src/downloader/downloader.rs

@@ -4,8 +4,8 @@ use tracing::debug;
 use hf_hub::api::tokio::{ApiBuilder, Progress};
 use indicatif::{ProgressBar, ProgressStyle};
 
-use crate::downloader::downloader::{DownloadError, Downloader};
 use crate::downloader::progress::{DownloadProgressManager, FileProgress};
+use crate::downloader::{DownloadError, Downloader};
 use crate::registry::model_registry::{CacheInfo, ModelInfo, ModelMetadata, ModelRegistry};
 use crate::utils::file::{self, format_model_name};
 

src/downloader/huggingface.rs

@@ -1,4 +1,31 @@
-#[allow(clippy::module_inception)]
-pub mod downloader;
+use core::fmt;
+
 pub mod huggingface;
 pub mod progress;
+
+#[derive(Debug)]
+pub enum DownloadError {
+    NetworkError(String),
+    AuthError(String),
+    ModelNotFound(String),
+    IoError(String),
+    ApiError(String),
+}
+
+impl std::error::Error for DownloadError {}
+
+impl fmt::Display for DownloadError {
+    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
+        match self {
+            DownloadError::NetworkError(e) => write!(f, "Network error: {}", e),
+            DownloadError::AuthError(e) => write!(f, "Authentication error: {}", e),
+            DownloadError::ModelNotFound(e) => write!(f, "Model not found: {}", e),
+            DownloadError::IoError(e) => write!(f, "IO error: {}", e),
+            DownloadError::ApiError(e) => write!(f, "API error: {}", e),
+        }
+    }
+}
+
+pub trait Downloader {
+    async fn download_model(&self, name: &str) -> Result<(), DownloadError>;
+}

src/downloader/mod.rs

@@ -1,5 +1,5 @@
+pub mod model_storage;
 pub mod sqlite;
-pub mod storage_trait;
 
+pub use model_storage::ModelStorage;
 pub use sqlite::SqliteStorage;
-pub use storage_trait::ModelStorage;