sqc 0.4.13

use super::super::{CertRule, RuleViolation};
use crate::manifest::{RuleCategory, Severity};
use crate::utility::cert_c::ast_utils::get_node_text;
use tree_sitter::Node;

pub struct Exp00C;

impl CertRule for Exp00C {
    fn rule_id(&self) -> &'static str {
        "EXP00-C"
    }

    fn description(&self) -> &'static str {
        "Use parentheses for precedence of operation"
    }

    fn severity(&self) -> Severity {
        Severity::Low
    }

    fn category(&self) -> RuleCategory {
        RuleCategory::Recommendation
    }

    fn cert_id(&self) -> &'static str {
        "EXP00-C"
    }

    fn check(&self, node: &Node, source: &str) -> Vec<RuleViolation> {
        let mut violations = Vec::new();

        // Check for binary expressions with precedence issues
        if node.kind() == "binary_expression" {
            if let Some(operator_node) = node.child_by_field_name("operator") {
                let operator = get_node_text(&operator_node, source);

                // Check if this is a comparison operator
                if is_comparison_operator(operator) {
                    // Check left operand for bitwise operators
                    if let Some(left) = node.child_by_field_name("left") {
                        if has_unparenthesized_bitwise_operator(&left, source) {
                            report_violation(&left, source, &mut violations);
                        }
                    }

                    // Check right operand for bitwise operators
                    if let Some(right) = node.child_by_field_name("right") {
                        if has_unparenthesized_bitwise_operator(&right, source) {
                            report_violation(&right, source, &mut violations);
                        }
                    }
                }
                // Check if this is a bitwise operator with comparison in operands or constant 0
                else if is_bitwise_operator(operator) {
                    // Check if either operand contains a comparison operator
                    if let Some(left) = node.child_by_field_name("left") {
                        if contains_comparison_operator(&left, source) {
                            report_violation(node, source, &mut violations);
                        }
                    }
                    if let Some(right) = node.child_by_field_name("right") {
                        if contains_comparison_operator(&right, source) {
                            report_violation(node, source, &mut violations);
                        }
                    }

                    // Check for bitwise AND/OR/XOR with constant 0
                    if operator == "&" || operator == "|" || operator == "^" {
                        if let Some(right) = node.child_by_field_name("right") {
                            let right_text = get_node_text(&right, source).trim();
                            if right_text == "0" {
                                report_violation(node, source, &mut violations);
                            }
                        }
                    }
                }
            }
        }

        // Recursively check child nodes
        for i in 0..node.child_count() {
            if let Some(child) = node.child(i) {
                violations.extend(self.check(&child, source));
            }
        }

        violations
    }
}

/// Check if the operator is a comparison operator
fn is_comparison_operator(op: &str) -> bool {
    matches!(op, "==" | "!=" | "<" | ">" | "<=" | ">=")
}

/// Check if the operator is a low-precedence bitwise operator
fn is_bitwise_operator(op: &str) -> bool {
    matches!(op, "&" | "|" | "^" | "<<" | ">>")
}

/// Check if a node contains an unparenthesized bitwise operator
fn has_unparenthesized_bitwise_operator(node: &Node, source: &str) -> bool {
    match node.kind() {
        "binary_expression" => {
            if let Some(operator_node) = node.child_by_field_name("operator") {
                let operator = get_node_text(&operator_node, source);
                is_bitwise_operator(operator)
            } else {
                false
            }
        }
        "parenthesized_expression" => {
            // If it's already parenthesized, it's safe
            false
        }
        _ => false,
    }
}

/// Check if a node or its children contain a comparison operator
fn contains_comparison_operator(node: &Node, source: &str) -> bool {
    match node.kind() {
        "binary_expression" => {
            if let Some(operator_node) = node.child_by_field_name("operator") {
                let operator = get_node_text(&operator_node, source);
                if is_comparison_operator(operator) {
                    return true;
                }
            }
            // Also check children recursively
            for i in 0..node.child_count() {
                if let Some(child) = node.child(i) {
                    if contains_comparison_operator(&child, source) {
                        return true;
                    }
                }
            }
            false
        }
        "parenthesized_expression" => {
            // Look inside parenthesized expressions to find comparisons
            for i in 0..node.child_count() {
                if let Some(child) = node.child(i) {
                    if contains_comparison_operator(&child, source) {
                        return true;
                    }
                }
            }
            false
        }
        _ => {
            // Check children for other node types
            for i in 0..node.child_count() {
                if let Some(child) = node.child(i) {
                    if contains_comparison_operator(&child, source) {
                        return true;
                    }
                }
            }
            false
        }
    }
}

/// Report a violation for operator precedence issues
fn report_violation(node: &Node, source: &str, violations: &mut Vec<RuleViolation>) {
    let start_point = node.start_position();
    let node_text = get_node_text(node, source);

    violations.push(RuleViolation {
        rule_id: "EXP00-C".to_string(),
        severity: Severity::Low,
        message: format!(
            "Use parentheses to clarify operator precedence: '{}'",
            node_text
        ),
        file_path: String::new(),
        line: start_point.row + 1,
        column: start_point.column + 1,
        suggestion: Some("Add parentheses to enforce the intended order of operations".to_string()),
        ..Default::default()
    });
}