argos/dmp-backend/core/src/main/java/eu/eudat/service/dmpblueprint/DmpBlueprintServiceImpl.java

package eu.eudat.service.dmpblueprint;

import eu.eudat.authorization.AuthorizationFlags;
import eu.eudat.authorization.Permission;
import eu.eudat.commons.enums.DmpBlueprintFieldCategory;
import eu.eudat.commons.enums.DmpBlueprintSystemFieldType;
import eu.eudat.commons.enums.IsActive;
import eu.eudat.commons.types.dmpblueprint.*;
import eu.eudat.commons.types.xml.XmlBuilder;
import eu.eudat.convention.ConventionService;
import eu.eudat.data.DmpBlueprintEntity;
import eu.eudat.model.DmpBlueprint;
import eu.eudat.model.builder.DmpBlueprintBuilder;
import eu.eudat.model.builder.dmpblueprintdefinition.DefinitionBuilder;
import eu.eudat.model.deleter.DmpBlueprintDeleter;
import eu.eudat.model.persist.DmpBlueprintPersist;
import eu.eudat.model.persist.dmpblueprintdefinition.*;
import gr.cite.commons.web.authz.service.AuthorizationService;
import gr.cite.tools.data.builder.BuilderFactory;
import gr.cite.tools.data.deleter.DeleterFactory;
import gr.cite.tools.exception.MyApplicationException;
import gr.cite.tools.exception.MyForbiddenException;
import gr.cite.tools.exception.MyNotFoundException;
import gr.cite.tools.exception.MyValidationException;
import gr.cite.tools.fieldset.BaseFieldSet;
import gr.cite.tools.fieldset.FieldSet;
import gr.cite.tools.logging.LoggerService;
import gr.cite.tools.logging.MapLogEntry;
import jakarta.persistence.EntityManager;
import org.jetbrains.annotations.NotNull;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.MessageSource;
import org.springframework.context.i18n.LocaleContextHolder;
import org.springframework.stereotype.Service;
import org.w3c.dom.Document;

import javax.management.InvalidApplicationException;
import java.time.Instant;
import java.util.ArrayList;
import java.util.List;
import java.util.UUID;

@Service
public class DmpBlueprintServiceImpl implements DmpBlueprintService {

    private static final LoggerService logger = new LoggerService(LoggerFactory.getLogger(DmpBlueprintServiceImpl.class));

    private final EntityManager entityManager;

    private final AuthorizationService authorizationService;

    private final DeleterFactory deleterFactory;

    private final BuilderFactory builderFactory;

    private final ConventionService conventionService;


    private final MessageSource messageSource;


    @Autowired
    public DmpBlueprintServiceImpl(
            EntityManager entityManager,
            AuthorizationService authorizationService,
            DeleterFactory deleterFactory,
            BuilderFactory builderFactory,
            ConventionService conventionService,
            MessageSource messageSource) {
        this.entityManager = entityManager;
        this.authorizationService = authorizationService;
        this.deleterFactory = deleterFactory;
        this.builderFactory = builderFactory;
        this.conventionService = conventionService;
        this.messageSource = messageSource;
    }

    public DmpBlueprint persist(DmpBlueprintPersist model, FieldSet fields) throws MyForbiddenException, MyValidationException, MyApplicationException, MyNotFoundException, InvalidApplicationException {
        logger.debug(new MapLogEntry("persisting data").And("model", model).And("fields", fields));

        this.authorizationService.authorizeForce(Permission.EditDmpBlueprint);

        Boolean isUpdate = this.conventionService.isValidGuid(model.getId());

        DmpBlueprintEntity data;
        if (isUpdate) {
            data = this.entityManager.find(DmpBlueprintEntity.class, model.getId());
            if (data == null)
                throw new MyNotFoundException(messageSource.getMessage("General_ItemNotFound", new Object[]{model.getId(), DmpBlueprint.class.getSimpleName()}, LocaleContextHolder.getLocale()));
        } else {
            data = new DmpBlueprintEntity();
            data.setId(UUID.randomUUID());
            data.setIsActive(IsActive.Active);
            data.setCreatedAt(Instant.now());
        }

        data.setLabel(model.getLabel());
        data.setStatus(model.getStatus());
        data.setUpdatedAt(Instant.now());

        Document document = XmlBuilder.getDocument();
        if (document == null) throw new InvalidApplicationException("Can not create document");
        document.appendChild(this.buildDefinitionEntity(model.getDefinition()).toXml(document));
        data.setDefinition(XmlBuilder.generateXml(document));
        
        if (isUpdate) this.entityManager.merge(data);
        else  this.entityManager.persist(data);

        this.entityManager.flush();

        return this.builderFactory.builder(DmpBlueprintBuilder.class).authorize(AuthorizationFlags.OwnerOrPermission).build(BaseFieldSet.build(fields, DmpBlueprint._id), data);
    }
    
    private @NotNull DefinitionEntity buildDefinitionEntity(DefinitionPersist persist){
        DefinitionEntity data = new DefinitionEntity();
        if (persist == null) return data;
        if (!this.conventionService.isListNullOrEmpty(persist.getSections())){
            data.setSections(new ArrayList<>());
            for (SectionPersist sectionPersist: persist.getSections()) {
                data.getSections().add(this.buildSectionEntity(sectionPersist));
            }
        }
        
        return data;
    }

    private @NotNull SectionEntity buildSectionEntity(SectionPersist persist){
        SectionEntity data = new SectionEntity();
        if (persist == null) return data;

        data.setId(persist.getId());
        data.setDescription(persist.getDescription());
        data.setLabel(persist.getLabel());
        data.setOrdinal(persist.getOrdinal());
        data.setHasTemplates(persist.getHasTemplates());
        if (!this.conventionService.isListNullOrEmpty(persist.getFields())){
            data.setFields(new ArrayList<>());
            for (FieldPersist fieldPersist: persist.getFields()) {
                data.getFields().add(this.buildExtraFieldEntity(fieldPersist));
            }
        }

        if (!this.conventionService.isListNullOrEmpty(persist.getDescriptionTemplates())){
            data.setDescriptionTemplates(new ArrayList<>());
            for (DescriptionTemplatePersist descriptionTemplatePersist: persist.getDescriptionTemplates()) {
                data.getDescriptionTemplates().add(this.buildDescriptionTemplateEntity(descriptionTemplatePersist));
            }
        }

        return data;
    }

    private @NotNull DescriptionTemplateEntity buildDescriptionTemplateEntity(DescriptionTemplatePersist persist){
        DescriptionTemplateEntity data = new DescriptionTemplateEntity();
        if (persist == null) return data;

        data.setId(persist.getId());
        data.setDescriptionTemplateId(persist.getDescriptionTemplateId());
        data.setLabel(persist.getLabel());
        data.setMaxMultiplicity(persist.getMaxMultiplicity());
        data.setMinMultiplicity(persist.getMinMultiplicity());

        return data;
    }

    private @NotNull FieldEntity buildExtraFieldEntity(FieldPersist persist){
        if (persist == null) return new ExtraFieldEntity();
        FieldEntity data;
        if (DmpBlueprintFieldCategory.Extra.equals(persist.getCategory())){
            ExtraFieldEntity dataTyped = new ExtraFieldEntity();
            dataTyped.setType(((ExtraFieldPersist)persist).getDataType());
            data = dataTyped;
        } else {
            SystemFieldEntity dataTyped = new SystemFieldEntity();
            dataTyped.setType(((SystemFieldPersist)persist).getSystemFieldType());
            data = dataTyped;
        }
        data.setId(persist.getId());
        data.setCategory(persist.getCategory());
        data.setLabel(persist.getLabel());
        data.setPlaceholder(persist.getPlaceholder());
        data.setDescription(persist.getDescription());
        data.setOrdinal(persist.getOrdinal());
        data.setRequired(persist.getRequired());

        return data;
    }

    public void deleteAndSave(UUID id) throws MyForbiddenException, InvalidApplicationException {
        logger.debug("deleting : {}", id);

        this.authorizationService.authorizeForce(Permission.DeleteDmpBlueprint);

        this.deleterFactory.deleter(DmpBlueprintDeleter.class).deleteAndSaveByIds(List.of(id));
    }

    public boolean fieldInBlueprint(DmpBlueprintEntity dmpBlueprintEntity, DmpBlueprintSystemFieldType type) {

        DefinitionEntity definition = new DefinitionEntity().fromXml(XmlBuilder.fromXml(dmpBlueprintEntity.getDefinition()).getDocumentElement());
        if (definition == null || definition.getSections() == null) return false;
        
        for(SectionEntity section: definition.getSections()){
            if (section.getFields() == null) continue;
            for(FieldEntity field: section.getFields()){
                if(field.getCategory().equals(DmpBlueprintFieldCategory.System)){
                    SystemFieldEntity systemField = (SystemFieldEntity)field;
                    if(systemField.getType().equals(type)){
                        return true;
                    }
                }
            }
        }
        return false;
    }

    public boolean fieldInBlueprint(UUID id, DmpBlueprintSystemFieldType type)  {
        DmpBlueprintEntity data = this.entityManager.find(DmpBlueprintEntity.class, id);
        if (data == null) throw new MyNotFoundException(messageSource.getMessage("General_ItemNotFound", new Object[]{id, DmpBlueprint.class.getSimpleName()}, LocaleContextHolder.getLocale()));
        return this.fieldInBlueprint(data, type);
    }

}