switchback-protobuf 0.0.1-0.dev.4

Protobuf parser for the switchback framework.
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

//! Descriptor set loading.

use crate::descriptor::FileDescriptorProto;
use crate::input::ResolvedInput;
use anyhow::{Context, Result};
use buffa::Message;
use buffa_descriptor::generated::descriptor::FileDescriptorSet;
use std::path::{Path, PathBuf};

pub fn load_descriptor_set(path: &Path) -> Result<(Vec<FileDescriptorProto>, Vec<String>)> {
    let bytes =
        std::fs::read(path).with_context(|| format!("read descriptor set {}", path.display()))?;
    let set = FileDescriptorSet::decode_from_slice(&bytes)
        .map_err(|e| anyhow::anyhow!("decode FileDescriptorSet {}: {e}", path.display()))?;
    let files = set.file;
    let names: Vec<String> = files.iter().filter_map(|f| f.name.clone()).collect();
    Ok((files, names))
}

#[allow(dead_code)]
pub fn read_request_stdin() -> Result<ResolvedInput> {
    use buffa_descriptor::generated::compiler::CodeGeneratorRequest;
    use std::io::Read;
    let mut stdin = Vec::new();
    std::io::stdin()
        .read_to_end(&mut stdin)
        .context("read CodeGeneratorRequest from stdin")?;
    let req = CodeGeneratorRequest::decode_from_slice(&stdin)
        .map_err(|e| anyhow::anyhow!("decode CodeGeneratorRequest: {e}"))?;
    Ok(ResolvedInput {
        proto_file: req.proto_file,
        file_to_generate: req.file_to_generate,
        proto_search_paths: Vec::new(),
        module_root: PathBuf::from("."),
    })
}