@Override
public <T extends ExtRepositoryObject> T getExtRepositoryObject(
ExtRepositoryObjectType<T> extRepositoryObjectType,
String extRepositoryFolderKey,
String title)
throws PortalException {
try {
StringBundler sb = new StringBundler();
sb.append("'");
sb.append(extRepositoryFolderKey);
sb.append("' in parents and title contains '");
sb.append(title);
sb.append(" and mimeType ");
if (extRepositoryObjectType.equals(ExtRepositoryObjectType.FOLDER)) {
sb.append("= ");
} else {
sb.append("!= ");
}
sb.append(_FOLDER_MIME_TYPE);
Drive drive = getDrive();
Drive.Files driveFiles = drive.files();
Drive.Files.List driveFilesList = driveFiles.list();
driveFilesList.setQ(sb.toString());
FileList fileList = driveFilesList.execute();
List<File> files = fileList.getItems();
if (files.isEmpty()) {
if (extRepositoryObjectType == ExtRepositoryObjectType.FOLDER) {
throw new NoSuchFolderException(title);
}
throw new NoSuchFileEntryException(title);
}
if (extRepositoryObjectType.equals(ExtRepositoryObjectType.FOLDER)) {
return (T) new GoogleDriveFolder(files.get(0), getRootFolderKey());
}
return (T) new GoogleDriveFileEntry(files.get(0));
} catch (IOException ioe) {
_log.error(ioe, ioe);
throw new SystemException(ioe);
}
}
@Override
public List<ExtRepositorySearchResult<?>> search(
SearchContext searchContext, Query query, ExtRepositoryQueryMapper extRepositoryQueryMapper)
throws PortalException {
try {
Drive drive = getDrive();
Drive.Files driveFiles = drive.files();
Drive.Files.List driveFilesList = driveFiles.list();
String searchQuery =
getSearchQuery(
searchContext.getKeywords(), searchContext.getFolderIds(), extRepositoryQueryMapper);
driveFilesList.setQ(searchQuery);
FileList fileList = driveFilesList.execute();
List<File> files = fileList.getItems();
List<ExtRepositorySearchResult<?>> extRepositorySearchResults = new ArrayList<>(files.size());
for (File file : files) {
if (_FOLDER_MIME_TYPE.equals(file.getMimeType())) {
GoogleDriveFolder googleDriveFolder = new GoogleDriveFolder(file, getRootFolderKey());
ExtRepositorySearchResult<GoogleDriveFolder> extRepositorySearchResult =
new ExtRepositorySearchResult<>(googleDriveFolder, 1.0f, StringPool.BLANK);
extRepositorySearchResults.add(extRepositorySearchResult);
} else {
GoogleDriveFileEntry googleDriveFileEntry = new GoogleDriveFileEntry(file);
ExtRepositorySearchResult<GoogleDriveFileEntry> extRepositorySearchResult =
new ExtRepositorySearchResult<>(googleDriveFileEntry, 1.0f, StringPool.BLANK);
extRepositorySearchResults.add(extRepositorySearchResult);
}
}
return extRepositorySearchResults;
} catch (IOException ioe) {
_log.error(ioe, ioe);
throw new SystemException(ioe);
}
}
@Override
public <T extends ExtRepositoryObject> List<T> getExtRepositoryObjects(
ExtRepositoryObjectType<T> extRepositoryObjectType, String extRepositoryFolderKey)
throws PortalException {
try {
Drive drive = getDrive();
Drive.Files driveFiles = drive.files();
Drive.Files.List driveFilesList = driveFiles.list();
StringBundler sb = new StringBundler();
if (extRepositoryFolderKey != null) {
sb.append("'");
sb.append(extRepositoryFolderKey);
sb.append("' in parents and ");
}
if (!extRepositoryObjectType.equals(ExtRepositoryObjectType.OBJECT)) {
sb.append("mimeType");
if (extRepositoryObjectType.equals(ExtRepositoryObjectType.FILE)) {
sb.append(" != '");
} else {
sb.append(" = '");
}
sb.append(_FOLDER_MIME_TYPE);
sb.append("' and ");
}
sb.append("trashed = false");
driveFilesList.setQ(sb.toString());
FileList fileList = driveFilesList.execute();
List<File> files = fileList.getItems();
List<T> extRepositoryObjects = new ArrayList<>();
GoogleDriveCache googleDriveCache = GoogleDriveCache.getInstance();
for (File file : files) {
if (_FOLDER_MIME_TYPE.equals(file.getMimeType())) {
extRepositoryObjects.add((T) new GoogleDriveFolder(file, getRootFolderKey()));
} else {
extRepositoryObjects.add((T) new GoogleDriveFileEntry(file));
}
googleDriveCache.put(file);
}
return extRepositoryObjects;
} catch (IOException ioe) {
_log.error(ioe, ioe);
throw new SystemException(ioe);
}
}
