DumpJudgmentsListSuccessRepresentationBuilder.java example

Explorer
saos-master
package pl.edu.icm.saos.api.dump.judgment;

import static pl.edu.icm.saos.api.ApiConstants.JUDGMENT_END_DATE;
import static pl.edu.icm.saos.api.ApiConstants.JUDGMENT_START_DATE;
import static pl.edu.icm.saos.api.ApiConstants.PAGE_NUMBER;
import static pl.edu.icm.saos.api.ApiConstants.PAGE_SIZE;
import static pl.edu.icm.saos.api.ApiConstants.SELF;
import static pl.edu.icm.saos.api.ApiConstants.SINCE_MODIFICATION_DATE;

import java.util.LinkedList;
import java.util.List;
import java.util.stream.Collectors;

import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.hateoas.Link;
import org.springframework.stereotype.Service;
import org.springframework.web.util.UriComponentsBuilder;

import pl.edu.icm.saos.api.ApiConstants;
import pl.edu.icm.saos.api.dump.judgment.item.representation.CommonCourtJudgmentItem;
import pl.edu.icm.saos.api.dump.judgment.item.representation.ConstitutionalTribunalJudgmentItem;
import pl.edu.icm.saos.api.dump.judgment.item.representation.JudgmentItem;
import pl.edu.icm.saos.api.dump.judgment.item.representation.SupremeCourtJudgmentItem;
import pl.edu.icm.saos.api.dump.judgment.mapping.DumpCommonCourtJudgmentItemMapper;
import pl.edu.icm.saos.api.dump.judgment.mapping.DumpConstitutionalTribunalJudgmentItemMapper;
import pl.edu.icm.saos.api.dump.judgment.mapping.DumpJudgmentItemMapper;
import pl.edu.icm.saos.api.dump.judgment.mapping.DumpSupremeCourtJudgmentItemMapper;
import pl.edu.icm.saos.api.dump.judgment.parameters.RequestDumpJudgmentsParameters;
import pl.edu.icm.saos.api.dump.judgment.views.DumpJudgmentsView;
import pl.edu.icm.saos.api.dump.judgment.views.DumpJudgmentsView.QueryTemplate;
import pl.edu.icm.saos.api.dump.judgment.views.DumpJudgmentsView.SinceModificationDateTemplate;
import pl.edu.icm.saos.api.search.parameters.Pagination;
import pl.edu.icm.saos.api.services.dates.DateMapping;
import pl.edu.icm.saos.api.services.representations.success.template.JudgmentDateFromTemplate;
import pl.edu.icm.saos.api.services.representations.success.template.JudgmentDateToTemplate;
import pl.edu.icm.saos.api.services.representations.success.template.PaginationTemplateFactory;
import pl.edu.icm.saos.api.services.representations.success.template.WithGeneratedTemplate;
import pl.edu.icm.saos.persistence.model.CommonCourtJudgment;
import pl.edu.icm.saos.persistence.model.ConstitutionalTribunalJudgment;
import pl.edu.icm.saos.persistence.model.Judgment;
import pl.edu.icm.saos.persistence.model.SupremeCourtJudgment;
import pl.edu.icm.saos.persistence.search.result.SearchResult;

/**
 * Provides functionality for building success object view for dump list of judgments.
 * @author pavtel
 */
@Service
public class DumpJudgmentsListSuccessRepresentationBuilder {

    @Autowired
    private DumpJudgmentItemMapper judgmentItemMapper;

    @Autowired
    private DumpCommonCourtJudgmentItemMapper commonCourtJudgmentItemMapper;

    @Autowired
    private DumpSupremeCourtJudgmentItemMapper supremeCourtJudgmentItemMapper;

    @Autowired
    private DumpConstitutionalTribunalJudgmentItemMapper constitutionalTribunalJudgmentItemMapper;

    @Autowired
    private DateMapping dateMapping;
    
    @Autowired
    private PaginationTemplateFactory paginationTemplateFactory;



    //------------------------ LOGIC --------------------------

    /**
     * Constructs {@link pl.edu.icm.saos.api.dump.judgment.views.DumpJudgmentsView DumpJudgmentsView}.
     * @param searchResult to process.
     * @param pagination to process.
     * @param requestDumpJudgmentsParameters used to create queryTemplate.
     * @param uriComponentsBuilder used to create links.
     * @return representation.
     */
    public DumpJudgmentsView build(SearchResult<Judgment> searchResult, Pagination pagination, RequestDumpJudgmentsParameters requestDumpJudgmentsParameters, UriComponentsBuilder uriComponentsBuilder){
        DumpJudgmentsView dumpJudgmentsView = new DumpJudgmentsView();
        dumpJudgmentsView.setItems(toItems(searchResult.getResultRecords()));

        String startDate = dateMapping.toISO8601Format(requestDumpJudgmentsParameters.getJudgmentStartDate());
        String endDate = dateMapping.toISO8601Format(requestDumpJudgmentsParameters.getJudgmentEndDate());
        String modificationDate = dateMapping.toStringWithZoneUTC(requestDumpJudgmentsParameters.getSinceModificationDate());

        dumpJudgmentsView.setLinks(toLinks(pagination, startDate, endDate, modificationDate, requestDumpJudgmentsParameters.isWithGenerated(), uriComponentsBuilder, searchResult.isMoreRecordsExist()));
        dumpJudgmentsView.setQueryTemplate(toQueryTemplate(pagination, startDate, endDate, modificationDate, requestDumpJudgmentsParameters.isWithGenerated()));

        return dumpJudgmentsView;
    }

    //------------------------ PRIVATE --------------------------
    private List<JudgmentItem> toItems(List<Judgment> judgments) {
        List<JudgmentItem> items = judgments.stream()
                .map(judgment -> toItem(judgment))
                .collect(Collectors.toList());

        return items;
    }

    private JudgmentItem toItem(Judgment judgment){
        JudgmentItem item = initializeItemViewAndFillSpecificFields(judgment);
        judgmentItemMapper.fillJudgmentFieldsToRepresentation(item, judgment);

        return item;
    }

    private JudgmentItem initializeItemViewAndFillSpecificFields(Judgment judgment){
        switch (judgment.getCourtType()){
            case COMMON:
                CommonCourtJudgment ccJudgment = (CommonCourtJudgment) judgment;
                CommonCourtJudgmentItem ccItem = new CommonCourtJudgmentItem();
                commonCourtJudgmentItemMapper.fillJudgmentsFieldsToItemRepresentation(ccItem, ccJudgment);
                return ccItem;
            case SUPREME:
                SupremeCourtJudgment scJudgment = (SupremeCourtJudgment) judgment;
                SupremeCourtJudgmentItem scItem = new SupremeCourtJudgmentItem();
                supremeCourtJudgmentItemMapper.fillJudgmentsFieldsToItemRepresentation(scItem, scJudgment);
                return scItem;
            case CONSTITUTIONAL_TRIBUNAL:
                ConstitutionalTribunalJudgment ctJudgment = (ConstitutionalTribunalJudgment) judgment;
                ConstitutionalTribunalJudgmentItem ctItem = new ConstitutionalTribunalJudgmentItem();
                constitutionalTribunalJudgmentItemMapper.fillJudgmentsFieldsToItemRepresentation(ctItem, ctJudgment);
                return ctItem;
            default:
                return new JudgmentItem();
        }
    }


    private List<Link> toLinks(Pagination pagination, String startDate, String endDate, String modificationDate, Boolean withGenerated, UriComponentsBuilder uriComponentsBuilder, boolean hasMore) {
        List<Link> links = new LinkedList<>();


        links.add(buildLink(pagination, startDate, endDate ,modificationDate, withGenerated, SELF, uriComponentsBuilder));

        if(pagination.hasPrevious()){
            links.add(buildLink(pagination.getPrevious(), startDate, endDate, modificationDate, withGenerated, Link.REL_PREVIOUS, uriComponentsBuilder));
        }

        if(hasMore){
            links.add(buildLink(pagination.getNext(), startDate, endDate, modificationDate, withGenerated, Link.REL_NEXT, uriComponentsBuilder));
        }


        return links;
    }


    private Link buildLink(Pagination pagination, String startDate, String endDate, String modificationDate, Boolean withGenerated, String relName, UriComponentsBuilder uriComponentsBuilder) {

        uriComponentsBuilder
                .replaceQueryParam(PAGE_SIZE, pagination.getPageSize())
                .replaceQueryParam(PAGE_NUMBER, pagination.getPageNumber());

        if(StringUtils.isNotBlank(startDate)){
            uriComponentsBuilder.replaceQueryParam(JUDGMENT_START_DATE, startDate);
        }

        if(StringUtils.isNotBlank(endDate)){
            uriComponentsBuilder.replaceQueryParam(JUDGMENT_END_DATE, endDate);
        }

        if(StringUtils.isNotBlank(modificationDate)){
            uriComponentsBuilder.replaceQueryParam(SINCE_MODIFICATION_DATE, modificationDate);
        }
        
        if (withGenerated != null) {
            uriComponentsBuilder.replaceQueryParam(ApiConstants.WITH_GENERATED, withGenerated);
        }
        
        String path = uriComponentsBuilder.build().encode().toUriString();
        return new Link(path, relName);
    }


    private QueryTemplate toQueryTemplate(Pagination pagination, String startDate, String endDate, String modificationDate, Boolean withGenerated) {

        QueryTemplate queryTemplate = new QueryTemplate();
        queryTemplate.setPageNumber(paginationTemplateFactory.createPageNumberTemplate(pagination));
        queryTemplate.setPageSize(paginationTemplateFactory.createPageSizeTemplate(pagination));
        queryTemplate.setJudgmentStartDate(new JudgmentDateFromTemplate(StringUtils.trimToEmpty(startDate)));
        queryTemplate.setJudgmentEndDate(new JudgmentDateToTemplate(StringUtils.trimToEmpty(endDate)));
        queryTemplate.setSinceModificationDate(new SinceModificationDateTemplate(StringUtils.trimToEmpty(modificationDate)));
        queryTemplate.setWithGenerated(new WithGeneratedTemplate(withGenerated));
        return queryTemplate;
    }

    //------------------------ SETTERS --------------------------

    public void setJudgmentItemMapper(DumpJudgmentItemMapper judgmentItemMapper) {
        this.judgmentItemMapper = judgmentItemMapper;
    }

    public void setCommonCourtJudgmentItemMapper(DumpCommonCourtJudgmentItemMapper commonCourtJudgmentItemMapper) {
        this.commonCourtJudgmentItemMapper = commonCourtJudgmentItemMapper;
    }

    public void setDateMapping(DateMapping dateMapping) {
        this.dateMapping = dateMapping;
    }

    public void setSupremeCourtJudgmentItemMapper(DumpSupremeCourtJudgmentItemMapper supremeCourtJudgmentItemMapper) {
        this.supremeCourtJudgmentItemMapper = supremeCourtJudgmentItemMapper;
    }

    public void setConstitutionalTribunalJudgmentItemMapper(DumpConstitutionalTribunalJudgmentItemMapper constitutionalTribunalJudgmentItemMapper) {
        this.constitutionalTribunalJudgmentItemMapper = constitutionalTribunalJudgmentItemMapper;
    }
}