replay 0.1.2

A simple HTTP request replay tool
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74


use std::sync::{Arc};

use clap::{Arg, Command};
use owo_colors::OwoColorize;
use proxy::start_server;
use tokio::sync::Mutex;

pub mod proxy;
pub mod replay;
pub mod store;

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
    const BANNER: &str = r#"

        ____             __
       / __ \___  ____  / /___ ___  __
      / /_/ / _ \/ __ \/ / __ `/ / / /
     / _, _/  __/ /_/ / / /_/ / /_/ /
    /_/ |_|\___/ .___/_/\__,_/\__, /
              /_/            /____/

    Sniff and replay HTTP requests and responses — perfect for mocking APIs during testing.
    "#;
    let matches = Command::new("replay")
        .version(env!("CARGO_PKG_VERSION"))
        .author("Tsiry Sandratraina <tsiry.sndr@rocksky.app>")
        .about(&format!("{}", BANNER.magenta()))
        .arg(
            Arg::new("target")
                .short('t')
                .long("target")
                .help("The target URL to replay the requests to")
        )
        .arg(
            Arg::new("listen")
                .short('l')
                .long("listen")
                .help("The address to listen on for incoming requests")
                .default_value("127.0.0.1:6677")
        )
        .subcommand(
            Command::new("mock")
                .about("Read mocks from replay_mock.json and start a replay server")
        )
        .get_matches();


    if let Some(_) = matches.subcommand_matches("mock") {
        let logs = store::load_logs_from_file(proxy::PROXY_LOG_FILE)?;
        let logs = Arc::new(Mutex::new(logs));
        let listen = matches.get_one::<String>("listen").unwrap();
        println!("Loaded {} mocks from {}", logs.lock().await.len().magenta(), proxy::PROXY_LOG_FILE.magenta());
        println!("Replay server is running on {}", listen.magenta());
        replay::start_replay_server(logs, listen).await?;
        return Ok(());
    }


    let target = matches.get_one::<String>("target");

    if target.is_none() {
        eprintln!("Error: Target URL is required");
        std::process::exit(1);
    }

    let target = target.unwrap();
    let listen = matches.get_one::<String>("listen").unwrap();

    start_server(target, listen).await?;

    Ok(())
}