Fix superfluous disable command for custom rules

Source/SwiftLintFramework/Models/Linter.swift

@@ -16,15 +16,15 @@ private struct LintResult {
 }
 
 private extension Rule {
-    static func superfluousDisableCommandViolations(regions: [Region],
+    func superfluousDisableCommandViolations(regions: [Region],
                                                     superfluousDisableCommandRule: SuperfluousDisableCommandRule?,
                                                     allViolations: [StyleViolation]) -> [StyleViolation] {
         guard regions.isNotEmpty, let superfluousDisableCommandRule else {
             return []
         }
 
         let regionsDisablingCurrentRule = regions.filter { region in
-            return region.isRuleDisabled(self.init())
+            return region.isRuleDisabled(self)
         }
         let regionsDisablingSuperfluousDisableRule = regions.filter { region in
             return region.isRuleDisabled(superfluousDisableCommandRule)
@@ -50,7 +50,7 @@ private extension Rule {
                 ruleDescription: type(of: superfluousDisableCommandRule).description,
                 severity: superfluousDisableCommandRule.configuration.severity,
                 location: region.start,
-                reason: superfluousDisableCommandRule.reason(for: self)
+                reason: superfluousDisableCommandRule.reason(for: Self.self)
             )
         }
     }
@@ -91,12 +91,17 @@ private extension Rule {
             return region?.isRuleEnabled(self) ?? true
         }
 
-        let ruleIDs = Self.description.allIdentifiers +
-            (superfluousDisableCommandRule.map({ type(of: $0) })?.description.allIdentifiers ?? []) +
+        let ruleIDs: [String]
+        if let customRules = self as? CustomRules {
+            ruleIDs = customRules.customRuleIdentifiers
+        } else {
+            ruleIDs = Self.description.allIdentifiers
+        }
+        let allRuleIDs = ruleIDs + (superfluousDisableCommandRule.map({ type(of: $0) })?.description.allIdentifiers ?? []) +
             [RuleIdentifier.all.stringRepresentation]
-        let ruleIdentifiers = Set(ruleIDs.map { RuleIdentifier($0) })
-
-        let superfluousDisableCommandViolations = Self.superfluousDisableCommandViolations(
+        let ruleIdentifiers = Set(allRuleIDs.map { RuleIdentifier($0) })
+        
+        let superfluousDisableCommandViolations = superfluousDisableCommandViolations(
             regions: regions.count > 1 ? file.regions(restrictingRuleIdentifiers: ruleIdentifiers) : regions,
             superfluousDisableCommandRule: superfluousDisableCommandRule,
             allViolations: violations

Source/SwiftLintFramework/Models/Region.swift

@@ -50,7 +50,12 @@ public struct Region: Equatable {
             return true
         }
 
-        let identifiersToCheck = type(of: rule).description.allIdentifiers
+        let identifiersToCheck: [String]
+        if let customRules = rule as? CustomRules {
+            identifiersToCheck = customRules.customRuleIdentifiers
+        } else {
+            identifiersToCheck = type(of: rule).description.allIdentifiers
+        }
         let regionIdentifiers = Set(disabledRuleIdentifiers.map { $0.stringRepresentation })
         return !regionIdentifiers.isDisjoint(with: identifiersToCheck)
     }

Source/SwiftLintFramework/Rules/Style/CustomRules.swift

@@ -79,6 +79,8 @@ struct CustomRules: Rule, ConfigurationProviderRule, CacheDescriptionProvider {
         kind: .style)
 
     var configuration = CustomRulesConfiguration()
+
+    var customRuleIdentifiers: [String] { configuration.customRuleConfigurations.map { $0.identifier } }
 
     func validate(file: SwiftLintFile) -> [StyleViolation] {
         var configurations = configuration.customRuleConfigurations

Tests/SwiftLintFrameworkTests/CustomRulesTests.swift

@@ -126,6 +126,25 @@ class CustomRulesTests: XCTestCase {
                                        reason: configs.0.message)])
     }
 
+    func testSuperfluousDisableCommandWithCustomRules() {
+        let customRulesConfiguration: [String:Any] = [
+            "custom1": [
+                "regex": "pattern",
+                "match_kinds": "comment"
+            ]
+        ]
+
+        let example = Example(
+            "// swiftlint:disable custom1\n",
+            configuration: customRulesConfiguration
+        ).skipWrappingInCommentTest()
+        let configuration = try! Configuration(dict: ["custom_rules": customRulesConfiguration])
+        let violations = violations(example, config: configuration)
+
+        XCTAssertEqual(violations.count, 1)
+        XCTAssertTrue(violations.allSatisfy { $0.ruleIdentifier == "superfluous_disable_command" })
+    }
+
     func testCustomRulesIncludedDefault() {
         // Violation detected when included is omitted.
         let (_, customRules) = getCustomRules()