SwiftLint/Source/SwiftLintFramework/Rules/LegacyConstantRule.swift

86 lines
2.9 KiB
Swift

//
// LegacyConstantRule.swift
// SwiftLint
//
// Created by Aaron McTavish on 12/01/2016.
// Copyright © 2016 Realm. All rights reserved.
//
import SourceKittenFramework
public struct LegacyConstantRule: CorrectableRule, ConfigProviderRule {
public var config = SeverityConfig(.Warning)
public init() { }
public static let description = RuleDescription(
identifier: "legacy_constant",
name: "Legacy Constant",
description: "Struct-scoped constants are preferred over legacy global constants.",
nonTriggeringExamples: [
Trigger("CGRect.infinite"),
Trigger("CGPoint.zero"),
Trigger("CGRect.zero"),
Trigger("CGSize.zero"),
Trigger("CGRect.null")
],
triggeringExamples: [
Trigger("↓CGRectInfinite"),
Trigger("↓CGPointZero"),
Trigger("↓CGRectZero"),
Trigger("↓CGSizeZero"),
Trigger("↓CGRectNull")
],
corrections: [
"CGRectInfinite\n": "CGRect.infinite\n",
"CGPointZero\n": "CGPoint.zero\n",
"CGRectZero\n": "CGRect.zero\n",
"CGSizeZero\n": "CGSize.zero\n",
"CGRectNull\n": "CGRect.null\n"
]
)
public func validateFile(file: File) -> [StyleViolation] {
let constants = ["CGRectInfinite", "CGPointZero", "CGRectZero", "CGSizeZero",
"CGRectNull"]
let pattern = "\\b(" + constants.joinWithSeparator("|") + ")\\b"
return file.matchPattern(pattern, withSyntaxKinds: [.Identifier]).map {
StyleViolation(ruleDescription: self.dynamicType.description,
severity: config.severity,
location: Location(file: file, characterOffset: $0.location))
}
}
public func correctFile(file: File) -> [Correction] {
let patterns = [
"CGRectInfinite": "CGRect.infinite",
"CGPointZero": "CGPoint.zero",
"CGRectZero": "CGRect.zero",
"CGSizeZero": "CGSize.zero",
"CGRectNull": "CGRect.null"
]
let description = self.dynamicType.description
var corrections = [Correction]()
var contents = file.contents
for (pattern, template) in patterns {
let matches = file.matchPattern(pattern, withSyntaxKinds: [.Identifier])
let regularExpression = regex(pattern)
for range in matches.reverse() {
contents = regularExpression.stringByReplacingMatchesInString(contents,
options: [], range: range, withTemplate: template)
let location = Location(file: file, characterOffset: range.location)
corrections.append(Correction(ruleDescription: description, location: location))
}
}
file.write(contents)
return corrections
}
}