#!/usr/bin/python

import argparse
import logging
import subprocess
import sys


def stop_err(logger, msg):
    logger.critical(msg)
    sys.exit()


def log_subprocess_output(logger, pipe):
    for line in iter(pipe.readline, b""):
        logger.debug(line.decode().rstrip())


def get_arg():
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "--alignment_report",
        dest="alignment_report",
        action="store",
        metavar="alignment_report",
        type=str,
    )
    parser.add_argument(
        "--dedup_report",
        dest="dedup_report",
        action="store",
        metavar="dedup_report",
        type=str,
    )
    parser.add_argument(
        "--splitting_report",
        dest="splitting_report",
        action="store",
        metavar="splitting_report",
        type=str,
    )
    parser.add_argument(
        "--mbias_report",
        dest="mbias_report",
        action="store",
        metavar="mbias_report",
        type=str,
    )
    parser.add_argument(
        "--nucleotide_report",
        dest="nucleotide_report",
        action="store",
        metavar="nucleotide_report",
        type=str,
    )
    parser.add_argument(
        "--output_html_report",
        dest="output_html_report",
        action="store",
        metavar="output_html_report",
        type=str,
    )
    parser.add_argument(
        "--log_report",
        dest="log_report",
        action="store",
        metavar="log_report",
        type=str,
    )
    args = parser.parse_args()
    return args


def __main__():
    args = get_arg()
    logger = logging.getLogger("bismark_deduplicate_wrapper")
    logger.setLevel(logging.DEBUG)
    ch = logging.StreamHandler(sys.stdout)
    if args.log_report:
        ch.setLevel(logging.WARNING)
        handler = logging.FileHandler(args.log_report)
        handler.setLevel(logging.DEBUG)
        logger.addHandler(handler)
    else:
        ch.setLevel(logging.DEBUG)
    logger.addHandler(ch)

    cmd = [
        "bismark2report",
        "--verbose",
        "--alignment_report",
        args.alignment_report,
        "--output",
        args.output_html_report,
    ]

    if args.dedup_report:
        cmd.extend(["--dedup_report", args.dedup_report])
    if args.splitting_report:
        cmd.extend(["--splitting_report", args.splitting_report])
    if args.mbias_report:
        cmd.extend(["--mbias_report", args.mbias_report])
    if args.nucleotide_report:
        cmd.extend(["--nucleotide_report", args.nucleotide_report])

    logger.info("Generating report with: '%s'", " ".join(cmd))
    process = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
    with process.stdout:
        log_subprocess_output(logger, process.stdout)
    exitcode = process.wait()
    if exitcode != 0:
        stop_err(
            logger,
            "Bismark pretty report error (also check the log file if any)!\n%s"
            % process.stderr,
        )


if __name__ == "__main__":
    __main__()