Files
SwiftLint/Source/SwiftLintFramework/Rules/RuleConfigurations/RegexConfiguration.swift
T
Pyry Jahkola 805b9ab3ba Add capture_group option to custom_rules
This option allows for more fine-grained placement of the location
marker for code violating a custom rule, e.g.:

```swift
print("Hello world.")
             ^~~~~
```

for this `.swiftlint.yml`:

```yaml
custom_rules:
  world_preceded_by_hello:
    name: "World Preceded by Hello"
    included: ".+\\.swift"
    message: "The word World predeced by the word hello should be capitalised."
    severity: warning
    regex: "(?i:hello)\\s+(world)"
    match_kinds: [string]
    capture_group: 1
```
2020-01-05 15:24:11 -08:00

90 lines
3.2 KiB
Swift

import Foundation
import SourceKittenFramework
public struct RegexConfiguration: RuleConfiguration, Hashable, CacheDescriptionProvider {
public let identifier: String
public var name: String?
public var message = "Regex matched."
public var regex: NSRegularExpression!
public var included: NSRegularExpression?
public var excluded: NSRegularExpression?
public var matchKinds = SyntaxKind.allKinds
public var severityConfiguration = SeverityConfiguration(.warning)
public var captureGroup: Int = 0
public var severity: ViolationSeverity {
return severityConfiguration.severity
}
public var consoleDescription: String {
return "\(severity.rawValue): \(regex.pattern)"
}
internal var cacheDescription: String {
let jsonObject: [String] = [
identifier,
name ?? "",
message,
regex.pattern,
included?.pattern ?? "",
excluded?.pattern ?? "",
matchKinds.map({ $0.rawValue }).sorted(by: <).joined(separator: ","),
severityConfiguration.consoleDescription
]
if let jsonData = try? JSONSerialization.data(withJSONObject: jsonObject),
let jsonString = String(data: jsonData, encoding: .utf8) {
return jsonString
}
queuedFatalError("Could not serialize regex configuration for cache")
}
public var description: RuleDescription {
return RuleDescription(identifier: identifier, name: name ?? identifier,
description: "", kind: .style)
}
public init(identifier: String) {
self.identifier = identifier
}
public mutating func apply(configuration: Any) throws {
guard let configurationDict = configuration as? [String: Any],
let regexString = configurationDict["regex"] as? String else {
throw ConfigurationError.unknownConfiguration
}
regex = try .cached(pattern: regexString)
if let includedString = configurationDict["included"] as? String {
included = try .cached(pattern: includedString)
}
if let excludedString = configurationDict["excluded"] as? String {
excluded = try .cached(pattern: excludedString)
}
if let name = configurationDict["name"] as? String {
self.name = name
}
if let message = configurationDict["message"] as? String {
self.message = message
}
if let matchKinds = [String].array(of: configurationDict["match_kinds"]) {
self.matchKinds = Set(try matchKinds.map({ try SyntaxKind(shortName: $0) }))
}
if let severityString = configurationDict["severity"] as? String {
try severityConfiguration.apply(configuration: severityString)
}
if let captureGroup = configurationDict["capture_group"] as? Int {
guard (0 ... regex.numberOfCaptureGroups).contains(captureGroup) else {
throw ConfigurationError.unknownConfiguration
}
self.captureGroup = captureGroup
}
}
public func hash(into hasher: inout Hasher) {
hasher.combine(identifier)
}
}