main.rs

mod error;
mod server;
mod storage;

use crate::error::MocksError;
use crate::server::Server;
use crate::storage::Storage;
use clap::Parser;
use std::net::{IpAddr, SocketAddr};

#[derive(Parser, Debug)]
#[command(version, about, long_about = None)]
struct Args {
    /// Path of json file for data storage
    file: String,

    /// Host
    #[arg(short = 'H', long, default_value = "localhost")]
    host: String,

    /// Port
    #[arg(short, long, default_value_t = 3000)]
    port: u16,

    /// No overwrite save to json file
    #[arg(long, default_value_t = false)]
    no_overwrite: bool,
}

#[tokio::main]
async fn main() -> Result<(), MocksError> {
    let args = Args::parse();
    let socket_addr = init(&args.host, args.port)?;

    println!("`mocks` started");
    println!("Press CTRL-C to stop");

    let url = format!("http://{}:{}", &args.host, args.port);
    let overwrite = !args.no_overwrite;

    print_startup_info(&url, &args.file, overwrite);

    let storage = Storage::new(&args.file, overwrite)?;
    Server::startup(socket_addr, &url, storage).await?;

    Ok(())
}

fn init(host: &str, port: u16) -> Result<SocketAddr, MocksError> {
    let ip_addr = if host == "localhost" {
        "127.0.0.1"
    } else {
        host
    };

    ip_addr
        .parse::<IpAddr>()
        .map(|ip| SocketAddr::from((ip, port)))
        .map_err(|e| MocksError::InvalidArgs(e.to_string()))
}

fn print_startup_info(url: &str, file: &str, overwrite: bool) {
    println!("\nIndex:\n{}", url);
    println!("\nStorage files:\n{}", file);
    println!("\nOverwrite:\n{}", if overwrite { "YES" } else { "NO" });
    println!();
}