//
// SwiftyUserDefaults
//
// Copyright (c) 2015-present Radosław Pietruszewski, Łukasz Mróz
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in all
// copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
// SOFTWARE.
//

import Foundation

/// Class important for saving and getting values from UserDefaults. Be careful when you
/// subclass your own!
open class DefaultsBridge<T> {

    public init() {}

    /// This method provides a way of saving your data in UserDefaults. Usually needed
    /// when you want to create your custom Bridge, so you'll have to override it.
    open func save(key: String, value: T?, userDefaults: UserDefaults) {
        fatalError("This Bridge wasn't subclassed! Please do so before using it in your type.")
    }

    /// This method provides a way of saving your data in UserDefaults. Usually needed
    /// when you want to create your custom Bridge, so you'll have to override it.
    open func get(key: String, userDefaults: UserDefaults) -> T? {
        fatalError("This Bridge wasn't subclassed! Please do so before using it in your type.")
    }

    /// Override this function if your data is represented differently in UserDefaults
    /// and you map it in save/get methods.
    ///
    /// For instance, if you store it as Data in UserDefaults, but your type is not Data in your
    /// defaults key, then you need to `return true` here and provide `deserialize(_:)` method as well.
    ///
    /// Similar if you store your array of type as e.g. `[String]` but the type you use is actually `[SomeClassThatHasOnlyOneStringProperty]`.
    ///
    /// See `DefaultsRawRepresentableBridge` or `DefaultsCodableBridge` for examples.
    open func isSerialized() -> Bool {
        return false
    }

    /// Override this function if you've returned `true` in `isSerialized()` method.
    ///
    /// See `isSerialized()` method description for more details.
    open func deserialize(_ object: Any) -> T? {
        fatalError("You set `isSerialized` to true, now you have to implement `deserialize` method.")
    }
}

open class DefaultsObjectBridge<T>: DefaultsBridge<T> {
    open override func save(key: String, value: T?, userDefaults: UserDefaults) {
        userDefaults.set(value, forKey: key)
    }

    open override func get(key: String, userDefaults: UserDefaults) -> T? {
        return userDefaults.object(forKey: key) as? T
    }
}

open class DefaultsArrayBridge<T: Collection>: DefaultsBridge<T> {
    open override func save(key: String, value: T?, userDefaults: UserDefaults) {
        userDefaults.set(value, forKey: key)
    }

    open override func get(key: String, userDefaults: UserDefaults) -> T? {
        return userDefaults.array(forKey: key) as? T
    }
}

open class DefaultsStringBridge: DefaultsBridge<String> {
    open override func save(key: String, value: String?, userDefaults: UserDefaults) {
        userDefaults.set(value, forKey: key)
    }

    open override func get(key: String, userDefaults: UserDefaults) -> String? {
        return userDefaults.string(forKey: key)
    }
}

open class DefaultsIntBridge: DefaultsBridge<Int> {
    open override func save(key: String, value: Int?, userDefaults: UserDefaults) {
        userDefaults.set(value, forKey: key)
    }

    open override func get(key: String, userDefaults: UserDefaults) -> Int? {
        if let int = userDefaults.number(forKey: key)?.intValue {
            return int
        }

        // Fallback for launch arguments
        if let string = userDefaults.object(forKey: key) as? String,
            let int = Int(string) {
            return int
        }

        return nil
    }
}

open class DefaultsDoubleBridge: DefaultsBridge<Double> {
    open override func save(key: String, value: Double?, userDefaults: UserDefaults) {
        userDefaults.set(value, forKey: key)
    }

    open override func get(key: String, userDefaults: UserDefaults) -> Double? {
        if let double = userDefaults.number(forKey: key)?.doubleValue {
            return double
        }

        // Fallback for launch arguments
        if let string = userDefaults.object(forKey: key) as? String,
            let double = Double(string) {
            return double
        }

        return nil
    }
}

open class DefaultsBoolBridge: DefaultsBridge<Bool> {
    open override func save(key: String, value: Bool?, userDefaults: UserDefaults) {
        userDefaults.set(value, forKey: key)
    }

    open override func get(key: String, userDefaults: UserDefaults) -> Bool? {
        // @warning we use number(forKey:) instead of bool(forKey:), because
        // bool(forKey:) will always return value, even if it's not set
        //
        // Now, let's see if there is value in defaults that converts to Bool first:
        if let bool = userDefaults.number(forKey: key)?.boolValue {
            return bool
        }

        // If not, fallback for values saved in a plist (e.g. for testing)
        // For instance, few of the string("YES", "true", "NO", "false") convert to Bool from a property list
        return (userDefaults.object(forKey: key) as? String)?.bool
    }
}

open class DefaultsDataBridge: DefaultsBridge<Data> {
    open override func save(key: String, value: Data?, userDefaults: UserDefaults) {
        userDefaults.set(value, forKey: key)
    }

    open override func get(key: String, userDefaults: UserDefaults) -> Data? {
        return userDefaults.data(forKey: key)
    }
}

open class DefaultsUrlBridge: DefaultsBridge<URL> {
    open override func save(key: String, value: URL?, userDefaults: UserDefaults) {
        userDefaults.set(value, forKey: key)
    }

    open override func get(key: String, userDefaults: UserDefaults) -> URL? {
        return userDefaults.url(forKey: key)
    }

    open override func isSerialized() -> Bool {
        return true
    }

    open override func deserialize(_ object: Any) -> URL? {
        if let object = object as? URL {
            return object
        }

        if let object = object as? Data {
            return NSKeyedUnarchiver.unarchiveObject(with: object) as? URL
        }

        if let object = object as? NSString {
            let path = object.expandingTildeInPath
            return URL(fileURLWithPath: path)
        }

        return nil
    }
}

open class DefaultsCodableBridge<T: Codable>: DefaultsBridge<T> {

    open override func save(key: String, value: T?, userDefaults: UserDefaults) {
        guard let value = value else {
            userDefaults.removeObject(forKey: key)
            return
        }
        userDefaults.set(encodable: value, forKey: key)
    }

    open override func get(key: String, userDefaults: UserDefaults) -> T? {
        guard let data = userDefaults.data(forKey: key) else {
            return nil
        }
        return deserialize(data)
    }

    open override func isSerialized() -> Bool {
        return true
    }

    open override func deserialize(_ object: Any) -> T? {
        guard let data = object as? Data else { return nil }

        return try? JSONDecoder().decode(T.self, from: data)
    }
}

open class DefaultsKeyedArchiverBridge<T>: DefaultsBridge<T> {

    open override func get(key: String, userDefaults: UserDefaults) -> T? {
        guard let data = userDefaults.data(forKey: key) else {
            return nil
        }
        return deserialize(data)
    }

    open override func save(key: String, value: T?, userDefaults: UserDefaults) {
        guard let value = value else {
            userDefaults.removeObject(forKey: key)
            return
        }

        // Needed because Quick/Nimble have min target 10.10...
        if #available(OSX 10.11, *) {
            userDefaults.set(NSKeyedArchiver.archivedData(withRootObject: value), forKey: key)
        } else {
            fatalError("Shouldn't really happen. We do not support macOS 10.10, if it happened to you please report your use-case on GitHub issues.")
        }
    }

    open override func isSerialized() -> Bool {
        return true
    }

    open override func deserialize(_ object: Any) -> T? {
        guard let data = object as? Data else { return nil }

        return NSKeyedUnarchiver.unarchiveObject(with: data) as? T
    }
}

open class DefaultsRawRepresentableBridge<T: RawRepresentable>: DefaultsBridge<T> {

    open override func get(key: String, userDefaults: UserDefaults) -> T? {
        guard let object = userDefaults.object(forKey: key) else { return nil }

        return deserialize(object)
    }

    open override func save(key: String, value: T?, userDefaults: UserDefaults) {
        userDefaults.set(value?.rawValue, forKey: key)
    }

    open override func isSerialized() -> Bool {
        return true
    }

    open override func deserialize(_ object: Any) -> T? {
        guard let rawValue = object as? T.RawValue else { return nil }

        return T(rawValue: rawValue)
    }
}

open class DefaultsRawRepresentableArrayBridge<T: Collection>: DefaultsBridge<T> where T.Element: RawRepresentable {

    open override func get(key: String, userDefaults: UserDefaults) -> T? {
        guard let object = userDefaults.array(forKey: key) else { return nil }

        return deserialize(object)
    }

    open override func save(key: String, value: T?, userDefaults: UserDefaults) {
        let raw = value?.map { $0.rawValue }

        userDefaults.set(raw, forKey: key)
    }

    open override func isSerialized() -> Bool {
        return true
    }

    open override func deserialize(_ object: Any) -> T? {
        guard let rawValue = object as? [T.Element.RawValue] else { return nil }

        return rawValue.compactMap { T.Element(rawValue: $0) } as? T
    }
}