argos/dmp-migration-tool/web/src/main/java/eu/old/eudat/migration/DatasetMigrationService.java

package eu.old.eudat.migration;

import com.fasterxml.jackson.core.JsonProcessingException;
import eu.eudat.commons.JsonHandlingService;
import eu.eudat.commons.XmlHandlingService;
import eu.eudat.commons.enums.*;
import eu.eudat.commons.types.description.FieldEntity;
import eu.eudat.commons.types.description.PropertyDefinitionEntity;
import eu.eudat.commons.types.description.PropertyDefinitionFieldSetEntity;
import eu.eudat.commons.types.descriptiontemplate.FieldSetEntity;
import eu.eudat.commons.types.dmp.DmpBlueprintValueEntity;
import eu.eudat.commons.types.dmp.DmpContactEntity;
import eu.eudat.commons.types.dmp.DmpPropertiesEntity;
import eu.eudat.commons.types.dmpblueprint.DefinitionEntity;
import eu.eudat.data.*;
import eu.eudat.model.Dmp;
import eu.eudat.model.DmpDescriptionTemplate;
import eu.eudat.query.DescriptionTemplateQuery;
import eu.eudat.query.DmpBlueprintQuery;
import eu.eudat.query.DmpDescriptionTemplateQuery;
import eu.eudat.query.DmpQuery;
import eu.old.eudat.data.dao.entities.DatasetDao;
import eu.old.eudat.data.entities.DMP;
import eu.old.eudat.data.entities.DMPDatasetProfile;
import eu.old.eudat.data.entities.Dataset;
import eu.old.eudat.logic.services.operations.DatabaseRepository;
import eu.old.eudat.models.data.datasetwizard.DatasetWizardModel;
import eu.old.eudat.models.data.dmp.DataManagementPlan;
import eu.old.eudat.queryable.QueryableList;
import gr.cite.tools.data.query.QueryFactory;
import gr.cite.tools.exception.MyApplicationException;
import gr.cite.tools.fieldset.BaseFieldSet;
import gr.cite.tools.logging.LoggerService;
import jakarta.persistence.EntityManager;
import jakarta.xml.bind.JAXBException;
import org.json.JSONObject;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Service;
import org.xml.sax.SAXException;

import javax.xml.parsers.ParserConfigurationException;
import java.io.IOException;
import java.time.Instant;
import java.util.*;
import java.util.stream.Collectors;

@Service
public class DatasetMigrationService {

    private static final LoggerService logger = new LoggerService(LoggerFactory.getLogger(DmpDatasetProfileMigrationService.class));
    private final DatabaseRepository databaseRepository;
    private final JsonHandlingService jsonHandlingService;
    private final QueryFactory queryFactory;
    private final XmlHandlingService xmlHandlingService;
    private static final int PageSize = 500;
    private static final boolean TestMode = false;
    private final EntityManager entityManager;

    public DatasetMigrationService(DatabaseRepository databaseRepository, JsonHandlingService jsonHandlingService, QueryFactory queryFactory, XmlHandlingService xmlHandlingService, EntityManager entityManager) {
        this.databaseRepository = databaseRepository;
        this.jsonHandlingService = jsonHandlingService;
        this.queryFactory = queryFactory;
        this.xmlHandlingService = xmlHandlingService;
        this.entityManager = entityManager;
    }

    public void migrate() throws IOException, JAXBException, ParserConfigurationException, InstantiationException, IllegalAccessException, SAXException {
        DatasetDao datasetDao = databaseRepository.getDatasetDao();
        long total = datasetDao.asQueryable().count();
        logger.debug("Migrate Dataset Total : " + total);
        int page = 0;

        List<Dataset> items;
        do {
            items = datasetDao.asQueryable().orderBy((builder, root) -> builder.asc(root.get("created"))).orderBy((builder, root) -> builder.asc(root.get("ID"))).skip(page * PageSize).take(PageSize).toList();
            if (items != null && !items.isEmpty()) {
                logger.debug("Migrate Dataset " + page * PageSize + " of " + total);

                List<DmpBlueprintEntity> dmpBlueprints = this.queryFactory.query(DmpBlueprintQuery.class).ids(items.stream().map(x-> x.getDmp().getProfile().getId()).distinct().toList()).collect();
                Map<UUID, DefinitionEntity> dmpBlueprintsMap = new HashMap<>();
                for (DmpBlueprintEntity dmpBlueprint : dmpBlueprints) {
                    DefinitionEntity definitionEntity = this.xmlHandlingService.fromXml(DefinitionEntity.class, dmpBlueprint.getDefinition());
                    dmpBlueprintsMap.put(dmpBlueprint.getId(), definitionEntity);
                }

                List<DmpDescriptionTemplateEntity> dmpDescriptionTemplateEntities = this.queryFactory.query(DmpDescriptionTemplateQuery.class).dmpIds(items.stream().map(x-> x.getDmp().getId()).distinct().toList()).collect();
                List<DescriptionTemplateEntity> descriptionTemplates = this.queryFactory.query(DescriptionTemplateQuery.class).ids(items.stream().map(x-> x.getProfile().getId()).distinct().toList()).collect();
                Map<UUID, UUID> creatorsByDmp = this.queryFactory.query(DmpQuery.class).ids(items.stream().map(x-> x.getDmp().getId()).distinct().toList()).collectAs(new BaseFieldSet().ensure(Dmp._id).ensure(Dmp._creator)).stream().collect(Collectors.toMap(DmpEntity::getId, DmpEntity::getCreatorId));;
                
                Map<UUID, eu.eudat.commons.types.descriptiontemplate.DefinitionEntity> descriptionTemplateDefinitionMap = new HashMap<>();
                for (DescriptionTemplateEntity descriptionTemplateEntity : descriptionTemplates){
                    descriptionTemplateDefinitionMap.put(descriptionTemplateEntity.getId(), this.xmlHandlingService.fromXml(eu.eudat.commons.types.descriptiontemplate.DefinitionEntity.class, descriptionTemplateEntity.getDefinition()));
                }
                for (Dataset item : items) {
                    //entityManager.detach(item);

                    //DatasetWizardModel model = new DatasetWizardModel();
                    //model.fromDataModel(item);

                    DefinitionEntity definition = dmpBlueprintsMap.getOrDefault(item.getDmp().getProfile().getId(), null);

                    if (definition == null || definition.getSections() == null || definition.getSections().size() <= item.getDmpSectionIndex()) {
                        logger.error("Migrate Dataset " + item.getId() + " cannot found section id for section  " + item.getDmpSectionIndex());
                        throw  new MyApplicationException("Migrate Dataset " + item.getId() + " cannot found section id for section  " + item.getDmpSectionIndex());
                    }
                    UUID sectionId = definition.getSections().get(item.getDmpSectionIndex()).getId();
                    if (sectionId == null) {
                        logger.error("Migrate Dataset " + item.getId() + " cannot found section id for section  " + item.getDmpSectionIndex());
                        throw  new MyApplicationException("Migrate Dataset " + item.getId() + " cannot found section id for section  " + item.getDmpSectionIndex());
                    }

                    List<DmpDescriptionTemplateEntity> itemDescriptionTemplates = this.getOrCreateDmpDescriptionTemplateEntity(item, sectionId, dmpDescriptionTemplateEntities);
                    
                    if (itemDescriptionTemplates.size() > 1) {
                        logger.error("Migrate Dataset " + item.getId() + " multiple DmpDescriptionTemplateEntity for section  " + item.getDmpSectionIndex());
                        //TODO CLEAN Duplicates dmp, DescriptionTemplate, SectionId
                    }
                    
                    DescriptionEntity data = new DescriptionEntity();
                    data.setId(item.getId());
                    data.setDescription(item.getDescription());
                    if (item.getCreator() != null) {
                        data.setCreatedById(item.getCreator().getId());
                    } else {
                        data.setCreatedById(creatorsByDmp.getOrDefault(item.getDmp().getId(), null));
                    }
                    data.setDmpId(item.getDmp().getId());
                    data.setLabel(item.getLabel());
                    data.setDmpDescriptionTemplateId(itemDescriptionTemplates.getFirst().getId());
                    data.setDescriptionTemplateId(item.getProfile().getId());
                    data.setCreatedAt(item.getCreated() != null ? item.getCreated().toInstant() : Instant.now());
                    data.setUpdatedAt(item.getModified() != null ? item.getModified().toInstant() : Instant.now());
                    if (item.getFinalizedAt() != null)
                        data.setFinalizedAt(item.getFinalizedAt().toInstant());
                    if (item.getStatus() == 99) {
                        data.setIsActive(IsActive.Inactive);
                        data.setStatus(DescriptionStatus.Canceled);
                    } else {
                        data.setIsActive(IsActive.Active);
                        data.setStatus(DescriptionStatus.of(item.getStatus()));
                    }

                    eu.eudat.commons.types.descriptiontemplate.DefinitionEntity descriptionTemplateDefinitionEntity = descriptionTemplateDefinitionMap.getOrDefault(item.getProfile().getId(), null);
                    data.setProperties(this.jsonHandlingService.toJson(this.buildPropertyDefinitionEntity(item, descriptionTemplateDefinitionEntity)));
                    

                    if (data.getCreatedById() == null){
                        logger.error("Migration skipped creator not found " + item.getId());
                        throw  new MyApplicationException("Migration skipped creator not found " + item.getId());
                    }

                    this.entityManager.persist(data);
                    this.entityManager.flush();
                }

                page++;
            }
        } while (items != null && !items.isEmpty() && !TestMode);
    }
    
    private List<DmpDescriptionTemplateEntity> getOrCreateDmpDescriptionTemplateEntity(Dataset item, UUID sectionId, List<DmpDescriptionTemplateEntity> dmpDescriptionTemplateEntities){
        List<DmpDescriptionTemplateEntity> itemDescriptionTemplates = dmpDescriptionTemplateEntities.stream().filter(x-> x.getDescriptionTemplateGroupId().equals(item.getProfile().getGroupId()) && x.getDmpId().equals(item.getDmp().getId()) && x.getSectionId().equals(sectionId)).toList();
        if (itemDescriptionTemplates.isEmpty()) {
            logger.error("Migrate Dataset " + item.getId() + " cannot found DmpDescriptionTemplateEntity for section  " + item.getDmpSectionIndex());
            if (dmpDescriptionTemplateEntities.stream().filter(x-> x.getDmpId().equals(item.getDmp().getId()) && x.getSectionId().equals(sectionId)).count() > 0) {
                DmpDescriptionTemplateEntity dmpDescriptionTemplateEntity = new DmpDescriptionTemplateEntity();
                dmpDescriptionTemplateEntity.setId(UUID.randomUUID());
                dmpDescriptionTemplateEntity.setDescriptionTemplateGroupId(item.getProfile().getGroupId());
                dmpDescriptionTemplateEntity.setDmpId(item.getDmp().getId());
                dmpDescriptionTemplateEntity.setCreatedAt(item.getCreated() != null ? item.getCreated().toInstant() : Instant.now());
                dmpDescriptionTemplateEntity.setUpdatedAt(item.getModified() != null ? item.getModified().toInstant() : Instant.now());
                dmpDescriptionTemplateEntity.setSectionId(sectionId);
                dmpDescriptionTemplateEntity.setIsActive(IsActive.Active);
                this.entityManager.persist(dmpDescriptionTemplateEntity);
                itemDescriptionTemplates = List.of(dmpDescriptionTemplateEntity);
            } else {
                throw new MyApplicationException("Migrate Dataset " + item.getId() + " cannot found DmpDescriptionTemplateEntity for section  " + item.getDmpSectionIndex());
            }
        }
        return itemDescriptionTemplates;
    }
    
    private PropertyDefinitionEntity buildPropertyDefinitionEntity(Dataset item, eu.eudat.commons.types.descriptiontemplate.DefinitionEntity descriptionTemplateDefinitionEntity) {
        if (item.getProperties() != null) return null;
        JSONObject jObject = new JSONObject(item.getProperties());
        Map<String, Object> properties = jObject.toMap();

        PropertyDefinitionEntity propertyDefinitionEntity = new PropertyDefinitionEntity();
        List<FieldSetEntity> fieldSetEntities = descriptionTemplateDefinitionEntity.getAllFieldSets();
        
//        for (String key : properties.keySet()) {
//            FieldSetEntity currentFieldSet = null;
//            eu.eudat.commons.types.descriptiontemplate.FieldEntity currentField = null;
//            for (FieldSetEntity fieldSetEntity : fieldSetEntities){
//                List<eu.eudat.commons.types.descriptiontemplate.FieldEntity> fieldEntities = fieldSetEntity.getFieldById(key);
//                if (!fieldEntities.isEmpty()){
//                    currentFieldSet = fieldSetEntity;
//                    currentField = fieldEntities.getFirst();
//                }
//            }
//            
//            if (currentFieldSet == null || currentField == null) throw new MyApplicationException("aaa");
//
//            if (!propertyDefinitionEntity.getFieldSets().containsKey(currentFieldSet.getId())){
//                PropertyDefinitionFieldSetEntity fieldSetEntity = new PropertyDefinitionFieldSetEntity();
//                fieldSetEntity.setItems(new ArrayList<>());
//                propertyDefinitionEntity.getFieldSets().put(currentFieldSet.getId(), fieldSetEntity);
//            }
//            propertyDefinitionEntity.getFieldSets().get(currentFieldSet.getId())
//            
//            FieldEntity fieldEntity = new FieldEntity();
//            fieldEntity.setKey(key);
//            fieldEntity.setValue(properties.get(key) != null ? properties.get(key).toString() : null);
//            fieldEntities.add(fieldEntity);
//            propertyDefinitionEntity.getFieldSets().put(key, fieldEntity);
//        }
//        propertyDefinitionEntity.setFields(fieldEntities);
        return propertyDefinitionEntity;

    }

}