use ruff_python_semantic::SemanticModel;
use rustc_hash::FxHashSet;

use ruff_macros::{ViolationMetadata, derive_message_formats};
use ruff_python_ast::comparable::ComparableExpr;
use ruff_python_ast::{self as ast, Expr, ExprCall, Stmt};
use ruff_text_size::Ranged;

use crate::Violation;
use crate::checkers::ast::Checker;

/// ## What it does
/// Checks for enums that contain duplicate values.
///
/// ## Why is this bad?
/// Enum values should be unique. Non-unique values are redundant and likely a
/// mistake.
///
/// ## Example
/// ```python
/// from enum import Enum
///
///
/// class Foo(Enum):
///     A = 1
///     B = 2
///     C = 1
/// ```
///
/// Use instead:
/// ```python
/// from enum import Enum
///
///
/// class Foo(Enum):
///     A = 1
///     B = 2
///     C = 3
/// ```
///
/// ## References
/// - [Python documentation: `enum.Enum`](https://docs.python.org/3/library/enum.html#enum.Enum)
#[derive(ViolationMetadata)]
pub(crate) struct NonUniqueEnums {
    value: String,
}

impl Violation for NonUniqueEnums {
    #[derive_message_formats]
    fn message(&self) -> String {
        let NonUniqueEnums { value } = self;
        format!("Enum contains duplicate value: `{value}`")
    }
}

/// PIE796
pub(crate) fn non_unique_enums(checker: &Checker, parent: &Stmt, body: &[Stmt]) {
    let semantic = checker.semantic();

    let Stmt::ClassDef(parent) = parent else {
        return;
    };

    if !parent.bases().iter().any(|expr| {
        semantic
            .resolve_qualified_name(expr)
            .is_some_and(|qualified_name| matches!(qualified_name.segments(), ["enum", "Enum"]))
    }) {
        return;
    }

    let mut seen_targets: FxHashSet<ComparableExpr> = FxHashSet::default();
    for stmt in body {
        let Stmt::Assign(ast::StmtAssign { value, .. }) = stmt else {
            continue;
        };

        if is_call_to_enum_auto(semantic, value) {
            continue;
        } else if let Expr::Tuple(ast::ExprTuple { elts, .. }) = value.as_ref() {
            if elts.iter().any(|elt| is_call_to_enum_auto(semantic, elt)) {
                continue;
            }
        }

        if checker.source_type.is_stub() && member_has_unknown_value(semantic, value) {
            continue;
        }

        let comparable = ComparableExpr::from(value);

        if !seen_targets.insert(comparable) {
            checker.report_diagnostic(
                NonUniqueEnums {
                    value: checker.generator().expr(value),
                },
                stmt.range(),
            );
        }
    }
}

fn is_call_to_enum_auto(semantic: &SemanticModel, expr: &Expr) -> bool {
    expr.as_call_expr().is_some_and(|call| {
        semantic
            .resolve_qualified_name(&call.func)
            .is_some_and(|qualified_name| matches!(qualified_name.segments(), ["enum", "auto"]))
    })
}

/// Whether the value is a bare ellipsis literal (`A = ...`)
/// or a casted one (`A = cast(SomeType, ...)`).
fn member_has_unknown_value(semantic: &SemanticModel, expr: &Expr) -> bool {
    match expr {
        Expr::EllipsisLiteral(_) => true,

        Expr::Call(ExprCall {
            func, arguments, ..
        }) => {
            if !semantic.match_typing_expr(func, "cast") {
                return false;
            }

            if !arguments.keywords.is_empty() {
                return false;
            }

            matches!(arguments.args.as_ref(), [_, Expr::EllipsisLiteral(_)])
        }

        _ => false,
    }
}
