Skip to content

Commit 1f67607

Browse files
egorzhdanegorzhdan
authored
Merge pull request #67865 from apple/egorzhdan/cfoptions
[cxx-interop] Import custom `NS_OPTIONS` correctly
2 parents 1542bb5 + f20617b commit 1f67607

File tree

5 files changed

+80
-58
lines changed

5 files changed

+80
-58
lines changed

‎lib/ClangImporter/ImportType.cpp

Lines changed: 24 additions & 37 deletions
Original file line numberDiff line numberDiff line change
@@ -2646,44 +2646,31 @@ ArgumentAttrs ClangImporter::Implementation::inferDefaultArgument(
26462646
}
26472647
} else if (const clang::TypedefType *typedefType =
26482648
type->getAs<clang::TypedefType>()) {
2649-
// Get the AvailabilityAttr that would be set from CF/NS_OPTIONS
2650-
if (importer::isUnavailableInSwift(typedefType->getDecl(), nullptr, true)) {
2651-
// If we've taken this branch it means we have an enum type, and it is
2652-
// likely an integer or NSInteger that is being used by NS/CF_OPTIONS to
2653-
// behave like a C enum in the presence of C++.
2654-
auto enumName = typedefType->getDecl()->getName();
2655-
ArgumentAttrs argumentAttrs(DefaultArgumentKind::None, true, enumName);
2656-
auto camelCaseWords = camel_case::getWords(enumName);
2657-
for (auto it = camelCaseWords.rbegin(); it != camelCaseWords.rend();
2658-
++it) {
2659-
auto word = *it;
2660-
auto next = std::next(it);
2661-
if (camel_case::sameWordIgnoreFirstCase(word, "options")) {
2662-
argumentAttrs.argumentKind = DefaultArgumentKind::EmptyArray;
2663-
return argumentAttrs;
2649+
clang::TypedefNameDecl *typedefDecl = typedefType->getDecl();
2650+
// Find the next decl in the same context. If this typedef is a part of an
2651+
// NS/CF_OPTIONS declaration, the next decl will be an enum.
2652+
auto declsInContext = typedefDecl->getDeclContext()->decls();
2653+
auto declIter = llvm::find(declsInContext, typedefDecl);
2654+
if (declIter != declsInContext.end())
2655+
declIter++;
2656+
if (declIter != declsInContext.end()) {
2657+
if (auto enumDecl = dyn_cast<clang::EnumDecl>(*declIter)) {
2658+
if (auto cfOptionsTy =
2659+
nameImporter.getContext()
2660+
.getClangModuleLoader()
2661+
->getTypeDefForCXXCFOptionsDefinition(enumDecl)) {
2662+
if (cfOptionsTy->getDecl() == typedefDecl) {
2663+
auto enumName = typedefDecl->getName();
2664+
ArgumentAttrs argumentAttrs(DefaultArgumentKind::None, true,
2665+
enumName);
2666+
for (auto word : llvm::reverse(camel_case::getWords(enumName))) {
2667+
if (camel_case::sameWordIgnoreFirstCase(word, "options")) {
2668+
argumentAttrs.argumentKind = DefaultArgumentKind::EmptyArray;
2669+
}
2670+
}
2671+
return argumentAttrs;
2672+
}
26642673
}
2665-
if (camel_case::sameWordIgnoreFirstCase(word, "units"))
2666-
return argumentAttrs;
2667-
if (camel_case::sameWordIgnoreFirstCase(word, "domain"))
2668-
return argumentAttrs;
2669-
if (camel_case::sameWordIgnoreFirstCase(word, "action"))
2670-
return argumentAttrs;
2671-
if (camel_case::sameWordIgnoreFirstCase(word, "event"))
2672-
return argumentAttrs;
2673-
if (camel_case::sameWordIgnoreFirstCase(word, "events") &&
2674-
next != camelCaseWords.rend() &&
2675-
camel_case::sameWordIgnoreFirstCase(*next, "control"))
2676-
return argumentAttrs;
2677-
if (camel_case::sameWordIgnoreFirstCase(word, "state"))
2678-
return argumentAttrs;
2679-
if (camel_case::sameWordIgnoreFirstCase(word, "unit"))
2680-
return argumentAttrs;
2681-
if (camel_case::sameWordIgnoreFirstCase(word, "position") &&
2682-
next != camelCaseWords.rend() &&
2683-
camel_case::sameWordIgnoreFirstCase(*next, "scroll"))
2684-
return argumentAttrs;
2685-
if (camel_case::sameWordIgnoreFirstCase(word, "edge"))
2686-
return argumentAttrs;
26872674
}
26882675
}
26892676
}

‎test/Interop/Cxx/enum/Inputs/c-enums-withOptions-omit.h

Lines changed: 39 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -1,39 +1,57 @@
1-
// Enum usage that is bitwise-able and assignable in C++, aka how CF_OPTIONS
2-
// does things.
3-
typedef int __attribute__((availability(swift, unavailable))) NSEnumerationOptions;
4-
enum : NSEnumerationOptions { NSEnumerationConcurrent, NSEnumerationReverse };
1+
typedef unsigned NSUInteger;
2+
3+
#define __CF_OPTIONS_ATTRIBUTES __attribute__((flag_enum,enum_extensibility(open)))
4+
#if (__cplusplus)
5+
#define CF_OPTIONS(_type, _name) __attribute__((availability(swift,unavailable))) _type _name; enum __CF_OPTIONS_ATTRIBUTES : _name
6+
#else
7+
#define CF_OPTIONS(_type, _name) enum __CF_OPTIONS_ATTRIBUTES _name : _type _name; enum _name : _type
8+
#endif
9+
10+
typedef CF_OPTIONS(NSUInteger, NSEnumerationOptions) {
11+
NSEnumerationConcurrent = (1UL << 0),
12+
NSEnumerationReverse = (1UL << 1),
13+
};
514

615
@interface NSSet
716
- (void)enumerateObjectsWithOptions:(NSEnumerationOptions)opts ;
817
@end
918

10-
typedef int __attribute__((availability(swift, unavailable))) NSOrderedCollectionDifferenceCalculationOptions;
11-
enum : NSOrderedCollectionDifferenceCalculationOptions {
19+
typedef CF_OPTIONS(NSUInteger, NSOrderedCollectionDifferenceCalculationOptions) {
1220
NSOrderedCollectionDifferenceCalculationOptions1,
1321
NSOrderedCollectionDifferenceCalculationOptions2
1422
};
1523

16-
typedef int __attribute__((availability(swift, unavailable))) NSCalendarUnit;
17-
enum : NSCalendarUnit { NSCalendarUnit1, NSCalendarUnit2 };
24+
typedef CF_OPTIONS(NSUInteger, NSCalendarUnit) {
25+
NSCalendarUnit1,
26+
NSCalendarUnit2
27+
};
1828

19-
typedef int __attribute__((availability(swift, unavailable))) NSSearchPathDomainMask;
20-
enum : NSSearchPathDomainMask { NSSearchPathDomainMask1, NSSearchPathDomainMask2 };
29+
typedef CF_OPTIONS(NSUInteger, NSSearchPathDomainMask) {
30+
NSSearchPathDomainMask1,
31+
NSSearchPathDomainMask2
32+
};
2133

22-
typedef int __attribute__((availability(swift, unavailable))) NSControlCharacterAction;
23-
enum : NSControlCharacterAction { NSControlCharacterAction1, NSControlCharacterAction2 };
34+
typedef CF_OPTIONS(NSUInteger, NSControlCharacterAction) {
35+
NSControlCharacterAction1,
36+
NSControlCharacterAction2
37+
};
2438

25-
typedef int __attribute__((availability(swift, unavailable))) UIControlState;
26-
enum : UIControlState { UIControlState1, UIControlState2 };
39+
typedef CF_OPTIONS(NSUInteger, UIControlState) {
40+
UIControlState1,
41+
UIControlState2
42+
};
2743

28-
typedef int __attribute__((availability(swift, unavailable))) UITableViewCellStateMask;
29-
enum : UITableViewCellStateMask { UITableViewCellStateMask1, UITableViewCellStateMask2 };
44+
typedef CF_OPTIONS(NSUInteger, UITableViewCellStateMask) {
45+
UITableViewCellStateMask1,
46+
UITableViewCellStateMask2
47+
};
3048

31-
typedef int __attribute__((availability(swift, unavailable))) UIControlEvents;
32-
enum : UIControlEvents { UIControlEvents1, UIControlEvents2 };
49+
typedef CF_OPTIONS(NSUInteger, UIControlEvents) {
50+
UIControlEvents1,
51+
UIControlEvents2
52+
};
3353

34-
typedef int __attribute__((availability(swift, unavailable)))
35-
UITableViewScrollPosition;
36-
enum : UITableViewScrollPosition {
54+
typedef CF_OPTIONS(NSUInteger, UITableViewScrollPosition) {
3755
UITableViewScrollPosition1,
3856
UITableViewScrollPosition2
3957
};

‎test/Interop/Cxx/objc-correctness/Inputs/customNSOptions.h

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
#include "CFOptions.h"
2+
3+
typedef CF_OPTIONS(unsigned, MyControlFlags) {
4+
MyControlFlagsNone = 0,
5+
MyControlFlagsFirst
6+
};

‎test/Interop/Cxx/objc-correctness/Inputs/module.modulemap

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,10 @@ module CxxClassWithNSStringInit {
99
requires cplusplus
1010
}
1111

12+
module CustomNSOptions {
13+
header "customNSOptions.h"
14+
}
15+
1216
module NSOptionsMangling {
1317
header "NSOptionsMangling.h"
1418
}

‎test/Interop/Cxx/objc-correctness/custom-nsoptions.swift

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,7 @@
1+
// RUN: %target-typecheck-verify-swift -verify-ignore-unknown -I %S/Inputs -enable-objc-interop -enable-experimental-cxx-interop
2+
// REQUIRES: objc_interop
3+
4+
import CustomNSOptions
5+
6+
let flags1: MyControlFlags = []
7+
let flags2: MyControlFlags = [.first]

0 commit comments

Comments
 (0)