1 use std::io::stdin;
2 use std::io::stdout;
3 use std::str;
4
5 use protobuf::descriptor::FileDescriptorProto;
6 use protobuf::plugin::*;
7 use protobuf::Message;
8 use protobuf_parse::ProtoPathBuf;
9
10 pub struct GenRequest<'a> {
11 pub file_descriptors: &'a [FileDescriptorProto],
12 pub files_to_generate: &'a [ProtoPathBuf],
13 pub parameter: &'a str,
14 }
15
16 pub struct GenResult {
17 pub name: String,
18 pub content: Vec<u8>,
19 }
20
plugin_main<F>(gen: F) -> anyhow::Result<()> where F: Fn(&GenRequest) -> anyhow::Result<Vec<GenResult>>,21 pub fn plugin_main<F>(gen: F) -> anyhow::Result<()>
22 where
23 F: Fn(&GenRequest) -> anyhow::Result<Vec<GenResult>>,
24 {
25 let req = CodeGeneratorRequest::parse_from_reader(&mut stdin()).unwrap();
26 let result = gen(&GenRequest {
27 file_descriptors: &req.proto_file,
28 files_to_generate: &req
29 .file_to_generate
30 .iter()
31 .map(|n| ProtoPathBuf::new(n.to_owned()))
32 .collect::<anyhow::Result<Vec<_>>>()?,
33 parameter: req.parameter(),
34 })?;
35 let mut resp = CodeGeneratorResponse::new();
36 resp.set_supported_features(code_generator_response::Feature::FEATURE_PROTO3_OPTIONAL as u64);
37 resp.file = result
38 .iter()
39 .map(|file| {
40 let mut r = code_generator_response::File::new();
41 r.set_name(file.name.to_string());
42 r.set_content(str::from_utf8(file.content.as_ref()).unwrap().to_string());
43 r
44 })
45 .collect();
46 resp.write_to_writer(&mut stdout()).unwrap();
47 Ok(())
48 }
49