package stats

import (
	"time"

	. "git.andr3h3nriqu3s.com/andr3/fyp/logic/db_types"
	. "git.andr3h3nriqu3s.com/andr3/fyp/logic/tasks/utils"
	. "git.andr3h3nriqu3s.com/andr3/fyp/logic/utils"
)

func handleTasksStats(handle *Handle) {
	type ModelTasksStatsRequest struct {
		ModelId string `json:"model_id" validate:"required"`
	}
	PostAuthJson(handle, "/stats/task/model/day", User_Normal, func(c *Context, dat *ModelTasksStatsRequest) *Error {
		model, err := GetBaseModel(c, dat.ModelId)
		if err == NotFoundError {
			return c.JsonBadRequest("Model not found!")
		} else if err != nil {
			return c.E500M("Failed to get model", err)
		}

		now := time.Now()
		now = time.Date(now.Year(), now.Month(), now.Day(), 0, 0, 0, 0, now.Location())

		tasks, err := GetDbMultitple[Task](c, "tasks where model_id=$1 and created_on > $2 order by created_on asc;", model.Id, now)
		if err != nil {
			return c.E500M("Failed to get task informations", err)
		}

		type DataPoint struct {
			Non_classfication_tasks_su  int `json:"nc_success"`
			Non_classfication_tasks_err int `json:"nc_error"`

			Classfication_su      int `json:"c_success"`
			Classfication_fal     int `json:"c_failure"`
			Classfication_un      int `json:"c_unknown"`
			Classfication_err     int `json:"c_error"`
			Classfication_pre     int `json:"c_pre_running"`
			Classfication_running int `json:"c_running"`
		}

		total := DataPoint{}
		hours := make([]DataPoint, 24)

		for i := 0; i < 24; i++ {
			hours[i] = DataPoint{}
		}

		for i := range tasks {
			task := tasks[i]
			hour := task.CreatedOn.Hour()
			if task.TaskType == int(TASK_TYPE_CLASSIFICATION) {
				if task.Status == 4 {
					if task.UserConfirmed == 1 {
						total.Classfication_su += 1
						hours[hour].Classfication_su += 1
					} else if task.UserConfirmed == -1 {
						total.Classfication_fal += 1
						hours[hour].Classfication_fal += 1
					} else {
						total.Classfication_un += 1
						hours[hour].Classfication_un += 1
					}
				} else if task.Status < 0 {
					total.Classfication_err += 1
					hours[hour].Classfication_err += 1
				} else if task.Status < 2 {
					total.Classfication_pre += 1
					hours[hour].Classfication_pre += 1
				} else if task.Status < 4 {
					total.Classfication_running += 1
					hours[hour].Classfication_running += 1
				}
			} else {
				if task.Status >= 0 {
					total.Non_classfication_tasks_su += 1
					hours[hour].Non_classfication_tasks_su += 1
				} else {
					total.Non_classfication_tasks_err += 1
					hours[hour].Non_classfication_tasks_err += 1
				}
			}
		}

		data := struct {
			Total DataPoint   `json:"total"`
			Hours []DataPoint `json:"hours"`
		}{
			Total: total,
			Hours: hours,
		}
		return c.SendJSON(data)
	})
}