public static FileVersion getLatestFileVersion(FileEntry fileEntry, boolean trusted)
throws PortalException, SystemException {
if (fileEntry instanceof LiferayFileEntry) {
LiferayFileEntry liferayFileEntry = (LiferayFileEntry) fileEntry;
DLFileEntry dlFileEntry = liferayFileEntry.getDLFileEntry();
return new LiferayFileVersion(dlFileEntry.getLatestFileVersion(trusted));
}
return fileEntry.getLatestFileVersion();
}
protected void importMetaData(
PortletDataContext portletDataContext,
Element fileEntryElement,
FileEntry fileEntry,
ServiceContext serviceContext)
throws Exception {
LiferayFileEntry liferayFileEntry = (LiferayFileEntry) fileEntry;
DLFileEntry dlFileEntry = liferayFileEntry.getDLFileEntry();
Map<Long, Long> dlFileEntryTypeIds =
(Map<Long, Long>) portletDataContext.getNewPrimaryKeysMap(DLFileEntryType.class);
long dlFileEntryTypeId =
MapUtil.getLong(
dlFileEntryTypeIds, dlFileEntry.getFileEntryTypeId(), dlFileEntry.getFileEntryTypeId());
DLFileEntryType existingDLFileEntryType =
DLFileEntryTypeLocalServiceUtil.fetchDLFileEntryType(dlFileEntryTypeId);
if (existingDLFileEntryType == null) {
serviceContext.setAttribute("fileEntryTypeId", -1);
return;
}
serviceContext.setAttribute("fileEntryTypeId", existingDLFileEntryType.getFileEntryTypeId());
List<DDMStructure> ddmStructures = existingDLFileEntryType.getDDMStructures();
for (DDMStructure ddmStructure : ddmStructures) {
Element structureFieldsElement =
(Element)
fileEntryElement.selectSingleNode(
"structure-fields[@structureUuid='".concat(ddmStructure.getUuid()).concat("']"));
if (structureFieldsElement == null) {
continue;
}
String path = structureFieldsElement.attributeValue("path");
DDMFormValues ddmFormValues = (DDMFormValues) portletDataContext.getZipEntryAsObject(path);
serviceContext.setAttribute(
DDMFormValues.class.getName() + ddmStructure.getStructureId(), ddmFormValues);
}
}
protected void exportMetaData(
PortletDataContext portletDataContext, Element fileEntryElement, FileEntry fileEntry)
throws Exception {
LiferayFileEntry liferayFileEntry = (LiferayFileEntry) fileEntry;
DLFileEntry dlFileEntry = liferayFileEntry.getDLFileEntry();
long fileEntryTypeId = dlFileEntry.getFileEntryTypeId();
DLFileEntryType dlFileEntryType =
DLFileEntryTypeLocalServiceUtil.fetchFileEntryType(fileEntryTypeId);
if ((dlFileEntryType == null) || !dlFileEntryType.isExportable()) {
return;
}
StagedModelDataHandlerUtil.exportReferenceStagedModel(
portletDataContext, fileEntry, dlFileEntryType, PortletDataContext.REFERENCE_TYPE_STRONG);
List<DDMStructure> ddmStructures = dlFileEntryType.getDDMStructures();
for (DDMStructure ddmStructure : ddmStructures) {
FileVersion fileVersion = fileEntry.getFileVersion();
DLFileEntryMetadata dlFileEntryMetadata =
DLFileEntryMetadataLocalServiceUtil.fetchFileEntryMetadata(
ddmStructure.getStructureId(), fileVersion.getFileVersionId());
if (dlFileEntryMetadata == null) {
continue;
}
Element structureFields = fileEntryElement.addElement("structure-fields");
String path =
ExportImportPathUtil.getModelPath(
ddmStructure, String.valueOf(dlFileEntryMetadata.getDDMStorageId()));
structureFields.addAttribute("path", path);
structureFields.addAttribute("structureUuid", ddmStructure.getUuid());
DDMFormValues ddmFormValues =
StorageEngineUtil.getDDMFormValues(dlFileEntryMetadata.getDDMStorageId());
portletDataContext.addZipEntry(path, ddmFormValues);
}
}
@Before
public void setUp() throws Exception {
FinderCacheUtil.clearCache();
_liveGroup = GroupTestUtil.addGroup();
_stagingGroup = GroupTestUtil.addGroup();
_fileEntry =
DLAppTestUtil.addFileEntry(
_stagingGroup.getGroupId(),
DLFolderConstants.DEFAULT_PARENT_FOLDER_ID,
ServiceTestUtil.randomString() + ".txt",
ServiceTestUtil.randomString(),
true);
LiferayFileEntry liferayFileEntry = (LiferayFileEntry) _fileEntry;
DLFileEntry dlFileEntry = liferayFileEntry.getDLFileEntry();
dlFileEntry.setLargeImageId(dlFileEntry.getFileEntryId());
DLFileEntryLocalServiceUtil.updateDLFileEntry(dlFileEntry);
TestReaderWriter testReaderWriter = new TestReaderWriter();
_portletDataContextExport =
PortletDataContextFactoryUtil.createExportPortletDataContext(
_stagingGroup.getCompanyId(),
_stagingGroup.getGroupId(),
new HashMap<String, String[]>(),
new Date(System.currentTimeMillis() - Time.HOUR),
new Date(),
testReaderWriter);
Element rootElement = SAXReaderUtil.createElement("root");
_portletDataContextExport.setExportDataRootElement(rootElement);
_stagingPrivateLayout =
LayoutTestUtil.addLayout(_stagingGroup.getGroupId(), ServiceTestUtil.randomString(), true);
_stagingPublicLayout =
LayoutTestUtil.addLayout(_stagingGroup.getGroupId(), ServiceTestUtil.randomString(), false);
_portletDataContextExport.setPlid(_stagingPublicLayout.getPlid());
_portletDataContextImport =
PortletDataContextFactoryUtil.createImportPortletDataContext(
_stagingGroup.getCompanyId(),
_stagingGroup.getGroupId(),
new HashMap<String, String[]>(),
new CurrentUserIdStrategy(TestPropsValues.getUser()),
testReaderWriter);
_portletDataContextImport.setImportDataRootElement(rootElement);
_livePublicLayout =
LayoutTestUtil.addLayout(_liveGroup.getGroupId(), ServiceTestUtil.randomString(), false);
_portletDataContextImport.setPlid(_livePublicLayout.getPlid());
_portletDataContextImport.setSourceGroupId(_stagingGroup.getGroupId());
rootElement.addElement("entry");
_referrerStagedModel =
JournalTestUtil.addArticle(
_stagingGroup.getGroupId(),
ServiceTestUtil.randomString(),
ServiceTestUtil.randomString());
}
protected static Object[] setUp(
long repositoryId, String titleSuffix, ServiceContext serviceContext) throws Exception {
Folder folder =
DLAppLocalServiceUtil.addFolder(
TestPropsValues.getUserId(),
repositoryId,
DLFolderConstants.DEFAULT_PARENT_FOLDER_ID,
"Folder A",
StringPool.BLANK,
serviceContext);
DLAppLocalServiceUtil.addFolder(
TestPropsValues.getUserId(),
repositoryId,
folder.getFolderId(),
"Folder B",
StringPool.BLANK,
serviceContext);
Folder folderC =
DLAppLocalServiceUtil.addFolder(
TestPropsValues.getUserId(),
repositoryId,
folder.getFolderId(),
"Folder C",
StringPool.BLANK,
serviceContext);
DLTrashServiceUtil.moveFolderToTrash(folderC.getFolderId());
FileEntry fileEntry =
addFileEntry(
_user.getUserId(),
repositoryId,
folder.getFolderId(),
"FE1.txt",
titleSuffix,
ContentTypes.TEXT_PLAIN,
DLFileEntryTypeConstants.FILE_ENTRY_TYPE_ID_BASIC_DOCUMENT);
LiferayFileEntry liferayFileEntry = (LiferayFileEntry) fileEntry;
DLFileEntry dlFileEntry = liferayFileEntry.getDLFileEntry();
dlFileEntry.setExtraSettings("hello=world");
dlFileEntry.setSmallImageId(_SMALL_IMAGE_ID);
dlFileEntry = DLFileEntryLocalServiceUtil.updateDLFileEntry(dlFileEntry);
DLFileVersion dlFileVersion = dlFileEntry.getFileVersion();
addFileEntry(
TestPropsValues.getUserId(),
repositoryId,
folder.getFolderId(),
"FE2.pdf",
titleSuffix,
ContentTypes.APPLICATION_PDF,
DLFileEntryTypeConstants.FILE_ENTRY_TYPE_ID_ALL);
fileEntry =
addFileEntry(
TestPropsValues.getUserId(),
repositoryId,
folder.getFolderId(),
"FE3.txt",
titleSuffix,
ContentTypes.TEXT_PLAIN,
DLFileEntryTypeConstants.FILE_ENTRY_TYPE_ID_ALL);
fileEntry =
DLAppServiceUtil.updateFileEntry(
fileEntry.getFileEntryId(),
"FE3.txt",
ContentTypes.TEXT_PLAIN,
"FE3.txt".concat(titleSuffix),
StringPool.BLANK,
StringPool.BLANK,
false,
RandomTestUtil.randomBytes(TikaSafeRandomizerBumper.INSTANCE),
serviceContext);
dlFileEntry = ((LiferayFileEntry) fileEntry).getDLFileEntry();
dlFileEntry.setDescription("FE3.txt");
DLFileEntryLocalServiceUtil.updateDLFileEntry(dlFileEntry);
DLFileVersion dlFileVersion3 = dlFileEntry.getFileVersion();
dlFileVersion3.setExtraSettings("hello=world");
DLFileVersionLocalServiceUtil.updateDLFileVersion(dlFileVersion3);
DLTrashServiceUtil.moveFileEntryToTrash(fileEntry.getFileEntryId());
return new Object[] {folder, dlFileVersion};
}
@Override
protected void doImportStagedModel(PortletDataContext portletDataContext, FileEntry fileEntry)
throws Exception {
long userId = portletDataContext.getUserId(fileEntry.getUserUuid());
if (!fileEntry.isDefaultRepository()) {
// References has been automatically imported, nothing to do here
return;
}
Map<Long, Long> folderIds =
(Map<Long, Long>) portletDataContext.getNewPrimaryKeysMap(Folder.class);
long folderId = MapUtil.getLong(folderIds, fileEntry.getFolderId(), fileEntry.getFolderId());
long[] assetCategoryIds =
portletDataContext.getAssetCategoryIds(DLFileEntry.class, fileEntry.getFileEntryId());
String[] assetTagNames =
portletDataContext.getAssetTagNames(DLFileEntry.class, fileEntry.getFileEntryId());
ServiceContext serviceContext =
portletDataContext.createServiceContext(fileEntry, DLFileEntry.class);
serviceContext.setAttribute("sourceFileName", "A." + fileEntry.getExtension());
serviceContext.setUserId(userId);
Element fileEntryElement = portletDataContext.getImportDataElement(fileEntry);
String binPath = fileEntryElement.attributeValue("bin-path");
InputStream is = null;
if (Validator.isNull(binPath) && portletDataContext.isPerformDirectBinaryImport()) {
try {
is = FileEntryUtil.getContentStream(fileEntry);
} catch (Exception e) {
if (_log.isWarnEnabled()) {
_log.warn("Unable to retrieve content for file entry " + fileEntry.getFileEntryId(), e);
}
return;
}
} else {
is = portletDataContext.getZipEntryAsInputStream(binPath);
}
if (is == null) {
if (_log.isWarnEnabled()) {
_log.warn("No file found for file entry " + fileEntry.getFileEntryId());
}
return;
}
importMetaData(portletDataContext, fileEntryElement, fileEntry, serviceContext);
FileEntry importedFileEntry = null;
String titleWithExtension = DLUtil.getTitleWithExtension(fileEntry);
String extension = fileEntry.getExtension();
String periodAndExtension = StringPool.PERIOD.concat(extension);
if (portletDataContext.isDataStrategyMirror()) {
FileEntry existingFileEntry =
fetchStagedModelByUuidAndGroupId(
fileEntry.getUuid(), portletDataContext.getScopeGroupId());
FileVersion fileVersion = fileEntry.getFileVersion();
if (existingFileEntry == null) {
String fileEntryTitle = fileEntry.getTitle();
FileEntry existingTitleFileEntry =
FileEntryUtil.fetchByR_F_T(
portletDataContext.getScopeGroupId(), folderId, fileEntryTitle);
if (existingTitleFileEntry != null) {
if ((fileEntry.getGroupId() == portletDataContext.getSourceGroupId())
&& portletDataContext.isDataStrategyMirrorWithOverwriting()) {
DLAppLocalServiceUtil.deleteFileEntry(existingTitleFileEntry.getFileEntryId());
} else {
boolean titleHasExtension = false;
if (fileEntryTitle.endsWith(periodAndExtension)) {
fileEntryTitle = FileUtil.stripExtension(fileEntryTitle);
titleHasExtension = true;
}
for (int i = 1; ; i++) {
fileEntryTitle += StringPool.SPACE + i;
titleWithExtension = fileEntryTitle + periodAndExtension;
existingTitleFileEntry =
FileEntryUtil.fetchByR_F_T(
portletDataContext.getScopeGroupId(), folderId, titleWithExtension);
if (existingTitleFileEntry == null) {
if (titleHasExtension) {
fileEntryTitle += periodAndExtension;
}
break;
}
}
}
}
serviceContext.setAttribute("fileVersionUuid", fileVersion.getUuid());
serviceContext.setUuid(fileEntry.getUuid());
importedFileEntry =
DLAppLocalServiceUtil.addFileEntry(
userId,
portletDataContext.getScopeGroupId(),
folderId,
titleWithExtension,
fileEntry.getMimeType(),
fileEntryTitle,
fileEntry.getDescription(),
null,
is,
fileEntry.getSize(),
serviceContext);
if (fileEntry.isInTrash()) {
importedFileEntry =
DLAppServiceUtil.moveFileEntryToTrash(importedFileEntry.getFileEntryId());
}
} else {
FileVersion latestExistingFileVersion = existingFileEntry.getLatestFileVersion(true);
boolean indexEnabled = serviceContext.isIndexingEnabled();
boolean deleteFileEntry = false;
boolean updateFileEntry = false;
if (!Validator.equals(fileVersion.getUuid(), latestExistingFileVersion.getUuid())) {
deleteFileEntry = true;
updateFileEntry = true;
} else {
InputStream existingFileVersionInputStream = null;
try {
existingFileVersionInputStream = latestExistingFileVersion.getContentStream(false);
} catch (Exception e) {
if (_log.isDebugEnabled()) {
_log.debug(e, e);
}
} finally {
if (existingFileVersionInputStream != null) {
existingFileVersionInputStream.close();
}
}
if (existingFileVersionInputStream == null) {
updateFileEntry = true;
}
}
try {
serviceContext.setIndexingEnabled(false);
if (updateFileEntry) {
DLFileVersion alreadyExistingFileVersion =
DLFileVersionLocalServiceUtil.getFileVersionByUuidAndGroupId(
fileVersion.getUuid(), existingFileEntry.getGroupId());
if (alreadyExistingFileVersion != null) {
serviceContext.setAttribute(
"existingDLFileVersionId", alreadyExistingFileVersion.getFileVersionId());
}
serviceContext.setUuid(fileVersion.getUuid());
importedFileEntry =
DLAppLocalServiceUtil.updateFileEntry(
userId,
existingFileEntry.getFileEntryId(),
titleWithExtension,
fileEntry.getMimeType(),
fileEntry.getTitle(),
fileEntry.getDescription(),
null,
false,
is,
fileEntry.getSize(),
serviceContext);
} else {
DLAppLocalServiceUtil.updateAsset(
userId, existingFileEntry,
latestExistingFileVersion, assetCategoryIds,
assetTagNames, null);
importedFileEntry = existingFileEntry;
}
if (importedFileEntry.getFolderId() != folderId) {
importedFileEntry =
DLAppLocalServiceUtil.moveFileEntry(
userId, importedFileEntry.getFileEntryId(), folderId, serviceContext);
}
if (importedFileEntry instanceof LiferayFileEntry) {
LiferayFileEntry liferayFileEntry = (LiferayFileEntry) importedFileEntry;
Indexer<DLFileEntry> indexer =
IndexerRegistryUtil.nullSafeGetIndexer(DLFileEntry.class);
indexer.reindex((DLFileEntry) liferayFileEntry.getModel());
}
if (deleteFileEntry && ExportImportThreadLocal.isStagingInProcess()) {
DLAppServiceUtil.deleteFileVersion(
latestExistingFileVersion.getFileEntryId(), latestExistingFileVersion.getVersion());
}
} finally {
serviceContext.setIndexingEnabled(indexEnabled);
}
}
} else {
try {
importedFileEntry =
DLAppLocalServiceUtil.addFileEntry(
userId,
portletDataContext.getScopeGroupId(),
folderId,
titleWithExtension,
fileEntry.getMimeType(),
fileEntry.getTitle(),
fileEntry.getDescription(),
null,
is,
fileEntry.getSize(),
serviceContext);
} catch (DuplicateFileException dfe) {
String title = fileEntry.getTitle();
String[] titleParts = title.split("\\.", 2);
title = titleParts[0] + StringUtil.randomString();
if (titleParts.length > 1) {
title += StringPool.PERIOD + titleParts[1];
}
if (!title.endsWith(periodAndExtension)) {
title += periodAndExtension;
}
importedFileEntry =
DLAppLocalServiceUtil.addFileEntry(
userId,
portletDataContext.getScopeGroupId(),
folderId,
title,
fileEntry.getMimeType(),
title,
fileEntry.getDescription(),
null,
is,
fileEntry.getSize(),
serviceContext);
}
}
if (portletDataContext.getBooleanParameter(
DLPortletDataHandler.NAMESPACE, "previews-and-thumbnails")) {
DLProcessorRegistryUtil.importGeneratedFiles(
portletDataContext, fileEntry, importedFileEntry, fileEntryElement);
}
portletDataContext.importClassedModel(fileEntry, importedFileEntry, DLFileEntry.class);
Map<Long, Long> fileEntryIds =
(Map<Long, Long>) portletDataContext.getNewPrimaryKeysMap(FileEntry.class);
fileEntryIds.put(fileEntry.getFileEntryId(), importedFileEntry.getFileEntryId());
}
@Override
protected void doExportStagedModel(PortletDataContext portletDataContext, FileEntry fileEntry)
throws Exception {
Element fileEntryElement = portletDataContext.getExportDataElement(fileEntry);
String fileEntryPath = ExportImportPathUtil.getModelPath(fileEntry);
if (!fileEntry.isDefaultRepository()) {
Repository repository = RepositoryLocalServiceUtil.getRepository(fileEntry.getRepositoryId());
StagedModelDataHandlerUtil.exportReferenceStagedModel(
portletDataContext, fileEntry, repository, PortletDataContext.REFERENCE_TYPE_STRONG);
portletDataContext.addClassedModel(fileEntryElement, fileEntryPath, fileEntry);
long portletRepositoryClassNameId =
PortalUtil.getClassNameId(PortletRepository.class.getName());
if (repository.getClassNameId() != portletRepositoryClassNameId) {
return;
}
}
if (fileEntry.getFolderId() != DLFolderConstants.DEFAULT_PARENT_FOLDER_ID) {
StagedModelDataHandlerUtil.exportReferenceStagedModel(
portletDataContext,
fileEntry,
fileEntry.getFolder(),
PortletDataContext.REFERENCE_TYPE_PARENT);
}
LiferayFileEntry liferayFileEntry = (LiferayFileEntry) fileEntry;
liferayFileEntry.setCachedFileVersion(fileEntry.getFileVersion());
if (!portletDataContext.isPerformDirectBinaryImport()) {
InputStream is = null;
try {
is = FileEntryUtil.getContentStream(fileEntry);
} catch (Exception e) {
if (_log.isWarnEnabled()) {
_log.warn("Unable to retrieve content for file entry " + fileEntry.getFileEntryId(), e);
}
}
if (is == null) {
fileEntryElement.detach();
return;
}
try {
String binPath = ExportImportPathUtil.getModelPath(fileEntry, fileEntry.getVersion());
portletDataContext.addZipEntry(binPath, is);
fileEntryElement.addAttribute("bin-path", binPath);
} finally {
try {
is.close();
} catch (IOException ioe) {
_log.error(ioe, ioe);
}
}
}
if (portletDataContext.getBooleanParameter(
DLPortletDataHandler.NAMESPACE, "previews-and-thumbnails")) {
DLProcessorRegistryUtil.exportGeneratedFiles(portletDataContext, fileEntry, fileEntryElement);
}
exportMetaData(portletDataContext, fileEntryElement, fileEntry);
portletDataContext.addClassedModel(
fileEntryElement, fileEntryPath, liferayFileEntry, DLFileEntry.class);
}