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 StagedModelDataHandlerUtil.importReferenceStagedModel(
097 portletDataContext, folder, BookmarksFolder.class,
098 folder.getParentFolderId());
099 }
100
101 Map<Long, Long> folderIds =
102 (Map<Long, Long>)portletDataContext.getNewPrimaryKeysMap(
103 BookmarksFolder.class);
104
105 long parentFolderId = MapUtil.getLong(
106 folderIds, folder.getParentFolderId(), folder.getParentFolderId());
107
108 ServiceContext serviceContext = portletDataContext.createServiceContext(
109 folder);
110
111 BookmarksFolder importedFolder = null;
112
113 if (portletDataContext.isDataStrategyMirror()) {
114 BookmarksFolder existingFolder =
115 BookmarksFolderLocalServiceUtil.
116 fetchBookmarksFolderByUuidAndGroupId(
117 folder.getUuid(), portletDataContext.getScopeGroupId());
118
119 if (existingFolder == null) {
120 serviceContext.setUuid(folder.getUuid());
121
122 importedFolder = BookmarksFolderLocalServiceUtil.addFolder(
123 userId, parentFolderId, folder.getName(),
124 folder.getDescription(), serviceContext);
125 }
126 else {
127 importedFolder = BookmarksFolderLocalServiceUtil.updateFolder(
128 userId, existingFolder.getFolderId(), parentFolderId,
129 folder.getName(), folder.getDescription(), false,
130 serviceContext);
131 }
132 }
133 else {
134 importedFolder = BookmarksFolderLocalServiceUtil.addFolder(
135 userId, parentFolderId, folder.getName(),
136 folder.getDescription(), serviceContext);
137 }
138
139 portletDataContext.importClassedModel(folder, importedFolder);
140 }
141
142 @Override
143 protected void doRestoreStagedModel(
144 PortletDataContext portletDataContext, BookmarksFolder folder)
145 throws Exception {
146
147 long userId = portletDataContext.getUserId(folder.getUserUuid());
148
149 BookmarksFolder existingFolder =
150 BookmarksFolderLocalServiceUtil.
151 fetchBookmarksFolderByUuidAndGroupId(
152 folder.getUuid(), portletDataContext.getScopeGroupId());
153
154 if ((existingFolder == null) || !existingFolder.isInTrash()) {
155 return;
156 }
157
158 TrashHandler trashHandler = existingFolder.getTrashHandler();
159
160 if (trashHandler.isRestorable(existingFolder.getFolderId())) {
161 trashHandler.restoreTrashEntry(
162 userId, existingFolder.getFolderId());
163 }
164 }
165
166 @Override
167 protected boolean validateMissingReference(
168 String uuid, long companyId, long groupId)
169 throws Exception {
170
171 BookmarksFolder folder =
172 BookmarksFolderLocalServiceUtil.
173 fetchBookmarksFolderByUuidAndGroupId(uuid, groupId);
174
175 if (folder == null) {
176 return false;
177 }
178
179 return true;
180 }
181
182 }