org.jboss.pressgang.ccms.provider.DBTranslatedTopicProvider Maven / Gradle / Ivy

Go to download
package org.jboss.pressgang.ccms.provider;

import javax.persistence.EntityManager;
import javax.persistence.criteria.CriteriaQuery;
import java.io.UnsupportedEncodingException;
import java.net.URLDecoder;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.jboss.pressgang.ccms.filter.TranslatedTopicFieldFilter;
import org.jboss.pressgang.ccms.filter.builder.TranslatedTopicDataFilterQueryBuilder;
import org.jboss.pressgang.ccms.filter.utils.EntityUtilities;
import org.jboss.pressgang.ccms.filter.utils.FilterUtilities;
import org.jboss.pressgang.ccms.model.Filter;
import org.jboss.pressgang.ccms.model.Tag;
import org.jboss.pressgang.ccms.model.TopicSourceUrl;
import org.jboss.pressgang.ccms.model.TopicToPropertyTag;
import org.jboss.pressgang.ccms.model.TranslatedTopicData;
import org.jboss.pressgang.ccms.model.TranslatedTopicString;
import org.jboss.pressgang.ccms.provider.listener.ProviderListener;
import org.jboss.pressgang.ccms.utils.constants.CommonFilterConstants;
import org.jboss.pressgang.ccms.wrapper.DBTranslatedTopicDataWrapper;
import org.jboss.pressgang.ccms.wrapper.LogMessageWrapper;
import org.jboss.pressgang.ccms.wrapper.PropertyTagInTopicWrapper;
import org.jboss.pressgang.ccms.wrapper.TagWrapper;
import org.jboss.pressgang.ccms.wrapper.TopicSourceURLWrapper;
import org.jboss.pressgang.ccms.wrapper.TranslatedTopicStringWrapper;
import org.jboss.pressgang.ccms.wrapper.TranslatedTopicWrapper;
import org.jboss.pressgang.ccms.wrapper.collection.CollectionWrapper;
import org.jboss.pressgang.ccms.wrapper.collection.UpdateableCollectionWrapper;

public class DBTranslatedTopicProvider extends DBDataProvider implements TranslatedTopicProvider {
    protected DBTranslatedTopicProvider(final EntityManager entityManager, final DBProviderFactory providerFactory, List listeners) {
        super(entityManager, providerFactory, listeners);
    }

    @Override
    public TranslatedTopicWrapper getTranslatedTopic(int id) {
        return getWrapperFactory().create(getEntity(TranslatedTopicData.class, id), false);
    }

    @Override
    public TranslatedTopicWrapper getTranslatedTopic(int id, Integer revision) {
        if (revision == null) {
            return getTranslatedTopic(id);
        } else {
            return getWrapperFactory().create(getRevisionEntity(TranslatedTopicData.class, id, revision), true);
        }
    }

    @Override
    public CollectionWrapper getTranslatedTopicTags(int id, Integer revision) {
        final DBTranslatedTopicDataWrapper topic = (DBTranslatedTopicDataWrapper) getTranslatedTopic(id, revision);
        if (topic == null) {
            return null;
        } else {
            return getWrapperFactory().createCollection(topic.unwrap().getTranslatedTopic().getEnversTopic(getEntityManager()).getTags(),
                    Tag.class, revision != null);
        }
    }

    @Override
    public CollectionWrapper getTranslatedTopicProperties(int id, Integer revision) {
        final DBTranslatedTopicDataWrapper topic = (DBTranslatedTopicDataWrapper) getTranslatedTopic(id, revision);
        if (topic == null) {
            return null;
        } else {
            return getWrapperFactory().createCollection(
                    topic.unwrap().getTranslatedTopic().getEnversTopic(getEntityManager()).getTopicToPropertyTags(),
                    TopicToPropertyTag.class, revision != null);
        }
    }

    @Override
    public CollectionWrapper getTranslatedTopicOutgoingRelationships(int id, Integer revision) {
        final DBTranslatedTopicDataWrapper topic = (DBTranslatedTopicDataWrapper) getTranslatedTopic(id, revision);
        if (topic == null) {
            return null;
        } else {
            return getWrapperFactory().createCollection(topic.unwrap().getOutgoingRelatedTranslatedTopicData(getEntityManager()),
                    TranslatedTopicData.class, revision != null);
        }
    }

    @Override
    public CollectionWrapper getTranslatedTopicIncomingRelationships(int id, Integer revision) {
        final DBTranslatedTopicDataWrapper topic = (DBTranslatedTopicDataWrapper) getTranslatedTopic(id, revision);
        if (topic == null) {
            return null;
        } else {
            return getWrapperFactory().createCollection(topic.unwrap().getIncomingRelatedTranslatedTopicData(getEntityManager()),
                    TranslatedTopicData.class, revision != null);
        }
    }

    @Override
    public CollectionWrapper getTranslatedTopicSourceUrls(int id, Integer revision) {
        final DBTranslatedTopicDataWrapper topic = (DBTranslatedTopicDataWrapper) getTranslatedTopic(id, revision);
        if (topic == null) {
            return null;
        } else {
            return getWrapperFactory().createCollection(
                    topic.unwrap().getTranslatedTopic().getEnversTopic(getEntityManager()).getTopicSourceUrls(), TopicSourceUrl.class,
                    revision != null);
        }
    }

    @Override
    public UpdateableCollectionWrapper getTranslatedTopicStrings(int id, Integer revision) {
        final DBTranslatedTopicDataWrapper topic = (DBTranslatedTopicDataWrapper) getTranslatedTopic(id, revision);
        if (topic == null) {
            return null;
        } else {
            final CollectionWrapper collection = getWrapperFactory().createCollection(
                    topic.unwrap().getTranslatedTopicStrings(), TranslatedTopicString.class, revision != null);
            return (UpdateableCollectionWrapper) collection;
        }
    }

    @Override
    public CollectionWrapper getTranslatedTopicRevisions(int id, Integer revision) {
        final List revisions = getRevisionList(TranslatedTopicData.class, id);
        return getWrapperFactory().createCollection(revisions, TranslatedTopicData.class, revision != null);
    }

    @Override
    public CollectionWrapper getTranslatedTopicsWithQuery(String query) {
        final String fixedQuery = query.replace("query;", "");
        final String[] queryValues = fixedQuery.split(";");
        final Map queryParameters = new HashMap();
        for (final String value : queryValues) {
            if (value.trim().isEmpty()) continue;
            String[] keyValue = value.split("=", 2);
            try {
                queryParameters.put(keyValue[0], URLDecoder.decode(keyValue[1], "UTF-8"));
            } catch (UnsupportedEncodingException e) {
                // Should support UTF-8, if not throw a runtime error.
                throw new RuntimeException(e);
            }
        }

        final Filter filter = EntityUtilities.populateFilter(getEntityManager(), queryParameters, CommonFilterConstants.FILTER_ID,
                CommonFilterConstants.MATCH_TAG, CommonFilterConstants.GROUP_TAG, CommonFilterConstants.CATEORY_INTERNAL_LOGIC,
                CommonFilterConstants.CATEORY_EXTERNAL_LOGIC, CommonFilterConstants.MATCH_LOCALE, new TranslatedTopicFieldFilter());

        final TranslatedTopicDataFilterQueryBuilder queryBuilder = new TranslatedTopicDataFilterQueryBuilder(getEntityManager());
        final CriteriaQuery criteriaQuery = FilterUtilities.buildQuery(filter, queryBuilder);

        return getWrapperFactory().createCollection(executeQuery(criteriaQuery), TranslatedTopicData.class, false);
    }

    @Override
    public TranslatedTopicWrapper createTranslatedTopic(TranslatedTopicWrapper translatedTopic) {
        return createTranslatedTopic(translatedTopic, null);
    }

    @Override
    public TranslatedTopicWrapper createTranslatedTopic(TranslatedTopicWrapper translatedTopic, LogMessageWrapper logMessage) {
        // Send the notification events
        notifyCreateEntity(translatedTopic);
        notifyLogMessage(logMessage);

        // Persist the new entity
        getEntityManager().persist(translatedTopic.unwrap());

        // Flush the changes to the database
        getEntityManager().flush();

        return translatedTopic;
    }

    @Override
    public CollectionWrapper createTranslatedTopics(
            CollectionWrapper translatedTopics) {
        // Send the notification events
        notifyCreateEntityCollection(translatedTopics);

        // Persist the new entities
        for (final TranslatedTopicWrapper topic : translatedTopics.getItems()) {
            getEntityManager().persist(topic.unwrap());
        }

        // Flush the changes to the database
        getEntityManager().flush();

        return translatedTopics;
    }

    @Override
    public TranslatedTopicWrapper updateTranslatedTopic(TranslatedTopicWrapper translatedTopic) {
        return updateTranslatedTopic(translatedTopic, null);
    }

    @Override
    public TranslatedTopicWrapper updateTranslatedTopic(TranslatedTopicWrapper translatedTopic, LogMessageWrapper logMessage) {
        // Send the notification events
        notifyUpdateEntity(translatedTopic);
        notifyLogMessage(logMessage);

        // Persist the changes
        getEntityManager().persist(translatedTopic.unwrap());

        // Flush the changes to the database
        getEntityManager().flush();

        return translatedTopic;
    }

    @Override
    public CollectionWrapper updateTranslatedTopics(
            CollectionWrapper translatedTopics) {
        // Send the notification events
        notifyUpdateEntityCollection(translatedTopics);

        // Persist the changes for each entity
        for (final TranslatedTopicWrapper topic : translatedTopics.getItems()) {
            getEntityManager().persist(topic.unwrap());
        }

        // Flush the changes to the database
        getEntityManager().flush();

        return translatedTopics;
    }

    @Override
    public boolean deleteTranslatedTopic(Integer id) {
        // Send the notification events
        notifyDeleteEntity(TranslatedTopicData.class, id);

        // Delete the entity
        final TranslatedTopicData topic = getEntityManager().find(TranslatedTopicData.class, id);
        getEntityManager().remove(topic);

        // Flush the changes to the database
        getEntityManager().flush();

        return true;
    }

    @Override
    public boolean deleteTranslatedTopics(List ids) {
        // Send the notification events
        notifyDeleteEntityCollection(TranslatedTopicData.class, ids);

        // Delete all the entities
        for (final Integer id : ids) {
            final TranslatedTopicData topic = getEntityManager().find(TranslatedTopicData.class, id);
            getEntityManager().remove(topic);
        }

        // Flush the changes to the database
        getEntityManager().flush();

        return true;
    }

    @Override
    public TranslatedTopicWrapper newTranslatedTopic() {
        return getWrapperFactory().create(new TranslatedTopicData(), false);
    }

    @Override
    public CollectionWrapper newTranslatedTopicCollection() {
        return getWrapperFactory().createCollection(new ArrayList(), TranslatedTopicData.class, false);
    }
}