org.jboss.pressgang.ccms.wrapper.DBFileWrapper 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.File;
import org.jboss.pressgang.ccms.model.LanguageFile;
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.DBLanguageFileCollectionWrapper;
import org.jboss.pressgang.ccms.wrapper.collection.UpdateableCollectionWrapper;
import org.jboss.pressgang.ccms.wrapper.collection.handler.DBLanguageFileCollectionHandler;

public class DBFileWrapper extends DBBaseEntityWrapper implements FileWrapper {
    private final DBLanguageFileCollectionHandler languageFileCollectionHandler;

    private final File file;

    public DBFileWrapper(final DBProviderFactory providerFactory, final File file, boolean isRevision) {
        super(providerFactory, isRevision, File.class);
        this.file = file;
        languageFileCollectionHandler = new DBLanguageFileCollectionHandler(file);
    }

    @Override
    protected File getEntity() {
        return file;
    }

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

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

    @Override
    public void setDescription(final String description) {
        getEntity().setDescription(description);
    }

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

    @Override
    public void setFilePath(final String filePath) {
        getEntity().setFilePath(filePath);
    }

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

    @Override
    public void setFilename(final String filename) {
        getEntity().setFileName(filename);
    }

    @Override
    public boolean isExplodeArchive() {
        return getEntity().getExplodeArchive();
    }

    @Override
    public void setExplodeArchive(boolean explodeArchive) {
        getEntity().setExplodeArchive(explodeArchive);
    }

    @Override
    public UpdateableCollectionWrapper getLanguageFiles() {
        final CollectionWrapper collection = getWrapperFactory().createCollection(getEntity().getLanguageFiles(),
                LanguageFile.class, isRevisionEntity(), languageFileCollectionHandler);
        return (UpdateableCollectionWrapper) collection;
    }

    @Override
    public void setLanguageFiles(UpdateableCollectionWrapper languageFiles) {
        if (languageFiles == null) return;
        final DBLanguageFileCollectionWrapper dbLanguageFiles = (DBLanguageFileCollectionWrapper) languageFiles;
        dbLanguageFiles.setHandler(languageFileCollectionHandler);

        // Only bother readjusting the collection if its a different collection than the current
        if (dbLanguageFiles.unwrap() != getEntity().getLanguageFiles()) {
            // Add new language files and skip any existing language files
            final Set currentLanguageFiles = new HashSet(getEntity().getLanguageFiles());
            final Collection newLanguageFiles = dbLanguageFiles.unwrap();
            for (final LanguageFile languageFile : newLanguageFiles) {
                if (currentLanguageFiles.contains(languageFile)) {
                    currentLanguageFiles.remove(languageFile);
                    continue;
                } else {
                    getEntity().addLanguageFile(languageFile);
                }
            }

            // Remove language files that should no longer exist in the collection
            for (final LanguageFile removeLanguageFile : currentLanguageFiles) {
                getEntity().removeLanguageFile(removeLanguageFile);
            }
        }
    }
}