iOS/Sources/Shared/Common/Extensions/Version+HA.swift

import Foundation
import Version

public extension Version {
    private static func replacements() throws -> [(regex: NSRegularExpression, replacement: String)] {
        try [
            (regex: NSRegularExpression(pattern: #"\.([a-zA-Z])"#), replacement: #"-$1"#),
            (regex: NSRegularExpression(pattern: #"([0-9])([a-zA-Z])"#), replacement: #"$1-$2"#),
        ]
    }

    init(hassVersion: String) throws {
        let sanitized = try Self.replacements().reduce(into: hassVersion) { result, pair in
            result = pair.regex.stringByReplacingMatches(
                in: result,
                range: NSRange(location: 0, length: result.count),
                withTemplate: pair.replacement
            )
        }

        let parser = VersionParser(strict: false)
        self = try parser.parse(string: sanitized)
    }

    func compare(buildOf other: Version) -> ComparisonResult {
        // Build can effectively be a sub-version
        guard let buildVersion = build.flatMap({ try? Version(hassVersion: $0) }),
              let otherBuildVersion = other.build.flatMap({ try? Version(hassVersion: $0) }) else {
            return .orderedAscending
        }

        if buildVersion < otherBuildVersion {
            return .orderedAscending
        } else if buildVersion == otherBuildVersion {
            return .orderedSame
        } else {
            return .orderedDescending
        }
    }
}