azure-functions-shared 0.4.0

Implementations shared between the azure-functions-codegen and azure-functions crates.
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
59

extern crate protoc_grpcio;

use std::env;
use std::fs;
use std::path::PathBuf;

const OUT_DIR_VAR: &'static str = "OUT_DIR";
const CACHE_DIR_NAME: &'static str = "cache";
const PROTOBUF_INPUT_FILE: &'static str = "FunctionRpc.proto";
const RUST_PROTOBUF_FILE: &'static str = "FunctionRpc.rs";
const RUST_GRPC_FILE: &'static str = "FunctionRpc_grpc.rs";

fn compile_protobufs(out_dir: &PathBuf, cache_dir: &PathBuf) {
    protoc_grpcio::compile_grpc_protos(&[PROTOBUF_INPUT_FILE], &["protobuf/src/proto"], &out_dir)
        .expect("Failed to compile gRPC definitions.");

    fs::copy(
        out_dir.join(RUST_PROTOBUF_FILE),
        cache_dir.join(RUST_PROTOBUF_FILE),
    )
    .expect(&format!("can't update cache file '{}'", RUST_PROTOBUF_FILE));

    fs::copy(out_dir.join(RUST_GRPC_FILE), cache_dir.join(RUST_GRPC_FILE))
        .expect(&format!("can't update cache file '{}'", RUST_GRPC_FILE));
}

fn use_cached_files(out_dir: &PathBuf, cache_dir: &PathBuf) {
    fs::copy(
        cache_dir.join(RUST_PROTOBUF_FILE),
        out_dir.join(RUST_PROTOBUF_FILE),
    )
    .expect(&format!(
        "can't copy cache file '{}' to output directory",
        RUST_PROTOBUF_FILE
    ));

    fs::copy(cache_dir.join(RUST_GRPC_FILE), out_dir.join(RUST_GRPC_FILE)).expect(&format!(
        "can't copy cache file '{}' to output directory",
        RUST_GRPC_FILE
    ));
}

fn main() {
    println!("cargo:rerun-if-changed=protobuf/src/proto/FunctionRpc.proto");

    let out_dir = PathBuf::from(env::var(OUT_DIR_VAR).unwrap());

    let cache_dir = env::current_dir()
        .expect("couldn't determine current working directory")
        .join(CACHE_DIR_NAME);

    fs::create_dir_all(&cache_dir).expect("failed to create cache directory");

    if cfg!(feature = "compile_protobufs") {
        compile_protobufs(&out_dir, &cache_dir);
    } else {
        use_cached_files(&out_dir, &cache_dir);
    }
}