""" MultiQC module to parse output from Samtools rmdup """

import logging
import re

from multiqc.plots import bargraph

# Initialise the logger
log = logging.getLogger(__name__)


class RmdupReportMixin:
    def parse_samtools_rmdup(self):
        """Find Samtools rmdup logs and parse their data"""

        self.samtools_rmdup = dict()
        for f in self.find_log_files("samtools/rmdup", filehandles=True):
            # Example below:
            # [bam_rmdupse_core] 26602816 / 103563641 = 0.2569 in library '   '
            dups_regex = "\[bam_rmdups?e?_core\] (\d+) / (\d+) = (\d+\.\d+) in library '(.*)'"
            s_name = f["s_name"]
            for line in f["f"]:
                match = re.search(dups_regex, line)
                if match:
                    library_name = match.group(4).strip()
                    if library_name != "":
                        s_name = library_name
                    if s_name in self.samtools_rmdup:
                        log.debug(f"Duplicate sample name found in {f['fn']}! Overwriting: {s_name}")
                    self.add_data_source(f, s_name)
                    self.samtools_rmdup[s_name] = {}
                    self.samtools_rmdup[s_name]["n_dups"] = int(match.group(1))
                    self.samtools_rmdup[s_name]["n_tot"] = int(match.group(2))
                    self.samtools_rmdup[s_name]["n_unique"] = int(match.group(2)) - int(match.group(1))
                    self.samtools_rmdup[s_name]["pct_dups"] = float(match.group(3)) * 100

        # Filter to strip out ignored sample names
        self.samtools_rmdup = self.ignore_samples(self.samtools_rmdup)

        if len(self.samtools_rmdup) == 0:
            return 0

        # Write parsed report data to a file
        self.write_data_file(self.samtools_rmdup, "multiqc_samtools_rmdup")

        # Superfluous function call to confirm that it is used in this module
        # Replace None with actual version if it is available
        self.add_software_version(None)

        # Make a bar plot showing duplicates
        keys = {
            "n_unique": {"name": "Non-duplicated reads"},
            "n_dups": {"name": "Duplicated reads"},
        }
        pconfig = {
            "id": "samtools_rmdup_plot",
            "title": "Samtools rmdup: Duplicate alignments",
            "ylab": "Number of reads",
            "yDecimals": False,
        }
        self.add_section(
            name="Duplicates removed",
            anchor="samtools-rmdup",
            plot=bargraph.plot(self.samtools_rmdup, keys, pconfig),
        )

        # Add a column to the General Stats table
        # General Stats Table
        stats_headers = {
            "pct_dups": {
                "title": "% Dups",
                "description": "Percent of duplicate alignments",
                "min": 0,
                "max": 100,
                "suffix": "%",
                "scale": "OrRd",
            }
        }
        self.general_stats_addcols(self.samtools_rmdup, stats_headers)

        return len(self.samtools_rmdup)
