SwiftLint/Source/SwiftLintFramework/Extensions/NSFileManager+SwiftLint.swift

//
//  NSFileManager+SwiftLint.swift
//  SwiftLint
//
//  Created by JP Simard on 5/28/15.
//  Copyright © 2015 Realm. All rights reserved.
//

import Foundation

extension FileManager {
    internal func filesToLintAtPath(_ path: String, rootDirectory: String? = nil) -> [String] {
        let rootPath = rootDirectory ?? currentDirectoryPath
        let absolutePath = path.bridge()
            .absolutePathRepresentation(rootDirectory: rootPath).bridge()
            .standardizingPath
        var isDirectory: ObjCBool = false
        guard fileExists(atPath: absolutePath, isDirectory: &isDirectory) else {
            return []
        }
        if isDirectory.boolValue {
            do {
                return try subpathsOfDirectory(atPath: absolutePath)
                    .map(absolutePath.bridge().appendingPathComponent).filter {
                        $0.bridge().isSwiftFile()
                }
            } catch {
                fatalError("Couldn't find files in \(absolutePath): \(error)")
            }
        } else if absolutePath.bridge().isSwiftFile() {
            return [absolutePath]
        }
        return []
    }
}