ContentSpecUtilities.java

/*
  Copyright 2011-2014 Red Hat, Inc

  This file is part of PressGang CCMS.

  PressGang CCMS is free software: you can redistribute it and/or modify
  it under the terms of the GNU Lesser General Public License as published by
  the Free Software Foundation, either version 3 of the License, or
  (at your option) any later version.

  PressGang CCMS is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  GNU Lesser General Public License for more details.

  You should have received a copy of the GNU Lesser General Public License
  along with PressGang CCMS.  If not, see <http://www.gnu.org/licenses/>.
*/

package org.jboss.pressgang.ccms.server.utils;

import javax.persistence.EntityManager;

import org.jboss.pressgang.ccms.contentspec.utils.CSTransformer;
import org.jboss.pressgang.ccms.model.contentspec.ContentSpec;
import org.jboss.pressgang.ccms.provider.ContentSpecProvider;
import org.jboss.pressgang.ccms.provider.DBProviderFactory;
import org.jboss.pressgang.ccms.wrapper.ContentSpecWrapper;
import org.jboss.resteasy.spi.InternalServerErrorException;
import org.jboss.resteasy.spi.NotFoundException;

public class ContentSpecUtilities extends org.jboss.pressgang.ccms.contentspec.utils.ContentSpecUtilities {

    private ContentSpecUtilities() {
    }

    public static ContentSpecWrapper getContentSpecEntity(final Integer id, final Integer revision, final DBProviderFactory providerFactory) {
        try {
            if (revision == null) {
                return providerFactory.getProvider(ContentSpecProvider.class).getContentSpec(id);
            } else {
                return providerFactory.getProvider(ContentSpecProvider.class).getContentSpec(id, revision);
            }
        } catch (org.jboss.pressgang.ccms.provider.exception.NotFoundException e) {
            throw new NotFoundException(e);
        } catch (org.jboss.pressgang.ccms.provider.exception.InternalServerErrorException e) {
            throw new InternalServerErrorException(e);
        }
    }

    public static org.jboss.pressgang.ccms.contentspec.ContentSpec getContentSpec(final Integer id, final Integer revision,
            final EntityManager entityManager) {
        final DBProviderFactory providerFactory = ProviderUtilities.getDBProviderFactory(entityManager);
        return getContentSpec(id, revision, providerFactory);
    }

    public static org.jboss.pressgang.ccms.contentspec.ContentSpec getContentSpec(final Integer id, final Integer revision,
            final DBProviderFactory providerFactory) {
        final ContentSpecWrapper entity = getContentSpecEntity(id, revision, providerFactory);
        return CSTransformer.transform(entity, providerFactory, false);
    }

    public static String getContentSpecText(final Integer id, final EntityManager entityManager) {
        return getContentSpecText(id, null, entityManager);
    }

    public static String getContentSpecText(final Integer id, final Integer revision, final EntityManager entityManager) {
        return getContentSpecText(id, revision, entityManager, true);
    }

    protected static String getContentSpecText(final Integer id, final Integer revision, final EntityManager entityManager, boolean fix) {
        final DBProviderFactory providerFactory = ProviderUtilities.getDBProviderFactory(entityManager);
        final ContentSpecWrapper entity = getContentSpecEntity(id, revision, providerFactory);

        if (fix && ((ContentSpec) entity.unwrap()).getFailedContentSpec() != null) {
            return fixFailedContentSpec(entity, null, false);
        } else {
            final org.jboss.pressgang.ccms.contentspec.ContentSpec contentSpec = CSTransformer.transform(entity, providerFactory, false);
            return contentSpec.toString(false);
        }
    }

    public static final String fixFailedContentSpec(final ContentSpec contentSpec) {
        return fixFailedContentSpec(contentSpec.getId(), contentSpec.getFailedContentSpec(), null, false);
    }
}