org.jboss.pressgang.ccms.wrapper.DBImageWrapper Maven / Gradle / Ivy

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

import java.util.Collection;
import java.util.HashSet;
import java.util.Set;

import org.jboss.pressgang.ccms.model.ImageFile;
import org.jboss.pressgang.ccms.model.LanguageImage;
import org.jboss.pressgang.ccms.provider.DBProviderFactory;
import org.jboss.pressgang.ccms.wrapper.base.DBBaseEntityWrapper;
import org.jboss.pressgang.ccms.wrapper.collection.CollectionWrapper;
import org.jboss.pressgang.ccms.wrapper.collection.DBLanguageImageCollectionWrapper;
import org.jboss.pressgang.ccms.wrapper.collection.UpdateableCollectionWrapper;
import org.jboss.pressgang.ccms.wrapper.collection.handler.DBLanguageImageCollectionHandler;

public class DBImageWrapper extends DBBaseEntityWrapper implements ImageWrapper {
    private final DBLanguageImageCollectionHandler languageImageCollectionHandler;

    private final ImageFile image;

    public DBImageWrapper(final DBProviderFactory providerFactory, final ImageFile image, boolean isRevision) {
        super(providerFactory, isRevision, ImageFile.class);
        this.image = image;
        languageImageCollectionHandler = new DBLanguageImageCollectionHandler(image);
    }

    @Override
    protected ImageFile getEntity() {
        return image;
    }

    @Override
    public void setId(Integer id) {
        getEntity().setImageFileId(id);
    }

    @Override
    public String getDescription() {
        return getEntity().getDescription();
    }

    @Override
    public UpdateableCollectionWrapper getLanguageImages() {
        final CollectionWrapper collection = getWrapperFactory().createCollection(getEntity().getLanguageImages(),
                LanguageImage.class, isRevisionEntity(), languageImageCollectionHandler);
        return (UpdateableCollectionWrapper) collection;
    }

    @Override
    public void setLanguageImages(UpdateableCollectionWrapper languageImages) {
        if (languageImages == null) return;
        final DBLanguageImageCollectionWrapper dbLanguageImages = (DBLanguageImageCollectionWrapper) languageImages;
        dbLanguageImages.setHandler(languageImageCollectionHandler);

        // Only bother readjusting the collection if its a different collection than the current
        if (dbLanguageImages.unwrap() != getEntity().getLanguageImages()) {
            // Add new language images and skip any existing language images
            final Set currentLanguageImages = new HashSet(getEntity().getLanguageImages());
            final Collection newLanguageImages = dbLanguageImages.unwrap();
            for (final LanguageImage languageImage : newLanguageImages) {
                if (currentLanguageImages.contains(languageImage)) {
                    currentLanguageImages.remove(languageImage);
                    continue;
                } else {
                    getEntity().addLanguageImage(languageImage);
                }
            }

            // Remove language images that should no longer exist in the collection
            for (final LanguageImage removeLanguageImage : currentLanguageImages) {
                getEntity().removeLanguageImage(removeLanguageImage);
            }
        }
    }
}