[SourceKit] Print custom attributes in interface-gen requests

Custom attributes were not printed because they are marked
'UserInaccesible'.

* Make CustomAttr 'RejectByParser' instead of 'UserInaccessible'
* Remove special treatment for Result Builder attributes
* Load implicit modules in module/header interface gen requests

rdar://79927502

Author: rintaro

include/swift/AST/Attr.def

@@ -494,7 +494,7 @@ SIMPLE_DECL_ATTR(_implementationOnly, ImplementationOnly,
   ABIStableToAdd | ABIStableToRemove | APIStableToAdd | APIStableToRemove,
   84)
 DECL_ATTR(_custom, Custom,
-  OnAnyDecl | UserInaccessible |
+  OnAnyDecl | RejectByParser |
   ABIStableToAdd | ABIStableToRemove | APIStableToAdd | APIStableToRemove,
   85)
 SIMPLE_DECL_ATTR(propertyWrapper, PropertyWrapper,

lib/AST/Attr.cpp

@@ -685,17 +685,11 @@ void DeclAttributes::print(ASTPrinter &Printer, const PrintOptions &Options,
   AttributeVector attributes;
   AttributeVector modifiers;
 
-  CustomAttr *FuncBuilderAttr = nullptr;
-  if (auto *VD = dyn_cast_or_null<ValueDecl>(D)) {
-    FuncBuilderAttr = VD->getAttachedResultBuilder();
-  }
   for (auto DA : llvm::reverse(FlattenedAttrs)) {
     // Always print result builder attribute.
-    bool isResultBuilderAttr = DA == FuncBuilderAttr;
     if (!Options.PrintImplicitAttrs && DA->isImplicit())
       continue;
     if (!Options.PrintUserInaccessibleAttrs &&
-        !isResultBuilderAttr &&
         DeclAttribute::isUserInaccessible(DA->getKind()))
       continue;
     if (Options.excludeAttrKind(DA->getKind()))

test/IDE/print_clang_objc_effectful_properties.swift

@@ -32,11 +32,11 @@
 // CHECK-NEXT:  var fromNullableHandler: String { get async }
 
 // CHECK:       @available(*, renamed: "getter:mainDogProp()")
-// CHECK-NEXT:  func getMainDog(_ completion: @escaping @Sendable (String) -> Void)
+// CHECK-NEXT:  func getMainDog(_ completion: @escaping @MainActor (String) -> Void)
 // CHECK-NEXT:  var mainDogProp: String { get async }
 
 // CHECK:       @available(*, renamed: "regularMainDog()")
-// CHECK-NEXT:  func regularMainDog(_ completion: @escaping @Sendable (String) -> Void)
+// CHECK-NEXT:  func regularMainDog(_ completion: @escaping @MainActor (String) -> Void)
 // CHECK-NEXT:  @discardableResult
 // CHECK-NEXT:  func regularMainDog() async -> String
 // CHECK: }

test/IDE/print_swift_module.swift

@@ -28,7 +28,7 @@ public func returnsAlias() -> Alias<Int> {
 }
 
 @resultBuilder
-struct BridgeBuilder {
+public struct BridgeBuilder {
     static func buildBlock(_: Any...) {}
 }
 

test/SourceKit/Inputs/concurrency/gen_concurrency.swift

@@ -2,4 +2,7 @@ class ClassWithAsyncAndHandler {
   @available(*, renamed: "foo(_:)")
   func foo(_ operation: String, completionHandler handler: @escaping (Int) -> Void) {}
   func foo(_ operation: String) async -> Int { 0 }
+
+  @MainActor
+  func mainActorMethod() {}
 }

test/SourceKit/Inputs/concurrency/header_concurrency.h

@@ -2,4 +2,6 @@
 
 @interface ClassWithHandlerMethod
 -(void)methodWithHandler:(NSString *)operation completionHandler:(void (^)(NSInteger))handler;
+
+-(void)mainActorMethod __attribute__((__swift_attr__("@UIActor")));
 @end

test/SourceKit/InterfaceGen/gen_custom_attributes.swift

@@ -0,0 +1,59 @@
+// REQUIRES: concurrency
+
+// RUN: %empty-directory(%t)
+
+// RUN: %sourcekitd-test -req=interface-gen %s -- %s -target %target-triple -module-name MyModule | %FileCheck %s --check-prefix=SWIFTSOURCE
+
+// RUN: %target-swift-frontend -emit-module -module-name MyModule -o %t/MyModule.swiftmodule %s
+// RUN: %sourcekitd-test -req=interface-gen -module MyModule -- -target %target-triple -I %t | %FileCheck %s --check-prefix=SWIFTMODULE
+// RUN: %sourcekitd-test -req=doc-info -module MyModule -- -target %target-triple -I %t | %FileCheck %s --check-prefix=DOCINFO
+
+@propertyWrapper
+public struct OrLess {
+  private var threshold: Int
+  private var number: Int = 0
+  init(wrappedValue: Int, threshold: Int) {
+    self.threshold = threshold
+    self.wrappedValue = wrappedValue
+  }
+  public var wrappedValue: Int {
+    get { return number }
+    set { number = min(newValue, 12) }
+  }
+}
+
+@resultBuilder
+public struct IntBuilder {
+    public static func buildBlock(_ val: Int) -> Int { val }
+}
+
+public struct TestStruct {
+
+  @MainActor public func mainActorMethod() {}
+  @MainActor(unsafe) public func mainActorUnsafeMethod() {}
+
+  @OrLess(threshold: 12) public var twelveOrLess = 13
+
+  @IntBuilder public var intValue: Int { 1 }
+}
+
+// SWIFTSOURCE: public struct TestStruct {
+// SWIFTSOURCE:     @MainActor public func mainActorMethod()
+// SWIFTSOURCE:     @MainActor public func mainActorUnsafeMethod()
+// SWIFTSOURCE:     @MyModule.OrLess public var twelveOrLess: Int { get set }
+// SWIFTSOURCE:     @MyModule.IntBuilder public var intValue: Int { get }
+// SWIFTSOURCE: }
+
+// SWIFTMODULE: public struct TestStruct {
+// SWIFTMODULE:     @MainActor public func mainActorMethod()
+// SWIFTMODULE:     @MainActor public func mainActorUnsafeMethod()
+// SWIFTMODULE:     @MyModule.OrLess public var twelveOrLess: Int
+// SWIFTMODULE:     @MyModule.IntBuilder public var intValue: Int { get }
+// SWIFTMODULE: }
+
+// DOCINFO: struct TestStruct {
+// DOCINFO:     @MainActor func mainActorMethod()
+// DOCINFO:     @MainActor func mainActorUnsafeMethod()
+// DOCINFO:     @MyModule.OrLess var twelveOrLess: Int
+// DOCINFO:     @MyModule.IntBuilder var intValue: Int { get }
+// DOCINFO: }

test/SourceKit/InterfaceGen/gen_objc_concurrency.swift

@@ -11,10 +11,15 @@
 // SWIFT-GEN-INTERFACE-NEXT:    internal func foo(_ operation: String, completionHandler handler: @escaping (Int) -> Void)
 // SWIFT-GEN-INTERFACE:         internal func foo(_ operation: String) async -> Int
 
+// SWIFT-GEN-INTERFACE:         @MainActor internal func mainActorMethod()
+
+
 // RUN: %sourcekitd-test -req=interface-gen -using-swift-args -header %S/../Inputs/concurrency/header_concurrency.h -- %s -Xfrontend -enable-objc-interop -import-objc-header %S/../Inputs/concurrency/header_concurrency.h -sdk %clang-importer-sdk | %FileCheck %s --check-prefix=OBJC-GEN-INTERFACE
 
 // But don't print @available if it was implicitly added to an imported Clang decl (rdar://76685011).
 // OBJC-GEN-INTERFACE-LABEL: class ClassWithHandlerMethod {
 // OBJC-GEN-INTERFACE-NOT:     @available
 // OBJC-GEN-INTERFACE:         func method(withHandler operation: String!, completionHandler handler: (@Sendable (Int) -> Void)!)
 // OBJC-GEN-INTERFACE:         func method(withHandler operation: String!) async -> Int
+
+// OBJC-GEN-INTERFACE:         @MainActor open func mainActorMethod()

test/attr/attributes.swift

@@ -342,3 +342,5 @@ enum E1 {
 
   func foo() -> @_nonEphemeral UnsafeMutableRawPointer? { return nil } // expected-error {{attribute can only be applied to declarations, not types}}
 }
+
+@_custom func testCustomAttribute() {} // expected-error {{unknown attribute '_custom'}}

tools/SourceKit/lib/SwiftLang/SwiftDocSupport.cpp

@@ -1086,6 +1086,12 @@ static bool reportModuleDocInfo(CompilerInvocation Invocation,
   ASTContext &Ctx = CI.getASTContext();
   registerIDERequestFunctions(Ctx.evaluator);
 
+  // Load implict imports so that Clang importer can use it.
+  for (auto unloadedImport :
+       CI.getMainModule()->getImplicitImportInfo().AdditionalUnloadedImports) {
+    (void)Ctx.getModule(unloadedImport.module.getModulePath());
+  }
+
   SourceTextInfo IFaceInfo;
   if (getModuleInterfaceInfo(Ctx, ModuleName, IFaceInfo))
     return true;