61 lines
2.4 KiB
Swift
61 lines
2.4 KiB
Swift
import SourceKittenFramework
|
|
|
|
public struct OverrideInExtensionRule: ConfigurationProviderRule, AutomaticTestableRule {
|
|
public var configuration = SeverityConfiguration(.warning)
|
|
|
|
public init() {}
|
|
|
|
public static let description = RuleDescription(
|
|
identifier: "override_in_extension",
|
|
name: "Override in Extension",
|
|
description: "Extensions shouldn't override declarations.",
|
|
kind: .lint,
|
|
isOptIn: true,
|
|
nonTriggeringExamples: [
|
|
"extension Person {\n var age: Int { return 42 }\n}\n",
|
|
"extension Person {\n func celebrateBirthday() {}\n}\n",
|
|
"class Employee: Person {\n override func celebrateBirthday() {}\n}\n",
|
|
"class Foo: NSObject {}\n" +
|
|
"extension Foo {\n" +
|
|
" override var description: String { return \"\" }\n" +
|
|
"}\n",
|
|
"struct Foo {\n" +
|
|
" class Bar: NSObject {}\n" +
|
|
"}\n" +
|
|
"extension Foo.Bar {\n" +
|
|
" override var description: String { return \"\" }\n" +
|
|
"}\n"
|
|
],
|
|
triggeringExamples: [
|
|
"extension Person {\n override ↓var age: Int { return 42 }\n}\n",
|
|
"extension Person {\n override ↓func celebrateBirthday() {}\n}\n"
|
|
]
|
|
)
|
|
|
|
public func validate(file: File) -> [StyleViolation] {
|
|
let collector = NamespaceCollector(dictionary: file.structure.dictionary)
|
|
let elements = collector.findAllElements(of: [.class, .struct, .enum, .extension])
|
|
|
|
let susceptibleNames = Set(elements.compactMap { $0.kind == .class ? $0.name : nil })
|
|
|
|
return elements
|
|
.filter { $0.kind == .extension && !susceptibleNames.contains($0.name) }
|
|
.flatMap { element in
|
|
return element.dictionary.substructure.compactMap { element -> Int? in
|
|
guard element.kind.flatMap(SwiftDeclarationKind.init) != nil,
|
|
element.enclosedSwiftAttributes.contains(.override),
|
|
let offset = element.offset else {
|
|
return nil
|
|
}
|
|
|
|
return offset
|
|
}
|
|
}
|
|
.map {
|
|
StyleViolation(ruleDescription: type(of: self).description,
|
|
severity: configuration.severity,
|
|
location: Location(file: file, byteOffset: $0))
|
|
}
|
|
}
|
|
}
|