Improve error handling for database path retrieval across Tauri compo…

…nents

- Refactored `get_database_path` function to return a `Result<String, io::Error>` for better error management.
- Updated calls to `get_database_path` in `get_chat_history`, `get_chat_history_by_session`, `get_chatgpt_response`, `get_session_id_list`, and `lib.rs` to handle potential errors gracefully.
- Removed direct panics from database path retrieval and directory creation by incorporating proper error handling.
- Ensured that the application can gracefully handle scenarios when the database path cannot be determined, improving overall robustness.

This update enhances the reliability of database-related operations within the application.

src-tauri/src/get_chat_history.rs

@@ -10,7 +10,7 @@ use tauri::async_runtime::block_on;
 pub fn get_chat_history() -> Result<Vec<RawDatabaseChatEntry>, String> {
     // Use block_on to run the async code synchronously
     block_on(async {
-        let database_path = get_database_path();
+        let database_path = get_database_path().map_err(|e| e.to_string())?;
         let mut conn = establish_connection(&database_path);
 
         // Pass the mutable reference to Diesel operations

src-tauri/src/get_chat_history_by_session.rs

@@ -13,7 +13,7 @@ pub fn get_chat_history_by_session(
     // Use block_on to run the async code synchronously
     block_on(async {
         // Use the helper function to get a mutable connection
-        let database_path = get_database_path();
+        let database_path = get_database_path().map_err(|e| e.to_string())?;
         let mut conn = establish_connection(&database_path);
 
         // Filter the chat history by the specific session_id

src-tauri/src/get_chatgpt_response.rs

@@ -15,7 +15,7 @@ pub async fn get_chatgpt_response(
     model: String,
     api_key: String,
 ) -> Result<ChatResponse, String> {
-    let database_path = get_database_path();
+    let database_path = get_database_path().map_err(|e| e.to_string())?;
     let mut conn = establish_connection(&database_path);
 
     let session_history = chat_histories

src-tauri/src/get_database_path.rs

@@ -1,14 +1,20 @@
+use dirs;
 use std::fs;
+use std::io;
 
-pub fn get_database_path() -> String {
-    let mut db_path = dirs::home_dir().expect("Failed to get home directory");
+pub fn get_database_path() -> Result<String, io::Error> {
+    // ホームディレクトリの取得
+    let mut db_path = dirs::home_dir()
+        .ok_or_else(|| io::Error::new(io::ErrorKind::NotFound, "Failed to get home directory"))?;
     db_path.push(".cuuri/chat.db");
 
+    // 親ディレクトリの作成
     if let Some(parent) = db_path.parent() {
-        fs::create_dir_all(parent).expect("Failed to create directories for database file");
+        fs::create_dir_all(parent)?;
     }
 
+    // データベースURLの生成
     let database_url = format!("sqlite://{}", db_path.to_string_lossy());
 
-    return database_url;
+    Ok(database_url) // 成功時はOkを返す
 }

src-tauri/src/get_session_id_list.rs

@@ -8,7 +8,7 @@ use tauri::async_runtime::block_on;
 #[tauri::command]
 pub fn get_session_id_list() -> Result<Vec<SessionId>, String> {
     block_on(async {
-        let database_path = get_database_path();
+        let database_path = get_database_path().map_err(|e| e.to_string())?;
         let mut conn = establish_connection(&database_path);
 
         let results = chat_histories

src-tauri/src/lib.rs

@@ -35,8 +35,8 @@ use std::env;
 pub fn run() {
     init_config_file();
 
-    let database_url = get_database_path();
-    let mut connection: diesel::SqliteConnection = establish_connection(&database_url);
+    let database_path: String = get_database_path().unwrap();
+    let mut connection: diesel::SqliteConnection = establish_connection(&database_path);
     run_migrations(&mut connection);
 
     tauri::Builder::default()