    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
    63
    64
    65
    66
    67
    68
    69
    70
    71
    72
    73
    74
    75
    76
    77
    78
    79
    80
    81
    82
    83
    84
    85
    86
    87
    88
    89
    90
    91

    // Copyright 2018 Guillaume Pinot (@TeXitoi) <texitoi@texitoi.eu>,
    // Kevin Knapp (@kbknapp) <kbknapp@gmail.com>, and
    // Ana Hobden (@hoverbear) <operator@hoverbear.org>
    //
    // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
    // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
    // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
    // option. This file may not be copied, modified, or distributed
    // except according to those terms.
    //
    // This work was derived from Structopt (https://github.com/TeXitoi/structopt)
    // commit#ea76fa1b1b273e65e3b0b1046643715b49bec51f which is licensed under the
    // MIT/Apache 2.0 license.

    use proc_macro2::{Span, TokenStream};
    use quote::quote;
    use syn::{Generics, Ident};

    use crate::item::Item;

    pub fn gen_for_struct(item: &Item, item_name: &Ident, generics: &Generics) -> TokenStream {
        let (impl_generics, ty_generics, where_clause) = generics.split_for_impl();

        let name = item.cased_name();
        let app_var = Ident::new("__clap_app", Span::call_site());

        let tokens = quote! {
            #[allow(dead_code, unreachable_code, unused_variables, unused_braces)]
            #[allow(
                clippy::style,
                clippy::complexity,
                clippy::pedantic,
                clippy::restriction,
                clippy::perf,
                clippy::deprecated,
                clippy::nursery,
                clippy::cargo,
                clippy::suspicious_else_formatting,
            )]
            #[deny(clippy::correctness)]
            impl #impl_generics clap::CommandFactory for #item_name #ty_generics #where_clause {
                fn command<'b>() -> clap::Command {
                    let #app_var = clap::Command::new(#name);
                    <Self as clap::Args>::augment_args(#app_var)
                }

                fn command_for_update<'b>() -> clap::Command {
                    let #app_var = clap::Command::new(#name);
                    <Self as clap::Args>::augment_args_for_update(#app_var)
                }
            }
        };

        tokens
    }

    pub fn gen_for_enum(item: &Item, item_name: &Ident, generics: &Generics) -> TokenStream {
        let (impl_generics, ty_generics, where_clause) = generics.split_for_impl();

        let name = item.cased_name();
        let app_var = Ident::new("__clap_app", Span::call_site());

        quote! {
            #[allow(dead_code, unreachable_code, unused_variables, unused_braces)]
            #[allow(
                clippy::style,
                clippy::complexity,
                clippy::pedantic,
                clippy::restriction,
                clippy::perf,
                clippy::deprecated,
                clippy::nursery,
                clippy::cargo,
                clippy::suspicious_else_formatting,
            )]
            #[deny(clippy::correctness)]
            impl #impl_generics clap::CommandFactory for #item_name #ty_generics #where_clause {
                fn command<'b>() -> clap::Command {
                    let #app_var = clap::Command::new(#name)
                        .subcommand_required(true)
                        .arg_required_else_help(true);
                    <Self as clap::Subcommand>::augment_subcommands(#app_var)
                }

                fn command_for_update<'b>() -> clap::Command {
                    let #app_var = clap::Command::new(#name);
                    <Self as clap::Subcommand>::augment_subcommands_for_update(#app_var)
                }
            }
        }
    }