SwiftLint/Source/SwiftLintFramework/Rules/FileLengthRule.swift

//
//  FileLengthRule.swift
//  SwiftLint
//
//  Created by JP Simard on 2015-05-16.
//  Copyright (c) 2015 Realm. All rights reserved.
//

import SourceKittenFramework

public struct FileLengthRule: ParameterizedRule {
    public init() {
        self.init(parameters: [
            RuleParameter(severity: .Warning, value: 400),
            RuleParameter(severity: .Error, value: 1000)
        ])
    }

    public init(parameters: [RuleParameter<Int>]) {
        self.parameters = parameters
    }

    public let identifier = "file_length"

    public let parameters: [RuleParameter<Int>]

    public func validateFile(file: File) -> [StyleViolation] {
        let lineCount = file.lines.count
        for parameter in parameters.reverse() {
            if lineCount > parameter.value {
                return [StyleViolation(type: .Length,
                    location: Location(file: file.path, line: lineCount),
                    severity: parameter.severity,
                    reason: "File should contain \(parameters.first!.value) lines or less: " +
                    "currently contains \(lineCount)",
                    ruleId: self.identifier)]
            }
        }
        return []
    }

    public let example = RuleExample(
        ruleName: "File Line Length Rule",
        ruleDescription: "Enforce maximum file length",
        nonTriggeringExamples: [],
        triggeringExamples: [],
        showExamples: false
    )
}