Skip to content

Commit

Permalink
Add unneeded_override rule
Browse files Browse the repository at this point in the history
This rule flags functions where the only thing they do is call their
super function and therefore could be omitted. For example:

```swift
override func foo() {
    super.foo()
}
```

This can get pretty complex since there are a lot of slight variations
the subclasses' functions can call the superclasses' functions with, but
this covers many of the cases. Ideally this would handle variable
overrides too but it doesn't currently.
  • Loading branch information
keith committed Jul 21, 2023
1 parent b53919d commit ad64295
Show file tree
Hide file tree
Showing 4 changed files with 264 additions and 0 deletions.
1 change: 1 addition & 0 deletions Source/SwiftLintBuiltInRules/Models/BuiltInRules.swift
Original file line number Diff line number Diff line change
Expand Up @@ -205,6 +205,7 @@ public let builtInRules: [Rule.Type] = [
UnavailableConditionRule.self,
UnavailableFunctionRule.self,
UnhandledThrowingTaskRule.self,
UnneededOverrideRule.self,
UnneededBreakInSwitchRule.self,
UnneededParenthesesInClosureArgumentRule.self,
UnneededSynthesizedInitializerRule.self,
Expand Down
101 changes: 101 additions & 0 deletions Source/SwiftLintBuiltInRules/Rules/Lint/UnneededOverrideRule.swift
Original file line number Diff line number Diff line change
@@ -0,0 +1,101 @@
import SwiftSyntax
import SwiftSyntaxBuilder

struct UnneededOverrideRule: ConfigurationProviderRule, SwiftSyntaxRule {
var configuration = SeverityConfiguration<Self>(.warning)

static let description = RuleDescription(
identifier: "unneeded_override",
name: "Remove empty overridden functions",
description: "Remove overridden functions that don't do anything except call their super",
kind: .lint,
nonTriggeringExamples: UnneededOverrideRule.nonTriggeringExamples,
triggeringExamples: UnneededOverrideRule.triggeringExamples
)

func makeVisitor(file: SwiftLintFile) -> ViolationsSyntaxVisitor {
Visitor(viewMode: .sourceAccurate)
}
}

private func simplify(_ node: SyntaxProtocol) -> ExprSyntaxProtocol? {
if let expr = node.as(AwaitExprSyntax.self) {
return expr.expression
} else if let expr = node.as(TryExprSyntax.self) {
// Assume using try! / try? changes behavior
if expr.questionOrExclamationMark != nil {
return nil
}

return expr.expression
} else if let expr = node.as(FunctionCallExprSyntax.self) {
return expr
} else if let stmt = node.as(ReturnStmtSyntax.self) {
return stmt.expression
}

return nil
}

private final class Visitor: ViolationsSyntaxVisitor {
override func visitPost(_ node: FunctionDeclSyntax) {
guard let modifiers = node.modifiers, modifiers.contains(where: { $0.name.text == "override" }) else {
return
}

// Assume having @available changes behavior
if node.attributes.contains(attributeNamed: "available") {
return
}

guard node.body?.statements.count == 1, let statement = node.body?.statements.first else {
return
}

// Extract the function call from other expressions like try / await / return.
// If this returns a non-super calling function that will get filtered out later
var syntax: ExprSyntaxProtocol? = simplify(statement.item)
while let nestedSyntax = syntax {
if nestedSyntax.as(FunctionCallExprSyntax.self) != nil {
break
}

syntax = simplify(nestedSyntax)
}

guard let call = syntax?.as(FunctionCallExprSyntax.self) else {
return
}

guard let member = call.calledExpression.as(MemberAccessExprSyntax.self) else {
return
}

guard member.base?.as(SuperRefExprSyntax.self) != nil else {
return
}

let overridenFunctionName = node.identifier.text
guard member.name.text == overridenFunctionName else {
return
}

// Assume any change in arguments passed means behavior was changed
let expectedArguments = node.signature.input.parameterList.map {
($0.firstName.text == "_" ? "" : $0.firstName.text, $0.secondName?.text ?? $0.firstName.text)
}
let actualArguments = call.argumentList.map {
($0.label?.text ?? "", $0.expression.as(IdentifierExprSyntax.self)?.identifier.text ?? "")
}

guard expectedArguments.count == actualArguments.count else {
return
}

for (lhs, rhs) in zip(expectedArguments, actualArguments) where lhs != rhs {
return
}

self.violations.append(node.positionAfterSkippingLeadingTrivia)
}
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,156 @@
extension UnneededOverrideRule {
static let nonTriggeringExamples = [
Example("""
class Foo {
override func bar() {
super.bar()
print("hi")
}
}
"""),
Example("""
class Foo {
@available(*, unavailable)
override func bar() {
super.bar()
}
}
"""),
Example("""
class Foo {
override func bar() {
super.bar()
super.bar()
}
}
"""),
Example("""
class Foo {
override func bar() throws {
// Doing a different variation of 'try' changes behavior
try! super.bar()
}
}
"""),
Example("""
class Foo {
override func bar() throws {
// Doing a different variation of 'try' changes behavior
try? super.bar()
}
}
"""),
Example("""
class Foo {
override func bar() async throws {
// Doing a different variation of 'try' changes behavior
await try! super.bar()
}
}
"""),
Example("""
class Foo {
override func bar(arg: Bool) {
// Flipping the argument changes behavior
super.bar(arg: !arg)
}
}
"""),
Example("""
class Foo {
override func bar(_ arg: Int) {
// Changing the argument changes behavior
super.bar(arg + 1)
}
}
"""),
Example("""
class Foo {
override func bar(_ arg: Int) {
// Not passing arguments because they have default values changes behavior
super.bar()
}
}
"""),
Example("""
class Foo {
override func bar(arg: Int, _ arg3: Bool) {
// Calling a super function with different argument labels changes behavior
super.bar(arg2: arg, arg3: arg3)
}
}
"""),
Example("""
class Foo {
override func bar(animated: Bool, completion: () -> Void) {
super.bar(animated: animated) {
// This likely changes behavior
}
}
}
"""),
Example("""
class Foo {
override func bar(animated: Bool, completion: () -> Void) {
super.bar(animated: animated, completion: {
// This likely changes behavior
})
}
}
"""),
]

static let triggeringExamples = [
Example("""
class Foo {
override func bar() {
super.bar()
}
}
"""),
Example("""
class Foo {
override func bar() {
return super.bar()
}
}
"""),
Example("""
class Foo {
override func bar() {
super.bar()
// comments don't affect this
}
}
"""),
Example("""
class Foo {
override func bar() async {
await super.bar()
}
}
"""),
Example("""
class Foo {
override func bar() throws {
try super.bar()
// comments don't affect this
}
}
"""),
Example("""
class Foo {
override func bar(arg: Bool) throws {
try super.bar(arg: arg)
}
}
"""),
Example("""
class Foo {
override func bar(animated: Bool, completion: () -> Void) {
super.bar(animated: animated, completion: completion)
}
}
"""),
]
}
6 changes: 6 additions & 0 deletions Tests/GeneratedTests/GeneratedTests.swift
Original file line number Diff line number Diff line change
Expand Up @@ -1232,6 +1232,12 @@ class UnneededBreakInSwitchRuleGeneratedTests: SwiftLintTestCase {
}
}

class UnneededOverrideRuleGeneratedTests: SwiftLintTestCase {
func testWithDefaultConfiguration() {
verifyRule(UnneededOverrideRule.description)
}
}

class UnneededParenthesesInClosureArgumentRuleGeneratedTests: SwiftLintTestCase {
func testWithDefaultConfiguration() {
verifyRule(UnneededParenthesesInClosureArgumentRule.description)
Expand Down

0 comments on commit ad64295

Please sign in to comment.