Merge pull request #2 from cclvi256/refactor/database-any

Refactor database connection from generic based to any

Author: cclvi256

Cargo.lock

@@ -3,7 +3,7 @@ name = "botcat-capoo"
 version = "0.2.2"
 edition = "2024"
 authors = ["cclvi256 <i@cclvi.cc>"]
-description = "A QQ bot implemented in Rust"
+description = "A NapCat based QQ bot implemented in Rust"
 keywords = ["qq", "bot", "napcat", "chatbot"]
 homepage = "https://github.com/cclvi256/botcat-capoo"
 repository = "https://github.com/cclvi256/botcat-capoo"
@@ -19,11 +19,14 @@ serde = { version = "1.0.219", features = ["derive"] }
 serde_json = "1.0.142"
 sqlx = { version = "0.8.6", features = [
 	"any",
+	"chrono",
 	"macros",
+	"migrate",
 	"mysql",
 	"postgres",
 	"runtime-tokio-rustls",
 	"sqlite",
+	"uuid",
 ] }
 thiserror = "2.0.14"
 tokio = { version = "1.47.1", features = ["macros", "rt-multi-thread"] }

Cargo.toml

@@ -7,9 +7,6 @@ host = "::"
 port = 8080
 owner = "Your QQ ID here"
 id = "Your Bot's QQ ID here"
-# Deprecated, use [botcat_capoo.database] section instead
-# If [botcat_capoo.database] is not specified, this will be used as a fallback
-database_url = "Your database URL here, e.g., sqlite://data/botcat_capoo.sqlite"
 
 [botcat_capoo.database]
 # Options: Postgres, Sqlite, MySql, Common abbreviations are supported, case insensitive

config/example.toml

@@ -64,12 +64,11 @@ pub async fn get_private_history(
 #[cfg(test)]
 mod tests {
     use super::*;
-    use crate::config::load_config;
 
     #[tokio::test]
     #[ignore = "Requires an API server in internal network"]
     async fn test_get_group_history() {
-        let config = load_config().expect("Failed to load config");
+        let config = AppConfig::init().expect("Failed to load config");
         println!("Loaded config");
 
         match get_group_history(&config, 738943282, 20, true).await {
@@ -85,7 +84,7 @@ mod tests {
     #[tokio::test]
     #[ignore = "Requires an API server in internal network"]
     async fn test_get_private_history() {
-        let config = load_config().expect("Failed to load config");
+        let config = AppConfig::init().expect("Failed to load config");
         println!("Loaded config");
 
         match get_private_history(&config, 46595749, 20, true).await {

src/api/history.rs

@@ -42,36 +42,35 @@ pub async fn group_poke(config: &AppConfig, group_id: u64, to: u64) {
 #[cfg(test)]
 mod tests {
     use super::*;
-    use crate::config::load_config;
 
     #[tokio::test]
     #[ignore = "Requires an API server in internal network"]
     async fn test_private_poke() {
-        let config = load_config().expect("Failed to load config");
+        let config = AppConfig::init().expect("Failed to load config");
         println!("Loaded config");
         private_poke(&config, 46595749, false).await;
     }
 
     #[tokio::test]
     #[ignore = "Requires an API server in internal network"]
     async fn test_private_self_poke() {
-        let config = load_config().expect("Failed to load config");
+        let config = AppConfig::init().expect("Failed to load config");
         println!("Loaded config");
         private_poke(&config, 46595749, true).await;
     }
 
     #[tokio::test]
     #[ignore = "Requires an API server in internal network"]
     async fn test_group_poke() {
-        let config = load_config().expect("Failed to load config");
+        let config = AppConfig::init().expect("Failed to load config");
         println!("Loaded config");
         group_poke(&config, 738943282, 46595749).await;
     }
 
     #[tokio::test]
     #[ignore = "Requires an API server in internal network"]
     async fn test_group_self_poke() {
-        let config = load_config().expect("Failed to load config");
+        let config = AppConfig::init().expect("Failed to load config");
         println!("Loaded config");
         group_poke(&config, 738943282, config.botcat_capoo.id).await;
     }

src/api/poke.rs

@@ -39,13 +39,13 @@ pub async fn send_private_message(config: &AppConfig, to: u64, message: Message)
 #[cfg(test)]
 mod tests {
     use super::*;
-    use crate::config::load_config;
+
     use crate::model::message::segment::MessageSegment;
 
     #[tokio::test]
     #[ignore = "Requires an API server in internal network"]
     async fn test_send_group_message() {
-        let config = load_config().expect("Failed to load config");
+        let config = AppConfig::init().expect("Failed to load config");
         println!("Loaded config");
         let message = vec![
             MessageSegment::At {
@@ -61,7 +61,7 @@ mod tests {
     #[tokio::test]
     #[ignore = "Requires an API server in internal network"]
     async fn test_send_private_message() {
-        let config = load_config().expect("Failed to load config");
+        let config = AppConfig::init().expect("Failed to load config");
         println!("Loaded config");
         let message = vec![MessageSegment::Text {
             text: "Hello, user!".to_string(),