[ASTPrinter] Print 'async' as a keyword

rdar://problem/73426591

Author: rintaro

lib/AST/ASTPrinter.cpp

@@ -2902,8 +2902,10 @@ void PrintAST::printFunctionParameters(AbstractFunctionDecl *AFD) {
   printParameterList(BodyParams, parameterListTypes,
                      AFD->argumentNameIsAPIByDefault());
 
-  if (AFD->hasAsync())
-    Printer << " " << "async";
+  if (AFD->hasAsync()) {
+    Printer << " ";
+    Printer.printKeyword("async", Options);
+  }
 
   if (AFD->hasThrows()) {
     if (AFD->getAttrs().hasAttribute<RethrowsAttr>())
@@ -4331,8 +4333,10 @@ class TypePrinter : public TypeVisitor<TypePrinter> {
     // If we're stripping argument labels from types, do it when printing.
     visitAnyFunctionTypeParams(T->getParams(), /*printLabels*/false);
 
-    if (T->isAsync())
-      Printer << " " << "async";
+    if (T->isAsync()) {
+      Printer << " ";
+      Printer.printKeyword("async", Options);
+    }
 
     if (T->isThrowing())
       Printer << " " << tok::kw_throws;
@@ -4374,8 +4378,10 @@ class TypePrinter : public TypeVisitor<TypePrinter> {
 
    visitAnyFunctionTypeParams(T->getParams(), /*printLabels*/true);
 
-    if (T->isAsync())
-      Printer << " " << "async";
+    if (T->isAsync()) {
+      Printer << " ";
+      Printer.printKeyword("async", Options);
+    }
 
     if (T->isThrowing())
       Printer << " " << tok::kw_throws;

lib/AST/TypeRepr.cpp

@@ -225,6 +225,10 @@ void FunctionTypeRepr::printImpl(ASTPrinter &Printer,
                                  const PrintOptions &Opts) const {
   Printer.callPrintStructurePre(PrintStructureKind::FunctionType);
   printTypeRepr(ArgsTy, Printer, Opts);
+  if (isAsync()) {
+    Printer << " ";
+    Printer.printKeyword("async", Opts);
+  }
   if (isThrowing()) {
     Printer << " ";
     Printer.printKeyword("throws", Opts);

test/SourceKit/CursorInfo/cursor_info_concurrency.swift

@@ -0,0 +1,45 @@
+// BEGIN MyModule.swift
+public actor class MyActor {
+  public func asyncFunc(fn: () async -> Void) async throws {}
+}
+
+func test(act: MyActor) async throws {
+    try await act.asyncFunc {}
+}
+
+// BEGIN App.swift
+import MyModule
+
+func test(act: MyActor) async throws {
+    try await act.asyncFunc {}
+}
+
+// RUN: %empty-directory(%t)
+// RUN: %{python} %utils/split_file.py -o %t %s
+
+// RUN: %empty-directory(%t/Modules)
+// RUN: %target-swift-frontend -emit-module -o %t/Modules/MyModule.swiftmodule -module-name MyModule %t/MyModule.swift -enable-experimental-concurrency 
+
+// RUN: %sourcekitd-test -req=cursor -pos=1:20 %t/MyModule.swift -- %t/MyModule.swift -Xfrontend -enable-experimental-concurrency | %FileCheck -check-prefix=ACTOR %s
+// RUN: %sourcekitd-test -req=cursor -pos=2:15 %t/MyModule.swift -- %t/MyModule.swift -Xfrontend -enable-experimental-concurrency | %FileCheck -check-prefix=FUNC %s
+// RUN: %sourcekitd-test -req=cursor -pos=5:16 %t/MyModule.swift -- %t/MyModule.swift -Xfrontend -enable-experimental-concurrency | %FileCheck -check-prefix=ACTOR %s
+// RUN: %sourcekitd-test -req=cursor -pos=6:19  %t/MyModule.swift -- %t/MyModule.swift -Xfrontend -enable-experimental-concurrency | %FileCheck -check-prefix=FUNC %s
+
+// ACTOR: <Declaration>public actor class MyActor</Declaration>
+// ACTOR: <decl.class><syntaxtype.keyword>public</syntaxtype.keyword> <syntaxtype.keyword>actor</syntaxtype.keyword> <syntaxtype.keyword>class</syntaxtype.keyword> <decl.name>MyActor</decl.name></decl.class>
+
+// FUNC: <Declaration>public func asyncFunc(fn: () async -&gt; <Type usr="s:s4Voida">Void</Type>) async throws</Declaration>
+// FUNC: <decl.function.method.instance><syntaxtype.keyword>public</syntaxtype.keyword> <syntaxtype.keyword>func</syntaxtype.keyword> <decl.name>asyncFunc</decl.name>(<decl.var.parameter><decl.var.parameter.argument_label>fn</decl.var.parameter.argument_label>: <decl.var.parameter.type>() <syntaxtype.keyword>async</syntaxtype.keyword> -&gt; <decl.function.returntype><ref.typealias usr="s:s4Voida">Void</ref.typealias></decl.function.returntype></decl.var.parameter.type></decl.var.parameter>) <syntaxtype.keyword>async</syntaxtype.keyword> <syntaxtype.keyword>throws</syntaxtype.keyword></decl.function.method.instance>
+
+// RUN: %sourcekitd-test -req=cursor -pos=3:16 %t/App.swift -- %t/App.swift -I %t/Modules -Xfrontend -enable-experimental-concurrency | %FileCheck -check-prefix=ACTOR_XMOD %s
+// RUN: %sourcekitd-test -req=cursor -pos=4:19 %t/App.swift -- %t/App.swift -I %t/Modules -Xfrontend -enable-experimental-concurrency | %FileCheck -check-prefix=FUNC_XMOD %s
+
+// ACTOR_XMOD: <Declaration>actor class MyActor</Declaration>
+// ACTOR_XMOD: <decl.class><syntaxtype.keyword>actor</syntaxtype.keyword> <syntaxtype.keyword>class</syntaxtype.keyword> <decl.name>MyActor</decl.name></decl.class>
+
+// FUNC_XMOD: <Declaration>func asyncFunc(fn: () async -&gt; <Type usr="s:s4Voida">Void</Type>) async throws</Declaration>
+// FUNC_XMOD: <decl.function.method.instance><syntaxtype.keyword>func</syntaxtype.keyword> <decl.name>asyncFunc</decl.name>(<decl.var.parameter><decl.var.parameter.argument_label>fn</decl.var.parameter.argument_label>: <decl.var.parameter.type>() <syntaxtype.keyword>async</syntaxtype.keyword> -&gt; <decl.function.returntype><ref.typealias usr="s:s4Voida">Void</ref.typealias></decl.function.returntype></decl.var.parameter.type></decl.var.parameter>) <syntaxtype.keyword>async</syntaxtype.keyword> <syntaxtype.keyword>throws</syntaxtype.keyword></decl.function.method.instance>
+
+
+
+