[Macro] Add the expression attribute for macro declarations

Author: DougGregor

include/swift/AST/Decl.h

@@ -8297,6 +8297,17 @@ class MissingMemberDecl : public Decl {
   }
 };
 
+/// The context in which a macro can be used, which determines the syntax it
+/// uses.
+enum class MacroContext: uint8_t {
+  /// An expression macro, referenced explicitly via "#stringify" or similar
+  /// in the source code.
+  Expression,
+};
+
+/// The contexts in which a particular macro declaration can be used.
+using MacroContexts = OptionSet<MacroContext>;
+
 /// Provides a declaration of a macro.
 ///
 /// Macros are declared within the source code with the `macro` introducer.
@@ -8348,6 +8359,9 @@ class MacroDecl : public GenericContext, public ValueDecl {
   /// Retrieve the interface type produced when expanding this macro.
   Type getResultInterfaceType() const;
 
+  /// Determine the contexts in which this macro can be applied.
+  MacroContexts getMacroContexts() const;
+
   static bool classof(const DeclContext *C) {
     if (auto D = C->getAsDecl())
       return classof(D);

include/swift/AST/MacroDefinition.h

@@ -25,12 +25,6 @@ namespace swift {
 /// Provides the definition of a macro.
 class MacroDefinition {
 public:
-  /// The kind of macro, which determines how it can be used in source code.
-  enum Kind: uint8_t {
-    /// An expression macro.
-    Expression,
-  };
-
   /// Describes a missing macro definition.
   struct MissingDefinition {
     Identifier externalModuleName;
@@ -50,19 +44,18 @@ class MacroDefinition {
     InProcess,
   };
 
-  Kind kind;
   ImplementationKind implKind;
 
 private:
   void *opaqueHandle;
 
-  MacroDefinition(Kind kind, ImplementationKind implKind, void *opaqueHandle)
-    : kind(kind), implKind(implKind), opaqueHandle(opaqueHandle) { }
+  MacroDefinition(ImplementationKind implKind, void *opaqueHandle)
+    : implKind(implKind), opaqueHandle(opaqueHandle) { }
 
 public:
   static MacroDefinition forUndefined() {
     return MacroDefinition{
-      Kind::Expression, ImplementationKind::Undefined, nullptr
+      ImplementationKind::Undefined, nullptr
     };
   }
 
@@ -72,8 +65,8 @@ class MacroDefinition {
       Identifier externalMacroTypeName
   );
 
-  static MacroDefinition forInProcess(Kind kind, void *opaqueHandle) {
-    return MacroDefinition{kind, ImplementationKind::InProcess, opaqueHandle};
+  static MacroDefinition forInProcess(void *opaqueHandle) {
+    return MacroDefinition{ImplementationKind::InProcess, opaqueHandle};
   }
 
   /// Return the opaque handle for an in-process macro definition.

lib/AST/Decl.cpp

@@ -9689,9 +9689,7 @@ MacroDefinition MacroDefinition::forMissing(
   auto def = ctx.AllocateObjectCopy(
     MissingDefinition{externalModuleName, externalMacroTypeName}
   );
-  return MacroDefinition{
-    Kind::Expression, ImplementationKind::Missing, def
-  };
+  return MacroDefinition{ImplementationKind::Missing, def};
 }
 
 NominalTypeDecl *

lib/Sema/TypeCheckAttr.cpp

@@ -159,6 +159,7 @@ class AttributeChecker : public AttributeVisitor<AttributeChecker> {
   IGNORED_ATTR(Preconcurrency)
   IGNORED_ATTR(BackDeploy)
   IGNORED_ATTR(Documentation)
+  IGNORED_ATTR(Expression)
 #undef IGNORED_ATTR
 
   void visitAlignmentAttr(AlignmentAttr *attr) {

lib/Sema/TypeCheckDeclOverride.cpp

@@ -1625,6 +1625,8 @@ namespace  {
     UNINTERESTING_ATTR(NoEagerMove)
 
     UNINTERESTING_ATTR(RuntimeMetadata)
+
+    UNINTERESTING_ATTR(Expression)
 #undef UNINTERESTING_ATTR
 
     void visitAvailableAttr(AvailableAttr *attr) {

lib/Sema/TypeCheckMacros.cpp

@@ -113,8 +113,7 @@ MacroDefinition MacroDefinitionRequest::evaluate(
         swift_ASTGen_destroyMacro(inProcess);
       });
 
-      return MacroDefinition::forInProcess(
-          MacroDefinition::Expression, inProcess);
+      return MacroDefinition::forInProcess(inProcess);
     }
   }
 #endif

test/IDE/complete_decl_attribute.swift

@@ -315,6 +315,7 @@ struct _S {
 // ON_MEMBER_LAST-DAG: Keyword/None:                       typeWrapper[#Declaration Attribute#]; name=typeWrapper
 // ON_MEMBER_LAST-DAG: Keyword/None:                       typeWrapperIgnored[#Declaration Attribute#]; name=typeWrapperIgnored
 // ON_MEMBER_LAST-DAG: Keyword/None:                       runtimeMetadata[#Declaration Attribute#]; name=runtimeMetadata
+// ON_MEMBER_LAST-DAG: Keyword/None:                       expression[#Declaration Attribute#]; name=expression
 // ON_MEMBER_LAST-NOT: Keyword
 // ON_MEMBER_LAST-DAG: Decl[Struct]/CurrModule:            MyStruct[#MyStruct#]; name=MyStruct
 // ON_MEMBER_LAST-DAG: Decl[Struct]/CurrModule/TypeRelation[Convertible]: MyPropertyWrapper[#MyPropertyWrapper#]; name=MyPropertyWrapper
@@ -387,6 +388,7 @@ func dummy2() {}
 // KEYWORD_LAST-DAG: Keyword/None:                       typeWrapper[#Declaration Attribute#]; name=typeWrapper
 // KEYWORD_LAST-DAG: Keyword/None:                       typeWrapperIgnored[#Declaration Attribute#]; name=typeWrapperIgnored
 // KEYWORD_LAST-DAG: Keyword/None:                       runtimeMetadata[#Declaration Attribute#]; name=runtimeMetadata
+// KEYWORD_LAST-DAG: Keyword/None:                       expression[#Declaration Attribute#]; name=expression
 // KEYWORD_LAST-NOT: Keyword
 // KEYWORD_LAST-DAG: Decl[Struct]/CurrModule:            MyStruct[#MyStruct#]; name=MyStruct
 // KEYWORD_LAST-DAG: Decl[Struct]/CurrModule/TypeRelation[Convertible]: MyPropertyWrapper[#MyPropertyWrapper#]; name=MyPropertyWrapper

utils/gyb_syntax_support/AttributeKinds.py

@@ -902,6 +902,11 @@ def __init__(self, name, swift_name=None):
                                   ABIBreakingToAdd, ABIBreakingToRemove,
                                   APIBreakingToAdd, APIBreakingToRemove,
                                   code=130),
+
+    SimpleDeclAttribute('expression', 'Expression',
+                        OnMacro,
+                        ABIStableToAdd, ABIStableToRemove, APIStableToAdd, APIBreakingToRemove,  # noqa: E501
+                        code=140),
 ]
 
 DEPRECATED_MODIFIER_KINDS = [