Add option to make unneeded_override rule affect initializers (#5270)

This commit is contained in:
Leonardo de Sousa Rodrigues
2023-11-30 21:35:53 +01:00
committed by GitHub
parent 8cb2cf9f85
commit eecb44db6e
4 changed files with 213 additions and 83 deletions
@@ -3,7 +3,7 @@ import SwiftSyntaxBuilder
@SwiftSyntaxRule(explicitRewriter: true)
struct UnneededOverrideRule: Rule {
var configuration = SeverityConfiguration<Self>(.warning)
var configuration = UnneededOverrideRuleConfiguration()
static let description = RuleDescription(
identifier: "unneeded_override",
@@ -16,6 +16,137 @@ struct UnneededOverrideRule: Rule {
)
}
private extension UnneededOverrideRule {
final class Visitor: ViolationsSyntaxVisitor<ConfigurationType> {
override func visitPost(_ node: FunctionDeclSyntax) {
if node.isUnneededOverride {
self.violations.append(node.positionAfterSkippingLeadingTrivia)
}
}
override func visitPost(_ node: InitializerDeclSyntax) {
if configuration.affectInits && node.isUnneededOverride {
self.violations.append(node.positionAfterSkippingLeadingTrivia)
}
}
}
final class Rewriter: ViolationsSyntaxRewriter {
override func visit(_ node: FunctionDeclSyntax) -> DeclSyntax {
guard node.isUnneededOverride else {
return super.visit(node)
}
return visitUnneededOverride(node)
}
override func visit(_ node: InitializerDeclSyntax) -> DeclSyntax {
guard node.isUnneededOverride else {
return super.visit(node)
}
return visitUnneededOverride(node)
}
private func visitUnneededOverride(_ node: some DeclSyntaxProtocol) -> DeclSyntax {
correctionPositions.append(node.positionAfterSkippingLeadingTrivia)
let expr: DeclSyntax = ""
return expr
.with(\.leadingTrivia, node.leadingTrivia)
.with(\.trailingTrivia, node.trailingTrivia)
}
}
}
private extension FunctionDeclSyntax {
var isUnneededOverride: Bool {
mayBeUnneededOverride(name: name.text)
}
}
private extension InitializerDeclSyntax {
var isUnneededOverride: Bool {
guard
optionalMark == nil, // init? can be overridden with init! and vice versa.
!modifiers.contains(keyword: .private) // An initializer can be hidden by overriding it.
else {
return false
}
return mayBeUnneededOverride(name: "init")
}
}
private protocol OverridableDecl: WithAttributesSyntax, WithModifiersSyntax {
var signature: FunctionSignatureSyntax { get }
var body: CodeBlockSyntax? { get }
}
private extension OverridableDecl {
/// Perform checks common to all overridable types of declarations.
func mayBeUnneededOverride(name: String) -> Bool {
guard modifiers.contains(keyword: .override), let statement = body?.statements.onlyElement else {
return false
}
// Assume having @available changes behavior.
if attributes.contains(attributeNamed: "available") {
return false
}
guard let call = extractFunctionCallSyntax(statement.item),
let member = call.calledExpression.as(MemberAccessExprSyntax.self),
member.base?.is(SuperExprSyntax.self) == true,
member.declName.baseName.text == name else {
return false
}
let declParameters = signature.parameterClause.parameters
if declParameters.contains(where: { $0.defaultValue != nil }) {
// Any default parameter might be a change to the super.
return false
}
// Assume any change in arguments passed means behavior was changed.
let expectedArguments = declParameters.map {
($0.firstName.text == "_" ? "" : $0.firstName.text, $0.secondName?.text ?? $0.firstName.text)
}
let actualArguments = call.arguments.map {
($0.label?.text ?? "", $0.expression.as(DeclReferenceExprSyntax.self)?.baseName.text ?? "")
}
guard expectedArguments.count == actualArguments.count else {
return false
}
for (lhs, rhs) in zip(expectedArguments, actualArguments) where lhs != rhs {
return false
}
return true
}
}
extension FunctionDeclSyntax: OverridableDecl {}
extension InitializerDeclSyntax: OverridableDecl {}
/// Extract the function call from other expressions like try / await / return.
///
/// If this returns a non-super calling function, it will get filtered out later.
private func extractFunctionCallSyntax(_ node: some SyntaxProtocol) -> FunctionCallExprSyntax? {
var syntax = simplify(node)
while let nestedSyntax = syntax {
if nestedSyntax.as(FunctionCallExprSyntax.self) != nil {
break
}
syntax = simplify(nestedSyntax)
}
return syntax?.as(FunctionCallExprSyntax.self)
}
private func simplify(_ node: some SyntaxProtocol) -> (any ExprSyntaxProtocol)? {
if let expr = node.as(AwaitExprSyntax.self) {
return expr.expression
@@ -34,85 +165,3 @@ private func simplify(_ node: some SyntaxProtocol) -> (any ExprSyntaxProtocol)?
return nil
}
private func extractFunctionCallSyntax(_ node: some SyntaxProtocol) -> FunctionCallExprSyntax? {
// Extract the function call from other expressions like try / await / return.
// If this returns a non-super calling function, it will get filtered out later.
var syntax = simplify(node)
while let nestedSyntax = syntax {
if nestedSyntax.as(FunctionCallExprSyntax.self) != nil {
break
}
syntax = simplify(nestedSyntax)
}
return syntax?.as(FunctionCallExprSyntax.self)
}
private extension UnneededOverrideRule {
final class Visitor: ViolationsSyntaxVisitor<ConfigurationType> {
override func visitPost(_ node: FunctionDeclSyntax) {
if isUnneededOverride(node) {
self.violations.append(node.positionAfterSkippingLeadingTrivia)
}
}
}
final class Rewriter: ViolationsSyntaxRewriter {
override func visit(_ node: FunctionDeclSyntax) -> DeclSyntax {
if isUnneededOverride(node) {
correctionPositions.append(node.positionAfterSkippingLeadingTrivia)
let expr: DeclSyntax = ""
return expr
.with(\.leadingTrivia, node.leadingTrivia)
.with(\.trailingTrivia, node.trailingTrivia)
}
return super.visit(node)
}
}
}
private func isUnneededOverride(_ node: FunctionDeclSyntax) -> Bool {
guard node.modifiers.contains(keyword: .override), let statement = node.body?.statements.onlyElement else {
return false
}
// Assume having @available changes behavior.
if node.attributes.contains(attributeNamed: "available") {
return false
}
let overridenFunctionName = node.name.text
guard let call = extractFunctionCallSyntax(statement.item),
let member = call.calledExpression.as(MemberAccessExprSyntax.self),
member.base?.is(SuperExprSyntax.self) == true,
member.declName.baseName.text == overridenFunctionName else {
return false
}
let declParameters = node.signature.parameterClause.parameters
if declParameters.contains(where: { $0.defaultValue != nil }) {
// Any default parameter might be a change to the super function.
return false
}
// Assume any change in arguments passed means behavior was changed.
let expectedArguments = declParameters.map {
($0.firstName.text == "_" ? "" : $0.firstName.text, $0.secondName?.text ?? $0.firstName.text)
}
let actualArguments = call.arguments.map {
($0.label?.text ?? "", $0.expression.as(DeclReferenceExprSyntax.self)?.baseName.text ?? "")
}
guard expectedArguments.count == actualArguments.count else {
return false
}
for (lhs, rhs) in zip(expectedArguments, actualArguments) where lhs != rhs {
return false
}
return true
}
@@ -0,0 +1,11 @@
import SwiftLintCore
@AutoApply
struct UnneededOverrideRuleConfiguration: SeverityBasedRuleConfiguration {
typealias Parent = UnneededOverrideRule
@ConfigurationElement(key: "severity")
private(set) var severityConfiguration = SeverityConfiguration<Parent>(.warning)
@ConfigurationElement(key: "affect_initializers")
private(set) var affectInits = false
}