use clippy_utils::diagnostics::span_lint_and_then;
use clippy_utils::macros::MacroCall;
use clippy_utils::source::{expand_past_previous_comma, snippet_opt};
use clippy_utils::{span_extract_comments, sym};
use rustc_ast::{FormatArgs, FormatArgsPiece};
use rustc_errors::Applicability;
use rustc_lint::{LateContext, LintContext};

use super::{PRINTLN_EMPTY_STRING, WRITELN_EMPTY_STRING};

pub(super) fn check(cx: &LateContext<'_>, format_args: &FormatArgs, macro_call: &MacroCall, name: &str) {
    if let [FormatArgsPiece::Literal(sym::LF)] = &format_args.template[..] {
        let is_writeln = name == "writeln";

        span_lint_and_then(
            cx,
            if is_writeln {
                WRITELN_EMPTY_STRING
            } else {
                PRINTLN_EMPTY_STRING
            },
            macro_call.span,
            format!("empty string literal in `{name}!`"),
            |diag| {
                if span_extract_comments(cx, macro_call.span).is_empty() {
                    let closing_delim = snippet_opt(cx, macro_call.span)
                        .and_then(|snippet| snippet.chars().last())
                        .filter(|ch| matches!(ch, ')' | ']' | '}'));

                    if let Some(closing_delim) = closing_delim {
                        let closing_paren = cx.sess().source_map().span_extend_to_prev_char_before(
                            macro_call.span.shrink_to_hi(),
                            closing_delim,
                            false,
                        );
                        let mut span = format_args.span.with_hi(closing_paren.lo());
                        if is_writeln {
                            span = expand_past_previous_comma(cx, span);
                        }

                        diag.span_suggestion(span, "remove the empty string", "", Applicability::MachineApplicable);
                    }
                } else {
                    // If there is a comment in the span of macro call, we don't provide an auto-fix suggestion.
                    diag.span_note(format_args.span, "remove the empty string");
                }
            },
        );
    }
}