DirectoryFileListCreator.cpp [plain text]
#include "config.h"
#include "DirectoryFileListCreator.h"
#include "FileChooser.h"
#include "FileList.h"
#include <wtf/CrossThreadCopier.h>
#include <wtf/FileMetadata.h>
#include <wtf/FileSystem.h>
namespace WebCore {
DirectoryFileListCreator::~DirectoryFileListCreator()
{
ASSERT(!m_completionHandler);
}
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));
}
}
static Ref<FileList> createFileList(const Vector<FileChooserFileInfo>& paths)
{
Vector<Ref<File>> fileObjects;
for (auto& info : paths) {
if (FileSystem::fileIsDirectory(info.path, FileSystem::ShouldFollowSymbolicLinks::No))
appendDirectoryFiles(info.path, FileSystem::pathGetFileName(info.path), fileObjects);
else
fileObjects.append(File::create(info.path, info.displayName));
}
return FileList::create(WTFMove(fileObjects));
}
DirectoryFileListCreator::DirectoryFileListCreator(CompletionHandler&& completionHandler)
: m_workQueue(WorkQueue::create("DirectoryFileListCreator Work Queue"))
, m_completionHandler(WTFMove(completionHandler))
{
}
void DirectoryFileListCreator::start(const Vector<FileChooserFileInfo>& paths)
{
m_workQueue->dispatch([this, protectedThis = makeRef(*this), paths = crossThreadCopy(paths)]() mutable {
callOnMainThread([this, protectedThis = WTFMove(protectedThis), fileList = createFileList(paths)]() mutable {
if (auto completionHandler = std::exchange(m_completionHandler, nullptr))
completionHandler(WTFMove(fileList));
});
});
}
void DirectoryFileListCreator::cancel()
{
m_completionHandler = nullptr;
m_workQueue = nullptr;
}
}