FileListCreator.cpp [plain text]
#include "config.h"
#include "FileListCreator.h"
#include "FileChooser.h"
#include "FileList.h"
#include "FileMetadata.h"
#include "FileSystem.h"
#include <wtf/CrossThreadCopier.h>
namespace WebCore {
FileListCreator::~FileListCreator()
{
ASSERT(!m_completionHander);
}
static void appendDirectoryFiles(const String& directory, const String& relativePath, Vector<Ref<File>>& fileObjects)
{
for (auto& childPath : FileSystem::listDirectory(directory, "*")) {
auto metadata = FileSystem::fileMetadata(childPath);
if (!metadata)
continue;
if (metadata.value().isHidden)
continue;
String childRelativePath = relativePath + "/" + FileSystem::pathGetFileName(childPath);
if (metadata.value().type == FileMetadata::Type::Directory)
appendDirectoryFiles(childPath, childRelativePath, fileObjects);
else if (metadata.value().type == FileMetadata::Type::File)
fileObjects.append(File::createWithRelativePath(childPath, childRelativePath));
}
}
FileListCreator::FileListCreator(const Vector<FileChooserFileInfo>& paths, ShouldResolveDirectories shouldResolveDirectories, CompletionHandler&& completionHandler)
{
if (shouldResolveDirectories == ShouldResolveDirectories::No)
completionHandler(createFileList<ShouldResolveDirectories::No>(paths));
else {
m_completionHander = WTFMove(completionHandler);
m_workQueue = WorkQueue::create("FileListCreator Work Queue");
m_workQueue->dispatch([this, protectedThis = makeRef(*this), paths = crossThreadCopy(paths)]() mutable {
auto fileList = createFileList<ShouldResolveDirectories::Yes>(paths);
callOnMainThread([this, protectedThis = WTFMove(protectedThis), fileList = WTFMove(fileList)]() mutable {
if (auto completionHander = WTFMove(m_completionHander))
completionHander(WTFMove(fileList));
});
});
}
}
template<FileListCreator::ShouldResolveDirectories shouldResolveDirectories>
Ref<FileList> FileListCreator::createFileList(const Vector<FileChooserFileInfo>& paths)
{
Vector<Ref<File>> fileObjects;
for (auto& info : paths) {
if (shouldResolveDirectories == ShouldResolveDirectories::Yes && FileSystem::fileIsDirectory(info.path, FileSystem::ShouldFollowSymbolicLinks::No))
appendDirectoryFiles(info.path, FileSystem::pathGetFileName(info.path), fileObjects);
else
fileObjects.append(File::createWithName(info.path, info.displayName));
}
return FileList::create(WTFMove(fileObjects));
}
void FileListCreator::cancel()
{
m_completionHander = nullptr;
m_workQueue = nullptr;
}
}