001
014
015 package com.liferay.portlet.bookmarks.lar;
016
017 import com.liferay.portal.kernel.exception.PortalException;
018 import com.liferay.portal.kernel.exception.SystemException;
019 import com.liferay.portal.kernel.lar.BaseStagedModelDataHandler;
020 import com.liferay.portal.kernel.lar.ExportImportPathUtil;
021 import com.liferay.portal.kernel.lar.PortletDataContext;
022 import com.liferay.portal.kernel.lar.StagedModelDataHandlerUtil;
023 import com.liferay.portal.kernel.util.MapUtil;
024 import com.liferay.portal.kernel.xml.Element;
025 import com.liferay.portal.service.ServiceContext;
026 import com.liferay.portlet.bookmarks.model.BookmarksFolder;
027 import com.liferay.portlet.bookmarks.model.BookmarksFolderConstants;
028 import com.liferay.portlet.bookmarks.service.BookmarksFolderLocalServiceUtil;
029
030 import java.util.Map;
031
032
036 public class BookmarksFolderStagedModelDataHandler
037 extends BaseStagedModelDataHandler<BookmarksFolder> {
038
039 public static final String[] CLASS_NAMES =
040 {BookmarksFolder.class.getName()};
041
042 @Override
043 public void deleteStagedModel(
044 String uuid, long groupId, String className, String extraData)
045 throws PortalException, SystemException {
046
047 BookmarksFolder folder =
048 BookmarksFolderLocalServiceUtil.
049 fetchBookmarksFolderByUuidAndGroupId(uuid, groupId);
050
051 if (folder != null) {
052 BookmarksFolderLocalServiceUtil.deleteFolder(folder);
053 }
054 }
055
056 @Override
057 public String[] getClassNames() {
058 return CLASS_NAMES;
059 }
060
061 @Override
062 public String getDisplayName(BookmarksFolder folder) {
063 return folder.getName();
064 }
065
066 @Override
067 protected void doExportStagedModel(
068 PortletDataContext portletDataContext, BookmarksFolder folder)
069 throws Exception {
070
071 if (folder.getParentFolderId() !=
072 BookmarksFolderConstants.DEFAULT_PARENT_FOLDER_ID) {
073
074 StagedModelDataHandlerUtil.exportReferenceStagedModel(
075 portletDataContext, folder, folder.getParentFolder(),
076 PortletDataContext.REFERENCE_TYPE_PARENT);
077 }
078
079 Element folderElement = portletDataContext.getExportDataElement(folder);
080
081 portletDataContext.addClassedModel(
082 folderElement, ExportImportPathUtil.getModelPath(folder), folder,
083 BookmarksPortletDataHandler.NAMESPACE);
084 }
085
086 @Override
087 protected void doImportStagedModel(
088 PortletDataContext portletDataContext, BookmarksFolder folder)
089 throws Exception {
090
091 long userId = portletDataContext.getUserId(folder.getUserUuid());
092
093 if (folder.getParentFolderId() !=
094 BookmarksFolderConstants.DEFAULT_PARENT_FOLDER_ID) {
095
096 String parentFolderPath = ExportImportPathUtil.getModelPath(
097 portletDataContext, BookmarksFolder.class.getName(),
098 folder.getParentFolderId());
099
100 BookmarksFolder parentFolder =
101 (BookmarksFolder)portletDataContext.getZipEntryAsObject(
102 parentFolderPath);
103
104 StagedModelDataHandlerUtil.importReferenceStagedModel(
105 portletDataContext, parentFolder);
106 }
107
108 Map<Long, Long> folderIds =
109 (Map<Long, Long>)portletDataContext.getNewPrimaryKeysMap(
110 BookmarksFolder.class);
111
112 long parentFolderId = MapUtil.getLong(
113 folderIds, folder.getParentFolderId(), folder.getParentFolderId());
114
115 ServiceContext serviceContext = portletDataContext.createServiceContext(
116 folder, BookmarksPortletDataHandler.NAMESPACE);
117
118 BookmarksFolder importedFolder = null;
119
120 if (portletDataContext.isDataStrategyMirror()) {
121 BookmarksFolder existingFolder =
122 BookmarksFolderLocalServiceUtil.
123 fetchBookmarksFolderByUuidAndGroupId(
124 folder.getUuid(), portletDataContext.getScopeGroupId());
125
126 if (existingFolder == null) {
127 serviceContext.setUuid(folder.getUuid());
128
129 importedFolder = BookmarksFolderLocalServiceUtil.addFolder(
130 userId, parentFolderId, folder.getName(),
131 folder.getDescription(), serviceContext);
132 }
133 else {
134 importedFolder = BookmarksFolderLocalServiceUtil.updateFolder(
135 userId, existingFolder.getFolderId(), parentFolderId,
136 folder.getName(), folder.getDescription(), false,
137 serviceContext);
138 }
139 }
140 else {
141 importedFolder = BookmarksFolderLocalServiceUtil.addFolder(
142 userId, parentFolderId, folder.getName(),
143 folder.getDescription(), serviceContext);
144 }
145
146 portletDataContext.importClassedModel(
147 folder, importedFolder, BookmarksPortletDataHandler.NAMESPACE);
148 }
149
150 @Override
151 protected boolean validateMissingReference(
152 String uuid, long companyId, long groupId)
153 throws Exception {
154
155 BookmarksFolder folder =
156 BookmarksFolderLocalServiceUtil.
157 fetchBookmarksFolderByUuidAndGroupId(uuid, groupId);
158
159 if (folder == null) {
160 return false;
161 }
162
163 return true;
164 }
165
166 }