#!/usr/bin/env python3

import os
import logging
import argparse
import csv


def process_result(result_folder):
    status = "success"
    description = ""
    test_results = []

    duplicate_log_path = "{}/duplicate_output.txt".format(result_folder)
    if not os.path.exists(duplicate_log_path):
        logging.info("No header duplicates check log on path %s", duplicate_log_path)
        return "exception", "No header duplicates check log", []
    elif os.stat(duplicate_log_path).st_size != 0:
        description += " Header duplicates check failed. "
        test_results.append(("Header duplicates check", "FAIL"))
        status = "failure"
    else:
        test_results.append(("Header duplicates check", "OK"))

    shellcheck_log_path = "{}/shellcheck_output.txt".format(result_folder)
    if not os.path.exists(shellcheck_log_path):
        logging.info("No shellcheck  log on path %s", shellcheck_log_path)
        return "exception", "No shellcheck log", []
    elif os.stat(shellcheck_log_path).st_size != 0:
        description += " Shellcheck check failed. "
        test_results.append(("Shellcheck ", "FAIL"))
        status = "failure"
    else:
        test_results.append(("Shellcheck", "OK"))

    style_log_path = "{}/style_output.txt".format(result_folder)
    if not os.path.exists(style_log_path):
        logging.info("No style check log on path %s", style_log_path)
        return "exception", "No style check log", []
    elif os.stat(style_log_path).st_size != 0:
        description += "Style check failed. "
        test_results.append(("Style check", "FAIL"))
        status = "failure"
    else:
        test_results.append(("Style check", "OK"))

    typos_log_path = "{}/typos_output.txt".format(result_folder)
    if not os.path.exists(typos_log_path):
        logging.info("No typos check log on path %s", typos_log_path)
        return "exception", "No typos check log", []
    elif os.stat(typos_log_path).st_size != 0:
        description += "Typos check failed. "
        test_results.append(("Typos check", "FAIL"))
        status = "failure"
    else:
        test_results.append(("Typos check", "OK"))

    whitespaces_log_path = "{}/whitespaces_output.txt".format(result_folder)
    if not os.path.exists(whitespaces_log_path):
        logging.info("No whitespaces check log on path %s", whitespaces_log_path)
        return "exception", "No whitespaces check log", []
    elif os.stat(whitespaces_log_path).st_size != 0:
        description += "Whitespaces check failed. "
        test_results.append(("Whitespaces check", "FAIL"))
        status = "failure"
    else:
        test_results.append(("Whitespaces check", "OK"))

    workflows_log_path = "{}/workflows_output.txt".format(result_folder)
    if not os.path.exists(workflows_log_path):
        logging.info("No workflows check log on path %s", style_log_path)
        return "exception", "No workflows check log", []
    elif os.stat(whitespaces_log_path).st_size != 0:
        description += "Workflows check failed. "
        test_results.append(("Workflows check", "FAIL"))
        status = "failure"
    else:
        test_results.append(("Workflows check", "OK"))

    if not description:
        description += "Style check success"

    return status, description, test_results


def write_results(results_file, status_file, results, status):
    with open(results_file, "w") as f:
        out = csv.writer(f, delimiter="\t")
        out.writerows(results)
    with open(status_file, "w") as f:
        out = csv.writer(f, delimiter="\t")
        out.writerow(status)


if __name__ == "__main__":
    logging.basicConfig(level=logging.INFO, format="%(asctime)s %(message)s")
    parser = argparse.ArgumentParser(
        description="ClickHouse script for parsing results of style check"
    )
    parser.add_argument("--in-results-dir", default="/test_output/")
    parser.add_argument("--out-results-file", default="/test_output/test_results.tsv")
    parser.add_argument("--out-status-file", default="/test_output/check_status.tsv")
    args = parser.parse_args()

    state, description, test_results = process_result(args.in_results_dir)
    logging.info("Result parsed")
    status = (state, description)
    write_results(args.out_results_file, args.out_status_file, test_results, status)
    logging.info("Result written")