PageImportTask.java

/*
 * JBoss, Home of Professional Open Source.
 * Copyright 2011, Red Hat, Inc., and individual contributors
 * as indicated by the @author tags. See the copyright.txt file in the
 * distribution for a full listing of individual contributors.
 *
 * This 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 2.1 of
 * the License, or (at your option) any later version.
 *
 * This software 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 this software; if not, write to the Free
 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 */

package org.exoplatform.portal.mop.management.exportimport;

import java.util.ArrayList;
import java.util.List;

import org.exoplatform.portal.config.DataStorage;
import org.exoplatform.portal.config.model.Page;
import org.exoplatform.portal.mop.SiteKey;
import org.exoplatform.portal.mop.importer.ImportMode;
import org.exoplatform.portal.mop.management.operations.MOPSiteProvider;
import org.exoplatform.portal.mop.management.operations.page.PageUtils;
import org.exoplatform.portal.mop.page.PageContext;
import org.exoplatform.portal.mop.page.PageKey;
import org.exoplatform.portal.mop.page.PageService;
import org.gatein.mop.api.workspace.Site;

/**
 * @author <a href="mailto:nscavell@redhat.com">Nick Scavelli</a>
 * @version $Revision$
 */
public class PageImportTask extends AbstractImportTask<Page.PageSet> {
    private final DataStorage dataStorage;
    private final PageService pageService;
    private final MOPSiteProvider siteProvider;
    private Page.PageSet rollbackSaves;
    private Page.PageSet rollbackDeletes;

    public PageImportTask(Page.PageSet data, SiteKey siteKey, DataStorage dataStorage, PageService pageService,
            MOPSiteProvider siteProvider) {
        super(data, siteKey);
        this.dataStorage = dataStorage;
        this.pageService = pageService;
        this.siteProvider = siteProvider;
    }

    @Override
    public void importData(ImportMode importMode) throws Exception {
        if (data == null || data.getPages() == null || data.getPages().isEmpty())
            return;

        Site site = siteProvider.getSite(siteKey);
        if (site == null)
            throw new Exception("Cannot import pages because site does not exist for " + siteKey);
        org.gatein.mop.api.workspace.Page pages = site.getRootPage().getChild("pages");
        int size = (pages == null) ? 0 : pages.getChildren().size();

        Page.PageSet dst = null;
        switch (importMode) {
            case CONSERVE:

                // No pages exist yet.
                if (size == 0) {
                    dst = data;
                    rollbackDeletes = data;
                } else {
                    dst = null;
                }
                break;
            case INSERT:

                // No pages exist yet.
                if (size == 0) {
                    dst = data;
                    rollbackDeletes = data;
                } else {
                    dst = new Page.PageSet();
                    rollbackDeletes = new Page.PageSet();
                    for (Page src : data.getPages()) {
                        if (pages.getChild(src.getName()) == null) {
                            dst.getPages().add(src);
                            rollbackDeletes.getPages().add(src);
                        }
                    }
                }
                break;
            case MERGE:

                // No pages exist yet.
                if (size == 0) {
                    dst = data;
                    rollbackDeletes = data;
                } else {
                    dst = new Page.PageSet();
                    rollbackSaves = new Page.PageSet();
                    rollbackDeletes = new Page.PageSet();
                    for (Page src : data.getPages()) {
                        dst.getPages().add(src);
                        PageKey pageKey = siteKey.page(src.getName());
                        if (pages.getChild(src.getName()) == null) {
                            rollbackDeletes.getPages().add(src);
                        } else {
                            PageContext pageContext = pageService.loadPage(pageKey);
                            Page existing = dataStorage.getPage(pageKey.format());
                            pageContext.update(existing);
                            rollbackSaves.getPages().add(PageUtils.copy(existing));
                        }
                    }
                }
                break;
            case OVERWRITE:

                // No pages exist yet.
                if (size == 0) {
                    dst = data;
                    rollbackDeletes = data;
                } else {
                    List<Page> list = PageUtils.getAllPages(dataStorage, pageService, siteKey).getPages();
                    rollbackSaves = new Page.PageSet();
                    rollbackSaves.setPages(new ArrayList<Page>(list.size()));
                    rollbackDeletes = new Page.PageSet();
                    for (Page page : list) {
                        Page copy = PageUtils.copy(page);
                        pageService.destroyPage(siteKey.page(page.getName()));
                        dataStorage.save();
                        rollbackSaves.getPages().add(copy);
                    }
                    for (Page src : data.getPages()) {
                        Page found = findPage(list, src);
                        if (found == null) {
                            rollbackDeletes.getPages().add(src);
                        }
                    }

                    dst = data;
                }
                break;
        }

        if (dst != null) {
            for (Page page : dst.getPages()) {
                pageService.savePage(new PageContext(siteKey.page(page.getName()), PageUtils.toPageState(page)));
                dataStorage.save(page);
                dataStorage.save();
            }
        }
    }

    @Override
    public void rollback() throws Exception {
        if (rollbackDeletes != null && !rollbackDeletes.getPages().isEmpty()) {
            for (Page page : rollbackDeletes.getPages()) {
                pageService.destroyPage(siteKey.page(page.getName()));
                dataStorage.save();
            }
        }
        if (rollbackSaves != null && !rollbackSaves.getPages().isEmpty()) {
            for (Page page : rollbackSaves.getPages()) {
                pageService.savePage(new PageContext(siteKey.page(page.getName()), PageUtils.toPageState(page)));
                dataStorage.save(page);
                dataStorage.save();
            }
        }
    }

    Page.PageSet getRollbackSaves() {
        return rollbackSaves;
    }

    Page.PageSet getRollbackDeletes() {
        return rollbackDeletes;
    }

    private static Page findPage(List<Page> pages, Page src) {
        Page found = null;
        for (Page page : pages) {
            if (src.getName().equals(page.getName())) {
                found = page;
            }
        }
        return found;
    }
}