summaryrefslogtreecommitdiff
path: root/src/main.rs
blob: 3e1388beb7d6331b7b44ba4eb439e15f452ca52d (plain) (blame)
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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
use std::fmt::Display;

use clap::{Parser, Subcommand};
use config::SETTINGS;
use error::CliError;
use requests::{start::start, device};
use reqwest::header::{HeaderMap, HeaderValue};
use serde::Deserialize;

mod config;
mod error;
mod requests;

/// webol http client
#[derive(Parser)]
#[command(author, version, about, long_about = None)]
struct Args {
    #[command(subcommand)]
    commands: Commands,
}

#[derive(Subcommand)]
enum Commands {
    Start {
        /// id of the device
        id: String,
        #[arg(short, long)]
        ping: Option<bool>
    },
    Device {
        #[command(subcommand)]
        devicecmd: DeviceCmd,
    }
}

#[derive(Subcommand)]
enum DeviceCmd {
    Add {
        id: String,
        mac: String,
        broadcast_addr: String,
        ip: String
    },
    Get {
        id: String,
    },
    Edit {
        id: String,
        mac: String,
        broadcast_addr: String,
        ip: String
    },
}

#[tokio::main]
async fn main() -> Result<(), CliError> {
    let cli = Args::parse();

    match cli.commands {
        Commands::Start { id, ping } => {
            start(id, ping.unwrap_or(true)).await?;
        },
        Commands::Device { devicecmd } => {
            match devicecmd {
                DeviceCmd::Add { id, mac, broadcast_addr, ip } => {
                    device::put(id, mac, broadcast_addr, ip).await?;
                },
                DeviceCmd::Get { id } => {
                    device::get(id).await?;
                },
                DeviceCmd::Edit { id, mac, broadcast_addr, ip } => {
                    device::post(id, mac, broadcast_addr, ip).await?;
                },
            }
        }
    }

    Ok(())
}

fn default_headers() -> Result<HeaderMap, CliError> {
    let mut map = HeaderMap::new();
    map.append("Accept-Content", HeaderValue::from_str("application/json").unwrap());
    map.append("Content-Type", HeaderValue::from_str("application/json").unwrap());
    map.append(
        "Authorization",
        HeaderValue::from_str(
            SETTINGS.get_string("key")
            .map_err(CliError::Config)?
            .as_str()
        ).unwrap()
    );

    Ok(map)
}

fn format_url(path: &str, protocol: Protocols) -> Result<String, CliError> {
    Ok(format!(
        "{}://{}/{}",
        protocol,
        SETTINGS.get_string("server").map_err(CliError::Config)?,
        path
    ))
}

enum Protocols {
    Http,
    Websocket,
}

impl Display for Protocols {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            Self::Http => f.write_str("http"),
            Self::Websocket => f.write_str("ws")
        }
    }
}

#[derive(Debug, Deserialize)]
struct ErrorResponse {
    error: String
}