package endpoints

import (
	"arbeitszeitmessung/helper"
	"arbeitszeitmessung/helper/paramParser"
	"arbeitszeitmessung/models"
	"bytes"
	"fmt"
	"log"
	"log/slog"
	"net/http"
	"time"

	"github.com/Dadido3/go-typst"
)

const DE_DATE string = "02.01.2006"

func convertDaysToTypst(days []models.IWorkDay, u models.User) ([]typstDay, error) {
	var typstDays []typstDay
	for _, day := range days {
		var thisTypstDay typstDay
		work, pause, overtime := day.GetTimesReal(u, models.WorktimeBaseWeek)
		thisTypstDay.Date = day.Date().Format(DE_DATE)
		thisTypstDay.Worktime = helper.FormatDurationFill(work, true)
		thisTypstDay.Pausetime = helper.FormatDurationFill(pause, true)
		thisTypstDay.Overtime = helper.FormatDurationFill(overtime, true)
		thisTypstDay.IsFriday = day.Date().Weekday() == time.Friday

		thisTypstDay.DayParts = convertDayToTypstDayParts(day, u)
		typstDays = append(typstDays, thisTypstDay)
	}
	return typstDays, nil
}

func convertDayToTypstDayParts(day models.IWorkDay, user models.User) []typstDayPart {
	var typstDayParts []typstDayPart
	if day.IsWorkDay() {
		workDay, _ := day.(*models.WorkDay)
		for i := 0; i < len(workDay.Bookings); i += 2 {
			var typstDayPart typstDayPart
			typstDayPart.BookingFrom = workDay.Bookings[i].Timestamp.Format("15:04")
			typstDayPart.BookingTo = workDay.Bookings[i+1].Timestamp.Format("15:04")
			typstDayPart.WorkType = workDay.Bookings[i].BookingType.Name
			typstDayPart.IsWorkDay = true
			typstDayParts = append(typstDayParts, typstDayPart)
		}
	} else {
		absentDay, _ := day.(*models.Absence)
		typstDayParts = append(typstDayParts, typstDayPart{IsWorkDay: false, WorkType: absentDay.AbwesenheitTyp.Name})
	}
	return typstDayParts
}

func renderPDF(data []typstData) (bytes.Buffer, error) {
	var markup bytes.Buffer
	var output bytes.Buffer

	if err := typst.InjectValues(&markup, map[string]any{"data": data}); err != nil {
		return output, err
	}

	// Import the template and invoke the template function with the custom data.
	// Show is used to replace the current document with whatever content the template function in `template.typ` returns.
	markup.WriteString(`
		#import "templates/abrechnung.typ": abrechnung
		#for d in data {
  			abrechnung(d.Meta, d.Days)
     	}
		`)

	// Compile the prepared markup with Typst and write the result it into `output.pdf`.

	typstCLI := typst.DockerExec{
		ContainerName: helper.GetEnv("TYPST_CONTAINER", "arbeitszeitmessung-doc-creator"),
	}
	if err := typstCLI.Compile(&markup, &output, nil); err != nil {
		return output, err
	}
	return output, nil
}

func PDFCreateController(w http.ResponseWriter, r *http.Request) {
	helper.RequiresLogin(Session, w, r)
	switch r.Method {
	case http.MethodGet:
		user, err := models.GetUserFromSession(Session, r.Context())
		if err != nil {
			log.Println("Error getting user!")
			return
		}
		pp := paramParser.New(r.URL.Query())
		startDate := pp.ParseTimestampFallback("start_date", time.DateOnly, time.Now())
		var members []models.User = make([]models.User, 0)
		output, err := createReports(user, members, startDate)
		if err != nil {
			slog.Warn("Could not create pdf report", slog.Any("Error", err))
		}

		w.Header().Set("Content-type", "application/pdf")
		output.WriteTo(w)
		w.WriteHeader(http.StatusOK)

	default:
		http.Error(w, "Method not allowed!", http.StatusMethodNotAllowed)
	}
}

func createReports(user models.User, employes []models.User, startDate time.Time) (bytes.Buffer, error) {
	if startDate.Day() > 1 {
		startDate = startDate.AddDate(0, 0, -(startDate.Day() - 1))
	}
	endDate := startDate.AddDate(0, 1, -1)
	return createEmployeReport(user, startDate, endDate)
}

func createEmployeReport(employee models.User, startDate, endDate time.Time) (bytes.Buffer, error) {
	targetHours := (employee.ArbeitszeitProWoche() / 5) * time.Duration(helper.GetWorkingDays(startDate, endDate))
	workingDays := models.GetDays(employee, startDate, endDate, false)

	slog.Debug("Baseline Working hours", "targetHours", targetHours.Hours())

	var actualHours time.Duration
	for _, day := range workingDays {
		actualHours += day.TimeWorkVirtual(employee)
	}
	worktimeBalance := actualHours - targetHours

	typstDays, err := convertDaysToTypst(workingDays, employee)
	if err != nil {
		log.Panicf("Failed to convert days!")
	}

	metadata := typstMetadata{
		EmployeeName:     fmt.Sprintf("%s %s", employee.Vorname, employee.Name),
		TimeRange:        fmt.Sprintf("%s - %s", startDate.Format(DE_DATE), endDate.Format(DE_DATE)),
		Overtime:         helper.FormatDurationFill(worktimeBalance, true),
		WorkTime:         helper.FormatDurationFill(actualHours, true),
		OvertimeTotal:    "",
		CurrentTimestamp: time.Now().Format("02.01.2006 - 15:04 Uhr"),
	}

	return renderPDF([]typstData{{Meta: metadata, Days: typstDays}, {Meta: metadata, Days: typstDays}})
}

func PDFHandler(w http.ResponseWriter, r *http.Request) {
	helper.RequiresLogin(Session, w, r)
	startDate := time.Now()

	if startDate.Day() > 1 {
		startDate = startDate.AddDate(0, 0, -(startDate.Day() - 1))
	}
	endDate := startDate.AddDate(0, 1, -1)

	user, err := models.GetUserFromSession(Session, r.Context())
	if err != nil {
		log.Println("Error getting user!")
	}

	//TODO: only accepted weeks

	weeks := models.GetDays(user, startDate, endDate, false)
	var aggregatedOvertime, aggregatedWorkTime time.Duration
	for _, day := range weeks {
		aggregatedOvertime += day.TimeOvertimeReal(user)
		aggregatedWorkTime += day.TimeWorkVirtual(user)
	}

	typstDays, err := convertDaysToTypst(weeks, user)
	if err != nil {
		log.Panicf("Failed to convert days!")
	}
	metadata := typstMetadata{
		EmployeeName:     fmt.Sprintf("%s %s", user.Vorname, user.Name),
		TimeRange:        fmt.Sprintf("%s - %s", startDate.Format(DE_DATE), endDate.Format(DE_DATE)),
		Overtime:         helper.FormatDurationFill(aggregatedOvertime, true),
		WorkTime:         helper.FormatDurationFill(aggregatedWorkTime, true),
		OvertimeTotal:    "",
		CurrentTimestamp: time.Now().Format("02.01.2006 - 15:04 Uhr"),
	}

	output, err := renderPDF([]typstData{{Meta: metadata, Days: typstDays}})
	if err != nil {
		slog.Warn("Could not create pdf report", slog.Any("Error", err))
	}

	w.Header().Set("Content-type", "application/pdf")
	output.WriteTo(w)
	w.WriteHeader(200)
}

type typstMetadata struct {
	TimeRange        string `json:"time-range"`
	EmployeeName     string `json:"employee-name"`
	WorkTime         string `json:"worktime"`
	Overtime         string `json:"overtime"`
	OvertimeTotal    string `json:"overtime-total"`
	CurrentTimestamp string `json:"current-timestamp"`
}

type typstDayPart struct {
	BookingFrom string `json:"booking-from"`
	BookingTo   string `json:"booking-to"`
	WorkType    string `json:"worktype"`
	IsWorkDay   bool   `json:"is-workday"`
}

type typstDay struct {
	Date      string         `json:"date"`
	DayParts  []typstDayPart `json:"day-parts"`
	Worktime  string         `json:"worktime"`
	Pausetime string         `json:"pausetime"`
	Overtime  string         `json:"overtime"`
	IsFriday  bool           `json:"is-weekend"`
}

type typstData struct {
	Meta typstMetadata `json:"meta"`
	Days []typstDay    `json:"days"`
}