swift-mirror/test/PrintAsObjC/extensions.swift

// Please keep this file in alphabetical order!

// RUN: %empty-directory(%t)
// RUN: %target-swift-frontend(mock-sdk: %clang-importer-sdk) -emit-module -o %t %s -disable-objc-attr-requires-foundation-module -Xllvm -sil-disable-pass=MandatoryARCOpts
// RUN: %target-swift-frontend(mock-sdk: %clang-importer-sdk) -parse-as-library %t/extensions.swiftmodule -typecheck -verify -emit-objc-header-path %t/extensions.h -import-objc-header %S/../Inputs/empty.h -disable-objc-attr-requires-foundation-module -Xllvm -sil-disable-pass=MandatoryARCOpts
// RUN: %FileCheck %s --input-file %t/extensions.h
// RUN: %FileCheck --check-prefix=NEGATIVE %s --input-file %t/extensions.h
// RUN: %check-in-clang %t/extensions.h

// REQUIRES: objc_interop

import Foundation
import AppKit
import objc_generics

// CHECK-NOT: AppKit

// CHECK-LABEL: @interface A1{{$}}
// CHECK-NEXT: init
// CHECK-NEXT: @end
@objc @objcMembers class A1 {}
// CHECK-EMPTY:
// NEGATIVE-NOT: @interface A1 (SWIFT_EXTENSION(extensions))
extension A1 {}

// CHECK-LABEL: @interface A2{{$}}
// CHECK-NEXT: init
// CHECK-NEXT: @end
// CHECK-EMPTY:
// CHECK-NEXT: @interface A2 (SWIFT_EXTENSION(extensions))
// CHECK-DAG: @property (nonatomic, readonly) NSInteger some;
// CHECK-NEXT: @end
// CHECK-EMPTY:
extension A2 {
  @objc var some: Int { return 1 }
}
@objc @objcMembers class A2 {}

// CHECK-NEXT: SWIFT_CLASS
// CHECK-NEXT: @interface A3{{$}}
// CHECK-NEXT: init
// CHECK-NEXT: @end
// CHECK-EMPTY:
@objc @objcMembers class A3 {}

// CHECK-NEXT: @interface A3 (SWIFT_EXTENSION(extensions))
// CHECK-NEXT: @property (nonatomic, readonly) NSInteger more;
// CHECK-NEXT: @end
// CHECK-EMPTY:
// CHECK-NEXT: @interface A3 (SWIFT_EXTENSION(extensions))
// CHECK-NEXT: @property (nonatomic, readonly) NSInteger some;
// CHECK-NEXT: @end
// CHECK-EMPTY:
extension A3 {
  @objc var some: Int { return 1 }
}
extension A3 {
  @objc var more: Int { return 10 }
}

// CHECK-NEXT: SWIFT_CLASS
// CHECK-NEXT: @interface A4{{$}}
// CHECK-NEXT: init
// CHECK-NEXT: @end
// CHECK-EMPTY:
@objc @objcMembers class A4 {}

// CHECK-NEXT: @interface A4 (SWIFT_EXTENSION(extensions))
// CHECK-NEXT: @end
// CHECK-EMPTY:
extension A4 {
  // CHECK-NEXT: SWIFT_CLASS
  // CHECK-NEXT: @interface Inner
  // CHECK-NEXT: init
  // CHECK-NEXT: @end
  // CHECK-EMPTY:
  @objc @objcMembers class Inner {}
}

// CHECK-NEXT: SWIFT_CLASS
// CHECK-NEXT: @interface A5{{$}}
// CHECK-NEXT: init
// CHECK-NEXT: @end
// CHECK-EMPTY:
@objc @objcMembers class A5 {}

// NEGATIVE-NOT: @interface A5 (SWIFT_EXTENSION(extensions))
extension A5 {
  var notObjC: NotObjC { return NotObjC() }
}

// Check that two otherwise tied extensions will print in alphabetical
// order by first member with a differing Swift name.

// CHECK-NEXT: SWIFT_CLASS
// CHECK-NEXT: @interface A6
@objc class A6 {}

extension A6 {
  @objc(skippedBool:) func skipped(_: Bool) {}
  @objc func def() {}
}
extension A6 {
  @objc(skippedInt:) func skipped(_: Int) {}
  @objc func abc() {}
}
// CHECK: @interface A6 (SWIFT_EXTENSION(extensions))
// CHECK-NEXT: - (void)skippedInt:
// CHECK-NEXT: - (void)abc
// CHECK-NEXT: @end
// CHECK-EMPTY:
// CHECK-NEXT: @interface A6 (SWIFT_EXTENSION(extensions))
// CHECK-NEXT: - (void)skippedBool:
// CHECK-NEXT: - (void)def
// CHECK-NEXT: @end
// CHECK-EMPTY:

// CHECK-NEXT: SWIFT_CLASS
// CHECK-NEXT: @interface CustomName{{$}}
// CHECK-NEXT: init
// CHECK-NEXT: @end
// CHECK-EMPTY:
@objc(CustomName)
@objcMembers
class ClassWithCustomName {
}

// CHECK-NEXT: @interface CustomName (SWIFT_EXTENSION(extensions))
// CHECK-NEXT: - (void)foo;
// CHECK-NEXT: @end
// CHECK-EMPTY:
extension ClassWithCustomName {
  @objc func foo() {}
}

// NEGATIVE-NOT: CGColor
extension CGColor {
  func anyOldMethod() {}
}

// CHECK-NEXT: @interface GenericClass<T> (SWIFT_EXTENSION(extensions))
// CHECK-NEXT: - (void)bar;
// CHECK-NEXT: @end
// CHECK-EMPTY:
extension GenericClass {
  @objc func bar() {}
}

// NEGATIVE-NOT: NotObjC
class NotObjC {}
extension NotObjC {}

// NEGATIVE-NOT: @interface NSObject{{$}}
// NEGATIVE-NOT: @class NSObject
// CHECK-LABEL: @interface NSObject (SWIFT_EXTENSION(extensions))
// CHECK-NEXT: @property (nonatomic, readonly) NSInteger some;
// CHECK-NEXT: @end
// CHECK-EMPTY:
extension NSObject {
  @objc var some: Int { return 1 }
}

// NEGATIVE-NOT: @class NSString;
// CHECK: @class NSColor;
// CHECK-NEXT: @interface NSString (SWIFT_EXTENSION(extensions))
// CHECK-NEXT: - (void)test;
// CHECK-NEXT: + (void)test2;
// CHECK-NEXT: + (NSString * _Nullable)fromColor:(NSColor * _Nonnull)color SWIFT_WARN_UNUSED_RESULT;
// CHECK-NEXT: @end
// CHECK-EMPTY:
extension NSString {
  @objc func test() {}
  @objc class func test2() {}

  @objc class func fromColor(_ color: NSColor) -> NSString? { return nil; }
}

// CHECK-NEXT: @protocol Pettable;
// CHECK-NEXT: @interface PettableContainer<T> (SWIFT_EXTENSION(extensions))
// CHECK-NEXT: - (PettableContainer<T> * _Nonnull)duplicate SWIFT_WARN_UNUSED_RESULT;
// CHECK-NEXT: - (PettableContainer<T> * _Nonnull)duplicate2 SWIFT_WARN_UNUSED_RESULT;
// CHECK-NEXT: - (PettableContainer<PettableOverextendedMetaphor *> * _Nonnull)duplicate3 SWIFT_WARN_UNUSED_RESULT;
// CHECK-NEXT: - (T _Nonnull)extract SWIFT_WARN_UNUSED_RESULT;
// CHECK-NEXT: - (T _Nullable)extract2 SWIFT_WARN_UNUSED_RESULT;
// CHECK-NEXT: @end
// CHECK-EMPTY:
extension PettableContainer {
  @objc func duplicate() -> PettableContainer { fatalError() }
  @objc func duplicate2() -> PettableContainer<T> { fatalError() }
  @objc func duplicate3() -> PettableContainer<PettableOverextendedMetaphor> { fatalError() }
  @objc func extract() -> T { fatalError() }
  @objc func extract2() -> T? { fatalError() }
}