weblock 0.1.0

A lightweight http proxy for putting a password onto your selfhosted apps
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

use clap::{Parser, Subcommand};

use crate::proxy::start_proxy;

mod helper;
mod proxy;

#[derive(Debug, Parser, Clone)]
#[command(about, version, long_about = None)]
struct Cli {
    name: String,
    #[command(subcommand)]
    action: Action,
}

#[derive(Subcommand, Debug, Clone)]
pub enum Action {
    Create {
        #[arg(short = 'i', long = "in")]
        inport: u32,
        #[arg(short = 'o', long = "out")]
        outport: u32,
        #[arg(short = 'p', long = "password")]
        password: String,
    },
}

#[cfg(unix)]
fn main() {
    let options = Cli::parse();

    match options.action {
        Action::Create {
            inport,
            outport,
            password,
        } => {
            create_async_proxy(inport, outport, password);
        }
    }
}

pub fn create_async_proxy(inport: u32, outport: u32, password: String) {
    let rt = tokio::runtime::Builder::new_multi_thread()
        .enable_all()
        .build()
        .unwrap();
    rt.block_on(async { start_proxy(inport, outport, password).await });
}