from __future__ import annotations

from cti import mitre

from cti.models import Indicator, Technique

def extract_techniques(indicators: list[Indicator]) -> list[Technique]:

    table: dict[str, Technique] = {}

    for indicator in indicators:

        sources = indicator.source.split(",")

        for technique_id in indicator.techniques:

            technique = table.get(technique_id)

            if technique is None:

                technique = Technique(

                    technique_id=technique_id,

                    name=mitre.name_for(technique_id),

                    tactic=mitre.tactic_for(technique_id),

                )

                table[technique_id] = technique

            technique.sources.update(sources)

            technique.indicator_count += 1

    return sorted(table.values(), key=lambda t: (-t.indicator_count, t.technique_id))

def coverage_report(techniques: list[Technique]) -> str:

    lines = [

        "# TTP coverage",

        "",

        "| Technique | Tactic | Name | Indicators | Sources |",

        "|---|---|---|---|---|",

    ]

    for technique in techniques:

        lines.append(

            f"| {technique.technique_id} | {technique.tactic} | {technique.name} "

            f"| {technique.indicator_count} | {', '.join(sorted(technique.sources))} |"

        )

    return "\n".join(lines) + "\n"