from django.shortcuts import render

from core.report_templates import ReportTemplates
from core.util import get_week_range, next_date
from .azure_auth import AzureUser
from .models import Berichtsheft
import datetime

# Create your views here.


def write_new_report(request):
    user = AzureUser(request)

    # TODO : Get template for user
    definition = ReportTemplates.get_template("weekly")

    # Get the latest year and week
    latest = user.reports().order_by("-year", "-week").first()
    if latest is not None:
        current_year, current_week = next_date(latest.year, latest.week)
    else:
        current_year, current_week, _ = datetime.datetime.today().isocalendar()

    start_date, end_date = get_week_range(current_year, current_week)

    if latest is not None:
        current_num = latest.num + 1
    else:
        current_num = 1

    # TODO : Cookies for persistent saves

    return render(
        request,
        "write.html",
        {
            "user": user,
            "year": current_year,
            "week": current_week,
            "start_date": start_date,
            "end_date": end_date,
            "current_num": current_num,
            "definition": definition,
        },
    )


def index(request):
    user = AzureUser(request)

    # Get all berichtshefte
    all_reports = user.reports()

    return render(request, "index.html", {"user": user, "reports": all_reports})


def test(request):
    return render(request, "test.html", {})