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.trash.TrashHandler;
024 import com.liferay.portal.kernel.util.MapUtil;
025 import com.liferay.portal.kernel.xml.Element;
026 import com.liferay.portal.service.ServiceContext;
027 import com.liferay.portlet.bookmarks.model.BookmarksFolder;
028 import com.liferay.portlet.bookmarks.model.BookmarksFolderConstants;
029 import com.liferay.portlet.bookmarks.service.BookmarksFolderLocalServiceUtil;
030
031 import java.util.Map;
032
033
037 public class BookmarksFolderStagedModelDataHandler
038 extends BaseStagedModelDataHandler<BookmarksFolder> {
039
040 public static final String[] CLASS_NAMES =
041 {BookmarksFolder.class.getName()};
042
043 @Override
044 public void deleteStagedModel(
045 String uuid, long groupId, String className, String extraData)
046 throws PortalException, SystemException {
047
048 BookmarksFolder folder =
049 BookmarksFolderLocalServiceUtil.
050 fetchBookmarksFolderByUuidAndGroupId(uuid, groupId);
051
052 if (folder != null) {
053 BookmarksFolderLocalServiceUtil.deleteFolder(folder);
054 }
055 }
056
057 @Override
058 public String[] getClassNames() {
059 return CLASS_NAMES;
060 }
061
062 @Override
063 public String getDisplayName(BookmarksFolder folder) {
064 return folder.getName();
065 }
066
067 @Override
068 protected void doExportStagedModel(
069 PortletDataContext portletDataContext, BookmarksFolder folder)
070 throws Exception {
071
072 if (folder.getParentFolderId() !=
073 BookmarksFolderConstants.DEFAULT_PARENT_FOLDER_ID) {
074
075 StagedModelDataHandlerUtil.exportReferenceStagedModel(
076 portletDataContext, folder, folder.getParentFolder(),
077 PortletDataContext.REFERENCE_TYPE_PARENT);
078 }
079
080 Element folderElement = portletDataContext.getExportDataElement(folder);
081
082 portletDataContext.addClassedModel(
083 folderElement, ExportImportPathUtil.getModelPath(folder), folder);
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);
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(folder, importedFolder);
147 }
148
149 @Override
150 protected void doRestoreStagedModel(
151 PortletDataContext portletDataContext, BookmarksFolder folder)
152 throws Exception {
153
154 long userId = portletDataContext.getUserId(folder.getUserUuid());
155
156 BookmarksFolder existingFolder =
157 BookmarksFolderLocalServiceUtil.
158 fetchBookmarksFolderByUuidAndGroupId(
159 folder.getUuid(), portletDataContext.getScopeGroupId());
160
161 if ((existingFolder == null) || !existingFolder.isInTrash()) {
162 return;
163 }
164
165 TrashHandler trashHandler = existingFolder.getTrashHandler();
166
167 if (trashHandler.isRestorable(existingFolder.getFolderId())) {
168 trashHandler.restoreTrashEntry(
169 userId, existingFolder.getFolderId());
170 }
171 }
172
173 @Override
174 protected boolean validateMissingReference(
175 String uuid, long companyId, long groupId)
176 throws Exception {
177
178 BookmarksFolder folder =
179 BookmarksFolderLocalServiceUtil.
180 fetchBookmarksFolderByUuidAndGroupId(uuid, groupId);
181
182 if (folder == null) {
183 return false;
184 }
185
186 return true;
187 }
188
189 }