101 lines
3.5 KiB
Swift
101 lines
3.5 KiB
Swift
import SwiftSyntax
|
|
|
|
struct StrongIBOutletRule: ConfigurationProviderRule, SwiftSyntaxCorrectableRule, OptInRule {
|
|
var configuration = SeverityConfiguration(.warning)
|
|
|
|
init() {}
|
|
|
|
static let description = RuleDescription(
|
|
identifier: "strong_iboutlet",
|
|
name: "Strong IBOutlet",
|
|
description: "@IBOutlets shouldn't be declared as weak",
|
|
kind: .lint,
|
|
nonTriggeringExamples: [
|
|
wrapExample("@IBOutlet var label: UILabel?"),
|
|
wrapExample("weak var label: UILabel?")
|
|
],
|
|
triggeringExamples: [
|
|
wrapExample("@IBOutlet ↓weak var label: UILabel?"),
|
|
wrapExample("@IBOutlet ↓unowned var label: UILabel!"),
|
|
wrapExample("@IBOutlet ↓weak var textField: UITextField?")
|
|
],
|
|
corrections: [
|
|
wrapExample("@IBOutlet ↓weak var label: UILabel?"):
|
|
wrapExample("@IBOutlet var label: UILabel?"),
|
|
wrapExample("@IBOutlet ↓unowned var label: UILabel!"):
|
|
wrapExample("@IBOutlet var label: UILabel!"),
|
|
wrapExample("@IBOutlet ↓weak var textField: UITextField?"):
|
|
wrapExample("@IBOutlet var textField: UITextField?")
|
|
]
|
|
)
|
|
|
|
func makeVisitor(file: SwiftLintFile) -> ViolationsSyntaxVisitor {
|
|
Visitor(viewMode: .sourceAccurate)
|
|
}
|
|
|
|
func makeRewriter(file: SwiftLintFile) -> ViolationsSyntaxRewriter? {
|
|
Rewriter(
|
|
locationConverter: file.locationConverter,
|
|
disabledRegions: disabledRegions(file: file)
|
|
)
|
|
}
|
|
}
|
|
|
|
private extension StrongIBOutletRule {
|
|
final class Visitor: ViolationsSyntaxVisitor {
|
|
override func visitPost(_ node: VariableDeclSyntax) {
|
|
if let violationPosition = node.violationPosition {
|
|
violations.append(violationPosition)
|
|
}
|
|
}
|
|
}
|
|
|
|
private final class Rewriter: SyntaxRewriter, ViolationsSyntaxRewriter {
|
|
private(set) var correctionPositions: [AbsolutePosition] = []
|
|
let locationConverter: SourceLocationConverter
|
|
let disabledRegions: [SourceRange]
|
|
|
|
init(locationConverter: SourceLocationConverter, disabledRegions: [SourceRange]) {
|
|
self.locationConverter = locationConverter
|
|
self.disabledRegions = disabledRegions
|
|
}
|
|
|
|
override func visit(_ node: VariableDeclSyntax) -> DeclSyntax {
|
|
guard let violationPosition = node.violationPosition,
|
|
let weakOrUnownedModifier = node.weakOrUnownedModifier,
|
|
let modifiers = node.modifiers,
|
|
!node.isContainedIn(regions: disabledRegions, locationConverter: locationConverter)
|
|
else {
|
|
return super.visit(node)
|
|
}
|
|
|
|
let newModifiers = ModifierListSyntax(modifiers.filter { $0 != weakOrUnownedModifier })
|
|
let newNode = node.withModifiers(newModifiers)
|
|
correctionPositions.append(violationPosition)
|
|
return super.visit(newNode)
|
|
}
|
|
}
|
|
}
|
|
|
|
private extension VariableDeclSyntax {
|
|
var violationPosition: AbsolutePosition? {
|
|
guard let keyword = weakOrUnownedKeyword, isIBOutlet else {
|
|
return nil
|
|
}
|
|
|
|
return keyword.positionAfterSkippingLeadingTrivia
|
|
}
|
|
|
|
var weakOrUnownedKeyword: TokenSyntax? {
|
|
weakOrUnownedModifier?.name
|
|
}
|
|
}
|
|
|
|
private func wrapExample(_ text: String, file: StaticString = #file, line: UInt = #line) -> Example {
|
|
return Example("""
|
|
class ViewController: UIViewController {
|
|
\(text)
|
|
}
|
|
""", file: file, line: line)
|
|
}
|