-module(elvis_ruleset).

-elvis([
    {elvis_style, abc_size, #{ignore => [{elvis_ruleset, elvis_style_rules, 0}]}},
    {elvis_style, code_complexity, #{ignore => [{elvis_ruleset, elvis_style_rules, 0}]}}
]).
-format(#{inline_items => none}).

-export([rules/1, load_custom/1, drop_custom/0, is_defined/1, custom_names/0]).

-spec load_custom(#{atom() => list()}) -> ok.
load_custom(RulesetNames) ->
    case custom_names() of
        [] ->
            _ = elvis_utils:debug("loading custom rulesets into state", []),
            lists:foreach(
                fun({RulesetName, NSNameDefs}) ->
                    Rules = [elvis_rule:from_tuple(NSNameDef) || NSNameDef <- NSNameDefs],
                    ok = persistent_term:put({elvis_ruleset, RulesetName}, Rules)
                end,
                maps:to_list(RulesetNames)
            );
        _ ->
            ok
    end.

-spec custom_names() -> [atom()].
custom_names() ->
    [RulesetName || {{elvis_ruleset, RulesetName}, _} <- persistent_term:get()].

-spec drop_custom() -> ok.
drop_custom() ->
    [persistent_term:erase(K) || {{elvis_ruleset, _} = K, _} <- persistent_term:get()],
    ok.

-spec is_defined(atom()) -> boolean().
is_defined(RulesetName) ->
    rules(RulesetName) =/= [].

-spec rules(Group :: atom()) -> [elvis_rule:t()].
rules(gitignore) ->
    gitignore_rules();
rules(hrl_files) ->
    hrl_files_rules();
rules(hrl_files_strict) ->
    hrl_files_strict_rules();
rules(erl_files) ->
    erl_files_rules();
rules(erl_files_strict) ->
    erl_files_strict_rules();
rules(beam_files) ->
    beam_files_rules();
rules(beam_files_strict) ->
    beam_files_strict_rules();
rules(rebar_config) ->
    rebar_config_rules();
rules(erl_files_test) ->
    erl_files_test_rules();
rules(RulesetName) ->
    persistent_term:get({elvis_ruleset, RulesetName}, []).

gitignore_rules() ->
    [
        elvis_rule:new(elvis_gitignore, forbidden_patterns),
        elvis_rule:new(elvis_gitignore, required_patterns)
    ].

hrl_files_rules() ->
    trim(erl_files_rules() ++ hrl_only_files_rules(), doesnt_work_on_hrl_files()).

hrl_only_files_rules() ->
    [
        elvis_rule:new(elvis_style, no_includes),
        elvis_rule:new(elvis_style, no_specs),
        elvis_rule:new(elvis_style, no_types)
    ].

hrl_files_strict_rules() ->
    trim(erl_files_strict_rules(), doesnt_work_on_hrl_files()).

doesnt_work_on_hrl_files() ->
    [
        elvis_rule:new(elvis_style, abc_size),
        elvis_rule:new(elvis_style, behaviour_spelling),
        elvis_rule:new(elvis_style, code_complexity),
        elvis_rule:new(elvis_style, dont_repeat_yourself),
        elvis_rule:new(elvis_style, export_used_types),
        elvis_rule:new(elvis_style, function_naming_convention),
        elvis_rule:new(elvis_style, no_god_modules),
        elvis_rule:new(elvis_style, no_invalid_dynamic_calls),
        elvis_rule:new(elvis_style, max_anonymous_function_arity),
        elvis_rule:new(elvis_style, max_anonymous_function_clause_length),
        elvis_rule:new(elvis_style, max_anonymous_function_length),
        elvis_rule:new(elvis_style, max_function_clause_length),
        elvis_rule:new(elvis_style, max_function_length),
        elvis_rule:new(elvis_style, ms_transform_included),
        elvis_rule:new(elvis_style, no_call),
        elvis_rule:new(elvis_style, no_common_caveats_call),
        elvis_rule:new(elvis_style, no_init_lists),
        elvis_rule:new(elvis_style, no_macros),
        elvis_rule:new(elvis_style, no_spec_with_records),
        elvis_rule:new(elvis_style, param_pattern_matching),
        elvis_rule:new(elvis_style, private_data_types),
        elvis_rule:new(elvis_style, state_record_and_type)
    ].

erl_files_rules() ->
    elvis_style_rules() ++ elvis_text_style_rules().

elvis_style_rules() ->
    [
        elvis_rule:new(elvis_style, atom_naming_convention),
        elvis_rule:new(elvis_style, behaviour_spelling),
        elvis_rule:new(elvis_style, consistent_variable_naming),
        elvis_rule:new(elvis_style, dont_repeat_yourself),
        elvis_rule:new(elvis_style, export_used_types),
        elvis_rule:new(elvis_style, function_naming_convention),
        elvis_rule:new(elvis_style, no_god_modules),
        elvis_rule:new(elvis_style, no_invalid_dynamic_calls),
        elvis_rule:new(elvis_style, macro_naming_convention),
        elvis_rule:new(elvis_style, max_anonymous_function_arity),
        elvis_rule:new(elvis_style, max_function_arity),
        elvis_rule:new(elvis_style, module_naming_convention),
        elvis_rule:new(elvis_style, no_deep_nesting),
        elvis_rule:new(elvis_style, no_author),
        elvis_rule:new(elvis_style, no_behavior_info),
        elvis_rule:new(elvis_style, no_block_expressions),
        elvis_rule:new(elvis_style, no_boolean_in_comparison),
        elvis_rule:new(elvis_style, no_catch_expressions),
        elvis_rule:new(elvis_style, no_debug_call),
        elvis_rule:new(elvis_style, no_dollar_space),
        elvis_rule:new(elvis_style, no_if_expression),
        elvis_rule:new(elvis_style, no_import),
        elvis_rule:new(elvis_style, no_match_in_condition),
        elvis_rule:new(elvis_style, no_nested_try_catch),
        elvis_rule:new(elvis_style, no_operation_on_same_value),
        elvis_rule:new(elvis_style, expression_can_be_simplified),
        elvis_rule:new(elvis_style, no_receive_without_timeout),
        elvis_rule:new(elvis_style, no_single_clause_case),
        elvis_rule:new(elvis_style, no_single_match_maybe),
        elvis_rule:new(elvis_style, no_space_after_pound),
        elvis_rule:new(elvis_style, no_space),
        elvis_rule:new(elvis_style, no_spec_with_records),
        elvis_rule:new(elvis_style, no_successive_maps),
        elvis_rule:new(elvis_style, no_throw),
        elvis_rule:new(elvis_style, numeric_format),
        elvis_rule:new(elvis_style, operator_spaces),
        elvis_rule:new(elvis_style, param_pattern_matching),
        elvis_rule:new(elvis_style, prefer_robot_butt),
        elvis_rule:new(elvis_style, private_data_types),
        elvis_rule:new(elvis_style, no_used_ignored_variables),
        elvis_rule:new(elvis_style, variable_naming_convention),
        elvis_rule:new(elvis_style, guard_operators),
        elvis_rule:new(elvis_style, simplify_anonymous_functions)
    ].

erl_files_test_rules() ->
    trim(
        elvis_style_rules(),
        [
            elvis_rule:new(elvis_style, dont_repeat_yourself),
            elvis_rule:new(elvis_style, no_god_modules)
        ]
    ).

elvis_text_style_rules() ->
    [
        elvis_rule:new(elvis_text_style, max_line_length),
        elvis_rule:new(elvis_text_style, no_tabs),
        elvis_rule:new(elvis_text_style, no_trailing_whitespace)
    ].

erl_files_strict_rules() ->
    erl_files_rules() ++ elvis_style_stricter_rules() ++ elvis_text_style_stricter_rules().

elvis_style_stricter_rules() ->
    [
        elvis_rule:new(elvis_style, abc_size),
        elvis_rule:new(elvis_style, always_shortcircuit),
        elvis_rule:new(elvis_style, code_complexity),
        elvis_rule:new(elvis_style, consistent_ok_error_spec),
        elvis_rule:new(elvis_style, generic_type),
        elvis_rule:new(elvis_style, max_anonymous_function_clause_length),
        elvis_rule:new(elvis_style, max_anonymous_function_length),
        elvis_rule:new(elvis_style, max_function_clause_length),
        elvis_rule:new(elvis_style, max_function_length),
        elvis_rule:new(elvis_style, max_module_length),
        elvis_rule:new(elvis_style, max_record_fields),
        elvis_rule:new(elvis_style, max_map_type_keys),
        elvis_rule:new(elvis_style, ms_transform_included),
        elvis_rule:new(elvis_style, no_call),
        elvis_rule:new(elvis_style, no_common_caveats_call),
        elvis_rule:new(elvis_style, no_init_lists),
        elvis_rule:new(elvis_style, no_macros),
        elvis_rule:new(elvis_style, prefer_strict_generators),
        elvis_rule:new(elvis_style, prefer_unquoted_atoms),
        elvis_rule:new(elvis_style, state_record_and_type),
        elvis_rule:new(elvis_style, prefer_sigils),
        elvis_rule:new(elvis_style, prefer_include),
        elvis_rule:new(elvis_style, strict_term_equivalence),
        elvis_rule:new(elvis_style, macro_definition_parentheses)
    ].

elvis_text_style_stricter_rules() ->
    [
        elvis_rule:new(elvis_text_style, no_redundant_blank_lines)
    ].

beam_files_rules() ->
    trim(erl_files_rules(), doesnt_work_on_beam_files()).

beam_files_strict_rules() ->
    trim(erl_files_strict_rules(), doesnt_work_on_beam_files()).

doesnt_work_on_beam_files() ->
    not_on_beam() ++ elvis_text_style_rules() ++ elvis_text_style_stricter_rules().

not_on_beam() ->
    [
        elvis_rule:new(elvis_style, abc_size),
        elvis_rule:new(elvis_style, code_complexity),
        elvis_rule:new(elvis_style, macro_naming_convention),
        elvis_rule:new(elvis_style, max_anonymous_function_clause_length),
        elvis_rule:new(elvis_style, max_anonymous_function_length),
        elvis_rule:new(elvis_style, max_function_clause_length),
        elvis_rule:new(elvis_style, max_function_length),
        elvis_rule:new(elvis_style, max_module_length),
        elvis_rule:new(elvis_style, no_dollar_space),
        elvis_rule:new(elvis_style, no_macros),
        elvis_rule:new(elvis_style, no_space_after_pound),
        elvis_rule:new(elvis_style, no_space),
        elvis_rule:new(elvis_style, numeric_format),
        elvis_rule:new(elvis_style, operator_spaces),
        elvis_rule:new(elvis_style, prefer_sigils)
    ].

rebar_config_rules() ->
    [
        elvis_rule:new(elvis_project, no_branch_deps),
        elvis_rule:new(elvis_project, protocol_for_deps)
    ].

trim(Ruleset, RulesToRemove) ->
    lists:filter(
        fun(Rule) ->
            lists:all(
                fun(RuleToRemove) ->
                    not elvis_rule:same(RuleToRemove, Rule)
                end,
                RulesToRemove
            )
        end,
        Ruleset
    ).