evitable-derive-core 0.5.0

Derive implementation for evitable
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
61
62

use evitable_syn_meta_ext::PathExt;
use proc_macro2::{Ident, Span};
use syn::{parse_str, PathArguments, PathSegment, Token, VisRestricted, Visibility};

fn create_super_vis(depth: usize) -> Visibility {
  let mut code = String::new();
  code.push_str("pub(in ");
  for i in 0..depth {
    if i > 0 {
      code.push_str("::");
    }

    code.push_str("super");
  }

  code.push_str(")");
  parse_str(&code).unwrap()
}

fn modify_vis(r: &VisRestricted, depth: usize) -> Visibility {
  let mut r = r.clone();
  let ident = r.path.single_ident();
  if r.in_token.is_none() && ident.is_some() {
    // path is either self, super or crate
    let ident = ident.unwrap();
    match ident.to_string().as_ref() {
      "self" => create_super_vis(depth),
      "super" => create_super_vis(depth + 1),
      "crate" => Visibility::Restricted(r),
      _ => unreachable!(),
    }
  } else {
    r.in_token = Some(Token![in](Span::call_site()));
    let ident = &r.path.segments[0].ident;
    match ident.to_string().as_ref() {
      "self" => {
        r.path.segments[0].ident = Ident::new("super", Span::call_site());
      }
      "super" => {
        let segment = PathSegment {
          ident: Ident::new("super", Span::call_site()),
          arguments: PathArguments::None,
        };
        r.path.segments.insert(0, segment);
      }
      _ => {
        /* any other path is root based, and can be kept as is */
        ()
      }
    }
    Visibility::Restricted(r)
  }
}

pub fn inherited(vis: &Visibility, depth: usize) -> Visibility {
  match vis {
    Visibility::Public(_) => vis.clone(),
    Visibility::Crate(_) => vis.clone(),
    Visibility::Inherited => create_super_vis(depth),
    Visibility::Restricted(r) => modify_vis(r, depth),
  }
}