wick-component-codegen 0.6.0

Code generator for wick components
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
60

use check_keyword::CheckKeyword;
use heck::{AsPascalCase, AsSnakeCase};
use itertools::Itertools;
use proc_macro2::{Ident, Span};
use wick_config::config::Binding;
use wick_interface_types::{EnumVariant, OperationSignature};

pub(crate) fn id(name: &str) -> Ident {
  if name.is_keyword() {
    Ident::new(&format!("{}_", name), Span::call_site())
  } else {
    Ident::new(name, Span::call_site())
  }
}

pub(crate) fn component_id<T>(r: &Binding<T>) -> String {
  format!("{}Component", &pascal(r.id()))
}

pub(crate) fn generic_config_id() -> String {
  "Config".to_owned()
}

pub(crate) fn snake(s: &str) -> String {
  AsSnakeCase(s).to_string()
}

pub(crate) fn pascal(s: &str) -> String {
  AsPascalCase(s).to_string()
}

pub(crate) fn op_wrapper_name(op: &OperationSignature) -> String {
  snake(&format!("{}_wrapper", op.name()))
}

pub(crate) fn get_typename_parts(name: &str) -> (Vec<&str>, &str) {
  let parts = name.split("::").collect_vec();
  let len = parts.len();
  let parts = parts.split_at(len - 1);
  (parts.0.to_vec(), parts.1[0])
}

pub(crate) fn _pathify_typename(name: &str) -> String {
  println!("name: {:?}", name);
  let (module_parts, item_part) = get_typename_parts(name);
  let mut path = module_parts
    .iter()
    .map(|p| format!("{}::", snake(p)))
    .collect::<String>();
  let name = pascal(item_part);

  path.push_str(&name);

  println!("structdef_name: {:?}", path);
  path
}

pub(crate) fn enumvariant_name(ty: &EnumVariant) -> String {
  pascal(&ty.name)
}