diff --git a/lib/ClangImporter/ImportType.cpp b/lib/ClangImporter/ImportType.cpp index fe915dd30a9dc..b8540d3e12607 100644 --- a/lib/ClangImporter/ImportType.cpp +++ b/lib/ClangImporter/ImportType.cpp @@ -2646,44 +2646,31 @@ ArgumentAttrs ClangImporter::Implementation::inferDefaultArgument( } } else if (const clang::TypedefType *typedefType = type->getAs()) { - // Get the AvailabilityAttr that would be set from CF/NS_OPTIONS - if (importer::isUnavailableInSwift(typedefType->getDecl(), nullptr, true)) { - // If we've taken this branch it means we have an enum type, and it is - // likely an integer or NSInteger that is being used by NS/CF_OPTIONS to - // behave like a C enum in the presence of C++. - auto enumName = typedefType->getDecl()->getName(); - ArgumentAttrs argumentAttrs(DefaultArgumentKind::None, true, enumName); - auto camelCaseWords = camel_case::getWords(enumName); - for (auto it = camelCaseWords.rbegin(); it != camelCaseWords.rend(); - ++it) { - auto word = *it; - auto next = std::next(it); - if (camel_case::sameWordIgnoreFirstCase(word, "options")) { - argumentAttrs.argumentKind = DefaultArgumentKind::EmptyArray; - return argumentAttrs; + clang::TypedefNameDecl *typedefDecl = typedefType->getDecl(); + // Find the next decl in the same context. If this typedef is a part of an + // NS/CF_OPTIONS declaration, the next decl will be an enum. + auto declsInContext = typedefDecl->getDeclContext()->decls(); + auto declIter = llvm::find(declsInContext, typedefDecl); + if (declIter != declsInContext.end()) + declIter++; + if (declIter != declsInContext.end()) { + if (auto enumDecl = dyn_cast(*declIter)) { + if (auto cfOptionsTy = + nameImporter.getContext() + .getClangModuleLoader() + ->getTypeDefForCXXCFOptionsDefinition(enumDecl)) { + if (cfOptionsTy->getDecl() == typedefDecl) { + auto enumName = typedefDecl->getName(); + ArgumentAttrs argumentAttrs(DefaultArgumentKind::None, true, + enumName); + for (auto word : llvm::reverse(camel_case::getWords(enumName))) { + if (camel_case::sameWordIgnoreFirstCase(word, "options")) { + argumentAttrs.argumentKind = DefaultArgumentKind::EmptyArray; + } + } + return argumentAttrs; + } } - if (camel_case::sameWordIgnoreFirstCase(word, "units")) - return argumentAttrs; - if (camel_case::sameWordIgnoreFirstCase(word, "domain")) - return argumentAttrs; - if (camel_case::sameWordIgnoreFirstCase(word, "action")) - return argumentAttrs; - if (camel_case::sameWordIgnoreFirstCase(word, "event")) - return argumentAttrs; - if (camel_case::sameWordIgnoreFirstCase(word, "events") && - next != camelCaseWords.rend() && - camel_case::sameWordIgnoreFirstCase(*next, "control")) - return argumentAttrs; - if (camel_case::sameWordIgnoreFirstCase(word, "state")) - return argumentAttrs; - if (camel_case::sameWordIgnoreFirstCase(word, "unit")) - return argumentAttrs; - if (camel_case::sameWordIgnoreFirstCase(word, "position") && - next != camelCaseWords.rend() && - camel_case::sameWordIgnoreFirstCase(*next, "scroll")) - return argumentAttrs; - if (camel_case::sameWordIgnoreFirstCase(word, "edge")) - return argumentAttrs; } } } diff --git a/test/Interop/Cxx/enum/Inputs/c-enums-withOptions-omit.h b/test/Interop/Cxx/enum/Inputs/c-enums-withOptions-omit.h index daeced9d2377f..81b407ed60e48 100644 --- a/test/Interop/Cxx/enum/Inputs/c-enums-withOptions-omit.h +++ b/test/Interop/Cxx/enum/Inputs/c-enums-withOptions-omit.h @@ -1,39 +1,57 @@ -// Enum usage that is bitwise-able and assignable in C++, aka how CF_OPTIONS -// does things. -typedef int __attribute__((availability(swift, unavailable))) NSEnumerationOptions; -enum : NSEnumerationOptions { NSEnumerationConcurrent, NSEnumerationReverse }; +typedef unsigned NSUInteger; + +#define __CF_OPTIONS_ATTRIBUTES __attribute__((flag_enum,enum_extensibility(open))) +#if (__cplusplus) +#define CF_OPTIONS(_type, _name) __attribute__((availability(swift,unavailable))) _type _name; enum __CF_OPTIONS_ATTRIBUTES : _name +#else +#define CF_OPTIONS(_type, _name) enum __CF_OPTIONS_ATTRIBUTES _name : _type _name; enum _name : _type +#endif + +typedef CF_OPTIONS(NSUInteger, NSEnumerationOptions) { + NSEnumerationConcurrent = (1UL << 0), + NSEnumerationReverse = (1UL << 1), +}; @interface NSSet - (void)enumerateObjectsWithOptions:(NSEnumerationOptions)opts ; @end -typedef int __attribute__((availability(swift, unavailable))) NSOrderedCollectionDifferenceCalculationOptions; -enum : NSOrderedCollectionDifferenceCalculationOptions { +typedef CF_OPTIONS(NSUInteger, NSOrderedCollectionDifferenceCalculationOptions) { NSOrderedCollectionDifferenceCalculationOptions1, NSOrderedCollectionDifferenceCalculationOptions2 }; -typedef int __attribute__((availability(swift, unavailable))) NSCalendarUnit; -enum : NSCalendarUnit { NSCalendarUnit1, NSCalendarUnit2 }; +typedef CF_OPTIONS(NSUInteger, NSCalendarUnit) { + NSCalendarUnit1, + NSCalendarUnit2 +}; -typedef int __attribute__((availability(swift, unavailable))) NSSearchPathDomainMask; -enum : NSSearchPathDomainMask { NSSearchPathDomainMask1, NSSearchPathDomainMask2 }; +typedef CF_OPTIONS(NSUInteger, NSSearchPathDomainMask) { + NSSearchPathDomainMask1, + NSSearchPathDomainMask2 +}; -typedef int __attribute__((availability(swift, unavailable))) NSControlCharacterAction; -enum : NSControlCharacterAction { NSControlCharacterAction1, NSControlCharacterAction2 }; +typedef CF_OPTIONS(NSUInteger, NSControlCharacterAction) { + NSControlCharacterAction1, + NSControlCharacterAction2 +}; -typedef int __attribute__((availability(swift, unavailable))) UIControlState; -enum : UIControlState { UIControlState1, UIControlState2 }; +typedef CF_OPTIONS(NSUInteger, UIControlState) { + UIControlState1, + UIControlState2 +}; -typedef int __attribute__((availability(swift, unavailable))) UITableViewCellStateMask; -enum : UITableViewCellStateMask { UITableViewCellStateMask1, UITableViewCellStateMask2 }; +typedef CF_OPTIONS(NSUInteger, UITableViewCellStateMask) { + UITableViewCellStateMask1, + UITableViewCellStateMask2 +}; -typedef int __attribute__((availability(swift, unavailable))) UIControlEvents; -enum : UIControlEvents { UIControlEvents1, UIControlEvents2 }; +typedef CF_OPTIONS(NSUInteger, UIControlEvents) { + UIControlEvents1, + UIControlEvents2 +}; -typedef int __attribute__((availability(swift, unavailable))) -UITableViewScrollPosition; -enum : UITableViewScrollPosition { +typedef CF_OPTIONS(NSUInteger, UITableViewScrollPosition) { UITableViewScrollPosition1, UITableViewScrollPosition2 }; diff --git a/test/Interop/Cxx/objc-correctness/Inputs/customNSOptions.h b/test/Interop/Cxx/objc-correctness/Inputs/customNSOptions.h new file mode 100644 index 0000000000000..b666f4d713027 --- /dev/null +++ b/test/Interop/Cxx/objc-correctness/Inputs/customNSOptions.h @@ -0,0 +1,6 @@ +#include "CFOptions.h" + +typedef CF_OPTIONS(unsigned, MyControlFlags) { + MyControlFlagsNone = 0, + MyControlFlagsFirst +}; diff --git a/test/Interop/Cxx/objc-correctness/Inputs/module.modulemap b/test/Interop/Cxx/objc-correctness/Inputs/module.modulemap index 00e4e886f9220..005f2287098e1 100644 --- a/test/Interop/Cxx/objc-correctness/Inputs/module.modulemap +++ b/test/Interop/Cxx/objc-correctness/Inputs/module.modulemap @@ -9,6 +9,10 @@ module CxxClassWithNSStringInit { requires cplusplus } +module CustomNSOptions { + header "customNSOptions.h" +} + module NSOptionsMangling { header "NSOptionsMangling.h" } diff --git a/test/Interop/Cxx/objc-correctness/custom-nsoptions.swift b/test/Interop/Cxx/objc-correctness/custom-nsoptions.swift new file mode 100644 index 0000000000000..6c5e398643d10 --- /dev/null +++ b/test/Interop/Cxx/objc-correctness/custom-nsoptions.swift @@ -0,0 +1,7 @@ +// RUN: %target-typecheck-verify-swift -verify-ignore-unknown -I %S/Inputs -enable-objc-interop -enable-experimental-cxx-interop +// REQUIRES: objc_interop + +import CustomNSOptions + +let flags1: MyControlFlags = [] +let flags2: MyControlFlags = [.first]