yngvi 0.1.0

Read, calculate and write weather data
Documentation 
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

use yngvi::{
    core::{FileReader, InMemWriter, NoopWriter, StdoutWriter, WeatherReadingSource, Writer},
    display::{DisplayReader, HidSource},
    influxdb::{Influx2Writer, InfluxWriter},
    rtl_433::{rtl_433_source, RTL433Reader},
};

pub fn find_reader(value: &String) -> Box<dyn Iterator<Item = WeatherReadingSource>> {
    match value.to_uppercase().as_str() {
        "ACURITE_DISPLAY" => {
            let source = HidSource::new(0x24c0, 0x003).expect("could not start HID Api");

            Box::new(DisplayReader::new(source))
        }
        "RTL_433" => {
            let source = rtl_433_source();

            Box::new(RTL433Reader::new(source))
        }
        "FILE" => {
            let path = var("SRC_FILE_PATH").expect("PATH not provided");

            Box::new(FileReader::new(path.as_str()))
        }
        _ => panic!("no reader defined. found {}", value),
    }
}

pub fn find_writer(value: &String) -> Box<dyn Writer + Send> {
    match value.to_uppercase().as_str() {
        "INFLUXDB" => Box::new(create_influx_writer()),
        "INFLUXDB2" => Box::new(create_influx2_writer()),
        "INMEMORY" => Box::new(InMemWriter::new()),
        "STDOUT" => Box::new(StdoutWriter::new()),
        "NOOP" => Box::new(NoopWriter::new()),
        _ => panic!("no writer defined. found {}", value),
    }
}

pub fn create_influx_writer() -> InfluxWriter {
    let url = var("DEST_INFLUXDB_URL").unwrap_or("http://localhost:8086".to_string());
    let database = var("DEST_INFLUXDB_DB").unwrap_or("weather".to_string());

    InfluxWriter::new(url, database)
}

pub fn create_influx2_writer() -> Influx2Writer {
    let url = var("DEST_INFLUXDB2_URL").unwrap_or("http://localhost:8086".to_string());
    let org = var("DEST_INFLUXDB2_ORG").expect("ORG not provided");
    let bucket = var("DEST_INFLUXDB2_BUCKET").expect("BUCKET not provided");
    let token = var("DEST_INFLUXDB2_TOKEN").expect("TOKEN not provided");

    Influx2Writer::new(url, org, bucket, token)
}

pub fn var(key: &str) -> Result<String, std::env::VarError> {
    std::env::var(format!("WS_{}", key))
}