//! Completion for attributes
//!
//! This module uses a bit of static metadata to provide completions
//! for built-in attributes.

use ra_syntax::{ast, AstNode, SyntaxKind};
use rustc_hash::FxHashSet;

use crate::completion::{
    completion_context::CompletionContext,
    completion_item::{CompletionItem, CompletionItemKind, CompletionKind, Completions},
};

pub(super) fn complete_attribute(acc: &mut Completions, ctx: &CompletionContext) -> Option<()> {
    let attribute = ctx.attribute_under_caret.as_ref()?;

    match (attribute.path(), attribute.input()) {
        (Some(path), Some(ast::AttrInput::TokenTree(token_tree)))
            if path.to_string() == "derive" =>
        {
            complete_derive(acc, ctx, token_tree)
        }
        (_, Some(ast::AttrInput::TokenTree(_token_tree))) => {}
        _ => complete_attribute_start(acc, ctx, attribute),
    }
    Some(())
}

fn complete_attribute_start(acc: &mut Completions, ctx: &CompletionContext, attribute: &ast::Attr) {
    for attr_completion in ATTRIBUTES {
        let mut item = CompletionItem::new(
            CompletionKind::Attribute,
            ctx.source_range(),
            attr_completion.label,
        )
        .kind(CompletionItemKind::Attribute);

        if let Some(lookup) = attr_completion.lookup {
            item = item.lookup_by(lookup);
        }

        match (attr_completion.snippet, ctx.config.snippet_cap) {
            (Some(snippet), Some(cap)) => {
                item = item.insert_snippet(cap, snippet);
            }
            _ => {}
        }

        if attribute.kind() == ast::AttrKind::Inner || !attr_completion.prefer_inner {
            acc.add(item);
        }
    }
}

struct AttrCompletion {
    label: &'static str,
    lookup: Option<&'static str>,
    snippet: Option<&'static str>,
    prefer_inner: bool,
}

impl AttrCompletion {
    const fn prefer_inner(self) -> AttrCompletion {
        AttrCompletion { prefer_inner: true, ..self }
    }
}

const fn attr(
    label: &'static str,
    lookup: Option<&'static str>,
    snippet: Option<&'static str>,
) -> AttrCompletion {
    AttrCompletion { label, lookup, snippet, prefer_inner: false }
}

const ATTRIBUTES: &[AttrCompletion] = &[
    attr("allow(…)", Some("allow"), Some("allow(${0:lint})")),
    attr("cfg_attr(…)", Some("cfg_attr"), Some("cfg_attr(${1:predicate}, ${0:attr})")),
    attr("cfg(…)", Some("cfg"), Some("cfg(${0:predicate})")),
    attr("deny(…)", Some("deny"), Some("deny(${0:lint})")),
    attr(r#"deprecated = "…""#, Some("deprecated"), Some(r#"deprecated = "${0:reason}""#)),
    attr("derive(…)", Some("derive"), Some(r#"derive(${0:Debug})"#)),
    attr(r#"doc = "…""#, Some("doc"), Some(r#"doc = "${0:docs}""#)),
    attr("feature(…)", Some("feature"), Some("feature(${0:flag})")).prefer_inner(),
    attr("forbid(…)", Some("forbid"), Some("forbid(${0:lint})")),
    // FIXME: resolve through macro resolution?
    attr("global_allocator", None, None).prefer_inner(),
    attr("ignore(…)", Some("ignore"), Some("ignore(${0:lint})")),
    attr("inline(…)", Some("inline"), Some("inline(${0:lint})")),
    attr(r#"link_name = "…""#, Some("link_name"), Some(r#"link_name = "${0:symbol_name}""#)),
    attr("link", None, None),
    attr("macro_export", None, None),
    attr("macro_use", None, None),
    attr(r#"must_use = "…""#, Some("must_use"), Some(r#"must_use = "${0:reason}""#)),
    attr("no_mangle", None, None),
    attr("no_std", None, None).prefer_inner(),
    attr("non_exhaustive", None, None),
    attr("panic_handler", None, None).prefer_inner(),
    attr("path = \"…\"", Some("path"), Some("path =\"${0:path}\"")),
    attr("proc_macro", None, None),
    attr("proc_macro_attribute", None, None),
    attr("proc_macro_derive(…)", Some("proc_macro_derive"), Some("proc_macro_derive(${0:Trait})")),
    attr("recursion_limit = …", Some("recursion_limit"), Some("recursion_limit = ${0:128}"))
        .prefer_inner(),
    attr("repr(…)", Some("repr"), Some("repr(${0:C})")),
    attr(
        "should_panic(…)",
        Some("should_panic"),
        Some(r#"should_panic(expected = "${0:reason}")"#),
    ),
    attr(
        r#"target_feature = "…""#,
        Some("target_feature"),
        Some("target_feature = \"${0:feature}\""),
    ),
    attr("test", None, None),
    attr("used", None, None),
    attr("warn(…)", Some("warn"), Some("warn(${0:lint})")),
    attr(
        r#"windows_subsystem = "…""#,
        Some("windows_subsystem"),
        Some(r#"windows_subsystem = "${0:subsystem}""#),
    )
    .prefer_inner(),
];

fn complete_derive(acc: &mut Completions, ctx: &CompletionContext, derive_input: ast::TokenTree) {
    if let Ok(existing_derives) = parse_derive_input(derive_input) {
        for derive_completion in DEFAULT_DERIVE_COMPLETIONS
            .into_iter()
            .filter(|completion| !existing_derives.contains(completion.label))
        {
            let mut label = derive_completion.label.to_owned();
            for dependency in derive_completion
                .dependencies
                .into_iter()
                .filter(|&&dependency| !existing_derives.contains(dependency))
            {
                label.push_str(", ");
                label.push_str(dependency);
            }
            acc.add(
                CompletionItem::new(CompletionKind::Attribute, ctx.source_range(), label)
                    .kind(CompletionItemKind::Attribute),
            );
        }

        for custom_derive_name in get_derive_names_in_scope(ctx).difference(&existing_derives) {
            acc.add(
                CompletionItem::new(
                    CompletionKind::Attribute,
                    ctx.source_range(),
                    custom_derive_name,
                )
                .kind(CompletionItemKind::Attribute),
            );
        }
    }
}

fn parse_derive_input(derive_input: ast::TokenTree) -> Result<FxHashSet<String>, ()> {
    match (derive_input.left_delimiter_token(), derive_input.right_delimiter_token()) {
        (Some(left_paren), Some(right_paren))
            if left_paren.kind() == SyntaxKind::L_PAREN
                && right_paren.kind() == SyntaxKind::R_PAREN =>
        {
            let mut input_derives = FxHashSet::default();
            let mut current_derive = String::new();
            for token in derive_input
                .syntax()
                .children_with_tokens()
                .filter_map(|token| token.into_token())
                .skip_while(|token| token != &left_paren)
                .skip(1)
                .take_while(|token| token != &right_paren)
            {
                if SyntaxKind::COMMA == token.kind() {
                    if !current_derive.is_empty() {
                        input_derives.insert(current_derive);
                        current_derive = String::new();
                    }
                } else {
                    current_derive.push_str(token.to_string().trim());
                }
            }

            if !current_derive.is_empty() {
                input_derives.insert(current_derive);
            }
            Ok(input_derives)
        }
        _ => Err(()),
    }
}

fn get_derive_names_in_scope(ctx: &CompletionContext) -> FxHashSet<String> {
    let mut result = FxHashSet::default();
    ctx.scope().process_all_names(&mut |name, scope_def| {
        if let hir::ScopeDef::MacroDef(mac) = scope_def {
            if mac.is_derive_macro() {
                result.insert(name.to_string());
            }
        }
    });
    result
}

struct DeriveCompletion {
    label: &'static str,
    dependencies: &'static [&'static str],
}

/// Standard Rust derives and the information about their dependencies
/// (the dependencies are needed so that the main derive don't break the compilation when added)
const DEFAULT_DERIVE_COMPLETIONS: &[DeriveCompletion] = &[
    DeriveCompletion { label: "Clone", dependencies: &[] },
    DeriveCompletion { label: "Copy", dependencies: &["Clone"] },
    DeriveCompletion { label: "Debug", dependencies: &[] },
    DeriveCompletion { label: "Default", dependencies: &[] },
    DeriveCompletion { label: "Hash", dependencies: &[] },
    DeriveCompletion { label: "PartialEq", dependencies: &[] },
    DeriveCompletion { label: "Eq", dependencies: &["PartialEq"] },
    DeriveCompletion { label: "PartialOrd", dependencies: &["PartialEq"] },
    DeriveCompletion { label: "Ord", dependencies: &["PartialOrd", "Eq", "PartialEq"] },
];

#[cfg(test)]
mod tests {
    use crate::completion::{test_utils::do_completion, CompletionItem, CompletionKind};
    use insta::assert_debug_snapshot;

    fn do_attr_completion(code: &str) -> Vec<CompletionItem> {
        do_completion(code, CompletionKind::Attribute)
    }

    #[test]
    fn empty_derive_completion() {
        assert_debug_snapshot!(
            do_attr_completion(
                    r"
                    #[derive(<|>)]
                    struct Test {}
                    ",
            ),
            @r###"
        [
            CompletionItem {
                label: "Clone",
                source_range: 9..9,
                delete: 9..9,
                insert: "Clone",
                kind: Attribute,
            },
            CompletionItem {
                label: "Copy, Clone",
                source_range: 9..9,
                delete: 9..9,
                insert: "Copy, Clone",
                kind: Attribute,
            },
            CompletionItem {
                label: "Debug",
                source_range: 9..9,
                delete: 9..9,
                insert: "Debug",
                kind: Attribute,
            },
            CompletionItem {
                label: "Default",
                source_range: 9..9,
                delete: 9..9,
                insert: "Default",
                kind: Attribute,
            },
            CompletionItem {
                label: "Eq, PartialEq",
                source_range: 9..9,
                delete: 9..9,
                insert: "Eq, PartialEq",
                kind: Attribute,
            },
            CompletionItem {
                label: "Hash",
                source_range: 9..9,
                delete: 9..9,
                insert: "Hash",
                kind: Attribute,
            },
            CompletionItem {
                label: "Ord, PartialOrd, Eq, PartialEq",
                source_range: 9..9,
                delete: 9..9,
                insert: "Ord, PartialOrd, Eq, PartialEq",
                kind: Attribute,
            },
            CompletionItem {
                label: "PartialEq",
                source_range: 9..9,
                delete: 9..9,
                insert: "PartialEq",
                kind: Attribute,
            },
            CompletionItem {
                label: "PartialOrd, PartialEq",
                source_range: 9..9,
                delete: 9..9,
                insert: "PartialOrd, PartialEq",
                kind: Attribute,
            },
        ]
        "###
        );
    }

    #[test]
    fn no_completion_for_incorrect_derive() {
        assert_debug_snapshot!(
            do_attr_completion(
                r"
                    #[derive{<|>)]
                    struct Test {}
                    ",
            ),
            @"[]"
        );
    }

    #[test]
    fn derive_with_input_completion() {
        assert_debug_snapshot!(
            do_attr_completion(
                    r"
                    #[derive(serde::Serialize, PartialEq, <|>)]
                    struct Test {}
                    ",
            ),
            @r###"
        [
            CompletionItem {
                label: "Clone",
                source_range: 38..38,
                delete: 38..38,
                insert: "Clone",
                kind: Attribute,
            },
            CompletionItem {
                label: "Copy, Clone",
                source_range: 38..38,
                delete: 38..38,
                insert: "Copy, Clone",
                kind: Attribute,
            },
            CompletionItem {
                label: "Debug",
                source_range: 38..38,
                delete: 38..38,
                insert: "Debug",
                kind: Attribute,
            },
            CompletionItem {
                label: "Default",
                source_range: 38..38,
                delete: 38..38,
                insert: "Default",
                kind: Attribute,
            },
            CompletionItem {
                label: "Eq",
                source_range: 38..38,
                delete: 38..38,
                insert: "Eq",
                kind: Attribute,
            },
            CompletionItem {
                label: "Hash",
                source_range: 38..38,
                delete: 38..38,
                insert: "Hash",
                kind: Attribute,
            },
            CompletionItem {
                label: "Ord, PartialOrd, Eq",
                source_range: 38..38,
                delete: 38..38,
                insert: "Ord, PartialOrd, Eq",
                kind: Attribute,
            },
            CompletionItem {
                label: "PartialOrd",
                source_range: 38..38,
                delete: 38..38,
                insert: "PartialOrd",
                kind: Attribute,
            },
        ]
        "###
        );
    }

    #[test]
    fn test_attribute_completion() {
        assert_debug_snapshot!(
        do_attr_completion(
                r"
                #[<|>]
                ",
        ),
            @r###"
        [
            CompletionItem {
                label: "allow(…)",
                source_range: 2..2,
                delete: 2..2,
                insert: "allow(${0:lint})",
                kind: Attribute,
                lookup: "allow",
            },
            CompletionItem {
                label: "cfg(…)",
                source_range: 2..2,
                delete: 2..2,
                insert: "cfg(${0:predicate})",
                kind: Attribute,
                lookup: "cfg",
            },
            CompletionItem {
                label: "cfg_attr(…)",
                source_range: 2..2,
                delete: 2..2,
                insert: "cfg_attr(${1:predicate}, ${0:attr})",
                kind: Attribute,
                lookup: "cfg_attr",
            },
            CompletionItem {
                label: "deny(…)",
                source_range: 2..2,
                delete: 2..2,
                insert: "deny(${0:lint})",
                kind: Attribute,
                lookup: "deny",
            },
            CompletionItem {
                label: "deprecated = \"…\"",
                source_range: 2..2,
                delete: 2..2,
                insert: "deprecated = \"${0:reason}\"",
                kind: Attribute,
                lookup: "deprecated",
            },
            CompletionItem {
                label: "derive(…)",
                source_range: 2..2,
                delete: 2..2,
                insert: "derive(${0:Debug})",
                kind: Attribute,
                lookup: "derive",
            },
            CompletionItem {
                label: "doc = \"…\"",
                source_range: 2..2,
                delete: 2..2,
                insert: "doc = \"${0:docs}\"",
                kind: Attribute,
                lookup: "doc",
            },
            CompletionItem {
                label: "forbid(…)",
                source_range: 2..2,
                delete: 2..2,
                insert: "forbid(${0:lint})",
                kind: Attribute,
                lookup: "forbid",
            },
            CompletionItem {
                label: "ignore(…)",
                source_range: 2..2,
                delete: 2..2,
                insert: "ignore(${0:lint})",
                kind: Attribute,
                lookup: "ignore",
            },
            CompletionItem {
                label: "inline(…)",
                source_range: 2..2,
                delete: 2..2,
                insert: "inline(${0:lint})",
                kind: Attribute,
                lookup: "inline",
            },
            CompletionItem {
                label: "link",
                source_range: 2..2,
                delete: 2..2,
                insert: "link",
                kind: Attribute,
            },
            CompletionItem {
                label: "link_name = \"…\"",
                source_range: 2..2,
                delete: 2..2,
                insert: "link_name = \"${0:symbol_name}\"",
                kind: Attribute,
                lookup: "link_name",
            },
            CompletionItem {
                label: "macro_export",
                source_range: 2..2,
                delete: 2..2,
                insert: "macro_export",
                kind: Attribute,
            },
            CompletionItem {
                label: "macro_use",
                source_range: 2..2,
                delete: 2..2,
                insert: "macro_use",
                kind: Attribute,
            },
            CompletionItem {
                label: "must_use = \"…\"",
                source_range: 2..2,
                delete: 2..2,
                insert: "must_use = \"${0:reason}\"",
                kind: Attribute,
                lookup: "must_use",
            },
            CompletionItem {
                label: "no_mangle",
                source_range: 2..2,
                delete: 2..2,
                insert: "no_mangle",
                kind: Attribute,
            },
            CompletionItem {
                label: "non_exhaustive",
                source_range: 2..2,
                delete: 2..2,
                insert: "non_exhaustive",
                kind: Attribute,
            },
            CompletionItem {
                label: "path = \"…\"",
                source_range: 2..2,
                delete: 2..2,
                insert: "path =\"${0:path}\"",
                kind: Attribute,
                lookup: "path",
            },
            CompletionItem {
                label: "proc_macro",
                source_range: 2..2,
                delete: 2..2,
                insert: "proc_macro",
                kind: Attribute,
            },
            CompletionItem {
                label: "proc_macro_attribute",
                source_range: 2..2,
                delete: 2..2,
                insert: "proc_macro_attribute",
                kind: Attribute,
            },
            CompletionItem {
                label: "proc_macro_derive(…)",
                source_range: 2..2,
                delete: 2..2,
                insert: "proc_macro_derive(${0:Trait})",
                kind: Attribute,
                lookup: "proc_macro_derive",
            },
            CompletionItem {
                label: "repr(…)",
                source_range: 2..2,
                delete: 2..2,
                insert: "repr(${0:C})",
                kind: Attribute,
                lookup: "repr",
            },
            CompletionItem {
                label: "should_panic(…)",
                source_range: 2..2,
                delete: 2..2,
                insert: "should_panic(expected = \"${0:reason}\")",
                kind: Attribute,
                lookup: "should_panic",
            },
            CompletionItem {
                label: "target_feature = \"…\"",
                source_range: 2..2,
                delete: 2..2,
                insert: "target_feature = \"${0:feature}\"",
                kind: Attribute,
                lookup: "target_feature",
            },
            CompletionItem {
                label: "test",
                source_range: 2..2,
                delete: 2..2,
                insert: "test",
                kind: Attribute,
            },
            CompletionItem {
                label: "used",
                source_range: 2..2,
                delete: 2..2,
                insert: "used",
                kind: Attribute,
            },
            CompletionItem {
                label: "warn(…)",
                source_range: 2..2,
                delete: 2..2,
                insert: "warn(${0:lint})",
                kind: Attribute,
                lookup: "warn",
            },
        ]
        "###
        );
    }

    #[test]
    fn test_attribute_completion_inside_nested_attr() {
        assert_debug_snapshot!(
        do_attr_completion(
                r"
                #[allow(<|>)]
                ",
        ),
            @r###"
        []
        "###
        );
    }

    #[test]
    fn test_inner_attribute_completion() {
        assert_debug_snapshot!(
        do_attr_completion(
                r"
                #![<|>]
                ",
        ),
            @r###"
        [
            CompletionItem {
                label: "allow(…)",
                source_range: 3..3,
                delete: 3..3,
                insert: "allow(${0:lint})",
                kind: Attribute,
                lookup: "allow",
            },
            CompletionItem {
                label: "cfg(…)",
                source_range: 3..3,
                delete: 3..3,
                insert: "cfg(${0:predicate})",
                kind: Attribute,
                lookup: "cfg",
            },
            CompletionItem {
                label: "cfg_attr(…)",
                source_range: 3..3,
                delete: 3..3,
                insert: "cfg_attr(${1:predicate}, ${0:attr})",
                kind: Attribute,
                lookup: "cfg_attr",
            },
            CompletionItem {
                label: "deny(…)",
                source_range: 3..3,
                delete: 3..3,
                insert: "deny(${0:lint})",
                kind: Attribute,
                lookup: "deny",
            },
            CompletionItem {
                label: "deprecated = \"…\"",
                source_range: 3..3,
                delete: 3..3,
                insert: "deprecated = \"${0:reason}\"",
                kind: Attribute,
                lookup: "deprecated",
            },
            CompletionItem {
                label: "derive(…)",
                source_range: 3..3,
                delete: 3..3,
                insert: "derive(${0:Debug})",
                kind: Attribute,
                lookup: "derive",
            },
            CompletionItem {
                label: "doc = \"…\"",
                source_range: 3..3,
                delete: 3..3,
                insert: "doc = \"${0:docs}\"",
                kind: Attribute,
                lookup: "doc",
            },
            CompletionItem {
                label: "feature(…)",
                source_range: 3..3,
                delete: 3..3,
                insert: "feature(${0:flag})",
                kind: Attribute,
                lookup: "feature",
            },
            CompletionItem {
                label: "forbid(…)",
                source_range: 3..3,
                delete: 3..3,
                insert: "forbid(${0:lint})",
                kind: Attribute,
                lookup: "forbid",
            },
            CompletionItem {
                label: "global_allocator",
                source_range: 3..3,
                delete: 3..3,
                insert: "global_allocator",
                kind: Attribute,
            },
            CompletionItem {
                label: "ignore(…)",
                source_range: 3..3,
                delete: 3..3,
                insert: "ignore(${0:lint})",
                kind: Attribute,
                lookup: "ignore",
            },
            CompletionItem {
                label: "inline(…)",
                source_range: 3..3,
                delete: 3..3,
                insert: "inline(${0:lint})",
                kind: Attribute,
                lookup: "inline",
            },
            CompletionItem {
                label: "link",
                source_range: 3..3,
                delete: 3..3,
                insert: "link",
                kind: Attribute,
            },
            CompletionItem {
                label: "link_name = \"…\"",
                source_range: 3..3,
                delete: 3..3,
                insert: "link_name = \"${0:symbol_name}\"",
                kind: Attribute,
                lookup: "link_name",
            },
            CompletionItem {
                label: "macro_export",
                source_range: 3..3,
                delete: 3..3,
                insert: "macro_export",
                kind: Attribute,
            },
            CompletionItem {
                label: "macro_use",
                source_range: 3..3,
                delete: 3..3,
                insert: "macro_use",
                kind: Attribute,
            },
            CompletionItem {
                label: "must_use = \"…\"",
                source_range: 3..3,
                delete: 3..3,
                insert: "must_use = \"${0:reason}\"",
                kind: Attribute,
                lookup: "must_use",
            },
            CompletionItem {
                label: "no_mangle",
                source_range: 3..3,
                delete: 3..3,
                insert: "no_mangle",
                kind: Attribute,
            },
            CompletionItem {
                label: "no_std",
                source_range: 3..3,
                delete: 3..3,
                insert: "no_std",
                kind: Attribute,
            },
            CompletionItem {
                label: "non_exhaustive",
                source_range: 3..3,
                delete: 3..3,
                insert: "non_exhaustive",
                kind: Attribute,
            },
            CompletionItem {
                label: "panic_handler",
                source_range: 3..3,
                delete: 3..3,
                insert: "panic_handler",
                kind: Attribute,
            },
            CompletionItem {
                label: "path = \"…\"",
                source_range: 3..3,
                delete: 3..3,
                insert: "path =\"${0:path}\"",
                kind: Attribute,
                lookup: "path",
            },
            CompletionItem {
                label: "proc_macro",
                source_range: 3..3,
                delete: 3..3,
                insert: "proc_macro",
                kind: Attribute,
            },
            CompletionItem {
                label: "proc_macro_attribute",
                source_range: 3..3,
                delete: 3..3,
                insert: "proc_macro_attribute",
                kind: Attribute,
            },
            CompletionItem {
                label: "proc_macro_derive(…)",
                source_range: 3..3,
                delete: 3..3,
                insert: "proc_macro_derive(${0:Trait})",
                kind: Attribute,
                lookup: "proc_macro_derive",
            },
            CompletionItem {
                label: "recursion_limit = …",
                source_range: 3..3,
                delete: 3..3,
                insert: "recursion_limit = ${0:128}",
                kind: Attribute,
                lookup: "recursion_limit",
            },
            CompletionItem {
                label: "repr(…)",
                source_range: 3..3,
                delete: 3..3,
                insert: "repr(${0:C})",
                kind: Attribute,
                lookup: "repr",
            },
            CompletionItem {
                label: "should_panic(…)",
                source_range: 3..3,
                delete: 3..3,
                insert: "should_panic(expected = \"${0:reason}\")",
                kind: Attribute,
                lookup: "should_panic",
            },
            CompletionItem {
                label: "target_feature = \"…\"",
                source_range: 3..3,
                delete: 3..3,
                insert: "target_feature = \"${0:feature}\"",
                kind: Attribute,
                lookup: "target_feature",
            },
            CompletionItem {
                label: "test",
                source_range: 3..3,
                delete: 3..3,
                insert: "test",
                kind: Attribute,
            },
            CompletionItem {
                label: "used",
                source_range: 3..3,
                delete: 3..3,
                insert: "used",
                kind: Attribute,
            },
            CompletionItem {
                label: "warn(…)",
                source_range: 3..3,
                delete: 3..3,
                insert: "warn(${0:lint})",
                kind: Attribute,
                lookup: "warn",
            },
            CompletionItem {
                label: "windows_subsystem = \"…\"",
                source_range: 3..3,
                delete: 3..3,
                insert: "windows_subsystem = \"${0:subsystem}\"",
                kind: Attribute,
                lookup: "windows_subsystem",
            },
        ]
        "###
        );
    }
}