HLTech / judge-d

import dev.hltech.dredd.domain.contracts.ServiceContracts;

import dev.hltech.dredd.domain.contracts.ServiceContractsRepository;

import dev.hltech.dredd.domain.environment.EnvironmentAggregate;

import dev.hltech.dredd.domain.environment.EnvironmentRepository;

import dev.hltech.dredd.interfaces.rest.contracts.ContractsMapper;

import dev.hltech.dredd.interfaces.rest.contracts.ServiceContractsDto;

import io.swagger.annotations.ApiOperation;

import io.swagger.annotations.ApiResponse;

import io.swagger.annotations.ApiResponses;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.http.MediaType;

import org.springframework.web.bind.annotation.GetMapping;

import org.springframework.web.bind.annotation.PathVariable;

import org.springframework.web.bind.annotation.RestController;

import java.util.HashMap;

import java.util.Set;

import java.util.stream.Collectors;

@RestController

public class InterrelationshipController {

    private final EnvironmentRepository environmentRepository;

    private final ServiceContractsRepository serviceContractsRepository;

    private final ContractsMapper contractsMapper;

    @GetMapping(value = "/interrelationship/{environment}", produces = MediaType.APPLICATION_JSON_VALUE)

    @ApiOperation(value = "Get interrelationship between services in given environment", nickname = "Validate against environment")

    @ApiResponses(value = {

        @ApiResponse(code = 200, message = "Success", response = String.class, responseContainer = "list"),

        @ApiResponse(code = 500, message = "Failure")

    })

    public InterrelationshipDto getInterrelationship (@PathVariable("environment") String env) {

        Set<ServiceContractsDto> serviceContractsSet =  environmentRepository.get(env).getAllServices().stream()

            .map(this::getServiceContracts)

            .map(contractsMapper::toDto)

            .collect(Collectors.toSet());

        return new InterrelationshipDto(env, serviceContractsSet);

    }

    private ServiceContracts getServiceContracts(EnvironmentAggregate.ServiceVersion service) {

        return serviceContractsRepository.find(service.getName(), service.getVersion())

            .orElseGet(() ->

                new ServiceContracts(service.getName(), service.getVersion(), new HashMap<>(), new HashMap<>()));

    }

}