#!/usr/bin/env python
#
# ===- check-analyzer-fixit.py - Static Analyzer test helper ---*- python -*-===#
#
# Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
# See https://llvm.org/LICENSE.txt for license information.
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
#
# ===------------------------------------------------------------------------===#
#
# This file copy-pasted mostly from the Clang-Tidy's 'check_clang_tidy.py'.
#
# ===------------------------------------------------------------------------===#

r"""
Clang Static Analyzer test helper
=================================

This script runs the Analyzer in fix-it mode and verify fixes, warnings, notes.

Usage:
  check-analyzer-fixit.py <source-file> <temp-file> [analyzer arguments]

Example:
  // RUN: %check-analyzer-fixit %s %t -analyzer-checker=core
"""

import argparse
import os
import re
import subprocess
import sys


def write_file(file_name, text):
    with open(file_name, "w") as f:
        f.write(text)


def run_test_once(args, extra_args):
    input_file_name = args.input_file_name
    temp_file_name = args.temp_file_name
    clang_analyzer_extra_args = extra_args

    file_name_with_extension = input_file_name
    _, extension = os.path.splitext(file_name_with_extension)
    if extension not in [".c", ".hpp", ".m", ".mm"]:
        extension = ".cpp"
    temp_file_name = temp_file_name + extension

    with open(input_file_name, "r") as input_file:
        input_text = input_file.read()

    # Remove the contents of the CHECK lines to avoid CHECKs matching on
    # themselves.  We need to keep the comments to preserve line numbers while
    # avoiding empty lines which could potentially trigger formatting-related
    # checks.
    cleaned_test = re.sub("// *CHECK-[A-Z0-9\-]*:[^\r\n]*", "//", input_text)
    write_file(temp_file_name, cleaned_test)

    original_file_name = temp_file_name + ".orig"
    write_file(original_file_name, cleaned_test)

    try:
        builtin_include_dir = subprocess.check_output(
            ["clang", "-print-file-name=include"], stderr=subprocess.STDOUT
        ).decode()
    except subprocess.CalledProcessError as e:
        print("Cannot print Clang include directory: " + e.output.decode())

    builtin_include_dir = os.path.normpath(builtin_include_dir)

    args = (
        [
            "clang",
            "-cc1",
            "-internal-isystem",
            builtin_include_dir,
            "-nostdsysteminc",
            "-analyze",
            "-analyzer-constraints=range",
            "-analyzer-config",
            "apply-fixits=true",
        ]
        + clang_analyzer_extra_args
        + ["-verify", temp_file_name]
    )

    print("Running " + str(args) + "...")

    try:
        clang_analyzer_output = subprocess.check_output(
            args, stderr=subprocess.STDOUT
        ).decode()
    except subprocess.CalledProcessError as e:
        print("Clang Static Analyzer test failed:\n" + e.output.decode())
        raise

    print(
        "----------------- Clang Static Analyzer output -----------------\n"
        + clang_analyzer_output
        + "\n--------------------------------------------------------------"
    )

    try:
        diff_output = subprocess.check_output(
            ["diff", "-u", original_file_name, temp_file_name], stderr=subprocess.STDOUT
        )
    except subprocess.CalledProcessError as e:
        diff_output = e.output

    print(
        "----------------------------- Fixes ----------------------------\n"
        + diff_output.decode()
        + "\n--------------------------------------------------------------"
    )

    try:
        subprocess.check_output(
            [
                "FileCheck",
                "-input-file=" + temp_file_name,
                input_file_name,
                "-check-prefixes=CHECK-FIXES",
                "-strict-whitespace",
            ],
            stderr=subprocess.STDOUT,
        )
    except subprocess.CalledProcessError as e:
        print("FileCheck failed:\n" + e.output.decode())
        raise


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("input_file_name")
    parser.add_argument("temp_file_name")

    args, extra_args = parser.parse_known_args()
    run_test_once(args, extra_args)


if __name__ == "__main__":
    main()