mirror of
https://github.com/realm/SwiftLint.git
synced 2026-05-07 20:12:49 +00:00
62 lines
2.4 KiB
Swift
62 lines
2.4 KiB
Swift
import SwiftSyntax
|
|
|
|
@SwiftSyntaxRule(optIn: true)
|
|
struct LastWhereRule: Rule {
|
|
var configuration = SeverityConfiguration<Self>(.warning)
|
|
|
|
static let description = RuleDescription(
|
|
identifier: "last_where",
|
|
name: "Last Where",
|
|
description: "Prefer using `.last(where:)` over `.filter { }.last` in collections",
|
|
kind: .performance,
|
|
nonTriggeringExamples: [
|
|
Example("kinds.filter(excludingKinds.contains).isEmpty && kinds.last == .identifier"),
|
|
Example("myList.last(where: { $0 % 2 == 0 })"),
|
|
Example("match(pattern: pattern).filter { $0.last == .identifier }"),
|
|
Example("(myList.filter { $0 == 1 }.suffix(2)).last"),
|
|
Example(#"collection.filter("stringCol = '3'").last"#),
|
|
],
|
|
triggeringExamples: [
|
|
Example("↓myList.filter { $0 % 2 == 0 }.last"),
|
|
Example("↓myList.filter({ $0 % 2 == 0 }).last"),
|
|
Example("↓myList.map { $0 + 1 }.filter({ $0 % 2 == 0 }).last"),
|
|
Example("↓myList.map { $0 + 1 }.filter({ $0 % 2 == 0 }).last?.something()"),
|
|
Example("↓myList.filter(someFunction).last"),
|
|
Example("↓myList.filter({ $0 % 2 == 0 })\n.last"),
|
|
Example("(↓myList.filter { $0 == 1 }).last"),
|
|
]
|
|
)
|
|
}
|
|
|
|
private extension LastWhereRule {
|
|
final class Visitor: ViolationsSyntaxVisitor<ConfigurationType> {
|
|
override func visitPost(_ node: MemberAccessExprSyntax) {
|
|
guard
|
|
node.declName.baseName.text == "last",
|
|
let functionCall = node.base?.asFunctionCall,
|
|
let calledExpression = functionCall.calledExpression.as(MemberAccessExprSyntax.self),
|
|
calledExpression.declName.baseName.text == "filter",
|
|
!functionCall.arguments.contains(where: \.expression.shouldSkip)
|
|
else {
|
|
return
|
|
}
|
|
|
|
violations.append(functionCall.positionAfterSkippingLeadingTrivia)
|
|
}
|
|
}
|
|
}
|
|
|
|
private extension ExprSyntax {
|
|
var shouldSkip: Bool {
|
|
if `is`(StringLiteralExprSyntax.self) {
|
|
return true
|
|
}
|
|
if let functionCall = `as`(FunctionCallExprSyntax.self),
|
|
let calledExpression = functionCall.calledExpression.as(DeclReferenceExprSyntax.self),
|
|
calledExpression.baseName.text == "NSPredicate" {
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
}
|