package grpc

import (
	"context"
	"git.sch9.ru/new_gate/ms-tester/internal/contests"
	contestv1 "git.sch9.ru/new_gate/ms-tester/pkg/go/gen/proto/contest/v1"
	"git.sch9.ru/new_gate/ms-tester/pkg/utils"
	"google.golang.org/grpc"
	"google.golang.org/protobuf/types/known/emptypb"
)

type ContestHandlers struct {
	contestv1.UnimplementedContestServiceServer

	contestUC contests.ContestUseCase
}

func NewContestHandlers(gserver *grpc.Server, contestUC contests.ContestUseCase) {
	handlers := &ContestHandlers{contestUC: contestUC}

	contestv1.RegisterContestServiceServer(gserver, handlers)
}

func (h *ContestHandlers) CreateContest(ctx context.Context, req *contestv1.CreateContestRequest) (*contestv1.CreateContestResponse, error) {
	id, err := h.contestUC.CreateContest(ctx, req.GetTitle())
	if err != nil {
		return nil, err
	}
	return &contestv1.CreateContestResponse{Id: id}, nil
}

func (h *ContestHandlers) ReadContest(ctx context.Context, req *contestv1.ReadContestRequest) (*contestv1.ReadContestResponse, error) {
	contest, err := h.contestUC.ReadContestById(ctx, req.GetId())
	if err != nil {
		return nil, err
	}
	return &contestv1.ReadContestResponse{Contest: &contestv1.ReadContestResponse_Contest{
		Id:        *contest.Id,
		Title:     *contest.Title,
		CreatedAt: utils.TimestampP(contest.CreatedAt),
		UpdatedAt: utils.TimestampP(contest.UpdatedAt),
	}}, nil
}

func (h *ContestHandlers) DeleteContest(ctx context.Context, req *contestv1.DeleteContestRequest) (*emptypb.Empty, error) {
	err := h.contestUC.DeleteContest(ctx, req.GetId())
	if err != nil {
		return nil, err
	}
	return &emptypb.Empty{}, nil
}

func (h *ContestHandlers) AddTask(ctx context.Context, req *contestv1.AddTaskRequest) (*contestv1.AddTaskResponse, error) {
	id, err := h.contestUC.AddTask(ctx, req.GetContestId(), req.GetProblemId())
	if err != nil {
		return nil, err
	}
	return &contestv1.AddTaskResponse{Id: id}, nil
}

func (h *ContestHandlers) DeleteTask(ctx context.Context, req *contestv1.DeleteTaskRequest) (*emptypb.Empty, error) {
	err := h.contestUC.DeleteTask(ctx, req.GetTaskId())
	if err != nil {
		return nil, err
	}
	return &emptypb.Empty{}, nil
}