swiftlang · swift-ci · Apr 25, 2017 · Apr 21, 2017 · Apr 25, 2017
diff --git a/lib/ClangImporter/ImportEnumInfo.cpp b/lib/ClangImporter/ImportEnumInfo.cpp
@@ -50,9 +50,32 @@ void EnumInfo::classifyEnum(ASTContext &ctx, const clang::EnumDecl *decl,
     nsErrorDomain = ctx.AllocateCopy(domainAttr->getErrorDomain()->getName());
     return;
   }
+  if (decl->hasAttr<clang::FlagEnumAttr>()) {
+    kind = EnumKind::Options;
+    return;
+  }
+  if (decl->hasAttr<clang::EnumExtensibilityAttr>()) {
+    // FIXME: Distinguish between open and closed enums.
+    kind = EnumKind::Enum;
+    return;
+  }
+
+  // If API notes have /removed/ a FlagEnum or EnumExtensibility attribute,
+  // then we don't need to check the macros.
+  for (auto *attr : decl->specific_attrs<clang::SwiftVersionedAttr>()) {
+    if (!attr->getVersion().empty())
+      continue;
+    if (isa<clang::FlagEnumAttr>(attr->getAttrToAdd()) ||
+        isa<clang::EnumExtensibilityAttr>(attr->getAttrToAdd())) {
+      kind = EnumKind::Unknown;
+      return;
+    }
+  }
 
   // Was the enum declared using *_ENUM or *_OPTIONS?
-  // FIXME: Use Clang attributes instead of groveling the macro expansion loc.
+  // FIXME: Stop using these once flag_enum and enum_extensibility
+  // have been adopted everywhere, or at least relegate them to Swift 3 mode
+  // only.
   auto loc = decl->getLocStart();
   if (loc.isMacroID()) {
     StringRef MacroName = pp.getImmediateMacroName(loc);

diff --git a/lib/PrintAsObjC/PrintAsObjC.cpp b/lib/PrintAsObjC/PrintAsObjC.cpp
@@ -2451,15 +2451,25 @@ class ModuleWriter {
            "#  define OBJC_DESIGNATED_INITIALIZER\n"
            "# endif\n"
            "#endif\n"
+           "#if !defined(SWIFT_ENUM_ATTR)\n"
+           "# if defined(__has_attribute) && "
+             "__has_attribute(enum_extensibility)\n"
+           "#  define SWIFT_ENUM_ATTR "
+             "__attribute__((enum_extensibility(open)))\n"
+           "# else\n"
+           "#  define SWIFT_ENUM_ATTR\n"
+           "# endif\n"
+           "#endif\n"
            "#if !defined(SWIFT_ENUM)\n"
            "# define SWIFT_ENUM(_type, _name) "
              "enum _name : _type _name; "
-             "enum SWIFT_ENUM_EXTRA _name : _type\n"
+             "enum SWIFT_ENUM_ATTR SWIFT_ENUM_EXTRA _name : _type\n"
            "# if defined(__has_feature) && "
              "__has_feature(generalized_swift_name)\n"
            "#  define SWIFT_ENUM_NAMED(_type, _name, SWIFT_NAME) "
              "enum _name : _type _name SWIFT_COMPILE_NAME(SWIFT_NAME); "
-             "enum SWIFT_COMPILE_NAME(SWIFT_NAME) SWIFT_ENUM_EXTRA _name : _type\n"
+             "enum SWIFT_COMPILE_NAME(SWIFT_NAME) SWIFT_ENUM_ATTR "
+             "SWIFT_ENUM_EXTRA _name : _type\n"
            "# else\n"
            "#  define SWIFT_ENUM_NAMED(_type, _name, SWIFT_NAME) "
              "SWIFT_ENUM(_type, _name)\n"

diff --git a/test/ClangImporter/Inputs/custom-modules/SwiftName.h b/test/ClangImporter/Inputs/custom-modules/SwiftName.h
@@ -1,6 +1,6 @@
 #define SWIFT_NAME(X) __attribute__((swift_name(#X)))
 
-#define SWIFT_ENUM(_type, _name) enum _name : _type _name; enum _name : _type
+#define SWIFT_ENUM(_type, _name) enum _name : _type _name; enum __attribute__((enum_extensibility(open))) _name : _type
 
 void drawString(const char *, int x, int y) SWIFT_NAME(drawString(_:x:y:));
 

diff --git a/test/ClangImporter/Inputs/enum-objc.h b/test/ClangImporter/Inputs/enum-objc.h
@@ -1,7 +1,7 @@
 @import Foundation;
 
 #define SWIFT_COMPILE_NAME(X) __attribute__((swift_name(X)))
-#define SWIFT_ENUM_NAMED(_type, _name, SWIFT_NAME) enum _name : _type _name SWIFT_COMPILE_NAME(SWIFT_NAME); enum SWIFT_COMPILE_NAME(SWIFT_NAME) SWIFT_ENUM_EXTRA _name : _type
+#define SWIFT_ENUM_NAMED(_type, _name, SWIFT_NAME) enum _name : _type _name SWIFT_COMPILE_NAME(SWIFT_NAME); enum SWIFT_COMPILE_NAME(SWIFT_NAME) __attribute__((enum_extensibility(open))) SWIFT_ENUM_EXTRA _name : _type
 
 typedef SWIFT_ENUM_NAMED(NSInteger, ObjCEnum, "SwiftEnum") {
     ObjCEnumOne = 1,

diff --git a/test/ClangImporter/enum.swift b/test/ClangImporter/enum.swift
@@ -11,6 +11,7 @@
 
 import Foundation
 import user_objc
+import enums_using_attributes
 
 // NS_ENUM
 var mince = NSRuncingMode.mince
@@ -148,6 +149,9 @@ var qualifiedName = NSRuncingMode.mince
 var topLevelCaseName = RuncingMince // expected-error{{}}
 #endif
 
+let _: EnumViaAttribute = .first
+let _: CFEnumWithAttr = .first
+
 // NS_OPTIONS
 var withMince: NSRuncingOptions = .enableMince
 var withQuince: NSRuncingOptions = .enableQuince
@@ -193,6 +197,8 @@ let audioComponentFlags2: FakeAudioComponentFlags = [.loadOutOfProcess]
 let objcFlags: objc_flags = [.taggedPointer, .swiftRefcount]
 
 let optionsWithSwiftName: NSOptionsAlsoGetSwiftName = .Case
+let optionsViaAttribute: OptionsViaAttribute = [.first, .second]
+let optionsViaAttribute2: CFOptionsWithAttr = [.first]
 
 // <rdar://problem/25168818> Don't import None members in NS_OPTIONS types
 #if !IRGEN
@@ -208,3 +214,10 @@ _ = EmptySet3.None
 // Just use this type, making sure that its case alias doesn't cause problems.
 // rdar://problem/30401506
 _ = EnumWithAwkwardDeprecations.normalCase1
+
+#if !IRGEN
+let _: UnknownEnumThanksToAPINotes = .first // expected-error {{has no member 'first'}}
+let _: UnknownOptionsThanksToAPINotes = .first // expected-error {{has no member 'first'}}
+#endif
+let _ = UnknownEnumThanksToAPINotesFirst
+let _ = UnknownOptionsThanksToAPINotesFirst
diff --git a/test/IDE/Inputs/print_clang_header_swift_name.h b/test/IDE/Inputs/print_clang_header_swift_name.h
@@ -2,28 +2,23 @@
 
 #define SWIFT_COMPILE_NAME(X) __attribute__((swift_name(X)))
 
-#define SWIFT_ENUM(_type, _name) enum _name : _type _name; enum SWIFT_ENUM_EXTRA _name : _type
+#define SWIFT_ENUM(_type, _name) enum _name : _type _name; enum __attribute__((enum_extensibility(open))) SWIFT_ENUM_EXTRA _name : _type
 
 typedef SWIFT_ENUM(NSInteger, Normal) {
     NormalOne = 0,
     NormalTwo,
     NormalThree
 };
 
-// FIXME (#618): Use SWIFT_ENUM_NAMED() when support for that lands
-#undef SWIFT_ENUM
-#define SWIFT_ENUM(_type, _name) enum _name : _type _name SWIFT_COMPILE_NAME(SWIFT_ENUM_NAME); enum SWIFT_COMPILE_NAME(SWIFT_ENUM_NAME) SWIFT_ENUM_EXTRA _name : _type
+#define SWIFT_ENUM_NAMED(_type, _name, _swiftName) enum _name : _type _name SWIFT_COMPILE_NAME(_swiftName); enum SWIFT_COMPILE_NAME(_swiftName) __attribute__((enum_extensibility(open))) SWIFT_ENUM_EXTRA _name : _type
 
-#define SWIFT_ENUM_NAME "SwiftEnum"
-typedef SWIFT_ENUM(NSInteger, ObjCEnum) {
+typedef SWIFT_ENUM_NAMED(NSInteger, ObjCEnum, "SwiftEnum") {
     ObjCEnumOne = 0,
     ObjCEnumTwo,
     ObjCEnumThree
 };
 
-#undef SWIFT_ENUM_NAME
-#define SWIFT_ENUM_NAME "SwiftEnumTwo"
-typedef SWIFT_ENUM(NSInteger, ObjCEnumTwo) {
+typedef SWIFT_ENUM_NAMED(NSInteger, ObjCEnumTwo, "SwiftEnumTwo") {
     // the following shouldn't have their prefixes stripped
     SwiftEnumTwoA,
     SwiftEnumTwoB,

diff --git a/test/IDE/Inputs/swift_name.h b/test/IDE/Inputs/swift_name.h
@@ -7,7 +7,7 @@
 #ifndef SWIFT_ENUM
 #  define SWIFT_ENUM(_type, _name)    \
   enum _name : _type _name;           \
-  enum SWIFT_ENUM_EXTRA _name : _type
+  enum __attribute__((enum_extensibility(open))) SWIFT_ENUM_EXTRA _name : _type
 #endif
 
 // Renaming global variables.

diff --git a/test/Inputs/clang-importer-sdk/usr/include/enums_using_attributes.apinotes b/test/Inputs/clang-importer-sdk/usr/include/enums_using_attributes.apinotes
@@ -0,0 +1,6 @@
+Name: enums_using_attributes
+Tags:
+- Name: UnknownEnumThanksToAPINotes
+  EnumKind: none
+- Name: UnknownOptionsThanksToAPINotes
+  EnumKind: none
diff --git a/test/Inputs/clang-importer-sdk/usr/include/enums_using_attributes.h b/test/Inputs/clang-importer-sdk/usr/include/enums_using_attributes.h
@@ -0,0 +1,18 @@
+#define CF_ENUM(_type, _name) enum _name : _type _name; \
+  enum __attribute__((enum_extensibility(open))) _name : _type
+#define CF_OPTIONS(_type, _name) enum _name : _type _name; \
+  enum __attribute__((enum_extensibility(open), flag_enum)) _name : _type
+
+typedef CF_ENUM(int, CFEnumWithAttr) {
+  CFEnumWithAttrFirst = 1,
+};
+typedef CF_ENUM(int, UnknownEnumThanksToAPINotes) {
+  UnknownEnumThanksToAPINotesFirst = 1,
+};
+
+typedef CF_OPTIONS(int, CFOptionsWithAttr) {
+  CFOptionsWithAttrFirst = 1,
+};
+typedef CF_OPTIONS(int, UnknownOptionsThanksToAPINotes) {
+  UnknownOptionsThanksToAPINotesFirst = 1,
+};
diff --git a/test/Inputs/clang-importer-sdk/usr/include/module.map b/test/Inputs/clang-importer-sdk/usr/include/module.map
@@ -98,6 +98,10 @@ module objc_structs {
   export *
 }
 
+module enums_using_attributes {
+  header "enums_using_attributes.h"
+}
+
 module errors {
   header "errors.h"
   export *

diff --git a/test/Inputs/clang-importer-sdk/usr/include/user_objc.h b/test/Inputs/clang-importer-sdk/usr/include/user_objc.h
@@ -29,6 +29,12 @@ typedef NS_ENUM(NSInteger, EnumWithAwkwardDeprecations) {
   EnumWithAwkward2BitProblems __attribute__((deprecated)) = EnumWithAwkwardNormalCase1,
 };
 
+enum __attribute__((enum_extensibility(open))) EnumViaAttribute {
+  EnumViaAttributeFirst = 1,
+  EnumViaAttributeSecond = 2
+};
+
+
 // From <AudioUnit/AudioComponent.h>
 // The interesting feature of this enum is that the common prefix before
 // taking the enum name itself into account extends past the underscore.
@@ -68,3 +74,8 @@ typedef CF_OPTIONS(UInt32, EmptySet2) {
 typedef CF_OPTIONS(UInt32, EmptySet3) {
   kEmptySet3None __attribute__((swift_name("None")))
 };
+
+enum __attribute__((flag_enum)) OptionsViaAttribute {
+  OptionsViaAttributeFirst = 1,
+  OptionsViaAttributeSecond = 2
+};