1 files changed, 73 insertions, 0 deletions
diff --git a/crates/atuin-server/src/bin/main.rs b/crates/atuin-server/src/bin/main.rs
new file mode 100644
index 00000000..960bdf6e
--- /dev/null
+++ b/crates/atuin-server/src/bin/main.rs
@@ -0,0 +1,73 @@
+#![forbid(unsafe_code)]
+
+use std::net::SocketAddr;
+
+use atuin_server::{Settings, example_config, launch, launch_metrics_server};
+use atuin_server_database::DbType;
+use atuin_server_postgres::Postgres;
+use atuin_server_sqlite::Sqlite;
+
+use clap::Parser;
+use eyre::{Context, Result, eyre};
+use tracing_subscriber::{EnvFilter, fmt, prelude::*};
+
+#[derive(Parser, Debug)]
+#[clap(
+    name = "atuin-server",
+    about = "Atuin sync server",
+    version,
+    infer_subcommands = true
+)]
+enum Cmd {
+    /// Start the server
+    Start {
+        /// The host address to bind
+        #[clap(long)]
+        host: Option<String>,
+
+        /// The port to bind
+        #[clap(long, short)]
+        port: Option<u16>,
+    },
+
+    /// Print server example configuration
+    DefaultConfig,
+}
+
+#[tokio::main]
+async fn main() -> Result<()> {
+    let cmd = Cmd::parse();
+
+    tracing_subscriber::registry()
+        .with(fmt::layer())
+        .with(EnvFilter::from_default_env())
+        .init();
+
+    tracing::trace!(command = ?cmd, "server command");
+
+    match cmd {
+        Cmd::Start { host, port } => {
+            let settings = Settings::new().wrap_err("could not load server settings")?;
+            let host = host.as_ref().unwrap_or(&settings.host).clone();
+            let port = port.unwrap_or(settings.port);
+            let addr = SocketAddr::new(host.parse()?, port);
+
+            if settings.metrics.enable {
+                tokio::spawn(launch_metrics_server(
+                    settings.metrics.host.clone(),
+                    settings.metrics.port,
+                ));
+            }
+
+            match settings.db_settings.db_type() {
+                DbType::Postgres => launch::<Postgres>(settings, addr).await,
+                DbType::Sqlite => launch::<Sqlite>(settings, addr).await,
+                DbType::Unknown => Err(eyre!("db_uri must start with postgres:// or sqlite://")),
+            }
+        }
+        Cmd::DefaultConfig => {
+            println!("{}", example_config());
+            Ok(())
+        }
+    }
+}