averello/swift-mirror
1
0
Fork 0
mirror of https://github.com/apple/swift.git synced 2025-12-14 20:36:38 +01:00
Code Issues Packages Projects Releases Wiki Activity

[Compile Time Constant Extraction] Extract listed Availability Attributes for buildLimitedAvailability in Result Builders

Browse Source
This commit is contained in:
Venkatesh Sriram
2024-09-26 14:11:35 -07:00
parent 352fb004b3
commit f04486db3e
3 changed files with 64 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
lib/ConstExtract/ConstExtract.cpp
View File

@@ -945,6 +945,7 @@ getResultBuilderElementFromASTNode(const ASTNode node) {
BuilderValue::ConditionalMember
getConditionalMemberFromIfStmt(const IfStmt *ifStmt) {
std::vector<PlatformVersionConstraintAvailabilitySpec> AvailabilityAttributes;
std::vector<std::shared_ptr<BuilderValue::BuilderMember>> IfElements;
std::vector<std::shared_ptr<BuilderValue::BuilderMember>> ElseElements;
if (auto thenBraceStmt = ifStmt->getThenStmt()) {
@@ -976,12 +977,24 @@ getConditionalMemberFromIfStmt(const IfStmt *ifStmt) {
}
for (auto elt : ifStmt->getCond()) {
if (elt.getKind() == StmtConditionElement::CK_Availability) {
for (auto *Q : elt.getAvailability()->getQueries()) {
if (auto *availability =
dyn_cast<PlatformVersionConstraintAvailabilitySpec>(Q)) {
AvailabilityAttributes.push_back(*availability);
}
}
memberKind = BuilderValue::LimitedAvailability;
break;
}
}
return BuilderValue::ConditionalMember(memberKind, IfElements, ElseElements);
if (AvailabilityAttributes.empty()) {
return BuilderValue::ConditionalMember(memberKind, IfElements,
ElseElements);
}
return BuilderValue::ConditionalMember(memberKind, AvailabilityAttributes,
IfElements, ElseElements);
}
BuilderValue::ArrayMember
@@ -1094,6 +1107,17 @@ void writeBuilderMember(
default: {
auto member = cast<BuilderValue::ConditionalMember>(Member);
if (auto availabilityAttributes = member->getAvailabilityAttributes()) {
JSON.attributeArray("availabilityAttributes", [&] {
for (auto elem : *availabilityAttributes) {
JSON.object([&] {
JSON.attribute("platform",
platformString(elem.getPlatform()).str());
JSON.attribute("minVersion", elem.getVersion().getAsString());
});
}
});
}
JSON.attributeArray("ifElements", [&] {
for (auto elem : member->getIfElements()) {
JSON.object([&] { writeBuilderMember(JSON, elem); });