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

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

import javax.persistence.EntityManager;
import java.util.List;

import org.jboss.pressgang.ccms.model.File;
import org.jboss.pressgang.ccms.model.LanguageFile;
import org.jboss.pressgang.ccms.provider.listener.ProviderListener;
import org.jboss.pressgang.ccms.wrapper.DBFileWrapper;
import org.jboss.pressgang.ccms.wrapper.FileWrapper;
import org.jboss.pressgang.ccms.wrapper.LanguageFileWrapper;
import org.jboss.pressgang.ccms.wrapper.collection.CollectionWrapper;

public class DBFileProvider extends DBDataProvider implements FileProvider {
    protected DBFileProvider(EntityManager entityManager, DBProviderFactory providerFactory, List listeners) {
        super(entityManager, providerFactory, listeners);
    }

    @Override
    public FileWrapper getFile(int id) {
        return getWrapperFactory().create(getEntity(File.class, id), false);
    }

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

    public CollectionWrapper getFileLanguageFiles(int id, Integer revision) {
        final DBFileWrapper file = (DBFileWrapper) getFile(id, revision);
        if (file == null) {
            return null;
        } else {
            return getWrapperFactory().createCollection(file.unwrap().getLanguageFiles(), LanguageFile.class, revision != null);
        }
    }

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