Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Lint] Add a rule to detect that type declarations are not capitalized #587

Merged
merged 3 commits into from
Aug 14, 2023
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
39 changes: 25 additions & 14 deletions Sources/SwiftFormat/Pipelines+Generated.swift
Original file line number Diff line number Diff line change
Expand Up @@ -34,6 +34,11 @@ class LintPipeline: SyntaxVisitor {
super.init(viewMode: .sourceAccurate)
}

override func visit(_ node: ActorDeclSyntax) -> SyntaxVisitorContinueKind {
visitIfEnabled(TypeNamesShouldBeCapitalized.visit, for: node)
return .visitChildren
}

override func visit(_ node: AsExprSyntax) -> SyntaxVisitorContinueKind {
visitIfEnabled(NeverForceUnwrap.visit, for: node)
return .visitChildren
Expand All @@ -42,6 +47,7 @@ class LintPipeline: SyntaxVisitor {
override func visit(_ node: AssociatedTypeDeclSyntax) -> SyntaxVisitorContinueKind {
visitIfEnabled(BeginDocumentationCommentWithOneLineSummary.visit, for: node)
visitIfEnabled(NoLeadingUnderscores.visit, for: node)
visitIfEnabled(TypeNamesShouldBeCapitalized.visit, for: node)
return .visitChildren
}

Expand All @@ -51,6 +57,7 @@ class LintPipeline: SyntaxVisitor {
visitIfEnabled(BeginDocumentationCommentWithOneLineSummary.visit, for: node)
visitIfEnabled(DontRepeatTypeInStaticProperties.visit, for: node)
visitIfEnabled(NoLeadingUnderscores.visit, for: node)
visitIfEnabled(TypeNamesShouldBeCapitalized.visit, for: node)
visitIfEnabled(UseTripleSlashForDocumentationComments.visit, for: node)
return .visitChildren
}
Expand Down Expand Up @@ -108,6 +115,7 @@ class LintPipeline: SyntaxVisitor {
visitIfEnabled(FullyIndirectEnum.visit, for: node)
visitIfEnabled(NoLeadingUnderscores.visit, for: node)
visitIfEnabled(OneCasePerLine.visit, for: node)
visitIfEnabled(TypeNamesShouldBeCapitalized.visit, for: node)
visitIfEnabled(UseTripleSlashForDocumentationComments.visit, for: node)
return .visitChildren
}
Expand Down Expand Up @@ -165,12 +173,22 @@ class LintPipeline: SyntaxVisitor {
return .visitChildren
}

override func visit(_ node: GenericSpecializationExprSyntax) -> SyntaxVisitorContinueKind {
visitIfEnabled(UseShorthandTypeNames.visit, for: node)
return .visitChildren
}

override func visit(_ node: IdentifierPatternSyntax) -> SyntaxVisitorContinueKind {
visitIfEnabled(IdentifiersMustBeASCII.visit, for: node)
visitIfEnabled(NoLeadingUnderscores.visit, for: node)
return .visitChildren
}

override func visit(_ node: IdentifierTypeSyntax) -> SyntaxVisitorContinueKind {
visitIfEnabled(UseShorthandTypeNames.visit, for: node)
return .visitChildren
}

override func visit(_ node: IfExprSyntax) -> SyntaxVisitorContinueKind {
visitIfEnabled(NoParensAroundConditions.visit, for: node)
return .visitChildren
Expand All @@ -194,13 +212,13 @@ class LintPipeline: SyntaxVisitor {
return .visitChildren
}

override func visit(_ node: MemberBlockSyntax) -> SyntaxVisitorContinueKind {
visitIfEnabled(AmbiguousTrailingClosureOverload.visit, for: node)
override func visit(_ node: MemberBlockItemListSyntax) -> SyntaxVisitorContinueKind {
visitIfEnabled(DoNotUseSemicolons.visit, for: node)
return .visitChildren
}

override func visit(_ node: MemberBlockItemListSyntax) -> SyntaxVisitorContinueKind {
visitIfEnabled(DoNotUseSemicolons.visit, for: node)
override func visit(_ node: MemberBlockSyntax) -> SyntaxVisitorContinueKind {
visitIfEnabled(AmbiguousTrailingClosureOverload.visit, for: node)
return .visitChildren
}

Expand All @@ -224,6 +242,7 @@ class LintPipeline: SyntaxVisitor {
visitIfEnabled(BeginDocumentationCommentWithOneLineSummary.visit, for: node)
visitIfEnabled(DontRepeatTypeInStaticProperties.visit, for: node)
visitIfEnabled(NoLeadingUnderscores.visit, for: node)
visitIfEnabled(TypeNamesShouldBeCapitalized.visit, for: node)
visitIfEnabled(UseTripleSlashForDocumentationComments.visit, for: node)
return .visitChildren
}
Expand All @@ -233,11 +252,6 @@ class LintPipeline: SyntaxVisitor {
return .visitChildren
}

override func visit(_ node: IdentifierTypeSyntax) -> SyntaxVisitorContinueKind {
visitIfEnabled(UseShorthandTypeNames.visit, for: node)
return .visitChildren
}

override func visit(_ node: SourceFileSyntax) -> SyntaxVisitorContinueKind {
visitIfEnabled(AlwaysUseLowerCamelCase.visit, for: node)
visitIfEnabled(AmbiguousTrailingClosureOverload.visit, for: node)
Expand All @@ -249,16 +263,12 @@ class LintPipeline: SyntaxVisitor {
return .visitChildren
}

override func visit(_ node: GenericSpecializationExprSyntax) -> SyntaxVisitorContinueKind {
visitIfEnabled(UseShorthandTypeNames.visit, for: node)
return .visitChildren
}

override func visit(_ node: StructDeclSyntax) -> SyntaxVisitorContinueKind {
visitIfEnabled(AllPublicDeclarationsHaveDocumentation.visit, for: node)
visitIfEnabled(BeginDocumentationCommentWithOneLineSummary.visit, for: node)
visitIfEnabled(DontRepeatTypeInStaticProperties.visit, for: node)
visitIfEnabled(NoLeadingUnderscores.visit, for: node)
visitIfEnabled(TypeNamesShouldBeCapitalized.visit, for: node)
visitIfEnabled(UseSynthesizedInitializer.visit, for: node)
visitIfEnabled(UseTripleSlashForDocumentationComments.visit, for: node)
return .visitChildren
Expand Down Expand Up @@ -300,6 +310,7 @@ class LintPipeline: SyntaxVisitor {
visitIfEnabled(AllPublicDeclarationsHaveDocumentation.visit, for: node)
visitIfEnabled(BeginDocumentationCommentWithOneLineSummary.visit, for: node)
visitIfEnabled(NoLeadingUnderscores.visit, for: node)
visitIfEnabled(TypeNamesShouldBeCapitalized.visit, for: node)
visitIfEnabled(UseTripleSlashForDocumentationComments.visit, for: node)
return .visitChildren
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -41,6 +41,7 @@ enum RuleRegistry {
"OnlyOneTrailingClosureArgument": true,
"OrderedImports": true,
"ReturnVoidInsteadOfEmptyTuple": true,
"TypeNamesShouldBeCapitalized": true,
"UseEarlyExits": false,
"UseLetInEveryBoundCaseVariable": true,
"UseShorthandTypeNames": true,
Expand Down
1 change: 1 addition & 0 deletions Sources/SwiftFormatRules/RuleNameCache+Generated.swift
Original file line number Diff line number Diff line change
Expand Up @@ -41,6 +41,7 @@ public let ruleNameCache: [ObjectIdentifier: String] = [
ObjectIdentifier(OnlyOneTrailingClosureArgument.self): "OnlyOneTrailingClosureArgument",
ObjectIdentifier(OrderedImports.self): "OrderedImports",
ObjectIdentifier(ReturnVoidInsteadOfEmptyTuple.self): "ReturnVoidInsteadOfEmptyTuple",
ObjectIdentifier(TypeNamesShouldBeCapitalized.self): "TypeNamesShouldBeCapitalized",
ObjectIdentifier(UseEarlyExits.self): "UseEarlyExits",
ObjectIdentifier(UseLetInEveryBoundCaseVariable.self): "UseLetInEveryBoundCaseVariable",
ObjectIdentifier(UseShorthandTypeNames.self): "UseShorthandTypeNames",
Expand Down
72 changes: 72 additions & 0 deletions Sources/SwiftFormatRules/TypeNamesShouldBeCapitalized.swift
Original file line number Diff line number Diff line change
@@ -0,0 +1,72 @@
//===----------------------------------------------------------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2023 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//

import SwiftFormatCore
import SwiftSyntax

/// `struct`, `class`, `enum` and `protocol` declarations should have a capitalized name.
///
/// Lint: Types with un-capitalized names will yield a lint error.
public final class TypeNamesShouldBeCapitalized : SyntaxLintRule {
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Can you also add ActorDeclSyntax, AssociatedTypeDeclSyntax, and TypeAliasDeclSyntax? I think that covers everything we'd want to apply this to.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Sure! Will do! I went conservative initially to not over do it :)

public override func visit(_ node: StructDeclSyntax) -> SyntaxVisitorContinueKind {
diagnoseNameConventionMismatch(node, name: node.name)
return .visitChildren
}

public override func visit(_ node: ClassDeclSyntax) -> SyntaxVisitorContinueKind {
diagnoseNameConventionMismatch(node, name: node.name)
return .visitChildren
}

public override func visit(_ node: EnumDeclSyntax) -> SyntaxVisitorContinueKind {
diagnoseNameConventionMismatch(node, name: node.name)
return .visitChildren
}

public override func visit(_ node: ProtocolDeclSyntax) -> SyntaxVisitorContinueKind {
diagnoseNameConventionMismatch(node, name: node.name)
return .visitChildren
}

public override func visit(_ node: ActorDeclSyntax) -> SyntaxVisitorContinueKind {
diagnoseNameConventionMismatch(node, name: node.name)
return .visitChildren
}

public override func visit(_ node: AssociatedTypeDeclSyntax) -> SyntaxVisitorContinueKind {
diagnoseNameConventionMismatch(node, name: node.name)
return .visitChildren
}

public override func visit(_ node: TypeAliasDeclSyntax) -> SyntaxVisitorContinueKind {
diagnoseNameConventionMismatch(node, name: node.name)
return .visitChildren
}

private func diagnoseNameConventionMismatch<T: DeclSyntaxProtocol>(_ type: T, name: TokenSyntax) {
let leadingUnderscores = name.text.prefix { $0 == "_" }
if let firstChar = name.text[leadingUnderscores.endIndex...].first,
firstChar.uppercased() != String(firstChar) {
diagnose(.capitalizeTypeName(name: name.text), on: type, severity: .convention)
}
}
}

extension Finding.Message {
public static func capitalizeTypeName(name: String) -> Finding.Message {
var capitalized = name
let leadingUnderscores = capitalized.prefix { $0 == "_" }
let charAt = leadingUnderscores.endIndex
capitalized.replaceSubrange(charAt...charAt, with: capitalized[charAt].uppercased())
return "type names should be capitalized: \(name) -> \(capitalized)"
}
}
128 changes: 128 additions & 0 deletions Tests/SwiftFormatRulesTests/TypeNamesShouldBeCapitalizedTests.swift
Original file line number Diff line number Diff line change
@@ -0,0 +1,128 @@
import SwiftFormatRules

final class TypeNamesShouldBeCapitalizedTests: LintOrFormatRuleTestCase {
func testConstruction() {
let input =
"""
struct a {}
class klassName {
struct subType {}
}
protocol myProtocol {}

extension myType {
struct innerType {}
}
"""

performLint(TypeNamesShouldBeCapitalized.self, input: input)

XCTAssertDiagnosed(.capitalizeTypeName(name: "a"))
XCTAssertDiagnosed(.capitalizeTypeName(name: "klassName"))
XCTAssertDiagnosed(.capitalizeTypeName(name: "subType"))
XCTAssertDiagnosed(.capitalizeTypeName(name: "myProtocol"))
XCTAssertNotDiagnosed(.capitalizeTypeName(name: "myType"))
XCTAssertDiagnosed(.capitalizeTypeName(name: "innerType"))
}

func testActors() {
let input =
"""
actor myActor {}
actor OtherActor {}
distributed actor greeter {}
distributed actor DistGreeter {}
"""

performLint(TypeNamesShouldBeCapitalized.self, input: input)

XCTAssertDiagnosed(.capitalizeTypeName(name: "myActor"))
XCTAssertNotDiagnosed(.capitalizeTypeName(name: "OtherActor"))
XCTAssertDiagnosed(.capitalizeTypeName(name: "greeter"))
XCTAssertNotDiagnosed(.capitalizeTypeName(name: "DistGreeter"))
}

func testAssociatedTypeandTypeAlias() {
let input =
"""
protocol P {
associatedtype kind
associatedtype OtherKind
}

typealias x = Int
typealias Y = String

struct MyType {
typealias data<T> = Y

func test() {
typealias Value<T> = Y
}
}
"""

performLint(TypeNamesShouldBeCapitalized.self, input: input)

XCTAssertDiagnosed(.capitalizeTypeName(name: "kind"))
XCTAssertNotDiagnosed(.capitalizeTypeName(name: "OtherKind"))
XCTAssertDiagnosed(.capitalizeTypeName(name: "x"))
XCTAssertNotDiagnosed(.capitalizeTypeName(name: "Y"))
XCTAssertDiagnosed(.capitalizeTypeName(name: "data"))
XCTAssertNotDiagnosed(.capitalizeTypeName(name: "Value"))
}

func testThatUnderscoredNamesAreDiagnosed() {
let input =
"""
protocol _p {
associatedtype _value
associatedtype __Value
}

protocol ___Q {
}

struct _data {
typealias _x = Int
}

struct _Data {}

actor _internalActor {}

enum __e {
}

enum _OtherE {
}

func test() {
class _myClass {}
do {
class _MyClass {}
}
}

distributed actor __greeter {}
distributed actor __InternalGreeter {}
"""

performLint(TypeNamesShouldBeCapitalized.self, input: input)

XCTAssertDiagnosed(.capitalizeTypeName(name: "_p"))
XCTAssertNotDiagnosed(.capitalizeTypeName(name: "___Q"))
XCTAssertDiagnosed(.capitalizeTypeName(name: "_value"))
XCTAssertNotDiagnosed(.capitalizeTypeName(name: "__Value"))
XCTAssertDiagnosed(.capitalizeTypeName(name: "_data"))
XCTAssertNotDiagnosed(.capitalizeTypeName(name: "_Data"))
XCTAssertDiagnosed(.capitalizeTypeName(name: "_x"))
XCTAssertDiagnosed(.capitalizeTypeName(name: "_internalActor"))
XCTAssertDiagnosed(.capitalizeTypeName(name: "__e"))
XCTAssertNotDiagnosed(.capitalizeTypeName(name: "__OtherE"))
XCTAssertDiagnosed(.capitalizeTypeName(name: "_myClass"))
XCTAssertNotDiagnosed(.capitalizeTypeName(name: "_MyClass"))
XCTAssertDiagnosed(.capitalizeTypeName(name: "__greeter"))
XCTAssertNotDiagnosed(.capitalizeTypeName(name: "__InternalGreeter"))
}
}