HLTech / judge-d

import com.hltech.judged.server.domain.contracts.Capability;

import com.hltech.judged.server.domain.contracts.Expectation;

import com.hltech.judged.server.domain.contracts.ServiceContracts;

import lombok.Data;

import java.io.Serializable;

import java.time.Instant;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import java.util.stream.Collectors;

import static com.google.common.collect.Maps.newHashMap;

@Data

public class ServiceContractsDto {

    private final String name;

    private final String version;

    private final Map<String, ContractDto> capabilities;

    private final Map<String, Map<String, ContractDto>> expectations;

    private final Instant publicationTime;

    public static ServiceContractsDto fromDomain(ServiceContracts serviceContracts) {

        return new ServiceContractsDto(

            serviceContracts.getName(),

            serviceContracts.getVersion(),

            mapCapabilitiesToDto(serviceContracts.getCapabilities()),

            mapExpectationsToDto(serviceContracts.getExpectations()),

            serviceContracts.getPublicationTime()

        );

    }

    private static Map<String, ServiceContractsDto.ContractDto> mapCapabilitiesToDto(List<Capability> capabilities) {

        return capabilities.stream()

            .collect(Collectors.toMap(

                Capability::getProtocol,

                capability -> new ServiceContractsDto.ContractDto(

                    capability.getContract().getValue(),

                    capability.getContract().getMimeType())

            ));

    }

    @Data

    static class ContractDto implements Serializable {

        private final String value;

        private final String mimeType;

    }

}