Query history from history server

sable_history/Cargo.toml

@@ -11,6 +11,7 @@ built = { version = "0.5", features = [ "git2" ] }
 sable_network = { path = "../sable_network" }
 sable_server = { path = "../sable_server" }
 
+futures = "0.3"
 tokio = { version = "1.14", features = [ "full" ] }
 serde = { version = "1", features = [ "derive" ] }
 serde_with = "1.11"
@@ -22,7 +23,9 @@ tracing = "0.1"
 anyhow = "1.0"
 clap = { version = "4.5", features = [ "derive" ] }
 chrono = "0.4"
+itertools = "0.10"
 uuid = { version = "1.9.1", features = ["v7", "fast-rng", "serde"] }
 
 diesel = { version = "2.2", features = [ "postgres", "chrono", "uuid" ] }
-diesel-async = { version = "0.5", features = [ "postgres" ] }
+diesel-async = { version = "0.5", features = [ "postgres", "tokio", "async-connection-wrapper" ] }
+diesel_migrations = "2.2.0"

sable_history/build.rs

@@ -1,3 +1,6 @@
 fn main() {
+    // https://docs.rs/diesel_migrations/2.2.0/diesel_migrations/macro.embed_migrations.html#automatic-rebuilds
+    println!("cargo:rerun-if-changed=migrations/");
+
     built::write_built_file().expect("Failed to acquire build-time information");
 }

sable_history/diesel.toml

@@ -4,6 +4,7 @@
 [print_schema]
 file = "src/schema.rs"
 custom_type_derives = ["diesel::query_builder::QueryId", "Clone"]
+import_types = ["crate::type::*"]
 
 [migrations_directory]
 dir = "migrations"

sable_history/migrations/2024-10-27-125826_reproducible_messages/down.sql

@@ -0,0 +1,4 @@
+ALTER TABLE
+    DROP COLUMN message_type MessageType NOT NULL;
+
+DROP TYPE "MessageType";

sable_history/migrations/2024-10-27-125826_reproducible_messages/up.sql

@@ -0,0 +1,4 @@
+CREATE TYPE "Message_Type" AS ENUM ('privmsg', 'notice');
+
+ALTER TABLE messages
+    ADD COLUMN message_type "Message_Type" NOT NULL;

sable_history/src/lib.rs

@@ -1,5 +1,8 @@
+mod pg_history_service;
+pub use pg_history_service::PgHistoryService;
 mod server;
 pub use server::*;
 
 mod models;
 mod schema;
+mod types;

sable_history/src/models/message.rs

@@ -10,4 +10,5 @@ pub struct Message {
     pub source_user: i32,
     pub target_channel: i64,
     pub text: String,
+    pub message_type: crate::types::MessageType,
 }