feat: update main.rs

Signed-off-by: kjuulh <contact@kjuulh.io>

cuddle-rust-service/crates/%%name%%/src/main.rs

@@ -1,26 +1,93 @@
-use std::{net::SocketAddr, ops::Deref, sync::Arc};
+mod state {
+    #[derive(Clone)]
+    pub struct State {}
 
-use anyhow::Context;
-use axum::extract::MatchedPath;
-use axum::http::Request;
-use axum::Router;
-use axum::routing::get;
-use clap::{Parser, Subcommand};
-use tower_http::trace::TraceLayer;
-
-#[derive(Parser)]
-#[command(author, version, about, long_about = None, subcommand_required = true)]
-struct Command {
-    #[command(subcommand)]
-    command: Option<Commands>,
+    impl State {
+        pub async fn new() -> anyhow::Result<Self> {
+            Ok(Self {})
+        }
+    }
 }
+pub use state::*;
 
-#[derive(Subcommand)]
-enum Commands {
-    Serve {
-        #[arg(env = "SERVICE_HOST", long, default_value = "127.0.0.1:3000")]
-        host: SocketAddr,
-    },
+pub mod cli {
+    use clap::{Parser, Subcommand};
+
+    use crate::{State, cli::serve::ServeCommand};
+
+    mod serve {
+        use std::net::SocketAddr;
+
+        use axum::{Router, extract::MatchedPath, http::Request, routing::get};
+        use tower_http::trace::TraceLayer;
+
+        use crate::State;
+
+        #[derive(clap::Parser)]
+        pub struct ServeCommand {
+            #[arg(env = "SERVICE_HOST", long, default_value = "127.0.0.1:3000")]
+            host: SocketAddr,
+        }
+
+        impl ServeCommand {
+            pub async fn execute(&self, state: &State) -> anyhow::Result<()> {
+                let app = Router::new()
+                    .route("/", get(root))
+                    .with_state(state.clone())
+                    .layer(
+                        TraceLayer::new_for_http().make_span_with(|request: &Request<_>| {
+                            // Log the matched route's path (with placeholders not filled in).
+                            // Use request.uri() or OriginalUri if you want the real path.
+                            let matched_path = request
+                                .extensions()
+                                .get::<MatchedPath>()
+                                .map(MatchedPath::as_str);
+
+                            tracing::info_span!(
+                                "http_request",
+                                method = ?request.method(),
+                                matched_path,
+                                some_other_field = tracing::field::Empty,
+                            )
+                        }), // ...
+                    );
+
+                tracing::info!("listening on {}", self.host);
+                let listener = tokio::net::TcpListener::bind(self.host).await.unwrap();
+                axum::serve(listener, app.into_make_service())
+                    .await
+                    .unwrap();
+
+                Ok(())
+            }
+        }
+
+        async fn root() -> &'static str {
+            "Hello, nostore!"
+        }
+    }
+
+    #[derive(Parser)]
+    #[command(author, version, about, long_about = None, subcommand_required = true)]
+    struct Command {
+        #[command(subcommand)]
+        command: Option<Commands>,
+    }
+
+    #[derive(Subcommand)]
+    enum Commands {
+        Serve(ServeCommand),
+    }
+
+    pub async fn execute() -> anyhow::Result<()> {
+        let cli = Command::parse();
+
+        let state = State::new().await?;
+
+        match cli.command.expect("a subcommand") {
+            Commands::Serve(cmd) => cmd.execute(&state).await,
+        }
+    }
 }
 
 #[tokio::main]
@@ -28,64 +95,7 @@ async fn main() -> anyhow::Result<()> {
     dotenv::dotenv().ok();
     tracing_subscriber::fmt::init();
 
-    let cli = Command::parse();
-
-    if let Some(Commands::Serve { host }) = cli.command {
-        tracing::info!("Starting service");
-
-        let state = SharedState(Arc::new(State::new().await?));
-
-        let app = Router::new()
-            .route("/", get(root))
-            .with_state(state.clone())
-            .layer(
-                TraceLayer::new_for_http().make_span_with(|request: &Request<_>| {
-                    // Log the matched route's path (with placeholders not filled in).
-                    // Use request.uri() or OriginalUri if you want the real path.
-                    let matched_path = request
-                        .extensions()
-                        .get::<MatchedPath>()
-                        .map(MatchedPath::as_str);
-
-                    tracing::info_span!(
-                        "http_request",
-                        method = ?request.method(),
-                        matched_path,
-                        some_other_field = tracing::field::Empty,
-                    )
-                }), // ...
-            );
-
-        tracing::info!("listening on {}", host);
-        let listener = tokio::net::TcpListener::bind(host).await.unwrap();
-        axum::serve(listener, app.into_make_service())
-            .await
-            .unwrap();
-    }
+    cli::execute().await?;
 
     Ok(())
 }
-
-async fn root() -> &'static str {
-    "Hello, %%name%%!"
-}
-
-#[derive(Clone)]
-pub struct SharedState(Arc<State>);
-
-impl Deref for SharedState {
-    type Target = Arc<State>;
-
-    fn deref(&self) -> &Self::Target {
-        &self.0
-    }
-}
-
-pub struct State {}
-
-impl State {
-    pub async fn new() -> anyhow::Result<Self> {
-        Ok(Self {})
-    }
-}
-

cuddle-rust-service/cuddle.yaml

@@ -1,6 +1,6 @@
-# yaml-language-server: $schema=https://git.front.kjuulh.io/kjuulh/cuddle/raw/branch/main/schemas/base.json
+# yaml-language-server: $schema=https://git.kjuulh.io/kjuulh/cuddle/raw/branch/main/schemas/base.json
 
-base: "git@git.front.kjuulh.io:kjuulh/cuddle-rust-service-plan.git"
+base: "git@git.kjuulh.io:kjuulh/cuddle-rust-service-plan.git"
 
 vars:
   service: "%%name%%"