@Override
public void postProcessSearchQuery(BooleanQuery searchQuery, SearchContext searchContext)
throws Exception {
Set<DDMStructure> ddmStructuresSet = new TreeSet<DDMStructure>();
long[] groupIds = searchContext.getGroupIds();
if ((groupIds != null) && (groupIds.length > 0)) {
List<DLFileEntryType> dlFileEntryTypes =
DLFileEntryTypeLocalServiceUtil.getFileEntryTypes(groupIds);
for (DLFileEntryType dlFileEntryType : dlFileEntryTypes) {
ddmStructuresSet.addAll(dlFileEntryType.getDDMStructures());
}
}
Group group = GroupLocalServiceUtil.getCompanyGroup(searchContext.getCompanyId());
DDMStructure tikaRawMetadataStructure =
DDMStructureLocalServiceUtil.fetchStructure(
group.getGroupId(),
PortalUtil.getClassNameId(RawMetadataProcessor.class),
"TikaRawMetadata");
if (tikaRawMetadataStructure != null) {
ddmStructuresSet.add(tikaRawMetadataStructure);
}
for (DDMStructure ddmStructure : ddmStructuresSet) {
addSearchDDMStruture(searchQuery, searchContext, ddmStructure);
}
String keywords = searchContext.getKeywords();
if (Validator.isNull(keywords)) {
addSearchTerm(searchQuery, searchContext, Field.DESCRIPTION, false);
addSearchTerm(searchQuery, searchContext, Field.TITLE, false);
addSearchTerm(searchQuery, searchContext, Field.USER_NAME, false);
}
addSearchTerm(searchQuery, searchContext, "extension", false);
addSearchTerm(searchQuery, searchContext, "fileEntryTypeId", false);
addSearchTerm(searchQuery, searchContext, "path", false);
LinkedHashMap<String, Object> params =
(LinkedHashMap<String, Object>) searchContext.getAttribute("params");
if (params != null) {
String expandoAttributes = (String) params.get("expandoAttributes");
if (Validator.isNotNull(expandoAttributes)) {
addSearchExpando(searchQuery, searchContext, expandoAttributes);
}
}
}
@Before
@Override
public void setUp() throws Exception {
super.setUp();
SimpleAction simpleAction = new AddDefaultDocumentLibraryStructuresAction();
String companyIdString = String.valueOf(TestPropsValues.getCompanyId());
simpleAction.run(new String[] {companyIdString});
List<DLFileEntryType> dlFileEntryTypes =
DLFileEntryTypeLocalServiceUtil.getFileEntryTypes(
PortalUtil.getCurrentAndAncestorSiteGroupIds(group.getGroupId()));
for (DLFileEntryType dlFileEntryType : dlFileEntryTypes) {
String name = dlFileEntryType.getName(LocaleUtil.getSiteDefault());
if (name.equals(DLFileEntryTypeConstants.NAME_CONTRACT)) {
_contractDLFileEntryTypeId = dlFileEntryType.getFileEntryTypeId();
}
}
ExpandoTable expandoTable =
ExpandoTableLocalServiceUtil.addDefaultTable(
PortalUtil.getDefaultCompanyId(), DLFileEntry.class.getName());
ExpandoColumnLocalServiceUtil.addColumn(
expandoTable.getTableId(),
_EXPANDO_ATTRIBUTE_NAME,
ExpandoColumnConstants.STRING,
StringPool.BLANK);
_serviceContext = getServiceContext();
FileEntry fileEntry =
DLAppServiceUtil.addFileEntry(
group.getGroupId(),
parentFolder.getFolderId(),
_SOURCE_FILE_NAME,
ContentTypes.APPLICATION_OCTET_STREAM,
_TITLE,
StringPool.BLANK,
StringPool.BLANK,
_DATA_VERSION_1,
_serviceContext);
_fileVersion =
DLFileVersionLocalServiceUtil.getFileVersion(
fileEntry.getFileEntryId(), DLFileEntryConstants.VERSION_DEFAULT);
}
@Before
public void setUp() throws Exception {
SimpleAction simpleAction = new AddDefaultDocumentLibraryStructuresAction();
String companyIdString = String.valueOf(TestPropsValues.getCompanyId());
simpleAction.run(new String[] {companyIdString});
_group = GroupTestUtil.addGroup();
_folder =
DLAppLocalServiceUtil.addFolder(
TestPropsValues.getUserId(),
_group.getGroupId(),
DLFolderConstants.DEFAULT_PARENT_FOLDER_ID,
"Folder A",
"",
ServiceContextTestUtil.getServiceContext(_group.getGroupId()));
_subfolder =
DLAppLocalServiceUtil.addFolder(
TestPropsValues.getUserId(),
_group.getGroupId(),
_folder.getFolderId(),
"SubFolder AA",
"",
ServiceContextTestUtil.getServiceContext(_group.getGroupId()));
_basicDocumentDLFileEntryType =
DLFileEntryTypeLocalServiceUtil.getFileEntryType(
DLFileEntryTypeConstants.FILE_ENTRY_TYPE_ID_BASIC_DOCUMENT);
_dlFileEntryTypes =
DLFileEntryTypeLocalServiceUtil.getFileEntryTypes(
PortalUtil.getCurrentAndAncestorSiteGroupIds(_group.getGroupId()));
for (DLFileEntryType dlFileEntryType : _dlFileEntryTypes) {
String name = dlFileEntryType.getName(LocaleUtil.getSiteDefault());
if (name.equals(DLFileEntryTypeConstants.NAME_CONTRACT)) {
_contractDLFileEntryType = dlFileEntryType;
} else if (name.equals(DLFileEntryTypeConstants.NAME_MARKETING_BANNER)) {
_marketingBannerDLFileEntryType = dlFileEntryType;
}
}
}
@Override
public void postProcessSearchQuery(BooleanQuery searchQuery, SearchContext searchContext)
throws Exception {
Set<DDMStructure> ddmStructuresSet = new TreeSet<DDMStructure>();
long[] groupIds = searchContext.getGroupIds();
if ((groupIds != null) && (groupIds.length > 0)) {
for (long groupId : searchContext.getGroupIds()) {
List<DLFileEntryType> dlFileEntryTypes =
DLFileEntryTypeLocalServiceUtil.getFileEntryTypes(groupId);
for (DLFileEntryType dlFileEntryType : dlFileEntryTypes) {
ddmStructuresSet.addAll(dlFileEntryType.getDDMStructures());
}
}
}
for (DDMStructure ddmStructure : ddmStructuresSet) {
addSearchDDMStruture(searchQuery, searchContext, ddmStructure);
}
addSearchTerm(searchQuery, searchContext, Field.USER_NAME, true);
addSearchTerm(searchQuery, searchContext, "extension", true);
addSearchTerm(searchQuery, searchContext, "fileEntryTypeId", false);
addSearchTerm(searchQuery, searchContext, "path", true);
LinkedHashMap<String, Object> params =
(LinkedHashMap<String, Object>) searchContext.getAttribute("params");
if (params != null) {
String expandoAttributes = (String) params.get("expandoAttributes");
if (Validator.isNotNull(expandoAttributes)) {
addSearchExpando(searchQuery, searchContext, expandoAttributes);
}
}
}
