swift-mirror/lib/AST/Attr.cpp

//===--- Attr.cpp - Swift Language Attr ASTs ------------------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2017 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//
//
//  This file implements routines relating to declaration attributes.
//
//===----------------------------------------------------------------------===//

#include "swift/AST/Attr.h"
#include "swift/AST/ASTContext.h"
#include "swift/AST/ASTPrinter.h"
#include "swift/AST/AvailabilityDomain.h"
#include "swift/AST/Decl.h"
#include "swift/AST/Expr.h"
#include "swift/AST/GenericEnvironment.h"
#include "swift/AST/IndexSubset.h"
#include "swift/AST/LazyResolver.h"
#include "swift/AST/Module.h"
#include "swift/AST/NameLookupRequests.h"
#include "swift/AST/ParameterList.h"
#include "swift/AST/TypeCheckRequests.h"
#include "swift/AST/TypeRepr.h"
#include "swift/AST/Types.h"
#include "swift/Basic/Assertions.h"
#include "swift/Basic/Defer.h"
#include "swift/Basic/QuotedString.h"
#include "swift/Strings.h"
#include "llvm/ADT/SmallSet.h"
#include "llvm/ADT/SmallString.h"
#include "llvm/ADT/StringSwitch.h"
#include "llvm/Support/ErrorHandling.h"
#include "llvm/Support/raw_ostream.h"
using namespace swift;

static_assert(IsTriviallyDestructible<DeclAttributes>::value,
              "DeclAttributes are BumpPtrAllocated; the d'tor is never called");

#define DECL_ATTR(Name, Id, ...) \
  static_assert(DeclAttrKind::Id <= DeclAttrKind::Last_DeclAttr); \
  static_assert(IsTriviallyDestructible<Id##Attr>::value, \
                "Attrs are BumpPtrAllocated; the destructor is never called"); \
  static_assert(DeclAttribute::hasOneBehaviorFor##Id( \
                DeclAttribute::ABIBreakingToAdd | DeclAttribute::ABIStableToAdd), \
                #Name " needs to specify either ABIBreakingToAdd or ABIStableToAdd"); \
  static_assert(DeclAttribute::hasOneBehaviorFor##Id( \
                DeclAttribute::ABIBreakingToRemove | DeclAttribute::ABIStableToRemove), \
                #Name " needs to specify either ABIBreakingToRemove or ABIStableToRemove"); \
  static_assert(DeclAttribute::hasOneBehaviorFor##Id( \
                DeclAttribute::APIBreakingToAdd | DeclAttribute::APIStableToAdd),\
                #Name " needs to specify either APIBreakingToAdd or APIStableToAdd"); \
  static_assert(DeclAttribute::hasOneBehaviorFor##Id( \
                DeclAttribute::APIBreakingToRemove | DeclAttribute::APIStableToRemove), \
                #Name " needs to specify either APIBreakingToRemove or APIStableToRemove"); \
  static_assert(DeclAttribute::hasOneBehaviorFor##Id(DeclAttribute::InABIAttrMask), \
                #Name " needs to specify exactly one of ForbiddenInABIAttr, UnconstrainedInABIAttr, EquivalentInABIAttr, or UnreachableInABIAttr");
#include "swift/AST/DeclAttr.def"

#define TYPE_ATTR(_, Id)                                                       \
  static_assert(TypeAttrKind::Id <= TypeAttrKind::Last_TypeAttr);
#include "swift/AST/TypeAttr.def"

LLVM_ATTRIBUTE_USED StringRef swift::getDeclAttrKindID(DeclAttrKind kind) {
    switch (kind) {
#define DECL_ATTR(_, CLASS, ...)                               \
  case DeclAttrKind::CLASS:                                    \
    return #CLASS;
#include "swift/AST/DeclAttr.def"
  }
}

StringRef swift::getAccessLevelSpelling(AccessLevel value) {
  switch (value) {
  case AccessLevel::Private: return "private";
  case AccessLevel::FilePrivate: return "fileprivate";
  case AccessLevel::Internal: return "internal";
  case AccessLevel::Package: return "package";
  case AccessLevel::Public: return "public";
  case AccessLevel::Open: return "open";
  }

  llvm_unreachable("Unhandled AccessLevel in switch.");
}

SourceLoc TypeAttribute::getStartLoc() const {
  switch (getKind()) {
#define TYPE_ATTR(_, CLASS)                                                    \
  case TypeAttrKind::CLASS:                                                    \
    return static_cast<const CLASS##TypeAttr *>(this)->getStartLocImpl();
#include "swift/AST/TypeAttr.def"
  }
  llvm_unreachable("bad kind");
}

SourceLoc TypeAttribute::getEndLoc() const {
  switch (getKind()) {
#define TYPE_ATTR(_, CLASS)                                                    \
  case TypeAttrKind::CLASS:                                                    \
    return static_cast<const CLASS##TypeAttr *>(this)->getEndLocImpl();
#include "swift/AST/TypeAttr.def"
  }
  llvm_unreachable("bad kind");
}

SourceRange TypeAttribute::getSourceRange() const {
  switch (getKind()) {
#define TYPE_ATTR(_, CLASS)                                                    \
  case TypeAttrKind::CLASS: {                                                  \
    auto attr = static_cast<const CLASS##TypeAttr *>(this);                    \
    return SourceRange(attr->getStartLocImpl(), attr->getEndLocImpl());        \
  }
#include "swift/AST/TypeAttr.def"
  }
  llvm_unreachable("bad kind");
}

/// Given a name like "autoclosure", return the type attribute ID that
/// corresponds to it.
///
std::optional<TypeAttrKind>
TypeAttribute::getAttrKindFromString(StringRef Str) {
  return llvm::StringSwitch<std::optional<TypeAttrKind>>(Str)
#define TYPE_ATTR(X, C) .Case(#X, TypeAttrKind::C)
#include "swift/AST/TypeAttr.def"
      .Default(std::nullopt);
}

bool TypeAttribute::isSilOnly(TypeAttrKind TK) {
  switch (TK) {
#define SIL_TYPE_ATTR(X, C) case TypeAttrKind::C:
#include "swift/AST/TypeAttr.def"
    return true;
  default:
    return false;
  }
}

/// Return the name (like "autoclosure") for an attribute ID.
const char *TypeAttribute::getAttrName(TypeAttrKind kind) {
  switch (kind) {
#define TYPE_ATTR(X, C)                                                        \
  case TypeAttrKind::C:                                                        \
    return #X;
#include "swift/AST/TypeAttr.def"
  }
  llvm_unreachable("unknown type attribute kind");
}

bool TypeAttribute::isUserInaccessible(TypeAttrKind DK) {
  // Currently we can base this off whether it is underscored or for SIL.
  // TODO: We could introduce a similar options scheme to DECL_ATTR if we ever
  // need a user-inaccessible non-underscored attribute.
  switch (DK) {
    // SIL attributes are always considered user-inaccessible.
#define SIL_TYPE_ATTR(SPELLING, C)                                             \
  case TypeAttrKind::C:                                                        \
    return true;
    // For non-SIL attributes, check whether the spelling is underscored.
#define TYPE_ATTR(SPELLING, C)                                                 \
  case TypeAttrKind::C:                                                        \
    return StringRef(#SPELLING).starts_with("_");
#include "swift/AST/TypeAttr.def"
  }
  llvm_unreachable("unhandled case in switch!");
}

TypeAttribute *TypeAttribute::createSimple(const ASTContext &context,
                                           TypeAttrKind kind,
                                           SourceLoc atLoc,
                                           SourceLoc attrLoc) {

  switch (kind) {
  // The simple cases should all be doing the exact same thing, and we
  // can reasonably hope that the optimizer will unify them so that this
  // function doesn't actually need a switch.
#define TYPE_ATTR(SPELLING, CLASS)                                             \
  case TypeAttrKind::CLASS:                                                    \
    llvm_unreachable("not a simple attribute");
#define SIMPLE_TYPE_ATTR(SPELLING, CLASS)                                      \
  case TypeAttrKind::CLASS:                                                    \
    return new (context) CLASS##TypeAttr(atLoc, attrLoc);
#include "swift/AST/TypeAttr.def"
  }
  llvm_unreachable("bad type attribute kind");
}

void TypeAttribute::dump() const {
  StreamPrinter P(llvm::errs());
  PrintOptions PO = PrintOptions::printDeclarations();
  print(P, PO);
}

void TypeAttribute::print(ASTPrinter &printer,
                          const PrintOptions &options) const {
  switch (getKind()) {
#define TYPE_ATTR(_, CLASS)
#define SIMPLE_TYPE_ATTR(_, CLASS) case TypeAttrKind::CLASS:
#include "swift/AST/TypeAttr.def"
    printer.printSimpleAttr(getAttrName(getKind()), /*needAt*/ true);
    return;

#define TYPE_ATTR(_, CLASS)                                                    \
  case TypeAttrKind::CLASS:                                                    \
    return cast<CLASS##TypeAttr>(this)->printImpl(printer, options);
#define SIMPLE_TYPE_ATTR(_, C)
#include "swift/AST/TypeAttr.def"
  }
  llvm_unreachable("bad kind");
}

void DifferentiableTypeAttr::printImpl(ASTPrinter &printer,
                                       const PrintOptions &options) const {
  printer.callPrintStructurePre(PrintStructureKind::BuiltinAttribute);
  printer.printAttrName("@differentiable");
  switch (getDifferentiability()) {
  case DifferentiabilityKind::Normal:
    break;
  case DifferentiabilityKind::Forward:
    printer << "(_forward)";
    break;
  case DifferentiabilityKind::Reverse:
    printer << "(reverse)";
    break;
  case DifferentiabilityKind::Linear:
    printer << "(_linear)";
    break;
  case DifferentiabilityKind::NonDifferentiable:
    llvm_unreachable("Unexpected case 'NonDifferentiable'");
  }
  printer << ' ';
  printer.printStructurePost(PrintStructureKind::BuiltinAttribute);
}

void ConventionTypeAttr::printImpl(ASTPrinter &printer,
                                   const PrintOptions &options) const {
  printer.callPrintStructurePre(PrintStructureKind::BuiltinAttribute);
  printer.printAttrName("@convention");
  printer << "(" << getConventionName();
  if (auto protocol = getWitnessMethodProtocol()) {
    printer << ": " << protocol;
  } else if (auto clangType = getClangType()) {
    printer << ", cType: " << QuotedString(*clangType);
  }
  printer << ")";
  printer.printStructurePost(PrintStructureKind::BuiltinAttribute);
}

void OpaqueReturnTypeOfTypeAttr::printImpl(ASTPrinter &printer,
                                           const PrintOptions &options) const {
  printer.callPrintStructurePre(PrintStructureKind::BuiltinAttribute);
  printer.printAttrName("@_opaqueReturnTypeOf");
  printer << "(" << QuotedString(getMangledName()) << ", " << getIndex() << ")";
  printer.printStructurePost(PrintStructureKind::BuiltinAttribute);
}

void OpenedTypeAttr::printImpl(ASTPrinter &printer,
                               const PrintOptions &options) const {
  printer.callPrintStructurePre(PrintStructureKind::BuiltinAttribute);
  printer.printAttrName("@opened");
  printer << "(\"" << getUUID() << "\"";
  if (auto constraintType = getConstraintType()) {
    printer << ", ";
    constraintType->print(printer, options);
  }
  printer << ")";
  printer.printStructurePost(PrintStructureKind::BuiltinAttribute);
}

void PackElementTypeAttr::printImpl(ASTPrinter &printer,
                                    const PrintOptions &options) const {
  printer.callPrintStructurePre(PrintStructureKind::BuiltinAttribute);
  printer.printAttrName("@pack_element");
  printer << "(\"" << getUUID() << "\")";
  printer.printStructurePost(PrintStructureKind::BuiltinAttribute);
}

const char *IsolatedTypeAttr::getIsolationKindName(IsolationKind kind) {
  switch (kind) {
  case IsolationKind::Dynamic: return "any";
  }
  llvm_unreachable("bad kind");
}

void IsolatedTypeAttr::printImpl(ASTPrinter &printer,
                                 const PrintOptions &options) const {
  printer.callPrintStructurePre(PrintStructureKind::BuiltinAttribute);
  printer.printAttrName("@isolated");
  printer << "(" << getIsolationKindName() << ")";
  printer.printStructurePost(PrintStructureKind::BuiltinAttribute);
}

/// Given a name like "inline", return the decl attribute ID that corresponds
/// to it.  Note that this is a many-to-one mapping, and that the identifier
/// passed in may only be the first portion of the attribute (e.g. in the case
/// of the 'unowned(unsafe)' attribute, the string passed in is 'unowned'.
///
/// Also note that this recognizes both attributes like '@inline' (with no @)
/// and decl modifiers like 'final'.
///
std::optional<DeclAttrKind>
DeclAttribute::getAttrKindFromString(StringRef Str) {
  return llvm::StringSwitch<std::optional<DeclAttrKind>>(Str)
#define DECL_ATTR(X, CLASS, ...) .Case(#X, DeclAttrKind::CLASS)
#define DECL_ATTR_ALIAS(X, CLASS) .Case(#X, DeclAttrKind::CLASS)
#include "swift/AST/DeclAttr.def"
      .Case(SPI_AVAILABLE_ATTRNAME, DeclAttrKind::Available)
      .Default(std::nullopt);
}

DeclAttribute *DeclAttribute::createSimple(const ASTContext &context,
                                           DeclAttrKind kind, SourceLoc atLoc,
                                           SourceLoc attrLoc) {
  switch (kind) {
    // The simple cases should all be doing the exact same thing, and we
    // can reasonably hope that the optimizer will unify them so that this
    // function doesn't actually need a switch.
#define DECL_ATTR(SPELLING, CLASS, ...)                                        \
  case DeclAttrKind::CLASS:                                                    \
    llvm_unreachable("not a simple attribute");
#define SIMPLE_DECL_ATTR(SPELLING, CLASS, ...)                                 \
  case DeclAttrKind::CLASS:                                                    \
    return new (context) CLASS##Attr(atLoc, attrLoc);
#include "swift/AST/DeclAttr.def"
  }
  llvm_unreachable("bad decl attribute kind");
}

/// Returns true if this attribute can appear on the specified decl.
bool DeclAttribute::canAttributeAppearOnDecl(DeclAttrKind DK, const Decl *D) {
  if ((getRequirements(DK) & OnAnyClangDecl) && D->hasClangNode())
    return true;
  return canAttributeAppearOnDeclKind(DK, D->getKind());
}

bool DeclAttribute::canAttributeAppearOnDeclKind(DeclAttrKind DAK, DeclKind DK) {
  auto Reqs = getRequirements(DAK);
  switch (DK) {
#define DECL(Id, Parent) case DeclKind::Id: return (Reqs & On##Id) != 0;
#include "swift/AST/DeclNodes.def"
  }
  llvm_unreachable("bad DeclKind");
}

void DeclAttribute::attachToDecl(Decl *D) {
  ASSERT(D);
  switch (getKind()) {
#define DECL_ATTR(_, CLASS, ...)                                               \
  case DeclAttrKind::CLASS:                                                    \
    return static_cast<CLASS##Attr *>(this)->attachToDeclImpl(D);
#include "swift/AST/DeclAttr.def"
  }
}

// Ensure that every DeclAttribute subclass implements its own CloneAttr.
static void checkDeclAttributeClones() {
#define DECL_ATTR(_,CLASS,...) \
  CLASS##Attr *(CLASS##Attr::*ptr##CLASS)(ASTContext &) const = &CLASS##Attr::clone; \
  (void)ptr##CLASS;
#include "swift/AST/DeclAttr.def"
}

DeclAttribute *DeclAttribute::clone(ASTContext &ctx) const {
  (void)checkDeclAttributeClones;
  switch (getKind()) {
#define DECL_ATTR(_,CLASS, ...) \
  case DeclAttrKind::CLASS: return static_cast<const CLASS##Attr *>(this)->clone(ctx);
#include "swift/AST/DeclAttr.def"
  }
}

bool DeclAttribute::canClone() const {
  switch (getKind()) {
#define DECL_ATTR(_,CLASS, ...)                                \
  case DeclAttrKind::CLASS:                                    \
    if (&CLASS##Attr::canClone == &DeclAttribute::canClone)    \
      return true;                                             \
    return static_cast<const CLASS##Attr *>(this)->canClone();
#include "swift/AST/DeclAttr.def"
  }
}

// Ensure that every DeclAttribute subclass implements its own isEquivalent().
static void checkDeclAttributeIsEquivalents() {
#define DECL_ATTR(_,CLASS,...) \
  bool(CLASS##Attr::*ptr##CLASS)(const CLASS##Attr *, Decl *) const = &CLASS##Attr::isEquivalent; \
  (void)ptr##CLASS;
#include "swift/AST/DeclAttr.def"
}

bool DeclAttribute::isEquivalent(const DeclAttribute *other, Decl *attachedTo) const {
  (void)checkDeclAttributeIsEquivalents;
  if (getKind() != other->getKind())
    return false;
  switch (getKind()) {
#define DECL_ATTR(_,CLASS, ...) \
  case DeclAttrKind::CLASS:\
    return static_cast<const CLASS##Attr *>(this)->isEquivalent(\
                static_cast<const CLASS##Attr *>(other), attachedTo);
#include "swift/AST/DeclAttr.def"
  }
}

const BackDeployedAttr *
DeclAttributes::getBackDeployed(const ASTContext &ctx,
                                bool forTargetVariant) const {
  const BackDeployedAttr *bestAttr = nullptr;

  for (auto attr : *this) {
    auto *backDeployedAttr = dyn_cast<BackDeployedAttr>(attr);
    if (!backDeployedAttr)
      continue;

    if (backDeployedAttr->isInvalid() ||
        !backDeployedAttr->isActivePlatform(ctx, forTargetVariant))
      continue;

    // We have an attribute that is active for the platform, but
    // is it more specific than our current best?
    if (!bestAttr ||
        inheritsAvailabilityFromPlatform(backDeployedAttr->getPlatform(),
                                         bestAttr->getPlatform())) {
      bestAttr = backDeployedAttr;
    }
  }

  return bestAttr;
}

void DeclAttributes::print(const Decl *D) const {
  StreamPrinter P(llvm::errs());
  PrintOptions PO = PrintOptions::printDeclarations();
  print(P, PO, D);
}

/// Returns true if the attribute can be presented as a short form available
/// attribute (e.g., as @available(iOS 8.0, *). The presentation requires an
/// introduction version and does not support deprecation, obsoletion, or
/// messages.
LLVM_READONLY
static bool isShortAvailable(const SemanticAvailableAttr &attr) {
  auto parsedAttr = attr.getParsedAttr();
  if (attr.isSPI())
    return false;

  if (!attr.getIntroduced().has_value())
    return false;

  if (attr.getDeprecated().has_value())
    return false;

  if (attr.getObsoleted().has_value())
    return false;

  if (!attr.getMessage().empty())
    return false;

  if (!attr.getRename().empty())
    return false;

  switch (parsedAttr->getKind()) {
  case AvailableAttr::Kind::NoAsync:
  case AvailableAttr::Kind::Deprecated:
  case AvailableAttr::Kind::Unavailable:
    return false;
  case AvailableAttr::Kind::Default:
    return true;
  }

  return true;
}

/// Return true when another availability attribute implies the same availability as this
/// attribute and so printing the attribute can be skipped to de-clutter the declaration
/// when printing the short form.
/// For example, iOS availability implies macCatalyst availability so if attributes for
/// both are present and they have the same 'introduced' version, we can skip printing an
/// explicit availability for macCatalyst.
static bool
isShortFormAvailabilityImpliedByOther(SemanticAvailableAttr Attr,
                                      ArrayRef<SemanticAvailableAttr> Others) {
  assert(isShortAvailable(Attr));

  auto platform = Attr.getDomain().getPlatformKind();
  for (auto other : Others) {
    auto otherPlatform = other.getDomain().getPlatformKind();
    if (platform == otherPlatform)
      continue;

    if (!inheritsAvailabilityFromPlatform(platform, otherPlatform))
      continue;

    if (Attr.getIntroduced() == other.getIntroduced())
      return true;
  }
  return false;
}

/// Print the short-form @available() attribute for an array of long-form
/// AvailableAttrs that can be represented in the short form.
/// For example, for:
///   @available(OSX, introduced: 10.10)
///   @available(iOS, introduced: 8.0)
/// this will print:
///   @available(OSX 10.10, iOS 8.0, *)
static void printShortFormAvailable(const Decl *D,
                                    ArrayRef<SemanticAvailableAttr> Attrs,
                                    ASTPrinter &Printer,
                                    const PrintOptions &Options,
                                    bool forAtSpecialize = false) {
  assert(!Attrs.empty());
  if (!forAtSpecialize)
    Printer << "@available(";

  bool isFirst = true;
  bool isPlatformAvailability = false;
  for (auto attr : Attrs) {
    auto domain = attr.getDomain();
    auto introduced = attr.getIntroduced();
    assert(introduced);

    // Avoid omitting available attribute when we are printing module interface.
    if (!Options.IsForSwiftInterface &&
        isShortFormAvailabilityImpliedByOther(attr, Attrs))
      continue;

    Printer << (isFirst ? "" : ", ");
    isFirst = false;

    if (domain.isPlatform())
      isPlatformAvailability = true;

    Printer << domain.getNameForAttributePrinting() << " "
            << introduced.value().getAsString();
  }

  if (isPlatformAvailability)
    Printer << ", *";

  if (!forAtSpecialize) {
    Printer << ")";
    Printer.printNewline();
  }
}

static void printShortFormBackDeployed(ArrayRef<const DeclAttribute *> Attrs,
                                       ASTPrinter &Printer,
                                       const PrintOptions &Options) {
  assert(!Attrs.empty());
  Printer << "@backDeployed(before: ";
  bool isFirst = true;

  for (auto *DA : Attrs) {
    if (!isFirst)
      Printer << ", ";
    auto *attr = cast<BackDeployedAttr>(DA);
    Printer << platformString(attr->getPlatform()) << " "
            << attr->getVersion().getAsString();
    isFirst = false;
  }
  Printer << ")";
  Printer.printNewline();
}

/// The kind of a parameter in a `wrt:` differentiation parameters clause:
/// either a differentiability parameter or a linearity parameter. Used for
/// printing `@differentiable`, `@derivative`, and `@transpose` attributes.
enum class DifferentiationParameterKind {
  /// A differentiability parameter, printed by name.
  /// Used for `@differentiable` and `@derivative` attribute.
  Differentiability,
  /// A linearity parameter, printed by index.
  /// Used for `@transpose` attribute.
  Linearity
};

/// Returns the differentiation parameters clause string for the given function,
/// parameter indices, parsed parameters, and differentiation parameter kind.
/// Use the parameter indices if specified; otherwise, use the parsed
/// parameters.
static std::string getDifferentiationParametersClauseString(
    const AbstractFunctionDecl *function, IndexSubset *parameterIndices,
    ArrayRef<ParsedAutoDiffParameter> parsedParams,
    DifferentiationParameterKind parameterKind) {
  assert(function);
  bool isInstanceMethod = function->isInstanceMember();
  bool isStaticMethod = function->isStatic();
  std::string result;
  llvm::raw_string_ostream printer(result);

  // Use the parameter indices, if specified.
  if (parameterIndices) {
    auto parameters = parameterIndices->getBitVector();
    auto parameterCount = parameters.count();
    printer << "wrt: ";
    if (parameterCount > 1)
      printer << '(';
    // Check if differentiating wrt `self`. If so, manually print it first.
    bool isWrtSelf =
        (isInstanceMethod ||
         (isStaticMethod &&
          parameterKind == DifferentiationParameterKind::Linearity)) &&
        parameters.test(parameters.size() - 1);
    if (isWrtSelf) {
      parameters.reset(parameters.size() - 1);
      printer << "self";
      if (parameters.any())
        printer << ", ";
    }
    // Print remaining differentiation parameters.
    interleave(parameters.set_bits(), [&](unsigned index) {
      switch (parameterKind) {
      // Print differentiability parameters by name.
      case DifferentiationParameterKind::Differentiability:
        printer << function->getParameters()->get(index)->getName().str();
        break;
      // Print linearity parameters by index.
      case DifferentiationParameterKind::Linearity:
        printer << index;
        break;
      }
    }, [&] { printer << ", "; });
    if (parameterCount > 1)
      printer << ')';
  }
  // Otherwise, use the parsed parameters.
  else if (!parsedParams.empty()) {
    printer << "wrt: ";
    if (parsedParams.size() > 1)
      printer << '(';
    interleave(parsedParams, [&](const ParsedAutoDiffParameter &param) {
      switch (param.getKind()) {
      case ParsedAutoDiffParameter::Kind::Named:
        printer << param.getName();
        break;
      case ParsedAutoDiffParameter::Kind::Self:
        printer << "self";
        break;
      case ParsedAutoDiffParameter::Kind::Ordered:
        auto *paramList = function->getParameters();
        assert(param.getIndex() <= paramList->size() &&
               "wrt parameter is out of range");
        auto *funcParam = paramList->get(param.getIndex());
        printer << funcParam->getNameStr();
        break;
      }
    }, [&] { printer << ", "; });
    if (parsedParams.size() > 1)
      printer << ')';
  }
  return printer.str();
}

/// Print the arguments of the given `@differentiable` attribute.
/// - If `omitWrtClause` is true, omit printing the `wrt:` differentiation
///   parameters clause.
static void printDifferentiableAttrArguments(
    const DifferentiableAttr *attr, ASTPrinter &printer,
    const PrintOptions &Options, const Decl *D, bool omitWrtClause = false) {
  // Create a temporary string for the attribute argument text.
  std::string attrArgText;
  llvm::raw_string_ostream stream(attrArgText);

  // Print comma if not leading clause.
  bool isLeadingClause = false;
  auto printCommaIfNecessary = [&] {
    if (isLeadingClause) {
      isLeadingClause = false;
      return;
    }
    stream << ", ";
  };

  // Print if the function is marked as linear.
  switch (attr->getDifferentiabilityKind()) {
  case DifferentiabilityKind::Normal:
    isLeadingClause = true;
    break;
  case DifferentiabilityKind::Forward:
    stream << "_forward";
    break;
  case DifferentiabilityKind::Reverse:
    stream << "reverse";
    break;
  case DifferentiabilityKind::Linear:
    stream << "_linear";
    break;
  case DifferentiabilityKind::NonDifferentiable:
    llvm_unreachable("Impossible case `NonDifferentiable`");
  }

  // If the declaration is not available, there is not enough context to print
  // the differentiability parameters or the 'where' clause, so just print the
  // differentiability kind if applicable (when not `Normal`).
  if (!D) {
    if (attr->getDifferentiabilityKind() != DifferentiabilityKind::Normal) {
      printer << '(' << stream.str() << ')';
    }
    return;
  }

  // Get original function.
  auto *original = dyn_cast<AbstractFunctionDecl>(D);
  // Handle stored/computed properties and subscript methods.
  if (auto *asd = dyn_cast<AbstractStorageDecl>(D))
    original = asd->getAccessor(AccessorKind::Get);
  assert(original && "Must resolve original declaration");

  // Print differentiation parameters clause, unless it is to be omitted.
  if (!omitWrtClause) {
    auto diffParamsString = getDifferentiationParametersClauseString(
        original, attr->getParameterIndices(), attr->getParsedParameters(),
        DifferentiationParameterKind::Differentiability);
    // Check whether differentiation parameter clause is empty.
    // Handles edge case where resolved parameter indices are unset and
    // parsed parameters are empty. This case should never trigger for
    // user-visible printing.
    if (!diffParamsString.empty()) {
      printCommaIfNecessary();
      stream << diffParamsString;
    }
  }
  // Print 'where' clause, if any.
  // First, filter out requirements satisfied by the original function's
  // generic signature. They should not be printed.
  ArrayRef<Requirement> derivativeRequirements;
  if (auto derivativeGenSig = attr->getDerivativeGenericSignature())
    derivativeRequirements = derivativeGenSig.getRequirements();
  auto requirementsToPrint =
    llvm::make_filter_range(derivativeRequirements, [&](Requirement req) {
        if (const auto &originalGenSig = original->getGenericSignature())
          if (originalGenSig->isRequirementSatisfied(req))
            return false;
        return true;
      });
  if (!requirementsToPrint.empty()) {
    if (!isLeadingClause)
      stream << ' ';
    stream << "where ";
    interleave(requirementsToPrint, [&](Requirement req) {
      if (const auto &originalGenSig = original->getGenericSignature())
        if (originalGenSig->isRequirementSatisfied(req))
          return;
      req.print(stream, Options);
    }, [&] {
      stream << ", ";
    });
  }

  // If the attribute argument text is empty, return. Do not print parentheses.
  if (stream.str().empty())
    return;

  // Otherwise, print the attribute argument text enclosed in parentheses.
  printer << '(' << stream.str() << ')';
}

/// Returns the `PlatformKind` referenced by \p attr if applicable, or
/// `std::nullopt` otherwise.
static std::optional<PlatformKind> referencedPlatform(const DeclAttribute *attr,
                                                      const Decl *D) {
  switch (attr->getKind()) {
  case DeclAttrKind::Available:
    if (auto semanticAttr = D->getSemanticAvailableAttr(
            static_cast<const AvailableAttr *>(attr))) {
      if (semanticAttr->isPlatformSpecific())
        return semanticAttr->getPlatform();
    }
    return std::nullopt;
  case DeclAttrKind::BackDeployed:
    return static_cast<const BackDeployedAttr *>(attr)->getPlatform();
  case DeclAttrKind::OriginallyDefinedIn:
    return static_cast<const OriginallyDefinedInAttr *>(attr)->getPlatform();
  default:
    return std::nullopt;
  }
}

/// Returns true if \p attr contains a reference to a `PlatformKind` that should
/// be considered SPI.
static bool referencesSPIPlatform(const DeclAttribute *attr, const Decl *D) {
  if (auto platform = referencedPlatform(attr, D))
    return isPlatformSPI(*platform);
  return false;
}

void DeclAttributes::print(ASTPrinter &Printer, const PrintOptions &Options,
                           const Decl *D) const {
  if (!DeclAttrs)
    return;

  SmallVector<const DeclAttribute *, 8> orderedAttributes(begin(), end());
  print(Printer, Options, orderedAttributes, D);
}

void DeclAttributes::print(ASTPrinter &Printer, const PrintOptions &Options,
                           ArrayRef<const DeclAttribute *> FlattenedAttrs,
                           const Decl *D) {
  using AttributeVector = SmallVector<const DeclAttribute *, 8>;

  // Process attributes in passes.
  SmallVector<SemanticAvailableAttr, 8> shortAvailableAttributes;
  std::optional<SemanticAvailableAttr> swiftVersionAvailableAttribute;
  std::optional<SemanticAvailableAttr> packageDescriptionVersionAvailableAttribute;
  AttributeVector backDeployedAttributes;
  AttributeVector longAttributes;
  AttributeVector attributes;
  AttributeVector modifiers;
  bool libraryLevelAPI =
      D && D->getASTContext().LangOpts.LibraryLevel == LibraryLevel::API;
  auto *SF = D ? D->getDeclContext()->getParentSourceFile() : nullptr;

  for (auto DA : llvm::reverse(FlattenedAttrs)) {
    // Don't skip implicit custom attributes. Custom attributes like global
    // actor isolation have critical semantic meaning and should never be
    // suppressed. Other custom attrs that can be suppressed, like macros,
    // are handled below.
    if (DA->getKind() != DeclAttrKind::Custom &&
        !Options.PrintImplicitAttrs && DA->isImplicit())
      continue;
    if (!Options.PrintUserInaccessibleAttrs &&
        DeclAttribute::isUserInaccessible(DA->getKind()))
      continue;
    if (Options.excludeAttr(DA))
      continue;

    // In the public interfaces of -library-level=api modules, skip attributes
    // that reference SPI platforms.
    if (Options.printPublicInterface() && libraryLevelAPI &&
        referencesSPIPlatform(DA, D))
      continue;

    // If we're supposed to suppress expanded macros, check whether this is
    // a macro.
    if (Options.SuppressExpandedMacros) {
      if (auto customAttr = dyn_cast<CustomAttr>(DA)) {
        if (customAttr->getResolvedMacro())
          continue;
      }
    }

    // If this attribute is only allowed because this is a Clang decl, don't
    // print it.
    if (D && D->hasClangNode()
        && !DeclAttribute::canAttributeAppearOnDeclKind(
                               DA->getKind(), D->getKind()))
      continue;

    // Be careful not to coalesce `@available(swift 5)` with other short
    // `available' attributes.
    if (auto *availableAttr = dyn_cast<AvailableAttr>(DA)) {
      auto semanticAttr = D->getSemanticAvailableAttr(availableAttr);
      if (!semanticAttr)
        continue;

      // In the public interfaces of -library-level=api modules, skip @available
      // attributes that refer to domains imported from @_spiOnly modules.
      if (Options.printPublicInterface() && libraryLevelAPI) {
        if (auto *domainDecl = semanticAttr->getDomain().getDecl()) {
          if (SF->getRestrictedImportKind(domainDecl->getModuleContext()) ==
              RestrictedImportKind::SPIOnly)
            continue;
        }
      }

      if (isShortAvailable(*semanticAttr)) {
        if (semanticAttr->isSwiftLanguageModeSpecific())
          swiftVersionAvailableAttribute.emplace(*semanticAttr);
        else if (semanticAttr->isPackageDescriptionVersionSpecific())
          packageDescriptionVersionAvailableAttribute.emplace(*semanticAttr);
        else
          shortAvailableAttributes.push_back(*semanticAttr);

        continue;
      }
    }

    AttributeVector &which = DA->isDeclModifier() ? modifiers :
                             isa<BackDeployedAttr>(DA) ? backDeployedAttributes :
                             DA->isLongAttribute() ? longAttributes :
                             attributes;
    which.push_back(DA);
  }

  if (swiftVersionAvailableAttribute)
    printShortFormAvailable(D, *swiftVersionAvailableAttribute, Printer, Options);
  if (packageDescriptionVersionAvailableAttribute)
    printShortFormAvailable(D, *packageDescriptionVersionAvailableAttribute, Printer, Options);
  if (!shortAvailableAttributes.empty())
    printShortFormAvailable(D, shortAvailableAttributes, Printer, Options);
  if (!backDeployedAttributes.empty())
    printShortFormBackDeployed(backDeployedAttributes, Printer, Options);

  for (auto DA : longAttributes)
    DA->print(Printer, Options, D);
  for (auto DA : attributes)
    DA->print(Printer, Options, D);
  for (auto DA : modifiers)
    DA->print(Printer, Options, D);
}

static bool attributeIsNotAtStart(const DeclAttribute *attr) {
  switch (attr->getKind()) {
  case DeclAttrKind::Rethrows:
  case DeclAttrKind::Reasync:
    return true;

  default:
    return false;
  }
}

SourceLoc DeclAttributes::getStartLoc(bool forModifiers) const {
  if (isEmpty())
    return SourceLoc();

  const DeclAttribute *lastAttr = nullptr;
  for (auto attr : *this) {
    if (attributeIsNotAtStart(attr))
      continue;

    if (attr->getRangeWithAt().Start.isValid() &&
        (!forModifiers || attr->isDeclModifier()))
      lastAttr = attr;
  }

  return lastAttr ? lastAttr->getRangeWithAt().Start : SourceLoc();
}

std::optional<const DeclAttribute *>
ParsedDeclAttrFilter::operator()(const DeclAttribute *Attr) const {
  if (Attr->isImplicit())
    return std::nullopt;

  auto declLoc = decl->getStartLoc();
  auto *mod = decl->getModuleContext();
  auto *declFile = mod->getSourceFileContainingLocation(declLoc);
  auto *attrFile = mod->getSourceFileContainingLocation(Attr->getLocation());
  if (!declFile || !attrFile)
    return std::nullopt;

  // Only attributes in the same buffer as the declaration they're attached to
  // are part of the original attribute list.
  if (declFile->getBufferID() != attrFile->getBufferID())
    return std::nullopt;

  return Attr;
}

bool SemanticAvailableAttr::isActive(ASTContext &ctx) const {
  return getDomain().isActive(ctx);
}

std::optional<SemanticAvailableAttr>
SemanticAvailableAttributes::Filter::operator()(
    const DeclAttribute *attr) const {
  auto availableAttr = dyn_cast<AvailableAttr>(attr);
  if (!availableAttr)
    return std::nullopt;

  if (availableAttr->isInvalid())
    return std::nullopt;

  auto semanticAttr = decl->getSemanticAvailableAttr(availableAttr);
  if (!semanticAttr)
    return std::nullopt;

  if (!includeInactive && !semanticAttr->isActive(decl->getASTContext()))
    return std::nullopt;

  return *semanticAttr;
}

static void printAvailableAttr(const Decl *D, const SemanticAvailableAttr &Attr,
                               ASTPrinter &Printer,
                               const PrintOptions &Options) {
  auto ParsedAttr = Attr.getParsedAttr();
  auto Domain = Attr.getDomain();

  // The parser rejects `@available(swift, unavailable)`, so when printing
  // attributes that are universally unavailable in Swift, we must print them
  // as universally unavailable instead.
  // FIXME: Reconsider this, it's a weird special case.
  if (Domain.isSwiftLanguageMode() && Attr.isUnconditionallyUnavailable())
    Printer << "*";
  else
    Printer << Domain.getNameForAttributePrinting();

  if (Attr.isUnconditionallyUnavailable())
    Printer << ", unavailable";
  else if (Attr.isUnconditionallyDeprecated())
    Printer << ", deprecated";
  else if (Attr.isNoAsync())
    Printer << ", noasync";

  if (Attr.getIntroduced())
    Printer << ", introduced: " << Attr.getIntroduced().value().getAsString();
  if (Attr.getDeprecated())
    Printer << ", deprecated: " << Attr.getDeprecated().value().getAsString();
  if (Attr.getObsoleted())
    Printer << ", obsoleted: " << Attr.getObsoleted().value().getAsString();

  if (!Attr.getRename().empty()) {
    Printer << ", renamed: \"" << Attr.getRename() << "\"";
  } else if (auto *VD = dyn_cast<ValueDecl>(D)) {
    if (auto *renamedDecl = VD->getRenamedDecl(ParsedAttr)) {
      Printer << ", renamed: \"";
      if (auto *Accessor = dyn_cast<AccessorDecl>(renamedDecl)) {
        SmallString<32> Name;
        llvm::raw_svector_ostream OS(Name);
        Accessor->printUserFacingName(OS);
        Printer << Name.str();
      } else {
        Printer << renamedDecl->getName();
      }
      Printer << "\"";
    }
  }

  // If there's no message, but this is specifically an imported
  // "unavailable in Swift" attribute, synthesize a message to look good in
  // the generated interface.
  if (!Attr.getMessage().empty()) {
    Printer << ", message: ";
    Printer.printEscapedStringLiteral(Attr.getMessage());
  } else if (Domain.isSwiftLanguageMode() &&
             Attr.isUnconditionallyUnavailable())
    Printer << ", message: \"Not available in Swift\"";
}

bool DeclAttribute::printImpl(ASTPrinter &Printer, const PrintOptions &Options,
                              const Decl *D) const {

  // Handle any attributes that are not printed at all before we make printer
  // callbacks.
  switch (getKind()) {
  case DeclAttrKind::ObjC:
    if (Options.PrintForSIL && isImplicit())
      return false;
    break;
  case DeclAttrKind::RawDocComment:
  case DeclAttrKind::ObjCBridged:
  case DeclAttrKind::SynthesizedProtocol:
  case DeclAttrKind::Rethrows:
  case DeclAttrKind::Reasync:
  case DeclAttrKind::Infix:
    return false;
  case DeclAttrKind::Override: {
    if (!Options.IsForSwiftInterface)
      break;
    // When we are printing Swift interface, we have to skip the override keyword
    // if the overridden decl is invisible from the interface. Otherwise, an error
    // will occur while building the Swift module because the overriding decl
    // doesn't override anything.
    // We couldn't skip every `override` keywords because they change the
    // ABI if the overridden decl is also publicly visible.
    // For public-override-internal case, having `override` doesn't have ABI
    // implication. Thus we can skip them.
    if (auto *VD = dyn_cast<ValueDecl>(D)) {
      if (auto *BD = VD->getOverriddenDecl()) {
        // If the overridden decl won't be printed, printing override will fail
        // the build of the interface file.
        if (!Options.shouldPrint(BD))
          return false;
        if (!BD->hasClangNode() &&
            !BD->getFormalAccessScope(VD->getDeclContext(),
                                      /*treatUsableFromInlineAsPublic*/ true)
                 .isPublic()) {
          return false;
        }
      }
    }
    break;
  }
  case DeclAttrKind::Custom: {
    auto attr = cast<CustomAttr>(this);
    if (auto type =
            D->getResolvedCustomAttrType(const_cast<CustomAttr *>(attr))) {
      // Print custom attributes only if the attribute decl is accessible.
      // FIXME: rdar://85477478 They should be rejected.
      if (auto attrDecl = type->getNominalOrBoundGenericNominal()) {
        if (attrDecl->getFormalAccess() < Options.AccessFilter) {
          return false;
        }
      }
    }

    if (!Options.IsForSwiftInterface)
      break;
    // For Swift interface, we should print result builder attributes
    // on parameter decls and on protocol requirements.
    // Printing the attribute elsewhere isn't ABI relevant.
    if (auto *VD = dyn_cast<ValueDecl>(D)) {
      if (VD->getAttachedResultBuilder() == this) {
        if (!isa<ParamDecl>(D) &&
            !((isa<VarDecl>(D) || isa<FuncDecl>(D)) &&
               isa<ProtocolDecl>(D->getDeclContext())))
          return false;
      }
    }
    break;
  }
  case DeclAttrKind::OriginallyDefinedIn: {
    auto Attr = cast<OriginallyDefinedInAttr>(this);
    auto Name = D->getDeclContext()->getParentModule()->getName().str();
    if (Options.IsForSwiftInterface && Attr->getManglingModuleName() == Name)
      return false;
    break;
  }
  default:
    break;
  }

  // Handle any decl-modifiers.
  // FIXME: Ideally we would handle decl modifiers as a special kind of
  // attribute, but for now it's simpler to treat them as a keyword in the
  // printer.
  switch (getKind()) {
    // Handle all of the SIMPLE_DECL_ATTRs.
#define SIMPLE_DECL_ATTR(X, CLASS, ...) case DeclAttrKind::CLASS:
#include "swift/AST/DeclAttr.def"
  case DeclAttrKind::Inline:
  case DeclAttrKind::AccessControl:
  case DeclAttrKind::ReferenceOwnership:
  case DeclAttrKind::Effects:
  case DeclAttrKind::Export:
  case DeclAttrKind::Optimize:
  case DeclAttrKind::Exclusivity:
  case DeclAttrKind::NonSendable:
  case DeclAttrKind::ObjCImplementation:
    if (getKind() == DeclAttrKind::Effects &&
        cast<EffectsAttr>(this)->getKind() == EffectsKind::Custom) {
      Printer.printAttrName("@_effects");
      Printer << "(" << cast<EffectsAttr>(this)->getCustomString() << ")";
    } else if (DeclAttribute::isDeclModifier(getKind())) {
      Printer.printKeyword(getAttrName(), Options);
    } else if (Options.IsForSwiftInterface &&
               getKind() == DeclAttrKind::ResultBuilder) {
      // Use @_functionBuilder in Swift interfaces to maintain backward
      // compatibility.
      Printer.printSimpleAttr("_functionBuilder", /*needAt=*/true);
    } else if (getKind() == DeclAttrKind::MainType && Options.PrintForSIL) {
      // Don't print into SIL. Necessary bits have already been generated.
      return false;
    } else {
      auto attrName = getAttrName();
      if (getKind() == DeclAttrKind::Inline &&
          cast<InlineAttr>(this)->getKind() == InlineKind::Always &&
          Options.SuppressInlineAlways) {
        attrName = "inline(__always)";
        Printer.printSimpleAttr(attrName, /*needAt=*/true);
        Printer << ' ';
        // Add @inlinable
        Printer.printSimpleAttr("inlinable", /*needAt=*/true);
      } else {
        Printer.printSimpleAttr(attrName, /*needAt=*/true);
      }
    }
    return true;

  case DeclAttrKind::SetterAccess:
    Printer.printKeyword(getAttrName(), Options, "(set)");
    return true;

  case DeclAttrKind::SPIAccessControl: {
    if (Options.printPublicInterface()) return false;

    auto spiAttr = static_cast<const SPIAccessControlAttr*>(this);
    interleave(spiAttr->getSPIGroups(),
               [&](Identifier spiName) {
                 Printer.printAttrName(getAttrName(), true);
                 Printer << "(" << spiName << ")";
               },
               [&] { Printer << " "; });
    return true;
  }

  default:
    break;
  }

  Printer.callPrintStructurePre(PrintStructureKind::BuiltinAttribute);
  SWIFT_DEFER {
    Printer.printStructurePost(PrintStructureKind::BuiltinAttribute);
  };

  switch (getKind()) {
  case DeclAttrKind::Semantics:
    Printer.printAttrName("@_semantics");
    Printer << "(\"" << cast<SemanticsAttr>(this)->Value << "\")";
    break;

  case DeclAttrKind::Alignment:
    Printer.printAttrName("@_alignment");
    Printer << "(" << cast<AlignmentAttr>(this)->getValue() << ")";
    break;

  case DeclAttrKind::AllowFeatureSuppression: {
    auto Attr = cast<AllowFeatureSuppressionAttr>(this);
    Printer.printAttrName(Attr->getInverted() ? "@_disallowFeatureSuppression"
                                              : "@_allowFeatureSuppression");
    Printer << "(";
    interleave(
        Attr->getSuppressedFeatures(),
        [&](Identifier ident) { Printer << ident; }, [&] { Printer << ", "; });
    Printer << ")";
    break;
  }

  case DeclAttrKind::SILGenName:
    Printer.printAttrName("@_silgen_name");
    Printer << "(\"" << cast<SILGenNameAttr>(this)->Name << "\")";
    break;

  case DeclAttrKind::OriginallyDefinedIn: {
    Printer.printAttrName("@_originallyDefinedIn");
    Printer << "(module: ";
    auto Attr = cast<OriginallyDefinedInAttr>(this);
    auto ManglingModuleName = Attr->getManglingModuleName();
    auto LinkerModuleName = Attr->getLinkerModuleName();
    Printer << "\"" << ManglingModuleName;
    ASSERT(!ManglingModuleName.empty());
    ASSERT(!LinkerModuleName.empty());
    if (LinkerModuleName != ManglingModuleName)
      Printer << ";" << LinkerModuleName;
    Printer << "\", " << platformString(Attr->getPlatform()) << " "
            << Attr->getMovedVersion().getAsString();
    Printer << ")";
    break;
  }

  case DeclAttrKind::Available: {
    auto Attr = D->getSemanticAvailableAttr(cast<AvailableAttr>(this));
    if (!Attr)
      return false;

    if (Options.printPublicInterface() && Attr->isSPI()) {
      assert(Attr->isPlatformSpecific());
      assert(Attr->getIntroduced().has_value());
      Printer.printAttrName("@available");
      Printer << "(";
      Printer << Attr->getDomain().getNameForAttributePrinting();
      Printer << ", unavailable)";
      break;
    }
    if (Attr->isEmbeddedSpecific()) {
      std::string atUnavailableInEmbedded =
          (llvm::Twine("@") + UNAVAILABLE_IN_EMBEDDED_ATTRNAME).str();
      Printer.printAttrName(atUnavailableInEmbedded);
      break;
    }

    if (Attr->isSPI()) {
      std::string atSPI = (llvm::Twine("@") + SPI_AVAILABLE_ATTRNAME).str();
      Printer.printAttrName(atSPI);
    } else {
      Printer.printAttrName("@available");
    }
    Printer << "(";
    printAvailableAttr(D, *Attr, Printer, Options);
    Printer << ")";
    break;
  }

  case DeclAttrKind::CDecl: {
    auto Attr = cast<CDeclAttr>(this);
    if (Attr->Underscored)
      Printer << "@_cdecl(\"" << cast<CDeclAttr>(this)->Name << "\")";
    else {
      Printer << "@c";
      if (!Attr->Name.empty())
        Printer << "(" << cast<CDeclAttr>(this)->Name << ")";
    }
    break;
  }

  case DeclAttrKind::Expose: {
    Printer.printAttrName("@_expose");
    auto Attr = cast<ExposeAttr>(this);
    switch (Attr->getExposureKind()) {
    case ExposureKind::Wasm:
      Printer << "(wasm";
      break;
    case ExposureKind::Cxx:
      Printer << "(Cxx";
      break;
    case ExposureKind::NotCxx:
      Printer << "(!Cxx";
      break;
    }
    if (!cast<ExposeAttr>(this)->Name.empty())
      Printer << ", \"" << cast<ExposeAttr>(this)->Name << "\"";
    Printer << ")";
    break;
  }

  case DeclAttrKind::Extern: {
    auto *Attr = cast<ExternAttr>(this);
    Printer.printAttrName("@_extern");
    Printer << "(";
    switch (Attr->getExternKind()) {
    case ExternKind::C:
      Printer << "c";
      // Symbol name can be omitted for C.
      if (auto cName = Attr->Name)
        Printer << ", \"" << *cName << "\"";
      break;
    case ExternKind::Wasm:
      Printer << "wasm";
      // @_extern(wasm) always has names.
      Printer << ", module: \"" << *Attr->ModuleName << "\"";
      Printer << ", name: \"" << *Attr->Name << "\"";
      break;
    }
    Printer << ")";
    break;
  }

  case DeclAttrKind::Section:
    Printer.printAttrName("@section");
    Printer << "(\"" << cast<SectionAttr>(this)->Name << "\")";
    break;

  case DeclAttrKind::ObjC: {
    Printer.printAttrName("@objc");
    llvm::SmallString<32> scratch;
    if (auto Name = cast<ObjCAttr>(this)->getName()) {
      if (!cast<ObjCAttr>(this)->isNameImplicit())
        Printer << "(" << Name->getString(scratch) << ")";
    }
    break;
  }

  case DeclAttrKind::PrivateImport: {
    Printer.printAttrName("@_private(sourceFile: \"");
    Printer << cast<PrivateImportAttr>(this)->getSourceFile() << "\")";
    break;
  }

  case DeclAttrKind::SwiftNativeObjCRuntimeBase: {
    auto *attr = cast<SwiftNativeObjCRuntimeBaseAttr>(this);
    Printer.printAttrName("@_swift_native_objc_runtime_base");
    Printer << "(" << attr->BaseClassName.str() << ")";
    break;
  }

  case DeclAttrKind::Specialized:
  case DeclAttrKind::Specialize: {
    auto *attr = cast<AbstractSpecializeAttr>(this);
    // Don't print the _specialize attribute if it is marked spi and we are
    // asked to skip SPI.
    if (Options.printPublicInterface() && !attr->getSPIGroups().empty())
      return false;

    Printer << "@" << getAttrName() << "(";

    // The non-underscored @specialize attribute currently does not support
    // parameters so lets not print them.
    if (!attr->isPublic()) {
      auto exported = attr->isExported() ? "true" : "false";
      auto kind = attr->isPartialSpecialization() ? "partial" : "full";
      auto target = attr->getTargetFunctionName();
      Printer << "exported: "<<  exported << ", ";
      for (auto id : attr->getSPIGroups()) {
        Printer << "spi: " << id << ", ";
      }
      Printer << "kind: " << kind << ", ";
      if (target)
        Printer << "target: " << target << ", ";
      SmallVector<SemanticAvailableAttr, 8> semanticAvailAttrs;
      for (auto availAttr : attr->getAvailableAttrs()) {
        if (auto semanticAttr = D->getSemanticAvailableAttr(availAttr))
          semanticAvailAttrs.push_back(*semanticAttr);
      }

      if (!semanticAvailAttrs.empty()) {
        Printer << "availability: ";
        if (semanticAvailAttrs.size() == 1) {
          printAvailableAttr(D, semanticAvailAttrs[0], Printer, Options);
          Printer << "; ";
        } else {
          printShortFormAvailable(D, semanticAvailAttrs, Printer, Options,
                                  true /*forAtSpecialize*/);
          Printer << "; ";
        }
      }
    }
    SmallVector<Requirement, 4> requirementsScratch;
    auto *FnDecl = dyn_cast_or_null<AbstractFunctionDecl>(D);
    auto specializedSig = attr->getSpecializedSignature(FnDecl);
    auto requirements = specializedSig.getRequirements();
    if (FnDecl && FnDecl->getGenericSignature()) {
      auto genericSig = FnDecl->getGenericSignature();

      if (auto sig = specializedSig) {
        requirementsScratch = sig.requirementsNotSatisfiedBy(genericSig);
        requirements = requirementsScratch;
      }
    }

    if (!requirements.empty()) {
      Printer << "where ";
    }

    interleave(requirements,
                [&](Requirement req) {
                  bool typeErased = false;
                  if (req.getKind() == RequirementKind::Layout &&
                      !attr->getTypeErasedParams().empty()) {
                    const auto &erasedParams = attr->getTypeErasedParams();
                    typeErased = std::any_of(erasedParams.begin(),
                                           erasedParams.end(),
                                           [&](Type t) { return t->isEqual(req.getFirstType()); });
                    if (typeErased)
                      Printer << "@_noMetadata ";
                  }

                  PrintOptions::OverrideScope scope(Options);
                  OVERRIDE_PRINT_OPTION(scope, PrintInternalLayoutName, typeErased);
                  req.print(Printer, Options);
                },
                [&] { Printer << ", "; });

    Printer << ")";
    break;
  }

  case DeclAttrKind::Implements: {
    Printer.printAttrName("@_implements");
    Printer << "(";
    auto *attr = cast<ImplementsAttr>(this);
    if (auto *proto = attr->getProtocol(D->getDeclContext()))
      proto->getDeclaredInterfaceType()->print(Printer, Options);
    else
      attr->getProtocolTypeRepr()->print(Printer, Options);
    Printer << ", " << attr->getMemberName() << ")";
    break;
  }

  case DeclAttrKind::ObjCRuntimeName: {
    Printer.printAttrName("@_objcRuntimeName");
    Printer << "(";
    auto *attr = cast<ObjCRuntimeNameAttr>(this);
    Printer << attr->Name;
    Printer << ")";
    break;
  }

  case DeclAttrKind::ClangImporterSynthesizedType: {
    Printer.printAttrName("@_clangImporterSynthesizedType");
    auto *attr = cast<ClangImporterSynthesizedTypeAttr>(this);
    Printer << "(originalTypeName: \"" << attr->originalTypeName
            << "\", manglingForKind: \"" << attr->getManglingName() << "\")";
    break;
  }

  case DeclAttrKind::DynamicReplacement: {
    Printer.printAttrName("@_dynamicReplacement");
    Printer << "(for: \"";
    auto *attr = cast<DynamicReplacementAttr>(this);
    Printer << attr->getReplacedFunctionName() << "\")";
    break;
  }

  case DeclAttrKind::TypeEraser: {
    Printer.printAttrName("@_typeEraser");
    Printer << "(";
    Printer.callPrintNamePre(PrintNameContext::Attribute);
    auto *attr = cast<TypeEraserAttr>(this);
    if (auto *repr = attr->getParsedTypeEraserTypeRepr())
      repr->print(Printer, Options);
    else if (auto proto = dyn_cast<ProtocolDecl>(D))
      attr->getResolvedType(proto)->print(Printer, Options);
    Printer.printNamePost(PrintNameContext::Attribute);
    Printer << ")";
    break;
  }

  case DeclAttrKind::Custom: {
    auto *attr = cast<CustomAttr>(this);
    attr->printCustomAttr(Printer, Options);
    break;
  }

  case DeclAttrKind::ProjectedValueProperty:
    Printer.printAttrName("@_projectedValueProperty");
    Printer << "(";
    Printer << cast<ProjectedValuePropertyAttr>(this)->ProjectionPropertyName;
    Printer << ")";
    break;

  case DeclAttrKind::Differentiable: {
    Printer.printAttrName("@differentiable");
    auto *attr = cast<DifferentiableAttr>(this);
    printDifferentiableAttrArguments(attr, Printer, Options, D);
    break;
  }

  case DeclAttrKind::Derivative: {
    Printer.printAttrName("@derivative");
    Printer << "(of: ";
    auto *attr = cast<DerivativeAttr>(this);
    if (auto *baseType = attr->getBaseTypeRepr())
      baseType->print(Printer, Options);
    attr->getOriginalFunctionName().print(Printer);
    auto *derivative = cast<AbstractFunctionDecl>(D);
    auto diffParamsString = getDifferentiationParametersClauseString(
        derivative, attr->getParameterIndices(), attr->getParsedParameters(),
        DifferentiationParameterKind::Differentiability);
    if (!diffParamsString.empty())
      Printer << ", " << diffParamsString;
    Printer << ')';
    break;
  }

  case DeclAttrKind::Transpose: {
    Printer.printAttrName("@transpose");
    Printer << "(of: ";
    auto *attr = cast<TransposeAttr>(this);
    if (auto *baseType = attr->getBaseTypeRepr())
      baseType->print(Printer, Options);
    attr->getOriginalFunctionName().print(Printer);
    auto *transpose = cast<AbstractFunctionDecl>(D);
    auto transParamsString = getDifferentiationParametersClauseString(
        transpose, attr->getParameterIndices(), attr->getParsedParameters(),
        DifferentiationParameterKind::Linearity);
    if (!transParamsString.empty())
      Printer << ", " << transParamsString;
    Printer << ')';
    break;
  }

  case DeclAttrKind::UnavailableFromAsync: {
    Printer.printAttrName("@_unavailableFromAsync");
    const UnavailableFromAsyncAttr *attr = cast<UnavailableFromAsyncAttr>(this);
    if (attr->hasMessage()) {
      Printer << "(message: \"";
      Printer << attr->Message;
      Printer << "\")";
    }
    break;
  }

  case DeclAttrKind::BackDeployed: {
    Printer.printAttrName("@backDeployed");
    Printer << "(before: ";
    auto Attr = cast<BackDeployedAttr>(this);
    Printer << platformString(Attr->getPlatform()) << " "
            << Attr->getVersion().getAsString();
    Printer << ")";
    break;
  }

  case DeclAttrKind::Nonisolated: {
    Printer.printAttrName("nonisolated");
    switch (cast<NonisolatedAttr>(this)->getModifier()) {
    case NonIsolatedModifier::None:
      break;
    case NonIsolatedModifier::Unsafe:
      Printer << "(unsafe)";
      break;
    case NonIsolatedModifier::NonSending:
      Printer << "(nonsending)";
      break;
    }
    break;
  }

  case DeclAttrKind::InheritActorContext: {
    Printer.printAttrName("@_inheritActorContext");
    switch (cast<InheritActorContextAttr>(this)->getModifier()) {
    case InheritActorContextModifier::None:
      break;
    case InheritActorContextModifier::Always:
      Printer << "(always)";
      break;
    }
    break;
  }

  case DeclAttrKind::MacroRole: {
    auto Attr = cast<MacroRoleAttr>(this);

    switch (Attr->getMacroSyntax()) {
    case MacroSyntax::Freestanding:
      Printer.printAttrName("@freestanding");
      break;

    case MacroSyntax::Attached:
      Printer.printAttrName("@attached");
      break;
    }
    Printer << "(";
    Printer << getMacroRoleString(Attr->getMacroRole());

    // Print conformances, if present.
    auto conformances = evaluateOrDefault(
        D->getASTContext().evaluator,
        ResolveMacroConformances{Attr, D},
        {});
    if (!conformances.empty()) {
      Printer << ", conformances: ";
      interleave(conformances,
                 [&](Type type) {
                   type.print(Printer, Options);
                 },
                 [&] {
                   Printer << ", ";
                 });
    }

    if (!Attr->getNames().empty()) {
      Printer << ", names: ";
      interleave(
          Attr->getNames(),
          [&](MacroIntroducedDeclName name) {
            Printer << getMacroIntroducedDeclNameString(name.getKind());
            if (macroIntroducedNameRequiresArgument(name.getKind())) {
              SmallString<32> buffer;
              StringRef nameText = name.getName().getString(buffer);
              bool shouldEscape =
                  !name.getName().isSpecial() &&
                  (escapeIdentifierInContext(name.getName().getBaseIdentifier(),
                                             PrintNameContext::Normal) ||
                   nameText == "$");
              Printer << "(";
              if (shouldEscape)
                Printer << "`";
              Printer << nameText;
              if (shouldEscape)
                Printer << "`";
              Printer << ")";
            }
          },
          [&] {
            Printer << ", ";
          }
      );
    }
    Printer << ")";
    break;
  }

  case DeclAttrKind::Documentation: {
    auto *attr = cast<DocumentationAttr>(this);

    Printer.printAttrName("@_documentation");
    Printer << "(";

    bool needs_comma = !attr->Metadata.empty() && attr->Visibility;

    if (attr->Visibility) {
      Printer << "visibility: ";
      Printer << getAccessLevelSpelling(*attr->Visibility);
    }

    if (needs_comma) {
      Printer << ", ";
    }

    if (!attr->Metadata.empty()) {
      Printer << "metadata: ";
      Printer << attr->Metadata;
    }

    Printer << ")";
    break;
  }

  case DeclAttrKind::RawLayout: {
    auto *attr = cast<RawLayoutAttr>(this);
    Printer.printAttrName("@_rawLayout");
    Printer << "(";
    
    if (auto sizeAndAlign = attr->getSizeAndAlignment()) {
      Printer << "size: " << sizeAndAlign->first
              << ", alignment: " << sizeAndAlign->second;
    } else if (auto type = attr->getScalarLikeType()) {
      Printer << "like: ";
      type->print(Printer, Options);
    } else if (auto array = attr->getArrayLikeTypeAndCount()) {
      Printer << "likeArrayOf: ";
      array->first->print(Printer, Options);
      Printer << ", count: ";
      array->second->print(Printer, Options);
    } else {
      llvm_unreachable("unhandled @_rawLayout form");
    }
    Printer << ")";
    break;
  }

  case DeclAttrKind::StorageRestrictions: {
    auto *attr = cast<StorageRestrictionsAttr>(this);
    Printer.printAttrName("@storageRestrictions");
    Printer << "(";

    auto initializes = attr->getInitializesNames();
    auto accesses = attr->getAccessesNames();

    bool needsComma = !initializes.empty() && !accesses.empty();

    if (!initializes.empty()) {
      Printer << "initializes: ";
      interleave(initializes, Printer, ", ");
    }

    if (needsComma)
      Printer << ", ";

    if (!accesses.empty()) {
      Printer << "accesses: ";
      interleave(accesses, Printer, ", ");
    }
    Printer << ")";
    break;
  }

  case DeclAttrKind::Lifetime: {
    auto *attr = cast<LifetimeAttr>(this);
    if (!attr->isUnderscored() || Options.SuppressLifetimes) {
      Printer << "@lifetime" << attr->getLifetimeEntry()->getString();
    } else {
      Printer << "@_lifetime" << attr->getLifetimeEntry()->getString();
    }
    break;
  }

  case DeclAttrKind::ABI: {
    auto *attr = cast<ABIAttr>(this);
    Printer << "@abi(";
    Decl *abiDecl = attr->abiDecl;
    if (abiDecl && Options.ExplodePatternBindingDecls
          && isa<PatternBindingDecl>(abiDecl) && D && isa<VarDecl>(D))
      abiDecl = cast<PatternBindingDecl>(abiDecl)
                    ->getVarAtSimilarStructuralPosition(
                                      const_cast<VarDecl *>(cast<VarDecl>(D)));
    if (abiDecl) {
      PrintOptions::OverrideScope scope(Options);

      // Don't print any attributes marked with `ForbiddenInABIAttr`.
      // (Reminder: There is manual logic in `PrintAST::printAttributes()`
      // to handle non-ABI attributes when `PrintImplicitAttrs` is set.)
      for (auto rawAttrKind : range(0, unsigned(DeclAttrKind::Last_DeclAttr))) {
        DeclAttrKind attrKind{rawAttrKind}; 
        if (!(DeclAttribute::getBehaviors(attrKind)
                & DeclAttribute::ForbiddenInABIAttr))
          continue;

        if (attrKind == DeclAttrKind::AccessControl)
          OVERRIDE_PRINT_OPTION(scope, PrintAccess, false);
        else
          scope.addExcludedAttr(attrKind);
      }

      abiDecl->print(Printer, Options);
    }
    Printer << ")";

    break;
  }

  case DeclAttrKind::Nonexhaustive: {
    auto *attr = cast<NonexhaustiveAttr>(this);
    Printer << "@nonexhaustive";
    switch (attr->getMode()) {
    case NonexhaustiveMode::Error:
      break;
    case NonexhaustiveMode::Warning:
      Printer << "(warn)";
      break;
    }
    break;
  }

#define SIMPLE_DECL_ATTR(X, CLASS, ...) case DeclAttrKind::CLASS:
#include "swift/AST/DeclAttr.def"
    llvm_unreachable("handled above");

  default:
    assert(DeclAttribute::isDeclModifier(getKind()) &&
           "handled above");
  }

  return true;
}

void DeclAttribute::print(ASTPrinter &Printer, const PrintOptions &Options,
                          const Decl *D) const {

  if (!printImpl(Printer, Options, D))
    return; // Nothing printed.

  if (isLongAttribute() && Options.PrintLongAttrsOnSeparateLines)
    Printer.printNewline();
  else
    Printer << " ";
}

void DeclAttribute::print(llvm::raw_ostream &OS, const Decl *D) const {
  StreamPrinter P(OS);
  print(P, PrintOptions(), D);
}

uint64_t DeclAttribute::getRequirements(DeclAttrKind DK) {
  switch (DK) {
#define DECL_ATTR(_, CLASS, REQUIREMENTS, BEHAVIORS, ...)                      \
  case DeclAttrKind::CLASS:                                                    \
    return REQUIREMENTS;
#include "swift/AST/DeclAttr.def"
  }
  llvm_unreachable("bad DeclAttrKind");
}

uint64_t DeclAttribute::getBehaviors(DeclAttrKind DK) {
  switch (DK) {
#define DECL_ATTR(_, CLASS, REQUIREMENTS, BEHAVIORS, ...)                      \
  case DeclAttrKind::CLASS:                                                    \
    return BEHAVIORS;
#include "swift/AST/DeclAttr.def"
  }
  return 0;
}

std::optional<Feature> DeclAttribute::getRequiredFeature(DeclAttrKind DK) {
  switch (DK) {
#define DECL_ATTR_FEATURE_REQUIREMENT(CLASS, FEATURE_NAME)                     \
  case DeclAttrKind::CLASS:                                                    \
    return Feature::FEATURE_NAME;
#include "swift/AST/DeclAttr.def"
  default:
    return std::nullopt;
  }
  llvm_unreachable("bad DeclAttrKind");
}

StringRef DeclAttribute::getAttrName() const {
  switch (getKind()) {
#define SIMPLE_DECL_ATTR(NAME, CLASS, ...)                                     \
  case DeclAttrKind::CLASS:                                                    \
    return #NAME;
#include "swift/AST/DeclAttr.def"
  case DeclAttrKind::ABI:
    return "abi";
  case DeclAttrKind::SILGenName:
    return "_silgen_name";
  case DeclAttrKind::Alignment:
    return "_alignment";
  case DeclAttrKind::CDecl:
    if (cast<CDeclAttr>(this)->Underscored)
      return "_cdecl";
    return "c";
  case DeclAttrKind::SwiftNativeObjCRuntimeBase:
    return "_swift_native_objc_runtime_base";
  case DeclAttrKind::Semantics:
    return "_semantics";
  case DeclAttrKind::Available:
    return "available";
  case DeclAttrKind::ObjC:
  case DeclAttrKind::ObjCRuntimeName:
    return "objc";
  case DeclAttrKind::ObjCImplementation:
    if (cast<ObjCImplementationAttr>(this)->isEarlyAdopter())
      return "_objcImplementation";
    return "implementation";
  case DeclAttrKind::DynamicReplacement:
    return "_dynamicReplacement";
  case DeclAttrKind::TypeEraser:
    return "_typeEraser";
  case DeclAttrKind::PrivateImport:
    return "_private";
  case DeclAttrKind::RestatedObjCConformance:
    return "_restatedObjCConformance";
  case DeclAttrKind::Inline: {
    switch (cast<InlineAttr>(this)->getKind()) {
    case InlineKind::Never:
      return "inline(never)";
    case InlineKind::AlwaysUnderscored:
      return "inline(__always)";
    case InlineKind::Always:
      return "inline(always)";
    }
    llvm_unreachable("Invalid inline kind");
  }
  case DeclAttrKind::NonSendable: {
    switch (cast<NonSendableAttr>(this)->Specificity) {
    case NonSendableKind::Specific:
      return "_nonSendable";
    case NonSendableKind::Assumed:
      return "_nonSendable(_assumed)";
    }
    llvm_unreachable("Invalid nonSendable kind");
  }
  case DeclAttrKind::Optimize: {
    switch (cast<OptimizeAttr>(this)->getMode()) {
    case OptimizationMode::NoOptimization:
      return "_optimize(none)";
    case OptimizationMode::ForSpeed:
      return "_optimize(speed)";
    case OptimizationMode::ForSize:
      return "_optimize(size)";
    default:
      llvm_unreachable("Invalid optimization kind");
    }
  }
  case DeclAttrKind::Exclusivity: {
    switch (cast<ExclusivityAttr>(this)->getMode()) {
    case ExclusivityAttr::Checked:
      return "exclusivity(checked)";
    case ExclusivityAttr::Unchecked:
      return "exclusivity(unchecked)";
    }
    llvm_unreachable("Invalid optimization kind");
  }
  case DeclAttrKind::Effects:
    switch (cast<EffectsAttr>(this)->getKind()) {
      case EffectsKind::ReadNone:
        return "_effects(readnone)";
      case EffectsKind::ReadOnly:
        return "_effects(readonly)";
      case EffectsKind::ReleaseNone:
        return "_effects(releasenone)";
      case EffectsKind::ReadWrite:
        return "_effects(readwrite)";
      case EffectsKind::Unspecified:
        return "_effects(unspecified)";
      case EffectsKind::Custom:
        return "_effects";
    }
  case DeclAttrKind::Export: {
    switch (cast<ExportAttr>(this)->exportKind) {
    case ExportKind::Interface:
      return "export(interface)";
    case ExportKind::Implementation:
      return "export(implementation)";
    }
    llvm_unreachable("Invalid export kind");
  }
  case DeclAttrKind::AccessControl:
  case DeclAttrKind::SetterAccess: {
    AccessLevel access = cast<AbstractAccessControlAttr>(this)->getAccess();
    return getAccessLevelSpelling(access);
  }
  case DeclAttrKind::SPIAccessControl:
    return "_spi";
  case DeclAttrKind::ReferenceOwnership:
    return keywordOf(cast<ReferenceOwnershipAttr>(this)->get());
  case DeclAttrKind::RawDocComment:
    return "<<raw doc comment>>";
  case DeclAttrKind::ObjCBridged:
    return "<<ObjC bridged>>";
  case DeclAttrKind::SynthesizedProtocol:
    return "<<synthesized protocol>>";
  case DeclAttrKind::Specialized:
    return "specialized";
  case DeclAttrKind::Specialize:
    return "_specialize";
  case DeclAttrKind::StorageRestrictions:
    return "storageRestrictions";
  case DeclAttrKind::Implements:
    return "_implements";
  case DeclAttrKind::ClangImporterSynthesizedType:
    return "_clangImporterSynthesizedType";
  case DeclAttrKind::Custom:
    return "<<custom>>";
  case DeclAttrKind::ProjectedValueProperty:
    return "_projectedValueProperty";
  case DeclAttrKind::OriginallyDefinedIn:
    return "_originallyDefinedIn";
  case DeclAttrKind::Differentiable:
    return "differentiable";
  case DeclAttrKind::Derivative:
    return "derivative";
  case DeclAttrKind::Transpose:
    return "transpose";
  case DeclAttrKind::UnavailableFromAsync:
    return "_unavailableFromAsync";
  case DeclAttrKind::BackDeployed:
    return "backDeployed";
  case DeclAttrKind::Expose:
    return "_expose";
  case DeclAttrKind::Section:
    return "section";
  case DeclAttrKind::Documentation:
    return "_documentation";
  case DeclAttrKind::Nonisolated:
    switch (cast<NonisolatedAttr>(this)->getModifier()) {
    case NonIsolatedModifier::None:
      return "nonisolated";
    case NonIsolatedModifier::Unsafe:
      return "nonisolated(unsafe)";
    case NonIsolatedModifier::NonSending:
      return "nonisolated(nonsending)";
    }
  case DeclAttrKind::InheritActorContext:
    switch (cast<InheritActorContextAttr>(this)->getModifier()) {
    case InheritActorContextModifier::None:
      return "_inheritActorContext";
    case InheritActorContextModifier::Always:
      return "_inheritActorContext(always)";
    }
  case DeclAttrKind::MacroRole:
    switch (cast<MacroRoleAttr>(this)->getMacroSyntax()) {
    case MacroSyntax::Freestanding:
      return "freestanding";

    case MacroSyntax::Attached:
      return "attached";
    }
  case DeclAttrKind::RawLayout:
    return "_rawLayout";
  case DeclAttrKind::Extern:
    return "_extern";
  case DeclAttrKind::AllowFeatureSuppression:
    if (cast<AllowFeatureSuppressionAttr>(this)->getInverted()) {
      return "_disallowFeatureSuppression";
    } else {
      return "_allowFeatureSuppression";
    }
  case DeclAttrKind::Lifetime:
    return cast<LifetimeAttr>(this)->isUnderscored() ? "_lifetime" : "lifetime";
  case DeclAttrKind::Nonexhaustive:
    return "nonexhaustive";
  }
  llvm_unreachable("bad DeclAttrKind");
}

ObjCAttr::ObjCAttr(SourceLoc atLoc, SourceRange baseRange,
                   std::optional<ObjCSelector> name, SourceRange parenRange,
                   ArrayRef<SourceLoc> nameLocs)
    : DeclAttribute(DeclAttrKind::ObjC, atLoc, baseRange, /*Implicit=*/false),
      NameData(nullptr) {
  if (name) {
    // Store the name.
    assert(name->getNumSelectorPieces() == nameLocs.size());
    NameData = name->getOpaqueValue();

    // Store location information.
    Bits.ObjCAttr.HasTrailingLocationInfo = true;
    getTrailingLocations()[0] = parenRange.Start;
    getTrailingLocations()[1] = parenRange.End;
    std::memcpy(getTrailingLocations().slice(2).data(), nameLocs.data(),
                nameLocs.size() * sizeof(SourceLoc));
  } else {
    Bits.ObjCAttr.HasTrailingLocationInfo = false;
  }

  Bits.ObjCAttr.ImplicitName = false;
}

ObjCAttr *ObjCAttr::create(ASTContext &Ctx, std::optional<ObjCSelector> name,
                           bool isNameImplicit) {
  return new (Ctx) ObjCAttr(name, isNameImplicit);
}

ObjCAttr *ObjCAttr::createUnnamed(ASTContext &Ctx, SourceLoc AtLoc,
                                  SourceLoc ObjCLoc) {
  return new (Ctx)
      ObjCAttr(AtLoc, SourceRange(ObjCLoc), std::nullopt, SourceRange(), {});
}

ObjCAttr *ObjCAttr::createUnnamedImplicit(ASTContext &Ctx) {
  return new (Ctx) ObjCAttr(std::nullopt, false);
}

ObjCAttr *ObjCAttr::createNullary(ASTContext &Ctx, SourceLoc AtLoc, 
                                  SourceLoc ObjCLoc, SourceLoc LParenLoc, 
                                  SourceLoc NameLoc, Identifier Name,
                                  SourceLoc RParenLoc) {
  void *mem = Ctx.Allocate(totalSizeToAlloc<SourceLoc>(3), alignof(ObjCAttr));
  return new (mem) ObjCAttr(AtLoc, SourceRange(ObjCLoc, RParenLoc),
                            ObjCSelector(Ctx, 0, Name),
                            SourceRange(LParenLoc, RParenLoc),
                            NameLoc);
}

ObjCAttr *ObjCAttr::createNullary(ASTContext &Ctx, Identifier Name,
                                  bool isNameImplicit) {
  return new (Ctx) ObjCAttr(ObjCSelector(Ctx, 0, Name), isNameImplicit);
}

ObjCAttr *ObjCAttr::createSelector(ASTContext &Ctx, SourceLoc AtLoc, 
                                   SourceLoc ObjCLoc, SourceLoc LParenLoc, 
                                   ArrayRef<SourceLoc> NameLocs,
                                   ArrayRef<Identifier> Names,
                                   SourceLoc RParenLoc) {
  assert(NameLocs.size() == Names.size());
  void *mem = Ctx.Allocate(totalSizeToAlloc<SourceLoc>(NameLocs.size() + 2),
                           alignof(ObjCAttr));
  return new (mem) ObjCAttr(AtLoc, SourceRange(ObjCLoc, RParenLoc),
                            ObjCSelector(Ctx, Names.size(), Names),
                            SourceRange(LParenLoc, RParenLoc),
                            NameLocs);
}

ObjCAttr *ObjCAttr::createSelector(ASTContext &Ctx, 
                                   ArrayRef<Identifier> Names,
                                   bool isNameImplicit) {
  return new (Ctx) ObjCAttr(ObjCSelector(Ctx, Names.size(), Names), 
                            isNameImplicit);
}

ArrayRef<SourceLoc> ObjCAttr::getNameLocs() const {
  if (!hasTrailingLocationInfo())
    return { };

  return getTrailingLocations().slice(2);
}

SourceLoc ObjCAttr::getLParenLoc() const {
  if (!hasTrailingLocationInfo())
    return SourceLoc();

  return getTrailingLocations()[0];
}

SourceLoc ObjCAttr::getRParenLoc() const {
  if (!hasTrailingLocationInfo())
    return SourceLoc();

  return getTrailingLocations()[1];
}

ObjCAttr *ObjCAttr::clone(ASTContext &context) const {
  auto attr = new (context) ObjCAttr(getName(), isNameImplicit());
  attr->setAddedByAccessNote(getAddedByAccessNote());
  return attr;
}

PrivateImportAttr::PrivateImportAttr(SourceLoc atLoc, SourceRange baseRange,
                                     StringRef sourceFile,
                                     SourceRange parenRange,
                                     bool implicit)
    : DeclAttribute(DeclAttrKind::PrivateImport, atLoc, baseRange, implicit),
      SourceFile(sourceFile) {}

PrivateImportAttr *PrivateImportAttr::create(ASTContext &Ctxt, SourceLoc AtLoc,
                                             SourceLoc PrivateLoc,
                                             SourceLoc LParenLoc,
                                             StringRef sourceFile,
                                             SourceLoc RParenLoc) {
  return new (Ctxt)
      PrivateImportAttr(AtLoc, SourceRange(PrivateLoc, RParenLoc), sourceFile,
                        SourceRange(LParenLoc, RParenLoc));
}

DynamicReplacementAttr::DynamicReplacementAttr(SourceLoc atLoc,
                                               SourceRange baseRange,
                                               DeclNameRef name,
                                               DeclNameLoc nameLoc,
                                               SourceRange parenRange)
    : DeclAttribute(DeclAttrKind::DynamicReplacement, atLoc, baseRange,
                    /*Implicit=*/false),
      ReplacedFunctionName(name),  ReplacedFunctionNameLoc(nameLoc) {
  Bits.DynamicReplacementAttr.HasTrailingLocationInfo = true;
  getTrailingLocations()[0] = parenRange.Start;
  getTrailingLocations()[1] = parenRange.End;
}

DynamicReplacementAttr *
DynamicReplacementAttr::create(ASTContext &Ctx, SourceLoc AtLoc,
                               SourceLoc DynReplLoc, SourceLoc LParenLoc,
                               DeclNameRef ReplacedFunction,
                               DeclNameLoc nameLoc, SourceLoc RParenLoc) {
  void *mem = Ctx.Allocate(totalSizeToAlloc<SourceLoc>(2),
                           alignof(DynamicReplacementAttr));
  return new (mem) DynamicReplacementAttr(
      AtLoc, SourceRange(DynReplLoc, RParenLoc), ReplacedFunction, nameLoc,
      SourceRange(LParenLoc, RParenLoc));
}

DynamicReplacementAttr *
DynamicReplacementAttr::create(ASTContext &Ctx, DeclNameRef name,
                               AbstractFunctionDecl *f) {
  return new (Ctx) DynamicReplacementAttr(name, f);
}

DynamicReplacementAttr *
DynamicReplacementAttr::create(ASTContext &Ctx, DeclNameRef name,
                               LazyMemberLoader *Resolver, uint64_t Data) {
  return new (Ctx) DynamicReplacementAttr(name, Resolver, Data);
}

SourceLoc DynamicReplacementAttr::getLParenLoc() const {
  return getTrailingLocations()[0];
}

SourceLoc DynamicReplacementAttr::getRParenLoc() const {
  return getTrailingLocations()[1];
}

TypeEraserAttr *TypeEraserAttr::create(ASTContext &ctx,
                                       SourceLoc atLoc, SourceRange range,
                                       TypeExpr *typeEraserExpr) {
  return new (ctx) TypeEraserAttr(atLoc, range, typeEraserExpr, nullptr, 0);
}

TypeEraserAttr *TypeEraserAttr::create(ASTContext &ctx,
                                       LazyMemberLoader *Resolver,
                                       uint64_t Data) {
  return new (ctx) TypeEraserAttr(SourceLoc(), SourceRange(),
                                  nullptr, Resolver, Data);
}

bool TypeEraserAttr::hasViableTypeEraserInit(ProtocolDecl *protocol) const {
  return evaluateOrDefault(protocol->getASTContext().evaluator,
                           TypeEraserHasViableInitRequest{
                               const_cast<TypeEraserAttr *>(this), protocol},
                           false);
}

TypeRepr *TypeEraserAttr::getParsedTypeEraserTypeRepr() const {
  return TypeEraserExpr ? TypeEraserExpr->getTypeRepr() : nullptr;
}

SourceLoc TypeEraserAttr::getLoc() const {
  return TypeEraserExpr ? TypeEraserExpr->getLoc() : SourceLoc();
}

Type TypeEraserAttr::getTypeWithoutResolving() const {
  return TypeEraserExpr ? TypeEraserExpr->getInstanceType() : Type();
}

Type TypeEraserAttr::getResolvedType(const ProtocolDecl *PD) const {
  auto &ctx = PD->getASTContext();
  return evaluateOrDefault(ctx.evaluator,
                           ResolveTypeEraserTypeRequest{
                               const_cast<ProtocolDecl *>(PD),
                               const_cast<TypeEraserAttr *>(this)},
                           ErrorType::get(ctx));
}

static bool eqTypes(Type first, Type second) {
  if (first.isNull() || second.isNull())
    return false;
  return first->getCanonicalType() == second->getCanonicalType();
}

bool TypeEraserAttr::isEquivalent(const TypeEraserAttr *other,
                                  Decl *attachedTo) const {
  // Only makes sense when attached to a protocol.
  auto proto = dyn_cast<ProtocolDecl>(attachedTo);
  if (!proto)
    return true;

  Type thisType = getResolvedType(proto),
       otherType = other->getResolvedType(proto);
  if (thisType.isNull() || otherType.isNull())
    return false;
  return thisType->getCanonicalType() == otherType->getCanonicalType();
}

Type RawLayoutAttr::getResolvedLikeType(StructDecl *sd) const {
  auto &ctx = sd->getASTContext();
  return evaluateOrDefault(ctx.evaluator,
                           ResolveRawLayoutTypeRequest{sd,
                              const_cast<RawLayoutAttr *>(this),
                              /*isLikeType*/ true},
                           ErrorType::get(ctx));
}

Type RawLayoutAttr::getResolvedCountType(StructDecl *sd) const {
  auto &ctx = sd->getASTContext();
  return evaluateOrDefault(ctx.evaluator,
                           ResolveRawLayoutTypeRequest{sd,
                              const_cast<RawLayoutAttr *>(this),
                              /*isLikeType*/ false},
                           ErrorType::get(ctx));
}

bool RawLayoutAttr::isEquivalent(const RawLayoutAttr *other,
                                 Decl *attachedTo) const {
  if (shouldMoveAsLikeType() != other->shouldMoveAsLikeType())
    return false;

  if (auto thisSizeAndAlignment = getSizeAndAlignment())
    return thisSizeAndAlignment == other->getSizeAndAlignment();

  auto SD = dyn_cast<StructDecl>(attachedTo);
  if (!SD)
    return true;

  if (auto thisScalarLikeType = getResolvedScalarLikeType(SD)) {
    auto otherScalarLikeType = other->getResolvedScalarLikeType(SD);
    return otherScalarLikeType && eqTypes(*thisScalarLikeType,
                                          *otherScalarLikeType);
  }

  if (auto thisArrayLikeTypes = getResolvedArrayLikeTypeAndCount(SD)) {
    auto otherArrayLikeTypes = other->getResolvedArrayLikeTypeAndCount(SD);
    return otherArrayLikeTypes
            && eqTypes(thisArrayLikeTypes->first, otherArrayLikeTypes->first)
            && eqTypes(thisArrayLikeTypes->second, otherArrayLikeTypes->second);
  }

  llvm_unreachable("unknown variant of RawLayoutAttr");
}

AvailableAttr::AvailableAttr(
    SourceLoc AtLoc, SourceRange Range,
    AvailabilityDomainOrIdentifier DomainOrIdentifier, SourceLoc DomainLoc,
    Kind Kind, StringRef Message, StringRef Rename,
    const llvm::VersionTuple &Introduced, SourceRange IntroducedRange,
    const llvm::VersionTuple &Deprecated, SourceRange DeprecatedRange,
    const llvm::VersionTuple &Obsoleted, SourceRange ObsoletedRange,
    bool Implicit, bool IsSPI)
    : DeclAttribute(DeclAttrKind::Available, AtLoc, Range, Implicit),
      DomainOrIdentifier(DomainOrIdentifier), DomainLoc(DomainLoc),
      Message(Message), Rename(Rename), Introduced(Introduced),
      IntroducedRange(IntroducedRange), Deprecated(Deprecated),
      DeprecatedRange(DeprecatedRange), Obsoleted(Obsoleted),
      ObsoletedRange(ObsoletedRange) {
  Bits.AvailableAttr.Kind = static_cast<uint8_t>(Kind);
  Bits.AvailableAttr.IsSPI = IsSPI;
  Bits.AvailableAttr.IsGroupMember = false;
  Bits.AvailableAttr.IsGroupTerminator = false;
  Bits.AvailableAttr.IsGroupedWithWildcard = false;
}

AvailableAttr *AvailableAttr::createUniversallyUnavailable(ASTContext &C,
                                                           StringRef Message,
                                                           StringRef Rename) {
  return new (C) AvailableAttr(
      SourceLoc(), SourceRange(), AvailabilityDomain::forUniversal(),
      SourceLoc(), Kind::Unavailable, Message, Rename,
      /*Introduced=*/{}, SourceRange(), /*Deprecated=*/{}, SourceRange(),
      /*Obsoleted=*/{}, SourceRange(),
      /*Implicit=*/false,
      /*SPI=*/false);
}

AvailableAttr *AvailableAttr::createUniversallyDeprecated(ASTContext &C,
                                                          StringRef Message,
                                                          StringRef Rename) {
  return new (C) AvailableAttr(
      SourceLoc(), SourceRange(), AvailabilityDomain::forUniversal(),
      SourceLoc(), Kind::Deprecated, Message, Rename,
      /*Introduced=*/{}, SourceRange(), /*Deprecated=*/{}, SourceRange(),
      /*Obsoleted=*/{}, SourceRange(),
      /*Implicit=*/false,
      /*SPI=*/false);
}

AvailableAttr *AvailableAttr::createUnavailableInSwift(ASTContext &C,
                                                       StringRef Message,
                                                       StringRef Rename) {
  return new (C) AvailableAttr(
      SourceLoc(), SourceRange(), AvailabilityDomain::forSwiftLanguageMode(),
      SourceLoc(), Kind::Unavailable, Message, Rename,
      /*Introduced=*/{}, SourceRange(), /*Deprecated=*/{}, SourceRange(),
      /*Obsoleted=*/{}, SourceRange(),
      /*Implicit=*/false,
      /*SPI=*/false);
}

AvailableAttr *AvailableAttr::createSwiftLanguageModeVersioned(
    ASTContext &C, StringRef Message, StringRef Rename,
    llvm::VersionTuple Introduced, llvm::VersionTuple Obsoleted) {
  return new (C) AvailableAttr(
      SourceLoc(), SourceRange(), AvailabilityDomain::forSwiftLanguageMode(),
      SourceLoc(), Kind::Default, Message, Rename, Introduced, SourceRange(),
      /*Deprecated=*/{}, SourceRange(), Obsoleted, SourceRange(),
      /*Implicit=*/false,
      /*SPI=*/false);
}

AvailableAttr *AvailableAttr::createPlatformVersioned(
    ASTContext &C, PlatformKind Platform, StringRef Message, StringRef Rename,
    llvm::VersionTuple Introduced, llvm::VersionTuple Deprecated,
    llvm::VersionTuple Obsoleted) {
  return new (C) AvailableAttr(
      SourceLoc(), SourceRange(), AvailabilityDomain::forPlatform(Platform),
      SourceLoc(), Kind::Default, Message, Rename, Introduced, SourceRange(),
      Deprecated, SourceRange(), Obsoleted, SourceRange(),
      /*Implicit=*/false,
      /*SPI=*/false);
}

AvailableAttr *AvailableAttr::createUnavailableInEmbedded(ASTContext &C,
                                                          SourceLoc AtLoc,
                                                          SourceRange Range) {
  return new (C) AvailableAttr(
      AtLoc, Range, AvailabilityDomain::forEmbedded(), SourceLoc(),
      AvailableAttr::Kind::Unavailable, "unavailable in embedded Swift",
      /*Rename=*/StringRef(),
      /*Introduced=*/llvm::VersionTuple(), /*IntroducedRange=*/{},
      /*Deprecated=*/llvm::VersionTuple(), /*DeprecatedRange=*/{},
      /*Obsoleted=*/llvm::VersionTuple(), /*ObsoletedRange=*/{},
      /*Implicit=*/false, /*IsSPI=*/false);
}

llvm::VersionTuple BackDeployedAttr::getVersion() const {
  return canonicalizePlatformVersion(getPlatform(), getParsedVersion());
}

bool BackDeployedAttr::isActivePlatform(const ASTContext &ctx,
                                        bool forTargetVariant) const {
  return isPlatformActive(Platform, ctx.LangOpts, forTargetVariant);
}

AvailableAttr *AvailableAttr::clone(ASTContext &C, bool implicit) const {
  return new (C) AvailableAttr(
      implicit ? SourceLoc() : AtLoc, implicit ? SourceRange() : getRange(),
      DomainOrIdentifier, implicit ? SourceLoc() : DomainLoc, getKind(),
      Message, Rename, Introduced, implicit ? SourceRange() : IntroducedRange,
      Deprecated, implicit ? SourceRange() : DeprecatedRange, Obsoleted,
      implicit ? SourceRange() : ObsoletedRange, implicit, isSPI());
}

bool AvailableAttr::isEquivalent(const AvailableAttr *other,
                                 Decl *attachedTo) const {
  return getRawIntroduced() == other->getRawIntroduced()
          && getRawDeprecated() == other->getRawDeprecated()
          && getRawObsoleted() == other->getRawObsoleted()
          && getMessage() == other->getMessage()
          && getRename() == other->getRename()
          && isSPI() == other->isSPI()
          && getKind() == other->getKind()
          && attachedTo->getSemanticAvailableAttr(this)->getDomain()
                ==  attachedTo->getSemanticAvailableAttr(other)->getDomain();
}

static StringRef parseManglingModuleName(StringRef OriginalModuleName) {
  auto index = OriginalModuleName.find(";");
  return index == StringRef::npos
      ? OriginalModuleName
      : OriginalModuleName.slice(0, index);
}

static StringRef parseLinkerModuleName(StringRef OriginalModuleName) {
  auto index = OriginalModuleName.find(";");
  return index == StringRef::npos
      ? OriginalModuleName
      : OriginalModuleName.slice(index + 1, OriginalModuleName.size());
}

OriginallyDefinedInAttr::OriginallyDefinedInAttr(
    SourceLoc AtLoc, SourceRange Range, StringRef OriginalModuleName,
    PlatformKind Platform, const llvm::VersionTuple MovedVersion, bool Implicit)
    : DeclAttribute(DeclAttrKind::OriginallyDefinedIn, AtLoc, Range, Implicit),
      ManglingModuleName(parseManglingModuleName(OriginalModuleName)),
      LinkerModuleName(parseLinkerModuleName(OriginalModuleName)),
      Platform(Platform), MovedVersion(MovedVersion) {}

llvm::VersionTuple OriginallyDefinedInAttr::getMovedVersion() const {
  return canonicalizePlatformVersion(getPlatform(), getParsedMovedVersion());
}

std::optional<OriginallyDefinedInAttr::ActiveVersion>
OriginallyDefinedInAttr::isActivePlatform(const ASTContext &ctx) const {
  OriginallyDefinedInAttr::ActiveVersion Result;
  Result.Platform = Platform;
  Result.Version = getMovedVersion();
  Result.ManglingModuleName = ManglingModuleName;
  Result.LinkerModuleName = LinkerModuleName;
  if (isPlatformActive(Platform, ctx.LangOpts, /*TargetVariant*/false)) {
    return Result;
  }

  // Also check if the platform is active by using target variant. This ensures
  // we emit linker directives for multiple platforms when building zippered
  // libraries.
  if (ctx.LangOpts.TargetVariant.has_value() &&
      isPlatformActive(Platform, ctx.LangOpts, /*TargetVariant*/true)) {
    Result.ForTargetVariant = true;
    return Result;
  }
  return std::nullopt;
}

OriginallyDefinedInAttr *OriginallyDefinedInAttr::clone(ASTContext &C,
                                                        bool implicit) const {
  return new (C) OriginallyDefinedInAttr(
      implicit ? SourceLoc() : AtLoc, implicit ? SourceRange() : getRange(),
      ManglingModuleName, LinkerModuleName, Platform, MovedVersion, implicit);
}

void ABIAttr::attachToDeclImpl(Decl *D) {
  // Register the relationship between `D` and `abiDecl`. This is necessary for
  // `ABIRoleInfo::ABIRoleInfo()` to determine that `abiDecl` is ABI-only and
  // locate its API counterpart.
  Decl *owner = D;

  // The ABIAttr on a VarDecl ought to point to its PBD.
  if (auto VD = dyn_cast<VarDecl>(owner)) {
    if (auto PBD = VD->getParentPatternBinding())
      owner = PBD;
  }

  auto record = [&](Decl *decl) {
    auto &evaluator = owner->getASTContext().evaluator;
    DeclABIRoleInfoRequest(decl).recordABIOnly(evaluator, owner);
  };

  if (auto abiPBD = dyn_cast<PatternBindingDecl>(abiDecl)) {
    // Add to *every* VarDecl in the ABI PBD, even ones that don't properly
    // match anything in the API PBD.
    for (auto i : range(abiPBD->getNumPatternEntries())) {
      abiPBD->getPattern(i)->forEachVariable(record);
    }
    return;
  }

  record(abiDecl);
}

bool AvailableAttr::isUnconditionallyUnavailable() const {
  switch (getKind()) {
  case Kind::Default:
  case Kind::Deprecated:
  case Kind::NoAsync:
    return false;
  case Kind::Unavailable:
    return true;
  }

  llvm_unreachable("Unhandled AvailableAttr::Kind in switch.");
}

bool AvailableAttr::isUnconditionallyDeprecated() const {
  switch (getKind()) {
  case Kind::Default:
  case Kind::Unavailable:
  case Kind::NoAsync:
    return false;
  case Kind::Deprecated:
    return true;
  }

  llvm_unreachable("Unhandled AvailableAttr::Kind in switch.");
}

bool AvailableAttr::isNoAsync() const {
  switch (getKind()) {
  case Kind::Default:
  case Kind::Deprecated:
  case Kind::Unavailable:
    return false;
  case Kind::NoAsync:
    return true;
  }

  llvm_unreachable("Unhandled AvailableAttr::Kind in switch.");
}

AbstractSpecializeAttr::AbstractSpecializeAttr(DeclAttrKind DK,
                               SourceLoc atLoc, SourceRange range,
                               TrailingWhereClause *clause,
                               bool exported,
                               SpecializationKind kind,
                               GenericSignature specializedSignature,
                               DeclNameRef targetFunctionName,
                               DeclNameLoc targetFunctionNameLoc,
                               ArrayRef<Identifier> spiGroups,
                               ArrayRef<AvailableAttr *> availableAttrs,
                               size_t typeErasedParamsCount)
    : DeclAttribute(DK, atLoc, range, /*Implicit=*/clause == nullptr),
      trailingWhereClause(clause), specializedSignature(specializedSignature),
      targetFunctionName(targetFunctionName),
      targetFunctionNameLoc(targetFunctionNameLoc),
      numSPIGroups(spiGroups.size()),
      numAvailableAttrs(availableAttrs.size()),
      numTypeErasedParams(typeErasedParamsCount),
      typeErasedParamsInitialized(false) {
  std::uninitialized_copy(spiGroups.begin(), spiGroups.end(),
                          getSubclassTrailingObjects<Identifier>());
  std::uninitialized_copy(availableAttrs.begin(), availableAttrs.end(),
                          getSubclassTrailingObjects<AvailableAttr *>());

  Bits.AbstractSpecializeAttr.exported = exported;
  Bits.AbstractSpecializeAttr.kind = unsigned(kind);
}

TrailingWhereClause *AbstractSpecializeAttr::getTrailingWhereClause() const {
  return trailingWhereClause;
}

SpecializeAttr *SpecializeAttr::create(ASTContext &Ctx, SourceLoc atLoc,
                                       SourceRange range,
                                       TrailingWhereClause *clause,
                                       bool exported, SpecializationKind kind,
                                       DeclNameRef targetFunctionName,
                                       DeclNameLoc targetFunctionNameLoc,
                                       ArrayRef<Identifier> spiGroups,
                                       ArrayRef<AvailableAttr *> availableAttrs,
                                       GenericSignature specializedSignature) {
  size_t typeErasedParamsCount = 0;
  if (clause != nullptr) {
    for (auto &req : clause->getRequirements()) {
      if (req.getKind() == RequirementReprKind::LayoutConstraint) {
        if (auto *attributedTy =
                dyn_cast<AttributedTypeRepr>(req.getSubjectRepr())) {
          if (attributedTy->has(TypeAttrKind::NoMetadata)) {
            typeErasedParamsCount += 1;
          }
        }
      }
    }
  }

  unsigned size = totalSizeToAlloc<Identifier, AvailableAttr *, Type>(
      spiGroups.size(), availableAttrs.size(), typeErasedParamsCount);
  void *mem = Ctx.Allocate(size, alignof(SpecializeAttr));

  return new (mem)
      SpecializeAttr(atLoc, range, clause, exported, kind, specializedSignature,
                     targetFunctionName, targetFunctionNameLoc, spiGroups,
                     availableAttrs, typeErasedParamsCount);
}

SpecializeAttr *SpecializeAttr::create(ASTContext &ctx, bool exported,
                                       SpecializationKind kind,
                                       ArrayRef<Identifier> spiGroups,
                                       ArrayRef<AvailableAttr *> availableAttrs,
                                       GenericSignature specializedSignature,
                                       DeclNameRef targetFunctionName) {
  unsigned size = totalSizeToAlloc<Identifier, AvailableAttr *, Type>(
      spiGroups.size(), availableAttrs.size(), 0);
  void *mem = ctx.Allocate(size, alignof(SpecializeAttr));
  return new (mem) SpecializeAttr(
      SourceLoc(), SourceRange(), nullptr, exported, kind, specializedSignature,
      targetFunctionName, DeclNameLoc(), spiGroups, availableAttrs, 0);
}

SpecializeAttr *SpecializeAttr::create(
    ASTContext &ctx, bool exported, SpecializationKind kind,
    ArrayRef<Identifier> spiGroups, ArrayRef<AvailableAttr *> availableAttrs,
    ArrayRef<Type> typeErasedParams, GenericSignature specializedSignature,
    DeclNameRef targetFunctionName, LazyMemberLoader *resolver,
    uint64_t data) {
  unsigned size = totalSizeToAlloc<Identifier, AvailableAttr *, Type>(
      spiGroups.size(), availableAttrs.size(), typeErasedParams.size());
  void *mem = ctx.Allocate(size, alignof(SpecializeAttr));
  auto *attr = new (mem) SpecializeAttr(
      SourceLoc(), SourceRange(), nullptr, exported, kind, specializedSignature,
      targetFunctionName, DeclNameLoc(), spiGroups, availableAttrs,
      typeErasedParams.size());
  attr->setTypeErasedParams(typeErasedParams);
  attr->setResolver(resolver, data);
  return attr;
}

// @specialize
//
SpecializedAttr *SpecializedAttr::create(ASTContext &Ctx, SourceLoc atLoc,
                                       SourceRange range,
                                       TrailingWhereClause *clause,
                                       bool exported, SpecializationKind kind,
                                       DeclNameRef targetFunctionName,
                                       DeclNameLoc targetFunctionNameLoc,
                                       ArrayRef<Identifier> spiGroups,
                                       ArrayRef<AvailableAttr *> availableAttrs,
                                       GenericSignature specializedSignature) {
  size_t typeErasedParamsCount = 0;
  if (clause != nullptr) {
    for (auto &req : clause->getRequirements()) {
      if (req.getKind() == RequirementReprKind::LayoutConstraint) {
        if (auto *attributedTy =
                dyn_cast<AttributedTypeRepr>(req.getSubjectRepr())) {
          if (attributedTy->has(TypeAttrKind::NoMetadata)) {
            typeErasedParamsCount += 1;
          }
        }
      }
    }
  }

  unsigned size = totalSizeToAlloc<Identifier, AvailableAttr *, Type>(
      spiGroups.size(), availableAttrs.size(), typeErasedParamsCount);
  void *mem = Ctx.Allocate(size, alignof(SpecializedAttr));

  return new (mem)
      SpecializedAttr(atLoc, range, clause, exported, kind, specializedSignature,
                     targetFunctionName, targetFunctionNameLoc, spiGroups,
                     availableAttrs, typeErasedParamsCount);
}

SpecializedAttr *SpecializedAttr::create(ASTContext &ctx, bool exported,
                                       SpecializationKind kind,
                                       ArrayRef<Identifier> spiGroups,
                                       ArrayRef<AvailableAttr *> availableAttrs,
                                       GenericSignature specializedSignature,
                                       DeclNameRef targetFunctionName) {
  unsigned size = totalSizeToAlloc<Identifier, AvailableAttr *, Type>(
      spiGroups.size(), availableAttrs.size(), 0);
  void *mem = ctx.Allocate(size, alignof(SpecializedAttr));
  return new (mem) SpecializedAttr(
      SourceLoc(), SourceRange(), nullptr, exported, kind,
      specializedSignature, targetFunctionName, DeclNameLoc(), spiGroups,
      availableAttrs, 0);
}

SpecializedAttr *SpecializedAttr::create(
    ASTContext &ctx, bool exported, SpecializationKind kind,
    ArrayRef<Identifier> spiGroups, ArrayRef<AvailableAttr *> availableAttrs,
    ArrayRef<Type> typeErasedParams, GenericSignature specializedSignature,
    DeclNameRef targetFunctionName, LazyMemberLoader *resolver,
    uint64_t data) {
  unsigned size = totalSizeToAlloc<Identifier, AvailableAttr *, Type>(
      spiGroups.size(), availableAttrs.size(), typeErasedParams.size());
  void *mem = ctx.Allocate(size, alignof(SpecializedAttr));
  auto *attr = new (mem) SpecializedAttr(
      SourceLoc(), SourceRange(), nullptr, exported, kind, specializedSignature,
      targetFunctionName, DeclNameLoc(), spiGroups, availableAttrs,
      typeErasedParams.size());
  attr->setTypeErasedParams(typeErasedParams);
  attr->setResolver(resolver, data);
  return attr;
}

ValueDecl * AbstractSpecializeAttr::getTargetFunctionDecl(const ValueDecl *onDecl) const {
  return evaluateOrDefault(onDecl->getASTContext().evaluator,
                           SpecializeAttrTargetDeclRequest{
                               onDecl, const_cast<AbstractSpecializeAttr *>(this)},
                           nullptr);
}

GenericSignature AbstractSpecializeAttr::getSpecializedSignature(
    const AbstractFunctionDecl *onDecl) const {
  return evaluateOrDefault(onDecl->getASTContext().evaluator,
                           SerializeAttrGenericSignatureRequest{
                               onDecl, const_cast<AbstractSpecializeAttr *>(this)},
                           nullptr);
}

/// Checks whether \p first and \p second have the same elements according to
/// \p eq , ignoring the order those elements are in but considering the number
/// of repetitions.
template<typename Element, typename Eq>
bool sameElements(ArrayRef<Element> first, ArrayRef<Element> second, Eq eq) {
  if (first.size() != second.size())
    return false;

  llvm::SmallSet<unsigned, 8> claimedSecondIndices;

  for (auto firstElem : first) {
    bool found = false;
    for (auto i : indices(second)) {
      if (!eq(firstElem, second[i]) || claimedSecondIndices.count(i))
        continue;

      found = true;
      claimedSecondIndices.insert(i);
      break;
    }

    if (!found)
      return false;
  }

  return true;
}

/// Checks whether \p first and \p second have the same elements, ignoring the
/// order those elements are in but considering the number of repetitions.
template<typename Element>
bool sameElements(ArrayRef<Element> first, ArrayRef<Element> second) {
  return sameElements(first, second, std::equal_to<Element>());
}

bool AbstractSpecializeAttr::isEquivalent(const AbstractSpecializeAttr *other,
                                  Decl *attachedTo) const {
  if (isExported() != other->isExported() ||
      getSpecializationKind() != other->getSpecializationKind() ||
      getTargetFunctionName() != other->getTargetFunctionName() ||
      specializedSignature.getCanonicalSignature() !=
        other->specializedSignature.getCanonicalSignature())
    return false;

  if (!sameElements(getSPIGroups(), other->getSPIGroups()))
    return false;

  SmallVector<CanType, 8> thisTypeErasedParams, otherTypeErasedParams;
  for (auto ty : getTypeErasedParams())
    thisTypeErasedParams.push_back(ty->getCanonicalType());
  for (auto ty : other->getTypeErasedParams())
    otherTypeErasedParams.push_back(ty->getCanonicalType());

  if (!sameElements(ArrayRef(thisTypeErasedParams),
                    ArrayRef(otherTypeErasedParams)))
    return false;

  return sameElements(getAvailableAttrs(), other->getAvailableAttrs(),
                      [=](AvailableAttr *thisAttr, AvailableAttr *otherAttr) {
    return thisAttr->isEquivalent(otherAttr, attachedTo);
  });
}

SPIAccessControlAttr::SPIAccessControlAttr(SourceLoc atLoc, SourceRange range,
                                           ArrayRef<Identifier> spiGroups)
    : DeclAttribute(DeclAttrKind::SPIAccessControl, atLoc, range,
                    /*Implicit=*/false),
      numSPIGroups(spiGroups.size()) {
  std::uninitialized_copy(spiGroups.begin(), spiGroups.end(),
                          getTrailingObjects());
}

SPIAccessControlAttr *
SPIAccessControlAttr::create(ASTContext &context,
                             SourceLoc atLoc,
                             SourceRange range,
                             ArrayRef<Identifier> spiGroups) {
  unsigned size = totalSizeToAlloc<Identifier>(spiGroups.size());
  void *mem = context.Allocate(size, alignof(SPIAccessControlAttr));
  return new (mem) SPIAccessControlAttr(atLoc, range, spiGroups);
}

SPIAccessControlAttr *SPIAccessControlAttr::clone(ASTContext &C,
                                                  bool implicit) const {
  auto *attr = SPIAccessControlAttr::create(
      C, implicit ? SourceLoc() : AtLoc, implicit ? SourceRange() : getRange(),
      getSPIGroups());
  attr->setImplicit(implicit);
  return attr;
}

bool SPIAccessControlAttr::isEquivalent(const SPIAccessControlAttr *other,
                                        Decl *attachedTo) const {
  return sameElements(getSPIGroups(), other->getSPIGroups());
}

DifferentiableAttr::DifferentiableAttr(bool implicit, SourceLoc atLoc,
                                       SourceRange baseRange,
                                       enum DifferentiabilityKind diffKind,
                                       ArrayRef<ParsedAutoDiffParameter> params,
                                       TrailingWhereClause *clause)
    : DeclAttribute(DeclAttrKind::Differentiable, atLoc, baseRange, implicit),
      DifferentiabilityKind(diffKind), NumParsedParameters(params.size()),
      WhereClause(clause) {
  assert((diffKind != DifferentiabilityKind::Normal &&
          diffKind != DifferentiabilityKind::Forward) &&
         "'Normal' and 'Forward' are not supported");
  std::copy(params.begin(), params.end(), getTrailingObjects());
}

DifferentiableAttr::DifferentiableAttr(Decl *original, bool implicit,
                                       SourceLoc atLoc, SourceRange baseRange,
                                       enum DifferentiabilityKind diffKind,
                                       IndexSubset *parameterIndices,
                                       GenericSignature derivativeGenSig)
    : DeclAttribute(DeclAttrKind::Differentiable, atLoc, baseRange, implicit),
      OriginalDeclaration(original), DifferentiabilityKind(diffKind) {
  assert((diffKind != DifferentiabilityKind::Normal &&
          diffKind != DifferentiabilityKind::Forward) &&
         "'Normal' and 'Forward' are not supported");
  setParameterIndices(parameterIndices);
  setDerivativeGenericSignature(derivativeGenSig);
}

DifferentiableAttr *
DifferentiableAttr::create(ASTContext &context, bool implicit,
                           SourceLoc atLoc, SourceRange baseRange,
                           enum DifferentiabilityKind diffKind,
                           ArrayRef<ParsedAutoDiffParameter> parameters,
                           TrailingWhereClause *clause) {
  unsigned size = totalSizeToAlloc<ParsedAutoDiffParameter>(parameters.size());
  void *mem = context.Allocate(size, alignof(DifferentiableAttr));
  return new (mem) DifferentiableAttr(implicit, atLoc, baseRange, diffKind,
                                      parameters, clause);
}

DifferentiableAttr *
DifferentiableAttr::create(AbstractFunctionDecl *original, bool implicit,
                           SourceLoc atLoc, SourceRange baseRange,
                           enum DifferentiabilityKind diffKind,
                           IndexSubset *parameterIndices,
                           GenericSignature derivativeGenSig) {
  auto &ctx = original->getASTContext();
  
  size_t size = totalSizeToAlloc<ParsedAutoDiffParameter>(0); 
  void *mem = ctx.Allocate(size, alignof(DifferentiableAttr));
  return new (mem) DifferentiableAttr(original, implicit, atLoc, baseRange,
                                      diffKind, parameterIndices,
                                      derivativeGenSig);
}

void DifferentiableAttr::attachToDeclImpl(Decl *originalDeclaration) {
  // FIXME: This doesn't properly handle PatternBindingDecls with multiple
  // VarDecls.
  assert(!OriginalDeclaration || OriginalDeclaration == originalDeclaration &&
         "Original declaration cannot have already been set");
  OriginalDeclaration = originalDeclaration;
}

bool DifferentiableAttr::hasBeenTypeChecked() const {
  return ParameterIndicesAndBit.getInt();
}

IndexSubset *DifferentiableAttr::getParameterIndices() const {
  assert(getOriginalDeclaration() &&
         "Original declaration must have been resolved");
  auto &ctx = getOriginalDeclaration()->getASTContext();
  return evaluateOrDefault(ctx.evaluator,
                           DifferentiableAttributeTypeCheckRequest{
                               const_cast<DifferentiableAttr *>(this)},
                           nullptr);
}

void DifferentiableAttr::setParameterIndices(IndexSubset *paramIndices) {
  assert(getOriginalDeclaration() &&
         "Original declaration must have been resolved");
  auto &ctx = getOriginalDeclaration()->getASTContext();
  ctx.evaluator.cacheOutput(
      DifferentiableAttributeTypeCheckRequest{
          const_cast<DifferentiableAttr *>(this)},
      std::move(paramIndices));
}

GenericEnvironment *DifferentiableAttr::getDerivativeGenericEnvironment(
    AbstractFunctionDecl *original) const {
  if (auto derivativeGenSig = getDerivativeGenericSignature())
    return derivativeGenSig.getGenericEnvironment();
  return original->getGenericEnvironment();
}

void DeclNameRefWithLoc::print(ASTPrinter &Printer) const {
  Printer << Name;
  if (AccessorKind)
    Printer << '.' << getAccessorLabel(*AccessorKind);
}

void DifferentiableAttr::print(llvm::raw_ostream &OS, const Decl *D,
                               bool omitWrtClause) const {
  StreamPrinter P(OS);
  P << "@" << getAttrName();
  printDifferentiableAttrArguments(this, P, PrintOptions(), D, omitWrtClause);
}

DerivativeAttr::DerivativeAttr(bool implicit, SourceLoc atLoc,
                               SourceRange baseRange, TypeRepr *baseTypeRepr,
                               DeclNameRefWithLoc originalName,
                               ArrayRef<ParsedAutoDiffParameter> params)
    : DeclAttribute(DeclAttrKind::Derivative, atLoc, baseRange, implicit),
      BaseTypeRepr(baseTypeRepr), OriginalFunctionName(std::move(originalName)),
      NumParsedParameters(params.size()) {
  std::copy(params.begin(), params.end(), getTrailingObjects());
}

DerivativeAttr::DerivativeAttr(bool implicit, SourceLoc atLoc,
                               SourceRange baseRange, TypeRepr *baseTypeRepr,
                               DeclNameRefWithLoc originalName,
                               IndexSubset *parameterIndices)
    : DeclAttribute(DeclAttrKind::Derivative, atLoc, baseRange, implicit),
      BaseTypeRepr(baseTypeRepr), OriginalFunctionName(std::move(originalName)),
      ParameterIndices(parameterIndices) {}

DerivativeAttr *
DerivativeAttr::create(ASTContext &context, bool implicit, SourceLoc atLoc,
                       SourceRange baseRange, TypeRepr *baseTypeRepr,
                       DeclNameRefWithLoc originalName,
                       ArrayRef<ParsedAutoDiffParameter> params) {
  unsigned size = totalSizeToAlloc<ParsedAutoDiffParameter>(params.size());
  void *mem = context.Allocate(size, alignof(DerivativeAttr));
  return new (mem) DerivativeAttr(implicit, atLoc, baseRange, baseTypeRepr,
                                  std::move(originalName), params);
}

DerivativeAttr *DerivativeAttr::create(ASTContext &context, bool implicit,
                                       SourceLoc atLoc, SourceRange baseRange,
                                       TypeRepr *baseTypeRepr,
                                       DeclNameRefWithLoc originalName,
                                       IndexSubset *parameterIndices) {
  void *mem = context.Allocate(sizeof(DerivativeAttr), alignof(DerivativeAttr));
  return new (mem) DerivativeAttr(implicit, atLoc, baseRange, baseTypeRepr,
                                  std::move(originalName), parameterIndices);
}

AbstractFunctionDecl *
DerivativeAttr::getOriginalFunction(ASTContext &context) const {
  return evaluateOrDefault(
      context.evaluator,
      DerivativeAttrOriginalDeclRequest{const_cast<DerivativeAttr *>(this)},
      nullptr);
}

void DerivativeAttr::setOriginalFunction(AbstractFunctionDecl *decl) {
  assert(!OriginalFunction && "cannot overwrite original function");
  OriginalFunction = decl;
}

void DerivativeAttr::setOriginalFunctionResolver(
    LazyMemberLoader *resolver, uint64_t resolverContextData) {
  assert(!OriginalFunction && "cannot overwrite original function");
  OriginalFunction = resolver;
  ResolverContextData = resolverContextData;
}

void DerivativeAttr::attachToDeclImpl(Decl *originalDeclaration) {
  // FIXME: This doesn't properly handle PatternBindingDecls with multiple
  // VarDecls.
  assert(!OriginalDeclaration || OriginalDeclaration == originalDeclaration &&
         "Original declaration cannot have already been set");
  OriginalDeclaration = originalDeclaration;
}

TransposeAttr::TransposeAttr(bool implicit, SourceLoc atLoc,
                             SourceRange baseRange, TypeRepr *baseTypeRepr,
                             DeclNameRefWithLoc originalName,
                             ArrayRef<ParsedAutoDiffParameter> params)
    : DeclAttribute(DeclAttrKind::Transpose, atLoc, baseRange, implicit),
      BaseTypeRepr(baseTypeRepr), OriginalFunctionName(std::move(originalName)),
      NumParsedParameters(params.size()) {
  std::uninitialized_copy(params.begin(), params.end(), getTrailingObjects());
}

TransposeAttr::TransposeAttr(bool implicit, SourceLoc atLoc,
                             SourceRange baseRange, TypeRepr *baseTypeRepr,
                             DeclNameRefWithLoc originalName,
                             IndexSubset *parameterIndices)
    : DeclAttribute(DeclAttrKind::Transpose, atLoc, baseRange, implicit),
      BaseTypeRepr(baseTypeRepr), OriginalFunctionName(std::move(originalName)),
      ParameterIndices(parameterIndices) {}

TransposeAttr *TransposeAttr::create(ASTContext &context, bool implicit,
                                     SourceLoc atLoc, SourceRange baseRange,
                                     TypeRepr *baseType,
                                     DeclNameRefWithLoc originalName,
                                     ArrayRef<ParsedAutoDiffParameter> params) {
  unsigned size = totalSizeToAlloc<ParsedAutoDiffParameter>(params.size());
  void *mem = context.Allocate(size, alignof(TransposeAttr));
  return new (mem) TransposeAttr(implicit, atLoc, baseRange, baseType,
                                 std::move(originalName), params);
}

TransposeAttr *TransposeAttr::create(ASTContext &context, bool implicit,
                                     SourceLoc atLoc, SourceRange baseRange,
                                     TypeRepr *baseType,
                                     DeclNameRefWithLoc originalName,
                                     IndexSubset *parameterIndices) {
  void *mem = context.Allocate(sizeof(TransposeAttr), alignof(TransposeAttr));
  return new (mem) TransposeAttr(implicit, atLoc, baseRange, baseType,
                                 std::move(originalName), parameterIndices);
}

StorageRestrictionsAttr::StorageRestrictionsAttr(
    SourceLoc AtLoc, SourceRange Range, ArrayRef<Identifier> initializes,
    ArrayRef<Identifier> accesses, bool Implicit)
    : DeclAttribute(DeclAttrKind::StorageRestrictions, AtLoc, Range, Implicit),
      NumInitializes(initializes.size()), NumAccesses(accesses.size()) {
  std::uninitialized_copy(initializes.begin(), initializes.end(),
                          getTrailingObjects());
  std::uninitialized_copy(accesses.begin(), accesses.end(),
                          getTrailingObjects() + NumInitializes);
}

StorageRestrictionsAttr *
StorageRestrictionsAttr::create(
    ASTContext &ctx, SourceLoc atLoc, SourceRange range,
    ArrayRef<Identifier> initializes, ArrayRef<Identifier> accesses) {
  unsigned size =
      totalSizeToAlloc<Identifier>(initializes.size() + accesses.size());
  void *mem = ctx.Allocate(size, alignof(StorageRestrictionsAttr));
  return new (mem) StorageRestrictionsAttr(atLoc, range, initializes, accesses,
                                           /*implicit=*/false);
}

bool StorageRestrictionsAttr::
isEquivalent(const StorageRestrictionsAttr *other, Decl *attachedTo) const {
  return sameElements(getAccessesNames(), other->getAccessesNames())
           && sameElements(getInitializesNames(), other->getInitializesNames());
}

ImplementsAttr::
ImplementsAttr(SourceLoc atLoc, SourceRange range,
               llvm::PointerUnion<TypeRepr *, DeclContext *> TyROrDC,
               DeclName MemberName, DeclNameLoc MemberNameLoc)
    : DeclAttribute(DeclAttrKind::Implements, atLoc, range, /*Implicit=*/false),
      TyROrDC(TyROrDC), MemberName(MemberName), MemberNameLoc(MemberNameLoc) {}

ImplementsAttr *ImplementsAttr::create(ASTContext &Ctx, SourceLoc atLoc,
                                       SourceRange range,
                                       TypeRepr *TyR,
                                       DeclName MemberName,
                                       DeclNameLoc MemberNameLoc) {
  void *mem = Ctx.Allocate(sizeof(ImplementsAttr), alignof(ImplementsAttr));
  return new (mem) ImplementsAttr(atLoc, range, TyR,
                                  MemberName, MemberNameLoc);
}

ImplementsAttr *ImplementsAttr::create(DeclContext *DC,
                                       ProtocolDecl *Proto,
                                       DeclName MemberName) {
  auto &ctx = DC->getASTContext();
  void *mem = ctx.Allocate(sizeof(ImplementsAttr), alignof(ImplementsAttr));
  auto *attr = new (mem) ImplementsAttr(SourceLoc(), SourceRange(), DC,
                                        MemberName, DeclNameLoc());
  ctx.evaluator.cacheOutput(ImplementsAttrProtocolRequest{attr, DC},
                            std::move(Proto));
  return attr;
}

ProtocolDecl *ImplementsAttr::getProtocol(DeclContext *dc) const {
  return evaluateOrDefault(dc->getASTContext().evaluator,
        ImplementsAttrProtocolRequest{this, dc}, nullptr);
}

std::optional<ProtocolDecl *>
ImplementsAttr::getCachedProtocol(DeclContext *dc) const {
  ImplementsAttrProtocolRequest request{this, dc};
  if (dc->getASTContext().evaluator.hasCachedResult(request))
    return getProtocol(dc);
  return std::nullopt;
}

bool ImplementsAttr::isEquivalent(const ImplementsAttr *other,
                                  Decl *attachedTo) const {
  auto DC = attachedTo->getDeclContext();
  return getMemberName() == other->getMemberName()
          && getProtocol(DC) == other->getProtocol(DC);
}

DeclContext *CustomAttrOwner::getDeclContext() const {
  ASSERT(!Owner.isNull());
  if (auto *D = getAsDecl())
    return D->getDeclContext();

  return Owner.dyn_cast<DeclContext *>();
}

CustomAttr::CustomAttr(SourceLoc atLoc, SourceRange range, TypeExpr *type,
                       CustomAttrOwner owner,
                       CustomAttributeInitializer *initContext,
                       ArgumentList *argList, bool implicit)
    : DeclAttribute(DeclAttrKind::Custom, atLoc, range, implicit),
      typeExpr(type), argList(argList), owner(owner), initContext(initContext) {
  assert(type);
  isArgUnsafeBit = false;
}

CustomAttr *CustomAttr::create(ASTContext &ctx, SourceLoc atLoc, TypeExpr *type,
                               CustomAttrOwner owner,
                               CustomAttributeInitializer *initContext,
                               ArgumentList *argList, bool implicit) {
  assert(type);
  SourceRange range(atLoc, type->getSourceRange().End);
  if (argList)
    range.End = argList->getEndLoc();

  return new (ctx)
      CustomAttr(atLoc, range, type, owner, initContext, argList, implicit);
}

void CustomAttr::attachToDeclImpl(Decl *D) {
  // Prefer to set the parent PatternBindingDecl as the owner for a VarDecl,
  // this ensures we can handle PBDs with multiple vars.
  if (auto *VD = dyn_cast<VarDecl>(D)) {
    if (auto *PBD = VD->getParentPatternBinding())
      D = PBD;
  }
  owner = D;
}

std::pair<UnqualifiedIdentTypeRepr *, DeclRefTypeRepr *>
CustomAttr::destructureMacroRef() {
  TypeRepr *typeRepr = getTypeRepr();
  if (!typeRepr)
    return {nullptr, nullptr};
  if (auto *unqualIdentType = dyn_cast<UnqualifiedIdentTypeRepr>(typeRepr))
    return {nullptr, unqualIdentType};
  if (auto *qualIdentType = dyn_cast<QualifiedIdentTypeRepr>(typeRepr)) {
    if (auto *base =
            dyn_cast<UnqualifiedIdentTypeRepr>(qualIdentType->getBase())) {
      if (!base->hasGenericArgList())
        return {base, qualIdentType};
    }
  }
  return {nullptr, nullptr};
}

ASTContext &CustomAttr::getASTContext() const {
  return getOwner().getDeclContext()->getASTContext();
}

bool CustomAttr::shouldPreferPropertyWrapperOverMacro() const {
  // If we have a VarDecl in a local context, prefer to use a property wrapper
  // if one exists. This is necessary since we don't properly support peer
  // declarations in local contexts, so want to use a property wrapper if one
  // exists.
  if (auto *D = getOwner().getAsDecl()) {
    if ((isa<VarDecl>(D) || isa<PatternBindingDecl>(D)) &&
        D->getDeclContext()->isLocalContext()) {
      return true;
    }
  }
  return false;
}

NominalTypeDecl *CustomAttr::getNominalDecl() const {
  auto &eval = getASTContext().evaluator;
  auto *mutThis = const_cast<CustomAttr *>(this);
  return evaluateOrDefault(eval, CustomAttrNominalRequest{mutThis}, nullptr);
}

MacroDecl *CustomAttr::getResolvedMacro() const {
  auto &eval = getASTContext().evaluator;
  auto *mutThis = const_cast<CustomAttr *>(this);

  auto declRef =
      evaluateOrDefault(eval, ResolveMacroRequest{mutThis}, ConcreteDeclRef());

  return dyn_cast_or_null<MacroDecl>(declRef.getDecl());
}

TypeRepr *CustomAttr::getTypeRepr() const { return typeExpr->getTypeRepr(); }
Type CustomAttr::getType() const { return typeExpr->getInstanceType(); }

void CustomAttr::resetTypeInformation(TypeExpr *info) { typeExpr = info; }

void CustomAttr::setType(Type ty) {
  assert(ty);
  typeExpr->setType(MetatypeType::get(ty));
}

bool CustomAttr::isArgUnsafe() const {
  if (isArgUnsafeBit)
    return true;

  auto args = getArgs();
  if (!args)
    return false;

  auto *unary = args->getUnlabeledUnaryExpr();
  if (!unary)
    return false;

  if (auto declRef = dyn_cast<UnresolvedDeclRefExpr>(unary)) {
    if (declRef->getName().isSimpleName("unsafe"))
      isArgUnsafeBit = true;
  }

  return isArgUnsafeBit;
}

bool CustomAttr::isEquivalent(const CustomAttr *other, Decl *attachedTo) const {
  // For the sake of both @abi checking and implementability, we're going to
  // ignore expressions in the arguments and initializer.

  return isArgUnsafe() == other->isArgUnsafe() && eqTypes(getType(),
                                                          other->getType());
}

MacroRoleAttr::MacroRoleAttr(SourceLoc atLoc, SourceRange range,
                             MacroSyntax syntax, SourceLoc lParenLoc,
                             MacroRole role,
                             ArrayRef<MacroIntroducedDeclName> names,
                             ArrayRef<Expr *> conformances, SourceLoc rParenLoc,
                             bool implicit)
    : DeclAttribute(DeclAttrKind::MacroRole, atLoc, range, implicit),
      syntax(syntax), role(role), numNames(names.size()),
      numConformances(conformances.size()), lParenLoc(lParenLoc),
      rParenLoc(rParenLoc) {
  auto *trailingNamesBuffer = getTrailingObjects<MacroIntroducedDeclName>();
  std::uninitialized_copy(names.begin(), names.end(), trailingNamesBuffer);

  auto *trailingConformancesBuffer = getTrailingObjects<Expr *>();
  std::uninitialized_copy(conformances.begin(), conformances.end(),
                          trailingConformancesBuffer);
}

MacroRoleAttr *MacroRoleAttr::create(ASTContext &ctx, SourceLoc atLoc,
                                     SourceRange range, MacroSyntax syntax,
                                     SourceLoc lParenLoc, MacroRole role,
                                     ArrayRef<MacroIntroducedDeclName> names,
                                     ArrayRef<Expr *> conformances,
                                     SourceLoc rParenLoc, bool implicit) {
  unsigned size = totalSizeToAlloc<MacroIntroducedDeclName, Expr *>(
      names.size(), conformances.size());
  auto *mem = ctx.Allocate(size, alignof(MacroRoleAttr));
  return new (mem) MacroRoleAttr(atLoc, range, syntax, lParenLoc, role, names,
                                 conformances, rParenLoc, implicit);
}

ArrayRef<MacroIntroducedDeclName> MacroRoleAttr::getNames() const {
  return {
    getTrailingObjects<MacroIntroducedDeclName>(),
    numNames
  };
}

ArrayRef<Expr *> MacroRoleAttr::getConformances() const {
  return {getTrailingObjects<Expr *>(), numConformances};
}

MutableArrayRef<Expr *> MacroRoleAttr::getConformances() {
  return {getTrailingObjects<Expr *>(), numConformances};
}

bool MacroRoleAttr::hasNameKind(MacroIntroducedDeclNameKind kind) const {
  return llvm::find_if(getNames(), [kind](MacroIntroducedDeclName name) {
    return name.getKind() == kind;
  }) != getNames().end();
}

StringRef ExternAttr::getCName(const ValueDecl *D) const {
  if (auto cName = this->Name)
    return cName.value();
  // If no name was specified, fall back on the Swift base name without mangling.
  // Base name is always available and non-empty for functions and variables.
  return D->getBaseIdentifier().str();
}

const ExternAttr *ExternAttr::find(const DeclAttributes &attrs, ExternKind kind) {
  for (const DeclAttribute *attr : attrs) {
    if (auto *externAttr = dyn_cast<ExternAttr>(attr)) {
      if (externAttr->getExternKind() == kind)
        return externAttr;
    }
  }
  return nullptr;
}

const DeclAttribute *
DeclAttributes::getEffectiveSendableAttr() const {
  const NonSendableAttr *assumedAttr = nullptr;

  for (auto attr : getAttributes<NonSendableAttr>()) {
    if (attr->Specificity == NonSendableKind::Specific)
      return attr;
    if (!assumedAttr)
      assumedAttr = attr;
  }

  if (auto sendableAttr = getAttribute<SendableAttr>())
    return sendableAttr;

  // ~Sendable on declarations imported from Objective-C.
  for (auto *attr : getAttributes<SynthesizedProtocolAttr>()) {
    if (attr->getProtocol()->isSpecificProtocol(KnownProtocolKind::Sendable) &&
        attr->isSuppressed())
      return nullptr;
  }

  return assumedAttr;
}

ArrayRef<VarDecl *> StorageRestrictionsAttr::getInitializesProperties(
    AccessorDecl *attachedTo) const {
  auto &ctx = attachedTo->getASTContext();
  return evaluateOrDefault(ctx.evaluator,
                           InitAccessorReferencedVariablesRequest{
                               const_cast<StorageRestrictionsAttr *>(this),
                               attachedTo, getInitializesNames()},
                           {});
}

ArrayRef<VarDecl *>
StorageRestrictionsAttr::getAccessesProperties(AccessorDecl *attachedTo) const {
  auto &ctx = attachedTo->getASTContext();
  return evaluateOrDefault(ctx.evaluator,
                           InitAccessorReferencedVariablesRequest{
                               const_cast<StorageRestrictionsAttr *>(this),
                               attachedTo, getAccessesNames()},
                           {});
}

AllowFeatureSuppressionAttr::AllowFeatureSuppressionAttr(
    SourceLoc atLoc, SourceRange range, bool implicit, bool inverted,
    ArrayRef<Identifier> features)
    : DeclAttribute(DeclAttrKind::AllowFeatureSuppression, atLoc, range,
                    implicit) {
  Bits.AllowFeatureSuppressionAttr.Inverted = inverted;
  Bits.AllowFeatureSuppressionAttr.NumFeatures = features.size();
  std::uninitialized_copy(features.begin(), features.end(),
                          getTrailingObjects());
}

AllowFeatureSuppressionAttr *AllowFeatureSuppressionAttr::create(
    ASTContext &ctx, SourceLoc atLoc, SourceRange range, bool implicit,
    bool inverted, ArrayRef<Identifier> features) {
  unsigned size = totalSizeToAlloc<Identifier>(features.size());
  auto *mem = ctx.Allocate(size, alignof(AllowFeatureSuppressionAttr));
  return new (mem)
      AllowFeatureSuppressionAttr(atLoc, range, implicit, inverted, features);
}

bool AllowFeatureSuppressionAttr::
isEquivalent(const AllowFeatureSuppressionAttr *other, Decl *attachedTo) const {
  if (getInverted() != other->getInverted())
    return false;

  return sameElements(getSuppressedFeatures(), other->getSuppressedFeatures());
}

LifetimeAttr *LifetimeAttr::create(ASTContext &context, SourceLoc atLoc,
                                   SourceRange baseRange, bool implicit,
                                   LifetimeEntry *entry, bool isUnderscored) {
  return new (context)
      LifetimeAttr(atLoc, baseRange, implicit, entry, isUnderscored);
}

std::string LifetimeAttr::getString() const {
  return (isUnderscored() ? std::string("@_lifetime")
                          : std::string("@lifetime")) +
         getLifetimeEntry()->getString();
}

bool LifetimeAttr::isEquivalent(const LifetimeAttr *other,
                                Decl *attachedTo) const {
  // FIXME: This is kind of cheesy.
  return getLifetimeEntry()->getString()
      == other->getLifetimeEntry()->getString();
}

void swift::simple_display(llvm::raw_ostream &out, const DeclAttribute *attr) {
  if (attr)
    attr->print(out);
}

static bool hasDeclAttribute(const LangOptions &langOpts,
                             llvm::StringRef attributeName) {
  std::optional<DeclAttrKind> kind =
      DeclAttribute::getAttrKindFromString(attributeName);
  if (!kind)
    return false;

  if (DeclAttribute::isUserInaccessible(*kind))
    return false;
  if (DeclAttribute::isDeclModifier(*kind))
    return false;
  if (DeclAttribute::shouldBeRejectedByParser(*kind))
    return false;
  if (DeclAttribute::isSilOnly(*kind))
    return false;
  if (DeclAttribute::isConcurrencyOnly(*kind))
    return false;

  if (auto feature = DeclAttribute::getRequiredFeature(*kind))
    if (!langOpts.hasFeature(*feature))
      return false;

  return true;
}

static bool hasTypeAttribute(const LangOptions &langOpts,
                             llvm::StringRef attributeName) {
  std::optional<TypeAttrKind> kind =
      TypeAttribute::getAttrKindFromString(attributeName);
  if (!kind)
    return false;

  if (TypeAttribute::isSilOnly(*kind))
    return false;

  return true;
}

bool swift::hasAttribute(const LangOptions &langOpts,
                         llvm::StringRef attributeName) {
  if (hasDeclAttribute(langOpts, attributeName))
    return true;

  if (hasTypeAttribute(langOpts, attributeName))
    return true;

  return false;
}