iscp-rs 1.0.1

iSCPv2 Client Library
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

use anyhow::Result;

#[cfg(feature = "gen")]
use anyhow::Context;

fn main() -> Result<()> {
    gen_proto()
}

#[cfg(not(feature = "gen"))]
fn gen_proto() -> Result<()> {
    Ok(())
}

#[cfg(feature = "gen")]
const PROTO_SRC_DIR: &str = "iscp-proto/proto";

#[cfg(feature = "gen")]
fn gen_proto() -> Result<()> {
    const PROTO_SRC_FILES: &str = "iscp-proto/proto/**/*.proto";
    const AUTOGEN_DIR: &str = "src/message/proto";

    let proto_files: Vec<_> = glob::glob(PROTO_SRC_FILES)
        .unwrap_or_else(|e| panic!("Failed to read {}: {}", PROTO_SRC_FILES, e))
        .filter_map(|res| res.ok())
        .collect();

    if std::path::Path::new(AUTOGEN_DIR).exists() {
        std::fs::remove_dir_all(AUTOGEN_DIR).context("Remove dir")?;
    }
    std::fs::create_dir(AUTOGEN_DIR).context("Create dir")?;

    prost_build::Config::new()
        .out_dir(AUTOGEN_DIR)
        .bytes(["."])
        .type_attribute(".", "#[derive(serde::Serialize, serde::Deserialize)]")
        .type_attribute("DataID", "#[derive(PartialOrd, Ord, Eq, Hash)]")
        .type_attribute("DataFilter", "#[derive(PartialOrd, Ord, Eq, Hash)]")
        .type_attribute("DataPoint", "#[derive(Eq)]")
        .type_attribute("DownstreamFilter", "#[derive(PartialOrd, Ord, Eq, Hash)]")
        .type_attribute("UpstreamInfo", "#[derive(Eq)]")
        .field_attribute("type", "#[serde(rename = \"type\")]")
        .compile_protos(&proto_files, &[PROTO_SRC_DIR])
        .context("Proto compile")?;

    process_generated_files().context("Process generated files")?;

    Ok(())
}

#[cfg(feature = "gen")]
// Copy proto files to dest_dir and preprocess files
fn process_generated_files() -> Result<()> {
    const PATH: &[&str] = &[
        "src/message/proto/iscp2.v1.rs",
        "src/message/proto/iscp2.v1.extensions.rs",
    ];

    let re = regex::Regex::new("r#type").unwrap();
    for path in PATH {
        let s = std::fs::read_to_string(path)?;
        let replacer = |_caps: &regex::Captures<'_>| "type_";
        let s = re.replace_all(&s, replacer);

        std::fs::write(path, s.as_bytes())?;
    }

    // Workaround for type_attribute
    let re = regex::Regex::new("pub enum DataIdOrAlias").unwrap();
    for path in PATH {
        let s = std::fs::read_to_string(path)?;
        let replacer = |_caps: &regex::Captures<'_>| {
            "#[derive(PartialOrd, Ord, Eq, Hash)]\n    pub enum DataIdOrAlias"
        };
        let s = re.replace_all(&s, replacer);

        std::fs::write(path, s.as_bytes())?;
    }

    Ok(())
}