feat: Add SurrealDB schema files and dependencies

.surrealdb

@@ -0,0 +1,3 @@
+[core]
+schema = "full"
+path = "./surrealdb"

Cargo.toml

@@ -14,10 +14,13 @@ panic = 'abort'
 codegen-units = 1
 
 [dependencies]
-axum = { version = "0.8.4", features = ["tracing"] }
+axum = { version = "0.8.4", features = ["tracing", "macros"] }
 config = "0.15.15"
 dotenvy = "0.15.7"
-reqwest = { version = "0.12.23", default-features = false, features = ["json", "rustls-tls"] }
+reqwest = { version = "0.12.23", default-features = false, features = [
+    "json",
+    "rustls-tls",
+] }
 secrecy = { version = "0.10.3", features = ["serde"] }
 serde = { version = "1.0.219", features = ["derive"] }
 serde-humantime = "0.1.1"
@@ -34,6 +37,7 @@ url = { version = "2.5.7", features = ["serde"] }
 validator = "0.20.0"
 serde_json = "1.0.143"
 rand = { version = "0.9.2", features = ["std_rng"] }
+include_dir = "0.7.4"
 
 [dev-dependencies]
 mime = "0.3.17"

src/domain/subscriber/mod.rs

@@ -1,9 +1,8 @@
 mod email;
 mod name;
 
-pub use email::SubscriberEmail;
-
 use crate::handlers::FormData;
+pub use email::SubscriberEmail;
 use name::SubscriberName;
 use validator::ValidationErrors;
 

src/errors.rs

@@ -5,7 +5,7 @@ pub type Result<T = ()> = std::result::Result<T, Error>;
 #[derive(thiserror::Error, Debug)]
 pub enum Error {
     #[error(transparent)]
-    SurrealDb(#[from] surrealdb::Error),
+    SurrealDb(#[from] Box<surrealdb::Error>),
     #[error("{0:?}")]
     Migrations(String),
     #[error(transparent)]

src/handlers/health_check.rs

@@ -1,12 +1,13 @@
-use crate::Result;
 use crate::model::ModelManager;
+use crate::{AppState, Result};
 use axum::extract::State;
 use reqwest::StatusCode;
 use std::sync::Arc;
 
+#[axum::debug_handler(state = AppState)]
 #[tracing::instrument(skip(mm))]
 pub async fn health(State(mm): State<Arc<ModelManager>>) -> Result<StatusCode> {
-    mm.db().await?.health().await?;
+    mm.db().await?.health().await.map_err(Box::new)?;
 
     Ok(StatusCode::OK)
 }

src/handlers/subscription.rs

@@ -1,21 +1,22 @@
-use crate::Config;
-use crate::domain::Subscriber;
-use crate::model::ModelManager;
-use crate::{Result, email_client::EmailClient};
+use crate::{
+    AppState, Config, Result, domain::Subscriber, email_client::EmailClient, model::ModelManager,
+};
 use axum::extract::Query;
 use axum::{Form, extract::State};
 use rand::Rng;
 use rand::distr::Alphanumeric;
 use reqwest::StatusCode;
 use serde::Deserialize;
 use std::sync::Arc;
+use url::Url;
 
 #[derive(Debug, Deserialize)]
 pub struct FormData {
     pub email: String,
     pub name: String,
 }
 
+#[axum::debug_handler(state = AppState)]
 #[tracing::instrument(skip(mm, config, email_client))]
 pub async fn subscribe(
     State(mm): State<Arc<ModelManager>>,
@@ -39,36 +40,20 @@ pub struct Params {
     token: String,
 }
 
+#[axum::debug_handler(state = AppState)]
 #[tracing::instrument(skip(mm))]
 pub async fn confirm(
     State(mm): State<Arc<ModelManager>>,
     Query(params): Query<Params>,
 ) -> Result<StatusCode> {
-    _ = mm
-        .db()
-        .await?
-        .query(
-            r#"
-        UPDATE subscriptions
-        SET status = 'CONFIRMED'
-        WHERE token = (
-            SELECT id
-            FROM ONLY subscription_tokens
-            WHERE token = $token_val
-        ).id
-        AND status = 'PENDING';
-    "#,
-        )
-        .bind(("token_val", params.token))
-        .await?
-        .take::<surrealdb::Value>(0)?;
+    mm.confirm_subscriber(params.token.clone()).await?;
 
     Ok(StatusCode::OK)
 }
 
 fn get_confirmation_token() -> String {
-    let mut rng = rand::rng();
-    std::iter::repeat_with(|| rng.sample(Alphanumeric))
+    rand::rng()
+        .sample_iter(&Alphanumeric)
         .map(char::from)
         .take(25)
         .collect()
@@ -80,16 +65,14 @@ async fn send_confirmation_email(
     subscriber: &Subscriber,
     token: &str,
 ) -> Result<()> {
-    let mut confirmation_link = config.base_url.join("subscriptions/confirm")?;
-    confirmation_link.set_query(Some(&format!("token={token}")));
+    let confirmation_link = get_confirmation_link(config, token)?;
 
     email_client
         .send_email(
             &subscriber.email,
             "Welcome!",
             &format!(
-                "Welcome to our newsletter!<br />\
-                            Click <a href=\"{}\">here</a> to confirm your subscription.",
+                "Welcome to our newsletter!<br />Click <a href=\"{}\">here</a> to confirm your subscription.",
                 confirmation_link
             ),
             &format!(
@@ -100,3 +83,10 @@ async fn send_confirmation_email(
         .await?;
     Ok(())
 }
+
+fn get_confirmation_link(config: &Config, token: &str) -> Result<Url> {
+    let mut confirmation_link = config.base_url.join("subscriptions/confirm")?;
+    confirmation_link.set_query(Some(&format!("token={token}")));
+
+    Ok(confirmation_link)
+}

src/main.rs

@@ -1,10 +1,9 @@
-use std::process;
-use subscriptions::Config;
+use subscriptions::{Config, Error};
 use tokio::net::TcpListener;
 use tracing_subscriber::prelude::*;
 
 #[tokio::main]
-async fn main() {
+async fn main() -> Result<(), Error> {
     // Load environment variables from .env file if exists
     dotenvy::dotenv_override().ok();
 
@@ -15,33 +14,19 @@ async fn main() {
         .init();
 
     // Initialize configuration
-    let config = Config::load().unwrap_or_else(|error| {
-        tracing::error!(%error, "Failed to load configuration");
-        process::exit(1);
-    });
+    let config = Config::load()?;
 
     // Initialize application
-    let (router, _) = subscriptions::init(config.clone())
-        .await
-        .unwrap_or_else(|error| {
-            tracing::error!(%error, "Failed to initialize application");
-            process::exit(1);
-        });
+    let (router, _) = subscriptions::init(config.clone()).await?;
 
     // Bind address
-    let listener = TcpListener::bind((config.host.clone(), config.port))
-        .await
-        .unwrap_or_else(|error| {
-            tracing::error!(%error, "Failed to bind address `{}:{}`", config.host, config.port);
-            process::exit(1);
-        });
+    let listener = TcpListener::bind((config.host.clone(), config.port)).await?;
 
     // Start server
     tracing::info!("Start listening on: http://{}:{}", config.host, config.port);
-    if let Err(error) = axum::serve(listener, router).await {
-        tracing::error!(%error, "Failed to start server");
-        process::exit(1);
-    }
+    axum::serve(listener, router).await?;
 
     tracing::info!("Server gracefully shutdown");
+
+    Ok(())
 }

src/model.rs

@@ -1,4 +1,5 @@
 use crate::{Error, Result, config::DatabaseConfig, domain};
+use include_dir::include_dir;
 use secrecy::ExposeSecret;
 use surrealdb::{Surreal, engine::any::Any, opt::auth::Database};
 use surrealdb_migrations::MigrationRunner;
@@ -44,8 +45,30 @@ impl ModelManager {
             .bind(("token_val", token.to_string()))
             .bind(("email", subscriber.email.as_ref().to_string()))
             .bind(("name", subscriber.name.as_ref().to_string()))
+            .await.map_err(Box::new)?
+            .check().map_err(Box::new)?;
+
+        Ok(())
+    }
+
+    pub async fn confirm_subscriber(&self, token: String) -> Result<()> {
+        self.db()
             .await?
-            .check()?;
+            .query(
+                r#"
+                UPDATE subscriptions
+                SET status = 'CONFIRMED'
+                WHERE token = (
+                    SELECT id
+                    FROM ONLY subscription_tokens
+                    WHERE token = $token_val
+                ).id
+                AND status = 'PENDING';
+            "#,
+            )
+            .bind(("token_val", token))
+            .await
+            .map_err(Box::new)?;
 
         Ok(())
     }
@@ -55,7 +78,9 @@ impl ModelManager {
         let db = Surreal::<Any>::init();
 
         tracing::info!("Connecting to database: {}", config.base_url.as_str());
-        db.connect(config.base_url.as_str()).await?;
+        db.connect(config.base_url.as_str())
+            .await
+            .map_err(Box::new)?;
 
         if config.base_url.scheme() == "mem" {
             db.query(format!("DEFINE NAMESPACE {}", config.namespace))
@@ -66,7 +91,8 @@ impl ModelManager {
                     config.username,
                     config.password.expose_secret()
                 ))
-                .await?;
+                .await
+                .map_err(Box::new)?;
         }
 
         db.signin(Database {
@@ -75,10 +101,12 @@ impl ModelManager {
             namespace: &config.namespace,
             database: &config.name,
         })
-        .await?;
+        .await
+        .map_err(Box::new)?;
 
         // Apply Migrations
         MigrationRunner::new(&db)
+            .load_files(&include_dir!("$CARGO_MANIFEST_DIR/surrealdb"))
             .up()
             .await
             .map_err(|e| Error::Migrations(e.to_string()))?;