file_reader.cc 2.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960
  1. // Copyright (c) 2011 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include "extensions/browser/file_reader.h"
  5. #include "base/bind.h"
  6. #include "base/callback_helpers.h"
  7. #include "base/files/file_util.h"
  8. #include "base/strings/stringprintf.h"
  9. #include "base/threading/thread_task_runner_handle.h"
  10. #include "extensions/browser/extension_file_task_runner.h"
  11. FileReader::FileReader(std::vector<extensions::ExtensionResource> resources,
  12. OptionalFileSequenceTask optional_file_sequence_task,
  13. DoneCallback done_callback)
  14. : resources_(std::move(resources)),
  15. optional_file_sequence_task_(std::move(optional_file_sequence_task)),
  16. done_callback_(std::move(done_callback)),
  17. origin_task_runner_(base::ThreadTaskRunnerHandle::Get()) {}
  18. void FileReader::Start() {
  19. extensions::GetExtensionFileTaskRunner()->PostTask(
  20. FROM_HERE, base::BindOnce(&FileReader::ReadFilesOnFileSequence, this));
  21. }
  22. FileReader::~FileReader() {}
  23. void FileReader::ReadFilesOnFileSequence() {
  24. DCHECK(
  25. extensions::GetExtensionFileTaskRunner()->RunsTasksInCurrentSequence());
  26. std::vector<std::unique_ptr<std::string>> data;
  27. data.reserve(resources_.size());
  28. absl::optional<std::string> error;
  29. for (const auto& resource : resources_) {
  30. data.push_back(std::make_unique<std::string>());
  31. std::string* file_data = data.back().get();
  32. bool success = base::ReadFileToString(resource.GetFilePath(), file_data);
  33. if (!success) {
  34. error =
  35. base::StringPrintf("Could not load file: '%s'.",
  36. resource.relative_path().AsUTF8Unsafe().c_str());
  37. // Clear `data` to avoid passing a partial result.
  38. data.clear();
  39. break;
  40. }
  41. if (optional_file_sequence_task_)
  42. optional_file_sequence_task_.Run(file_data);
  43. }
  44. // Release any potentially-bound references from the file sequence task.
  45. optional_file_sequence_task_.Reset();
  46. origin_task_runner_->PostTask(
  47. FROM_HERE, base::BindOnce(std::move(done_callback_), std::move(data),
  48. std::move(error)));
  49. }