embedded-cli-macros 0.2.1

Macros for embedded-cli lib
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

use darling::Result;
use proc_macro2::TokenStream;
use quote::quote;

use super::{model::Command, TargetType};

#[cfg(feature = "autocomplete")]
pub fn derive_autocomplete(target: &TargetType, commands: &[Command]) -> Result<TokenStream> {
    let command_count = commands.len();
    let command_names: Vec<String> = commands.iter().map(|c| c.name.to_string()).collect();

    let ident = target.ident();
    let named_lifetime = target.named_lifetime();

    let output = quote! {
        impl #named_lifetime _cli::service::Autocomplete for #ident #named_lifetime {
            fn autocomplete(
                request: _cli::autocomplete::Request<'_>,
                autocompletion: &mut _cli::autocomplete::Autocompletion<'_>,
            ) {
                const NAMES: &[&str; #command_count] = &[#(#command_names),*];
                if let _cli::autocomplete::Request::CommandName(name) = request {
                    NAMES
                        .iter()
                        .skip_while(|n| !n.starts_with(name))
                        .take_while(|n| n.starts_with(name))
                        .for_each(|n| {
                            // SAFETY: n starts with name, so name cannot be longer
                            let autocompleted = unsafe { n.get_unchecked(name.len()..) };
                            autocompletion.merge_autocompletion(autocompleted)
                        });
                }
            }
        }
    };

    Ok(output)
}

#[allow(unused_variables)]
#[cfg(not(feature = "autocomplete"))]
pub fn derive_autocomplete(target: &TargetType, commands: &[Command]) -> Result<TokenStream> {
    let ident = target.ident();
    let named_lifetime = target.named_lifetime();

    let output = quote! {
        impl #named_lifetime _cli::service::Autocomplete for #ident #named_lifetime { }
    };

    Ok(output)
}