package server

import (
	"context"

	"github.com/gofiber/fiber/v2"
	"gitlab.mareshq.com/hq/yggdrasil/pkg/training"
)

type APIHandlers struct {
	trainingRepository training.TrainingRepository
}

func NewAPIHandlers(trainingRepository training.TrainingRepository) *APIHandlers {
	return &APIHandlers{
		trainingRepository: trainingRepository,
	}
}

func (h *APIHandlers) ListTrainings(ctx context.Context, req ListTrainingsRequestObject) (ListTrainingsResponseObject, error) {
	trainings, err := h.trainingRepository.FindAll()
	if err != nil {
		return ListTrainings500ApplicationProblemPlusJSONResponse{
			InternalErrorApplicationProblemPlusJSONResponse: InternalErrorApplicationProblemPlusJSONResponse{
				Status: fiber.StatusInternalServerError,
				Title:  "Internal Server Error: Failed to list trainings",
				Detail: err.Error(),
			}}, nil
	}

	data := make([]Training, len(trainings))
	for idx, training := range trainings {
		data[idx] = Training{
			Id:   training.ID,
			Name: training.Name,
			Days: training.Days,
		}
	}

	return ListTrainings200JSONResponse(data), nil
}

func (h *APIHandlers) CreateTraining(ctx context.Context, req CreateTrainingRequestObject) (CreateTrainingResponseObject, error) {
	t := training.Training{
		Name: req.Body.Name,
		Days: req.Body.Days,
	}

	err := h.trainingRepository.Create(&t)
	if err != nil {
		return CreateTraining500ApplicationProblemPlusJSONResponse{
			InternalErrorApplicationProblemPlusJSONResponse: InternalErrorApplicationProblemPlusJSONResponse{
				Status: fiber.StatusInternalServerError,
				Title:  "Internal Server Error: Failed to create training",
				Detail: err.Error(),
			}}, nil
	}

	return CreateTraining201JSONResponse{
		Id:   t.ID,
		Name: t.Name,
		Days: t.Days,
	}, nil
}

func (h *APIHandlers) DeleteTraining(ctx context.Context, req DeleteTrainingRequestObject) (DeleteTrainingResponseObject, error) {
	err := h.trainingRepository.Delete(req.TrainingID)
	if err == training.ErrTrainingNotFound {
		return DeleteTraining404ApplicationProblemPlusJSONResponse{
			NotFoundErrorApplicationProblemPlusJSONResponse: NotFoundErrorApplicationProblemPlusJSONResponse{
				Status: fiber.StatusNotFound,
				Title:  "Not Found: Training not found",
			}}, nil
	} else if err != nil {
		return DeleteTraining500ApplicationProblemPlusJSONResponse{
			InternalErrorApplicationProblemPlusJSONResponse: InternalErrorApplicationProblemPlusJSONResponse{
				Status: fiber.StatusInternalServerError,
				Title:  "Internal Server Error: Failed to delete training",
				Detail: err.Error(),
			}}, nil
	}

	return DeleteTraining204Response{}, nil
}

func (h *APIHandlers) GetTraining(ctx context.Context, req GetTrainingRequestObject) (GetTrainingResponseObject, error) {
	t, err := h.trainingRepository.FindByID(req.TrainingID)
	if err == training.ErrTrainingNotFound {
		return GetTraining404ApplicationProblemPlusJSONResponse{
			NotFoundErrorApplicationProblemPlusJSONResponse: NotFoundErrorApplicationProblemPlusJSONResponse{
				Status: fiber.StatusNotFound,
				Title:  "Not Found: Training not found",
			}}, nil
	} else if err != nil {
		return GetTraining500ApplicationProblemPlusJSONResponse{
			InternalErrorApplicationProblemPlusJSONResponse: InternalErrorApplicationProblemPlusJSONResponse{
				Status: fiber.StatusInternalServerError,
				Title:  "Internal Server Error: Failed to get training",
				Detail: err.Error(),
			}}, nil
	}

	return GetTraining200JSONResponse{
		Id:   t.ID,
		Name: t.Name,
		Days: t.Days,
	}, nil
}

func (h *APIHandlers) UpdateTraining(ctx context.Context, req UpdateTrainingRequestObject) (UpdateTrainingResponseObject, error) {
	t := training.Training{
		ID:   req.TrainingID,
		Name: req.Body.Name,
		Days: req.Body.Days,
	}

	err := h.trainingRepository.Update(&t)
	if err != nil {
		return UpdateTraining500ApplicationProblemPlusJSONResponse{
			InternalErrorApplicationProblemPlusJSONResponse: InternalErrorApplicationProblemPlusJSONResponse{
				Status: fiber.StatusInternalServerError,
				Title:  "Internal Server Error: Failed to update training",
				Detail: err.Error(),
			}}, nil
	}

	return UpdateTraining200JSONResponse{
		Id:   t.ID,
		Name: t.Name,
		Days: t.Days,
	}, nil
}