golem-rib 1.3.2-dev.2

Parser for Golem's Rib language
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

use cargo_metadata::MetadataCommand;
use miette::miette;
use protox::prost::Message;
use std::env;
use std::path::PathBuf;

fn main() -> Result<(), Box<dyn std::error::Error>> {
    let golem_wasm_root = find_package_root("golem-wasm");

    let file_descriptors = protox::compile(
        [
            "proto/golem/rib/compiler_output.proto",
            "proto/golem/rib/expr.proto",
            "proto/golem/rib/function_name.proto",
            "proto/golem/rib/instance_type.proto",
            "proto/golem/rib/ir.proto",
            "proto/golem/rib/rib_byte_code.proto",
            "proto/golem/rib/rib_input.proto",
            "proto/golem/rib/rib_output.proto",
            "proto/golem/rib/type_name.proto",
            "proto/golem/rib/worker_functions_in_rib.proto",
        ],
        [&format!("{golem_wasm_root}/proto"), &"proto".to_string()],
    )?;

    let out_dir = PathBuf::from(env::var("OUT_DIR").unwrap());
    let fd_path = out_dir.join("services.bin");

    std::fs::write(fd_path, file_descriptors.encode_to_vec())?;

    tonic_prost_build::configure()
        .build_server(true)
        .extern_path(".wasm.rpc", "::golem_wasm::protobuf")
        .include_file("mod.rs")
        .compile_fds(file_descriptors)
        .map_err(|e| miette!(e))?;

    Ok(())
}

fn find_package_root(name: &str) -> String {
    let metadata = MetadataCommand::new()
        .manifest_path("./Cargo.toml")
        .exec()
        .unwrap();
    let package = metadata
        .packages
        .iter()
        .find(|p| p.name.as_str() == name)
        .unwrap();
    package.manifest_path.parent().unwrap().to_string()
}