AST: Add AnyFunctionType::hasEffect()

Author: slavapestov

include/swift/AST/Types.h

@@ -76,6 +76,7 @@ class TypeAliasDecl;
 class TypeDecl;
 class NominalTypeDecl;
 class GenericTypeDecl;
+enum class EffectKind : uint8_t;
 class EnumDecl;
 class EnumElementDecl;
 class SILFunctionType;
@@ -3159,6 +3160,8 @@ class AnyFunctionType : public TypeBase {
 
   bool isThrowing() const { return getExtInfo().isThrowing(); }
 
+  bool hasEffect(EffectKind kind) const;
+
   bool isDifferentiable() const { return getExtInfo().isDifferentiable(); }
   DifferentiabilityKind getDifferentiabilityKind() const {
     return getExtInfo().getDifferentiabilityKind();

lib/AST/Effects.cpp

@@ -20,12 +20,21 @@
 #include "swift/AST/Decl.h"
 #include "swift/AST/ProtocolConformanceRef.h"
 #include "swift/AST/Type.h"
+#include "swift/AST/Types.h"
 #include "swift/AST/TypeCheckRequests.h"
 #include "llvm/ADT/ArrayRef.h"
 #include "llvm/Support/raw_ostream.h"
 
 using namespace swift;
 
+bool AnyFunctionType::hasEffect(EffectKind kind) const {
+  switch (kind) {
+  case EffectKind::Throws: return getExtInfo().isThrowing();
+  case EffectKind::Async: return getExtInfo().isAsync();
+  }
+  llvm_unreachable("Bad effect kind");
+}
+
 void swift::simple_display(llvm::raw_ostream &out, const EffectKind kind) {
   switch (kind) {
   case EffectKind::Throws: out << "throws"; return;

lib/Sema/TypeCheckEffects.cpp

@@ -35,12 +35,7 @@ static bool hasFunctionParameterWithEffect(EffectKind kind, Type type) {
 
   // Only consider function types with this effect.
   if (auto fnType = type->getAs<AnyFunctionType>()) {
-    auto extInfo = fnType->getExtInfo();
-    switch (kind) {
-    case EffectKind::Throws: return extInfo.isThrowing();
-    case EffectKind::Async: return extInfo.isAsync();
-    }
-    llvm_unreachable("Bad effect kind");
+    return fnType->hasEffect(kind);
   }
 
   // Look through tuples.