pro.taskana.rest.resource.mapper.ClassificationMapper Maven / Gradle / Ivy

Go to download
package pro.taskana.rest.resource.mapper;

import static org.springframework.hateoas.mvc.ControllerLinkBuilder.linkTo;
import static org.springframework.hateoas.mvc.ControllerLinkBuilder.methodOn;

import java.time.Instant;

import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import pro.taskana.Classification;
import pro.taskana.ClassificationService;
import pro.taskana.exceptions.NotAuthorizedException;
import pro.taskana.impl.ClassificationImpl;
import pro.taskana.rest.ClassificationController;
import pro.taskana.rest.resource.ClassificationResource;

/**
 * Transforms {@link Classification} to its resource counterpart {@link ClassificationResource} and vice versa.
 */
@Component
public class ClassificationMapper {

    @Autowired
    ClassificationService classificationService;

    public ClassificationResource toResource(Classification classification) {
        ClassificationResource resource = new ClassificationResource();
        BeanUtils.copyProperties(classification, resource);
        // need to be set by hand, because they are named different, or have different types
        resource.setClassificationId(classification.getId());
        resource.setCreated(classification.getCreated().toString());
        resource.setModified(classification.getModified().toString());
        return addLinks(resource, classification);
    }

    public Classification toModel(ClassificationResource classificationResource) throws NotAuthorizedException {
        ClassificationImpl classification = (ClassificationImpl) classificationService.newClassification(
            classificationResource.domain, classificationResource.key, classificationResource.type);
        BeanUtils.copyProperties(classificationResource, classification);

        classification.setId(classificationResource.getClassificationId());
        classification.setCreated(Instant.parse(classificationResource.getCreated()));
        classification.setModified(Instant.parse(classificationResource.getModified()));
        return classification;
    }

    private ClassificationResource addLinks(ClassificationResource resource, Classification classification) {
        resource.add(
            linkTo(methodOn(ClassificationController.class).getClassification(classification.getId()))
                .withSelfRel());
        resource.add(
            linkTo(methodOn(ClassificationController.class).getClassification(classification.getKey(),
                classification.getDomain()))
                    .withRel("getClassificationByKeyAndDomain"));
        resource.add(
            linkTo(methodOn(ClassificationController.class).getClassifications()).withRel("getAllClassifications"));
        resource.add(
            linkTo(methodOn(ClassificationController.class).createClassification(resource))
                .withRel("createClassification"));
        resource.add(
            linkTo(methodOn(ClassificationController.class).updateClassification(resource))
                .withRel("updateClassification"));
        return resource;
    }
}