package com.telephoners.krakyournet.ctf.resources;

import com.telephoners.krakyournet.ctf.beans.Flag;
import com.telephoners.krakyournet.ctf.beans.Solution;
import com.telephoners.krakyournet.ctf.beans.Team;
import com.telephoners.krakyournet.ctf.beans.User;
import com.telephoners.krakyournet.ctf.beans.tasks.Task;
import com.telephoners.krakyournet.ctf.exceptions.InvalidSolutionException;
import com.telephoners.krakyournet.ctf.repositories.SolutionsRepository;
import com.telephoners.krakyournet.ctf.repositories.TasksRepository;
import com.telephoners.krakyournet.ctf.repositories.TeamsRepository;
import io.dropwizard.auth.Auth;

import javax.inject.Inject;
import javax.inject.Singleton;
import javax.ws.rs.*;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Response;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

@Singleton
@Path(value = "/solutions")
@Produces(MediaType.APPLICATION_JSON)
public class SolutionsResource
{
    private final TeamsRepository teamsRepository;
    private final SolutionsRepository solutionsRepository;
    private final TasksRepository tasksRepository;

    @Inject
    public SolutionsResource(SolutionsRepository solutionsRepository, TasksRepository tasksRepository,
                             TeamsRepository teamsRepository)
    {
        this.solutionsRepository = solutionsRepository;
        this.teamsRepository = teamsRepository;
        this.tasksRepository = tasksRepository;
    }

    @POST
    @Path("/{task_level}")
    public Response submitSolution(@Auth User user,
                                   @PathParam("task_level") int taskLevel,
                                   String userFlag) throws Exception
    {
        Task task = tasksRepository.getByLevel(taskLevel);
        Flag flag = task.getFlags()
                .stream()
                .filter(taskFlag -> taskFlag.getValue().equals(userFlag))
                .findFirst()
                .orElseThrow(InvalidSolutionException::new);

        solutionsRepository.submitSolution(new Solution(teamsRepository.getTeamByUser(user), task, flag));
        return Response.ok().entity(flag).build();
    }

    @GET
    @Path("/my/completed")
    public List<Integer> getTeamCompletedTasks(@Auth User user)
    {
        return solutionsRepository.getCompletedTasks(teamsRepository.getTeamByUser(user));
    }

    @GET
    @Path("/all/completed")
    public Map<String, List<Integer>> getAllTeamsCompletedTasks(@Auth User user)
    {
        return teamsRepository.getAll()
                .stream()
                .collect(Collectors.toMap(
                        Team::getName,
                        solutionsRepository::getCompletedTasks
                ));
    }

    @GET
    @Path("/all")
    public Map<String, Map<Integer, List<Flag>>> getTeamsSolutions()
    {
        return teamsRepository.getAll()
                .stream()
                .collect(Collectors.toMap(
                        Team::getName,
                        solutionsRepository::getTeamSolutions
                ));
    }

}
