001    /**
002     * Copyright (c) 2000-2013 Liferay, Inc. All rights reserved.
003     *
004     * This library is free software; you can redistribute it and/or modify it under
005     * the terms of the GNU Lesser General Public License as published by the Free
006     * Software Foundation; either version 2.1 of the License, or (at your option)
007     * any later version.
008     *
009     * This library is distributed in the hope that it will be useful, but WITHOUT
010     * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
011     * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
012     * details.
013     */
014    
015    package com.liferay.portlet.bookmarks.lar;
016    
017    import com.liferay.portal.kernel.lar.BaseStagedModelDataHandler;
018    import com.liferay.portal.kernel.lar.ExportImportPathUtil;
019    import com.liferay.portal.kernel.lar.PortletDataContext;
020    import com.liferay.portal.kernel.util.MapUtil;
021    import com.liferay.portal.kernel.xml.Element;
022    import com.liferay.portal.service.ServiceContext;
023    import com.liferay.portlet.bookmarks.model.BookmarksFolder;
024    import com.liferay.portlet.bookmarks.model.BookmarksFolderConstants;
025    import com.liferay.portlet.bookmarks.service.BookmarksFolderLocalServiceUtil;
026    import com.liferay.portlet.bookmarks.service.persistence.BookmarksFolderUtil;
027    
028    import java.util.Map;
029    
030    /**
031     * @author Mate Thurzo
032     * @author Daniel Kocsis
033     */
034    public class BookmarksFolderStagedModelDataHandler
035            extends BaseStagedModelDataHandler<BookmarksFolder> {
036    
037            public static final String[] CLASS_NAMES =
038                    {BookmarksFolder.class.getName()};
039    
040            @Override
041            public String[] getClassNames() {
042                    return CLASS_NAMES;
043            }
044    
045            @Override
046            protected void doExportStagedModel(
047                            PortletDataContext portletDataContext, BookmarksFolder folder)
048                    throws Exception {
049    
050                    if (folder.getParentFolderId() !=
051                                    BookmarksFolderConstants.DEFAULT_PARENT_FOLDER_ID) {
052    
053                            exportStagedModel(portletDataContext, folder.getParentFolder());
054                    }
055    
056                    Element folderElement =
057                            portletDataContext.getExportDataStagedModelElement(folder);
058    
059                    portletDataContext.addClassedModel(
060                            folderElement, ExportImportPathUtil.getModelPath(folder), folder,
061                            BookmarksPortletDataHandler.NAMESPACE);
062            }
063    
064            @Override
065            protected void doImportStagedModel(
066                            PortletDataContext portletDataContext, BookmarksFolder folder)
067                    throws Exception {
068    
069                    long userId = portletDataContext.getUserId(folder.getUserUuid());
070    
071                    Map<Long, Long> folderIds =
072                            (Map<Long, Long>)portletDataContext.getNewPrimaryKeysMap(
073                                    BookmarksFolder.class);
074    
075                    long parentFolderId = MapUtil.getLong(
076                            folderIds, folder.getParentFolderId(), folder.getParentFolderId());
077    
078                    if ((parentFolderId !=
079                                    BookmarksFolderConstants.DEFAULT_PARENT_FOLDER_ID) &&
080                            (parentFolderId == folder.getParentFolderId())) {
081    
082                            String parentFolderPath = ExportImportPathUtil.getModelPath(
083                                    portletDataContext, BookmarksFolder.class.getName(),
084                                    parentFolderId);
085    
086                            BookmarksFolder parentFolder =
087                                    (BookmarksFolder)portletDataContext.getZipEntryAsObject(
088                                            parentFolderPath);
089    
090                            importStagedModel(portletDataContext, parentFolder);
091    
092                            parentFolderId = MapUtil.getLong(
093                                    folderIds, folder.getParentFolderId(),
094                                    folder.getParentFolderId());
095                    }
096    
097                    ServiceContext serviceContext = portletDataContext.createServiceContext(
098                            folder, BookmarksPortletDataHandler.NAMESPACE);
099    
100                    BookmarksFolder importedFolder = null;
101    
102                    if (portletDataContext.isDataStrategyMirror()) {
103                            BookmarksFolder existingFolder = BookmarksFolderUtil.fetchByUUID_G(
104                                    folder.getUuid(), portletDataContext.getScopeGroupId());
105    
106                            if (existingFolder == null) {
107                                    serviceContext.setUuid(folder.getUuid());
108    
109                                    importedFolder = BookmarksFolderLocalServiceUtil.addFolder(
110                                            userId, parentFolderId, folder.getName(),
111                                            folder.getDescription(), serviceContext);
112                            }
113                            else {
114                                    importedFolder = BookmarksFolderLocalServiceUtil.updateFolder(
115                                            userId, existingFolder.getFolderId(), parentFolderId,
116                                            folder.getName(), folder.getDescription(), false,
117                                            serviceContext);
118                            }
119                    }
120                    else {
121                            importedFolder = BookmarksFolderLocalServiceUtil.addFolder(
122                                    userId, parentFolderId, folder.getName(),
123                                    folder.getDescription(), serviceContext);
124                    }
125    
126                    portletDataContext.importClassedModel(
127                            folder, importedFolder, BookmarksPortletDataHandler.NAMESPACE);
128            }
129    
130    }