#!/usr/bin/env python3
# Copyright 2023 BentoML Team. All rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

from __future__ import annotations

from decimal import ROUND_DOWN
from decimal import Decimal
from pathlib import Path

import orjson

PRECISION = Decimal(".01")

ROOT = Path(__file__).resolve().parent.parent

def main() -> int:
  coverage_summary = ROOT / "coverage-summary.json"

  coverage_data = orjson.loads(coverage_summary.read_text(encoding="utf-8"))
  total_data = coverage_data.pop("total")

  lines = ["\n", "Package | Statements\n", "------- | ----------\n",]

  for package, data in sorted(coverage_data.items()):
    statements_covered = data["statements_covered"]
    statements = data["statements"]

    rate = Decimal(statements_covered) / Decimal(statements) * 100
    rate = rate.quantize(PRECISION, rounding=ROUND_DOWN)
    lines.append(f"{package} | {100 if rate == 100 else rate}% ({statements_covered} / {statements})\n")

  total_statements_covered = total_data["statements_covered"]
  total_statements = total_data["statements"]
  total_rate = Decimal(total_statements_covered) / Decimal(total_statements) * 100
  total_rate = total_rate.quantize(PRECISION, rounding=ROUND_DOWN)
  color = "ok" if float(total_rate) >= 95 else "critical"
  lines.insert(0, f"![Code Coverage](https://img.shields.io/badge/coverage-{total_rate}%25-{color}?style=flat)\n")

  lines.append(f"**Summary** | {100 if total_rate == 100 else total_rate}% "
                f"({total_statements_covered} / {total_statements})\n")

  coverage_report = ROOT / "coverage-report.md"
  with coverage_report.open("w", encoding="utf-8") as f:
    f.write("".join(lines))
  return 0

if __name__ == "__main__":
  raise SystemExit(main())