1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
// Copyright 2023 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.
#include "src/buildtool/execution_api/utils/subobject.hpp"
#ifndef BOOTSTRAP_BUILD_TOOL
#include "src/buildtool/compatibility/compatibility.hpp"
#include "src/buildtool/execution_api/bazel_msg/bazel_msg_factory.hpp"
#include "src/buildtool/execution_api/common/tree_reader_utils.hpp"
#include "src/buildtool/logging/log_level.hpp"
#include "src/buildtool/logging/logger.hpp"
auto RetrieveSubPathId(Artifact::ObjectInfo object_info,
IExecutionApi const& api,
const std::filesystem::path& sub_path)
-> std::optional<Artifact::ObjectInfo> {
std::filesystem::path sofar{};
for (auto const& segment : sub_path) {
if (object_info.type != ObjectType::Tree) {
Logger::Log(LogLevel::Warning,
"Non-tree found at path '{}', cannot follow to '{}'",
sofar.string(),
segment.string());
break;
}
auto data = api.RetrieveToMemory(object_info);
if (not data) {
Logger::Log(LogLevel::Error,
"Failed to retrieve artifact {} at path '{}'",
object_info.ToString(),
sofar.string());
return std::nullopt;
}
if (Compatibility::IsCompatible()) {
auto directory =
BazelMsgFactory::MessageFromString<bazel_re::Directory>(*data);
if (not directory) {
Logger::Log(LogLevel::Warning,
"Failed to parse directory message at path '{}'",
sofar.string());
break;
}
std::optional<Artifact::ObjectInfo> new_object_info{};
if (not TreeReaderUtils::ReadObjectInfos(
*directory,
[&new_object_info, &segment](auto path, auto info) {
if (path == segment) {
new_object_info = info;
}
return true;
})) {
Logger::Log(LogLevel::Warning,
"Failed to process directory message at path '{}'",
sofar.string());
break;
}
if (not new_object_info) {
Logger::Log(LogLevel::Warning,
"Entry {} not found at path '{}'",
segment.string(),
sofar.string());
break;
}
object_info = *new_object_info;
}
else {
auto entries = GitRepo::ReadTreeData(
*data,
HashFunction::ComputeTreeHash(*data).Bytes(),
[](auto const& /*unused*/) { return true; },
/*is_hex_id=*/false);
if (not entries) {
Logger::Log(LogLevel::Warning,
"Failed to parse tree {} at path '{}'",
object_info.ToString(),
sofar.string());
break;
}
std::optional<Artifact::ObjectInfo> new_object_info{};
if (not TreeReaderUtils::ReadObjectInfos(
*entries,
[&new_object_info, &segment](auto path, auto info) {
if (path == segment) {
new_object_info = info;
}
return true;
})) {
Logger::Log(LogLevel::Warning,
"Failed to process tree entries at path '{}'",
sofar.string());
break;
}
if (not new_object_info) {
Logger::Log(LogLevel::Warning,
"Entry {} not found at path '{}'",
segment.string(),
sofar.string());
break;
}
object_info = *new_object_info;
}
sofar /= segment;
}
return object_info;
}
#endif
|
