Fix root directory upload to honor tree invariant.

1 files changed, 64 insertions, 0 deletions

diff --git a/src/buildtool/execution_api/bazel_msg/blob_tree.hpp b/src/buildtool/execution_api/bazel_msg/blob_tree.hpp
new file mode 100644
index 00000000..786e3c55
--- /dev/null
+++ b/src/buildtool/execution_api/bazel_msg/blob_tree.hpp
@@ -0,0 +1,64 @@
+// Copyright 2022 Huawei Cloud Computing Technology Co., Ltd.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#ifndef INCLUDED_SRC_BUILDTOOL_EXECUTION_API_BAZEL_MSG_BLOB_TREE_HPP
+#define INCLUDED_SRC_BUILDTOOL_EXECUTION_API_BAZEL_MSG_BLOB_TREE_HPP
+
+#include <filesystem>
+#include <memory>
+#include <optional>
+#include <utility>
+#include <vector>
+
+#include "gsl-lite/gsl-lite.hpp"
+#include "src/buildtool/compatibility/native_support.hpp"
+#include "src/buildtool/execution_api/bazel_msg/bazel_blob.hpp"
+#include "src/buildtool/execution_api/bazel_msg/directory_tree.hpp"
+
+class BlobTree;
+using BlobTreePtr = gsl::not_null<std::shared_ptr<BlobTree>>;
+
+/// \brief Tree-like blob container to enable tree-invariant satisfying blob
+/// upload.
+class BlobTree {
+  public:
+    BlobTree(BazelBlob blob, std::vector<BlobTreePtr> nodes)
+        : blob_{std::move(blob)} {
+        nodes_.reserve(nodes.size());
+        for (auto& node : nodes) {
+            nodes_.emplace_back(std::move(node));
+        }
+    }
+
+    [[nodiscard]] auto Blob() const noexcept -> BazelBlob { return blob_; }
+    [[nodiscard]] auto IsTree() const noexcept -> bool {
+        return NativeSupport::IsTree(blob_.digest.hash());
+    }
+
+    /// \brief Create a `BlobTree` from a `DirectoryTree`.
+    [[nodiscard]] static auto FromDirectoryTree(
+        DirectoryTreePtr const& tree,
+        std::filesystem::path const& parent = "") noexcept
+        -> std::optional<BlobTreePtr>;
+
+    [[nodiscard]] auto begin() const noexcept { return nodes_.begin(); }
+    [[nodiscard]] auto end() const noexcept { return nodes_.end(); }
+    [[nodiscard]] auto size() const noexcept { return nodes_.size(); }
+
+  private:
+    BazelBlob blob_;
+    std::vector<BlobTreePtr> nodes_;
+};
+
+#endif  // INCLUDED_SRC_BUILDTOOL_EXECUTION_API_BAZEL_MSG_BLOB_TREE_HPP