use std::collections::HashSet;
use proc_macro2::TokenStream;
use crate::{Attributes, Service};
#[derive(Debug)]
pub struct CodeGenBuilder {
emit_package: bool,
compile_well_known_types: bool,
attributes: Attributes,
build_transport: bool,
disable_comments: HashSet<String>,
}
impl CodeGenBuilder {
pub fn new() -> Self {
Default::default()
}
pub fn emit_package(&mut self, enable: bool) -> &mut Self {
self.emit_package = enable;
self
}
pub fn attributes(&mut self, attributes: Attributes) -> &mut Self {
self.attributes = attributes;
self
}
pub fn build_transport(&mut self, build_transport: bool) -> &mut Self {
self.build_transport = build_transport;
self
}
pub fn compile_well_known_types(&mut self, enable: bool) -> &mut Self {
self.compile_well_known_types = enable;
self
}
pub fn disable_comments(&mut self, disable_comments: HashSet<String>) -> &mut Self {
self.disable_comments = disable_comments;
self
}
pub fn generate_client(&self, service: &impl Service, proto_path: &str) -> TokenStream {
crate::client::generate_internal(
service,
self.emit_package,
proto_path,
self.compile_well_known_types,
self.build_transport,
&self.attributes,
&self.disable_comments,
)
}
pub fn generate_server(&self, service: &impl Service, proto_path: &str) -> TokenStream {
crate::server::generate_internal(
service,
self.emit_package,
proto_path,
self.compile_well_known_types,
&self.attributes,
&self.disable_comments,
)
}
}
impl Default for CodeGenBuilder {
fn default() -> Self {
Self {
emit_package: true,
compile_well_known_types: false,
attributes: Attributes::default(),
build_transport: true,
disable_comments: HashSet::default(),
}
}
}