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

//
//  NSRegularExpression+SwiftLint.swift
//  SwiftLint
//
//  Created by Scott Hoyt on 1/21/16.
//  Copyright © 2016 Realm. All rights reserved.
//

import Foundation

#if os(Linux)
#if !swift(>=3.1)
public typealias NSRegularExpression = RegularExpression
#endif
public typealias NSTextCheckingResult = TextCheckingResult
#endif

private var regexCache = [RegexCacheKey: NSRegularExpression]()
private let regexCacheLock = NSLock()

private struct RegexCacheKey: Hashable {
    let pattern: String
    let options: NSRegularExpression.Options

    var hashValue: Int {
        return pattern.hashValue ^ options.rawValue.hashValue
    }

    static func == (lhs: RegexCacheKey, rhs: RegexCacheKey) -> Bool {
        return lhs.options == rhs.options && lhs.pattern == rhs.pattern
    }
}

extension NSRegularExpression {
    internal static func cached(pattern: String, options: Options? = nil) throws -> NSRegularExpression {
        let options = options ?? [.anchorsMatchLines, .dotMatchesLineSeparators]
        let key = RegexCacheKey(pattern: pattern, options: options)
        regexCacheLock.lock()
        defer { regexCacheLock.unlock() }
        if let result = regexCache[key] {
            return result
        }

        let result = try NSRegularExpression(pattern: pattern, options: options)
        regexCache[key] = result
        return result
    }
}