use rustc_hash::FxHashMap;

use ruff_macros::{ViolationMetadata, derive_message_formats};
use ruff_python_ast::Stmt;
use ruff_python_semantic::{
    Binding, BindingKind, Imported, NodeId, Scope, ScopeId, SemanticModel, ShadowedBinding,
    analyze::{typing::is_type_checking_block, visibility},
};
use ruff_source_file::SourceRow;
use ruff_text_size::Ranged;

use crate::checkers::ast::Checker;
use crate::fix::edits;
use crate::preview::{
    is_annotated_assignment_redefinition_enabled, is_f811_shadowing_in_type_checking_enabled,
};
use crate::{Fix, FixAvailability, Violation};

/// ## What it does
/// Checks for variable definitions that redefine (or "shadow") unused
/// variables.
///
/// ## Why is this bad?
/// Redefinitions of unused names are unnecessary and often indicative of a
/// mistake.
///
/// ## Example
///
/// ```python
/// import foo
/// import bar
/// import foo  # Redefinition of unused `foo` from line 1
/// ```
///
/// Use instead:
///
/// ```python
/// import foo
/// import bar
/// ```
///
/// ## Preview
/// When [preview] is enabled, this rule also flags annotated variable
/// redeclarations. For example, `bar: int = 1` followed by `bar: int = 2`
/// will be flagged as a redefinition of an unused variable, whereas plain
/// reassignments like `bar = 1` followed by `bar = 2` remain unflagged.
///
/// In preview, this rule will also flag redefinitions in `typing.TYPE_CHECKING` blocks:
///
/// ```python
/// import typing
/// from collections.abc import Sequence
///
/// if typing.TYPE_CHECKING:
///     # Redefinition of unused `Sequence` from line 2
///     from collections.abc import Sequence
/// ```
///
/// Use instead:
///
/// ```python
/// from collections.abc import Sequence
/// ```
///
/// [preview]: https://docs.astral.sh/ruff/preview/
///
/// ## Options
///
/// This rule ignores dummy variables, as determined by:
///
/// - `lint.dummy-variable-rgx`
#[derive(ViolationMetadata)]
#[violation_metadata(stable_since = "v0.0.171")]
pub(crate) struct RedefinedWhileUnused {
    pub name: String,
    pub row: SourceRow,
}

impl Violation for RedefinedWhileUnused {
    const FIX_AVAILABILITY: FixAvailability = FixAvailability::Sometimes;

    #[derive_message_formats]
    fn message(&self) -> String {
        let RedefinedWhileUnused { name, row } = self;
        format!("Redefinition of unused `{name}` from {row}")
    }

    fn fix_title(&self) -> Option<String> {
        let RedefinedWhileUnused { name, .. } = self;
        Some(format!("Remove definition: `{name}`"))
    }
}

/// F811
pub(crate) fn redefined_while_unused(checker: &Checker, scope_id: ScopeId, scope: &Scope) {
    // Index the redefined bindings by statement.
    let mut redefinitions: FxHashMap<Option<NodeId>, Vec<EntryInfo<'_>>> = FxHashMap::default();

    for (name, binding_id) in scope.bindings() {
        for shadow in checker.semantic().shadowed_bindings(scope_id, binding_id) {
            // If the shadowing binding is a loop variable, abort, to avoid overlap
            // with F402.
            let binding = &checker.semantic().bindings[shadow.binding_id()];
            if binding.kind.is_loop_var() {
                continue;
            }

            // If the shadowed binding is used, abort.
            let shadowed = &checker.semantic().bindings[shadow.shadowed_id()];
            if shadowed.is_used() {
                continue;
            }

            // If the shadowing binding isn't considered a "redefinition" of the
            // shadowed binding, abort — unless both are annotated assignments
            // and preview mode is enabled (see #23802).
            if !binding.redefines(shadowed) {
                if !(is_annotated_assignment_redefinition_enabled(checker.settings())
                    && is_annotated_assignment(binding, checker.semantic())
                    && is_annotated_assignment(shadowed, checker.semantic()))
                {
                    continue;
                }
            }

            if !is_f811_shadowing_in_type_checking_enabled(checker.settings()) {
                let shadowed_in_type_checking = shadowed
                    .source
                    .is_some_and(|source| is_in_type_checking_block(checker, source));
                let binding_in_type_checking = binding
                    .source
                    .is_some_and(|source| is_in_type_checking_block(checker, source));

                if (shadowed_in_type_checking || binding_in_type_checking)
                    && !(shadowed_in_type_checking && binding_in_type_checking)
                {
                    continue;
                }
            }

            if shadow.same_scope() {
                // If the symbol is a dummy variable, abort, unless the shadowed
                // binding is an import.
                if !matches!(
                    shadowed.kind,
                    BindingKind::Import(..)
                        | BindingKind::FromImport(..)
                        | BindingKind::SubmoduleImport(..)
                        | BindingKind::FutureImport
                ) && checker.settings().dummy_variable_rgx.is_match(name)
                {
                    continue;
                }

                let Some(node_id) = shadowed.source else {
                    continue;
                };

                // If this is an overloaded function, abort.
                if shadowed.kind.is_function_definition() {
                    if checker
                        .semantic()
                        .statement(node_id)
                        .as_function_def_stmt()
                        .is_some_and(|function| {
                            visibility::is_overload(&function.decorator_list, checker.semantic())
                        })
                    {
                        continue;
                    }
                }
            } else {
                // A binding in a class body creates a class attribute; it doesn't rebind names
                // in an enclosing module or function scope.
                if scope.kind.is_class() {
                    continue;
                }

                // Only enforce cross-scope shadowing for imports.
                if !matches!(
                    shadowed.kind,
                    BindingKind::Import(..)
                        | BindingKind::FromImport(..)
                        | BindingKind::SubmoduleImport(..)
                        | BindingKind::FutureImport
                ) {
                    continue;
                }

                // Don't flag explicit re-exports (e.g., `from x import y as y`).
                // A binding in a nested scope (like a class attribute) doesn't
                // invalidate a module-level re-export.
                if shadowed.is_explicit_export() {
                    continue;
                }
            }

            // If the bindings are in different forks, abort.
            if shadowed.source.is_none_or(|left| {
                binding
                    .source
                    .is_none_or(|right| bindings_in_different_forks(checker, &shadow, left, right))
            }) {
                continue;
            }

            redefinitions
                .entry(binding.source)
                .or_default()
                .push(EntryInfo::new(shadowed, binding));
        }
    }

    // Create a fix for each source statement.
    let mut fixes = FxHashMap::default();
    #[expect(
        clippy::iter_over_hash_type,
        reason = "each source statement gets an independent fix entry"
    )]
    for entries in redefinitions.values() {
        let Some(source) = entries.iter().find_map(|info| info.runtime_import) else {
            continue;
        };

        let member_names = entries
            .iter()
            .filter_map(|info| {
                if let Some(shadowed_import) = info.shadowed.as_any_import() {
                    if let Some(import) = info.binding.as_any_import() {
                        if shadowed_import.qualified_name() == import.qualified_name() {
                            return Some(import.member_name());
                        }
                    }
                }
                None
            })
            .collect::<Vec<_>>();

        if !member_names.is_empty() {
            let statement = checker.semantic().statement(source);
            let parent = checker.semantic().parent_statement(source);
            let Ok(edit) = edits::remove_unused_imports(
                member_names.iter().map(AsRef::as_ref),
                statement,
                parent,
                checker.locator(),
                checker.stylist(),
                checker.indexer(),
            ) else {
                continue;
            };

            fixes.insert(
                source,
                Fix::safe_edit(edit).isolate(Checker::isolation(
                    checker.semantic().parent_statement_id(source),
                )),
            );
        }
    }

    // Create diagnostics for each statement.
    #[expect(
        clippy::iter_over_hash_type,
        reason = "iteration order does not affect the diagnostics or fixes produced"
    )]
    for entries in redefinitions.values() {
        for info in entries {
            let name = info.binding.name(checker.source());

            let mut diagnostic = checker.report_diagnostic(
                RedefinedWhileUnused {
                    name: name.to_string(),
                    row: checker.compute_source_row(info.shadowed.start()),
                },
                info.binding.range(),
            );
            diagnostic.add_primary_tag(ruff_db::diagnostic::DiagnosticTag::Unnecessary);

            diagnostic.secondary_annotation(
                format_args!("previous definition of `{name}` here"),
                info.shadowed,
            );

            diagnostic.set_primary_annotation_message(format_args!("`{name}` redefined here"));

            if let Some(range) = info.binding.parent_range(checker.semantic()) {
                diagnostic.set_parent(range.start());
            }

            if let Some(source) = info.runtime_import {
                if let Some(fix) = fixes.get(&source) {
                    diagnostic.set_fix(fix.clone());
                }
            }
        }
    }
}

fn is_annotated_assignment(binding: &Binding, semantic: &SemanticModel) -> bool {
    binding
        .statement(semantic)
        .is_some_and(Stmt::is_ann_assign_stmt)
}

struct EntryInfo<'a> {
    shadowed: &'a Binding<'a>,
    binding: &'a Binding<'a>,
    runtime_import: Option<NodeId>,
}

impl<'a> EntryInfo<'a> {
    fn new(shadowed: &'a Binding<'a>, binding: &'a Binding<'a>) -> Self {
        let runtime_import = binding.source.or(shadowed.source);

        Self {
            shadowed,
            binding,
            runtime_import,
        }
    }
}

fn bindings_in_different_forks(
    checker: &Checker,
    shadow: &ShadowedBinding,
    left: NodeId,
    right: NodeId,
) -> bool {
    let left_ = is_in_type_checking_block(checker, left);
    let right_ = is_in_type_checking_block(checker, right);

    if left_ ^ right_ {
        let left_binding = &checker.semantic().bindings[shadow.shadowed_id()];
        let right_binding = &checker.semantic().bindings[shadow.binding_id()];

        if left_binding.scope == right_binding.scope
            && let (Some(left_import), Some(right_import)) =
                (left_binding.as_any_import(), right_binding.as_any_import())
            && left_import.qualified_name() == right_import.qualified_name()
        {
            let (runtime_import, type_checking_import) =
                if left_ { (right, left) } else { (left, right) };
            return !(checker
                .semantic()
                .dominates(runtime_import, type_checking_import)
                && checker.semantic().statement(runtime_import).start()
                    < checker.semantic().statement(type_checking_import).start());
        }
    }

    !checker.semantic().same_branch(left, right)
}

#[inline]
fn is_in_type_checking_block(checker: &Checker, node_id: NodeId) -> bool {
    checker.semantic().statements(node_id).any(|stmt| {
        stmt.as_if_stmt()
            .is_some_and(|if_stmt| is_type_checking_block(if_stmt, checker.semantic()))
    })
}
