staxman-old/src/main.rs

use crate::http::response::response_interceptor;
use anyhow::{anyhow, Result};
use axum::middleware::from_fn;
use bollard::Docker;
use clap::Parser;
use node::git::{GitConfig, ThreadSafeRepository};
use std::{net::SocketAddr, path::PathBuf};
use tokio::fs;

mod http;
mod nix;
mod node;
mod route;

/// GitOps+WebUI for arion-based stacks
#[derive(Parser, Debug)]
#[command(author, version, about, long_about = None)]
struct Args {
    /// Path to root of stacks
    #[arg(short = 'd', long = "stack-dir", env = "STAX_DIR")]
    stack_dir: PathBuf,

    /// Address:port to bind
    #[arg(short, long, default_value = "0.0.0.0:3000", env = "STAX_BIND")]
    bind: SocketAddr,

    /// Path to arion
    #[arg(
        long = "arion-bin",
        env = "STAX_ARION_BIN",
        default_value = "/run/current-system/sw/bin/arion"
    )]
    arion_binary: PathBuf,

    #[arg(
        long = "git-author",
        default_value = "staxman <staxman@example.tld>",
        env = "STAX_GIT_AUTHOR"
    )]
    git_author: String,
}

#[derive(Clone)]
pub struct AppState {
    pub stack_dir: PathBuf,
    pub arion_bin: PathBuf,
    pub docker: Docker,
    pub gitconfig: GitConfig,
    pub repository: ThreadSafeRepository,
}

#[tokio::main]
async fn main() -> Result<()> {
    // Initialize logging and env
    _ = dotenvy::dotenv();
    tracing_subscriber::fmt::init();

    // Try to connect to docker server
    let docker = Docker::connect_with_local_defaults()?;
    // Ping to make sure it works
    let version = docker.version().await?;
    tracing::info!("docker version: {}", version.version.unwrap());

    // Parse args
    let args = Args::parse();
    tracing::info!("listening on {}", &args.bind);

    let (author_name, author_email) = parse_author(&args.git_author)?;
    let gitconfig = GitConfig {
        author_name,
        author_email,
    };

    // Make sure stack arg exists and is properly initialized
    fs::create_dir_all(&args.stack_dir).await?;
    let repository = ThreadSafeRepository::ensure_repository(&gitconfig, &args.stack_dir)?;

    let state = AppState {
        stack_dir: args.stack_dir,
        arion_bin: args.arion_binary,
        docker,
        gitconfig,
        repository,
    };

    let app = route::router()
        .with_state(state)
        .layer(from_fn(response_interceptor));

    axum::Server::bind(&args.bind)
        .serve(app.into_make_service())
        .await?;

    Ok(())
}

fn parse_author(git_author: &str) -> Result<(String, String)> {
    match git_author.split_once('<') {
        Some((name, email)) => Ok((
            name.trim().to_string(),
            email.trim_end_matches(">").trim().to_string(),
        )),
        None => Err(anyhow!(
            "invalid git author format (email must be specified)"
        )),
    }
}