Storm/Sources/StORM/StORMExtensions.swift

//
//  StORMExtensions.swift
//  StORM
//
//  Created by Ryan Coyne on 11/22/17.
//

import SwiftMoment
import Foundation

//MARK: - Optionals
extension Optional {
    /// This returns whether or not the optional is equal to nil.
    var isNil : Bool {
        return self == nil
    }
    /// This returns whether or not the optional is not equal to nil.
    var isNotNil : Bool {
        return self != nil
    }
    /// This returns the optional unwrapped into a boolean value.
    var boolValue : Bool? {
        if self.isNil {
            return nil
        }
        switch self {
        case is String, is String?:
            return Bool(self as! String)
        case is Int, is Int?:
            return Bool(self as! Int)
        default:
            return nil
        }
    }
    /// This returns the optionally wrapped object as a string value.
    var stringValue : String? {
        return self as? String
    }
    /// This returns the optionally wrapped object as a dictionary value.
    var dicValue : [String:Any]! {
        return self as? [String:Any] ?? [:]
    }
    /// This returns the optionally wrapped object as an array of dictionaries...value.
    var arrayDicValue : [[String:Any]]! {
        return self as? [[String:Any]] ?? []
    }
    /// This returns the optionally wrapped object as an integer value.
    var intValue : Int? {
        return self as? Int
    }
    /// This returns the optionally wrapped object as a double value.
    var doubleValue : Double? {
        return self as? Double
    }
    /// This returns the optionally wrapped object as a float value.
    var floatValue : Float? {
        return self as? Float
    }
    /// This returns the optionally wrapped object as a URL value.
    var urlValue : URL? {
        if self.isNil {
            return nil
        }
        switch self {
        case is String, is Optional<String>:
            return URL(string: self.stringValue!)
        case is URL, is URL?:
            return self as? URL
        default:
            return nil
        }
    }
}
//MARK: - Boolean
extension Bool {
    init(_ integerValue : Int) {
        if integerValue == 0 {
            self.init(false)
        } else {
            self.init(true)
        }
    }
}
//MARK: - Mirror Array:
extension Array where Iterator.Element == Mirror {
    /// This function will automatically return the created/modified values IF your model has a variable with that name.  Created/modified are intended to be integer values of UTC time since 1970.
    /// Createdby/Modifiedby fields are automatically supported at the database level in PostgresStORM.
    func allChildren(includingNilValues: Bool = false, primaryKey: String? = nil) -> [Mirror.Child] {
        var allChild : [Mirror.Child] = []
        for mirror in self {
            mirror.children.forEach({ (child) in
                // Make sure our child has a label & the string describing the value is not nil. (Making optionals supported)
                if !includingNilValues {
                    if child.label.isNotNil, String(describing: child.value) != "nil" {
                        // If we default a created/modified integer to zero we need to overwrite it here:
                        if child.label! == "created" || child.label! == "modified" {
                            var mutableChild = child
                            mutableChild.value = Int(utc().epoch())
                            allChild.append(mutableChild)
                        } else {
                            allChild.append(child)
                        }
                        // Automatic created & modified fields:
                    } else if child.label.isNotNil, child.label == "created" || child.label == "modified" {
                        var mutableChild = child
                        mutableChild.value = Int(utc().epoch())
                        allChild.append(mutableChild)
                    }
                } else {
                    if child.label.isNotNil {
                        if child.label! == "created" || child.label! == "modified" {
                            var mutableChild = child
                            mutableChild.value = Int(utc().epoch())
                            allChild.append(mutableChild)
                        } else {
                            allChild.append(child)
                        }
                    }
                }
            })
        }
        // Lets make sure if the primaryKey is set, it is the first object returned for asData/asDataDic & firstAsKey functions:
        if let keyLabel = primaryKey, allChild.first?.label != keyLabel {
            if let index = allChild.index(where: { (child) -> Bool in
                return child.label == keyLabel
            }) {
                allChild.move(at: index, to: 0)
            }
        }
        return allChild
    }
}

extension Array {
    /// This removes & inserts the object at the old index, to the new specified index.
    mutating func move(at oldIndex: Int, to newIndex: Int) {
        self.insert(self.remove(at: oldIndex), at: newIndex)
    }
}