mollysocket 1.7.1

MollySocket allows getting signal notifications via UnifiedPush.
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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99

use clap::{ArgAction, Parser, Subcommand};
use qrcode::QrcodeCommand;
use std::{env, path::PathBuf};
use vapid::VapidCommand;

use crate::cli::{connection::ConnectionCommand, test::TestCommand};
use crate::config;

mod connection;
mod qrcode;
mod server;
mod test;
mod vapid;

#[derive(Parser)]
#[command(author, version, about, long_about = None)]
#[command(infer_subcommands = true)]
struct Cli {
    /// Sets a custom config file
    #[arg(short, long, value_name = "FILE")]
    config: Option<PathBuf>,

    /// Verbosity level
    #[arg(short, action = ArgAction::Count)]
    verbose: u8,

    #[command(subcommand)]
    command: Command,
}

#[derive(Subcommand)]
enum Command {
    /// Run webserver and websockets
    Server {},

    /// Generate and test VAPID keys
    Vapid {
        #[command(subcommand)]
        command: VapidCommand,
    },

    /// Print mollysocket link URL and show the associated QR code
    QRCode {
        #[command(subcommand)]
        command: QrcodeCommand,
    },

    /// Add, remove and list connections
    Connection {
        #[command(subcommand)]
        command: ConnectionCommand,
    },

    /// Test account and endpoint validity
    Test {
        #[command(subcommand)]
        command: TestCommand,
    },
}

pub async fn cli() {
    let cli = Cli::parse();

    match cli.verbose {
        0 => (),
        1 => match env::var("RUST_LOG") {
            Ok(v) if (v.as_str() == "trace" || v.as_str() == "debug") => (),
            _ => env::set_var("RUST_LOG", "info"),
        },
        2 => match env::var("RUST_LOG") {
            Ok(v) if (v.as_str() == "trace") => (),
            _ => env::set_var("RUST_LOG", "debug"),
        },
        _ => env::set_var("RUST_LOG", "trace"),
    }

    match &cli.command {
        Command::Server {} => (),
        Command::Vapid { .. } => (),
        _ => {
            if env::var("RUST_LOG").is_err() {
                env::set_var("RUST_LOG", "info");
            }
        }
    }
    env_logger::init();

    log::debug!("env_logger initialized");

    config::load_config(cli.config);

    match &cli.command {
        Command::Server {} => server::server().await,
        Command::QRCode { command } => qrcode::qrcode(command),
        Command::Connection { command } => connection::connection(command).await,
        Command::Test { command } => test::test(command).await,
        Command::Vapid { command } => vapid::vapid(command),
    }
}