use std::{collections::BTreeMap, path::Path};
use itertools::Itertools;
use proc_macro2::TokenStream;
use quote::quote;
use ploidy_core::codegen::{IntoCode, write_to_disk};
mod cargo;
mod cfg;
mod client;
mod config;
mod derives;
mod enum_;
mod ext;
mod graph;
mod inlines;
mod naming;
mod operation;
mod primitive;
mod query;
mod ref_;
mod resource;
mod schema;
mod statics;
mod struct_;
mod tagged;
mod types;
mod untagged;
#[cfg(test)]
mod tests;
pub use cargo::*;
pub use cfg::*;
pub use client::*;
pub use config::*;
pub use graph::*;
pub use naming::*;
pub use operation::*;
pub use primitive::*;
pub use query::*;
pub use resource::*;
pub use schema::*;
pub use statics::*;
pub use types::*;
pub fn write_types_to_disk(output: &Path, graph: &CodegenGraph<'_>) -> miette::Result<()> {
for schema in graph.schemas() {
let code = CodegenSchemaType::new(graph, &schema).into_code();
write_to_disk(output, code)?;
}
write_to_disk(output, CodegenTypesModule::new(graph))?;
Ok(())
}
pub fn write_client_to_disk(output: &Path, graph: &CodegenGraph<'_>) -> miette::Result<()> {
let ops_by_resource: BTreeMap<_, Vec<_>> =
graph.operations().fold(BTreeMap::default(), |mut map, op| {
let resource = graph.resource_for(&op);
map.entry(resource).or_default().push(op);
map
});
for (ident, ops) in &ops_by_resource {
write_to_disk(output, CodegenResource::new(graph, *ident, ops))?;
}
let idents = ops_by_resource.keys().copied().collect_vec();
write_to_disk(output, CodegenClientModule::new(graph, &idents))?;
Ok(())
}
pub fn doc_attrs(description: &str) -> TokenStream {
let lines = textwrap::wrap(description, 80)
.into_iter()
.map(|line| quote!(#[doc = #line]));
quote! { #(#lines)* }
}