| // Copyright 2021 The Fuchsia Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #ifndef SRC_STORAGE_FSHOST_COPIER_H_ |
| #define SRC_STORAGE_FSHOST_COPIER_H_ |
| |
| #include <lib/zx/status.h> |
| |
| #include <filesystem> |
| #include <list> |
| #include <memory> |
| #include <string> |
| #include <variant> |
| #include <vector> |
| |
| #include <fbl/unique_fd.h> |
| |
| namespace fshost { |
| |
| class Copier { |
| public: |
| struct File { |
| std::string name; |
| std::string contents; |
| }; |
| struct Directory; |
| using DirectoryEntry = std::variant<File, Directory>; |
| using DirectoryEntries = std::list<DirectoryEntry>; |
| struct Directory { |
| std::string name; |
| DirectoryEntries entries; |
| }; |
| |
| // Reads all the data at |root_fd| except for the files and directories that match |
| // |excluded_paths|. |
| static zx::status<Copier> Read(fbl::unique_fd root_fd, |
| const std::vector<std::filesystem::path>& excluded_paths = {}); |
| |
| Copier() = default; |
| Copier(Copier&&) = default; |
| Copier& operator=(Copier&&) = default; |
| |
| // Writes all data to the given root fd. |
| zx_status_t Write(fbl::unique_fd root_fd) const; |
| |
| // Inserts a file into the in-memory structure creating parent directories as necessary. |
| // Returns an error if the file already exists or a directory could not be created because a file |
| // with the same name already exists. |
| zx::status<> InsertFile(const std::filesystem::path& path, std::string contents); |
| |
| const DirectoryEntries& entries() const { return entries_; } |
| |
| bool empty() const { return entries_.empty(); } |
| |
| private: |
| DirectoryEntries entries_; |
| }; |
| |
| } // namespace fshost |
| |
| #endif // SRC_STORAGE_FSHOST_COPIER_H_ |