//
//  CoreStoreObject+Querying.swift
//  CoreStore
//
//  Copyright © 2017 John Rommel Estropia
//
//  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 CoreData
import Foundation


// MARK: - DynamicObject

public extension DynamicObject where Self: CoreStoreObject {
    
    /**
     Extracts the keyPath string from a `CoreStoreObject.Value` property.
     ```
     let keyPath: String = Person.keyPath { $0.nickname }
     ```
     */
    public static func keyPath<O, V>(_ attribute: (Self) -> ValueContainer<O>.Required<V>) -> String  {
        
        return attribute(self.meta).keyPath
    }
    
    /**
     Extracts the keyPath string from a `CoreStoreObject.Value` property.
     ```
     let keyPath: String = Person.keyPath { $0.nickname }
     ```
     */
    public static func keyPath<O, V>(_ attribute: (Self) -> ValueContainer<O>.Optional<V>) -> String  {
        
        return attribute(self.meta).keyPath
    }
    
    /**
     Extracts the keyPath string from a `CoreStoreObject.Relationship` property.
     ```
     let keyPath: String = Person.keyPath { $0.pets }
     ```
     */
    public static func keyPath<O, D>(_ relationship: (Self) -> RelationshipContainer<O>.ToOne<D>) -> String  {
        
        return relationship(self.meta).keyPath
    }
    
    /**
     Extracts the keyPath string from a `CoreStoreObject.Relationship` property.
     ```
     let keyPath: String = Person.keyPath { $0.pets }
     ```
     */
    public static func keyPath<O, D>(_ relationship: (Self) -> RelationshipContainer<O>.ToManyOrdered<D>) -> String  {
        
        return relationship(self.meta).keyPath
    }
    
    /**
     Extracts the keyPath string from a `CoreStoreObject.Relationship` property.
     ```
     let keyPath: String = Person.keyPath { $0.pets }
     ```
     */
    public static func keyPath<O, D>(_ relationship: (Self) -> RelationshipContainer<O>.ToManyUnordered<D>) -> String  {
        
        return relationship(self.meta).keyPath
    }
    
    /**
     Creates a `Where` clause from a `CoreStoreObject.Value` property.
     ```
     let person = CoreStore.fetchOne(From<Person>(), Person.where { $0.nickname == "John" })
     ```
     */
    public static func `where`(_ condition: (Self) -> Where) -> Where  {
        
        return condition(self.meta)
    }
    
    /**
     Creates an `OrderBy` clause from a `CoreStoreObject.Value` property.
     ```
     let person = CoreStore.fetchAll(From<Person>(), Person.orderBy(ascending: { $0.age }))
     ```
     */
    public static func orderBy<O, V>(ascending attribute: (Self) -> ValueContainer<O>.Required<V>) -> OrderBy  {
        
        return OrderBy(.ascending(attribute(self.meta).keyPath))
    }
    
    /**
     Creates an `OrderBy` clause from a `CoreStoreObject.Value` property.
     ```
     let person = CoreStore.fetchAll(From<Person>(), Person.orderBy(ascending: { $0.age }))
     ```
     */
    public static func orderBy<O, V>(ascending attribute: (Self) -> ValueContainer<O>.Optional<V>) -> OrderBy  {
        
        return OrderBy(.ascending(attribute(self.meta).keyPath))
    }
    
    /**
     Creates an `OrderBy` clause from a `CoreStoreObject.Value` property.
     ```
     let person = CoreStore.fetchAll(From<Person>(), Person.orderBy(descending: { $0.age }))
     ```
     */
    public static func orderBy<O, V>(descending attribute: (Self) -> ValueContainer<O>.Required<V>) -> OrderBy  {
        
        return OrderBy(.descending(attribute(self.meta).keyPath))
    }
    
    /**
     Creates an `OrderBy` clause from a `CoreStoreObject.Value` property.
     ```
     let person = CoreStore.fetchAll(From<Person>(), Person.orderBy(descending: { $0.age }))
     ```
     */
    public static func orderBy<O, V>(descending attribute: (Self) -> ValueContainer<O>.Optional<V>) -> OrderBy  {
        
        return OrderBy(.descending(attribute(self.meta).keyPath))
    }
    
    
    // MARK: Deprecated
    
    @available(*, deprecated, renamed: "orderBy(ascending:)")
    public static func ascending<O, V>(_ attribute: (Self) -> ValueContainer<O>.Optional<V>) -> OrderBy  {
        
        return OrderBy(.ascending(attribute(self.meta).keyPath))
    }
    
    @available(*, deprecated, renamed: "orderBy(descending:)")
    public static func descending<O, V>(_ attribute: (Self) -> ValueContainer<O>.Optional<V>) -> OrderBy  {
        
        return OrderBy(.descending(attribute(self.meta).keyPath))
    }
}


// MARK: - ValueContainer.Required

public extension ValueContainer.Required {
    
    /**
     Creates a `Where` clause from a `CoreStoreObject.Value` property.
     ```
     let person = CoreStore.fetchOne(From<Person>(), Person.where { $0.nickname == "John" })
     ```
     */
    @inline(__always)
    public static func == (_ attribute: ValueContainer<O>.Required<V>, _ value: V) -> Where {
        
        return Where(attribute.keyPath, isEqualTo: value)
    }
    
    /**
     Creates a `Where` clause from a `CoreStoreObject.Value` property.
     ```
     let person = CoreStore.fetchOne(From<Person>(), Person.where { $0.nickname != "John" })
     ```
     */
    @inline(__always)
    public static func != (_ attribute: ValueContainer<O>.Required<V>, _ value: V) -> Where {
        
        return !Where(attribute.keyPath, isEqualTo: value)
    }
    
    /**
     Creates a `Where` clause from a `CoreStoreObject.Value` property.
     ```
     let person = CoreStore.fetchOne(From<Person>(), Person.where { $0.age < 20 })
     ```
     */
    @inline(__always)
    public static func < (_ attribute: ValueContainer<O>.Required<V>, _ value: V) -> Where {
        
        return Where("%K < %@", attribute.keyPath, value)
    }
    
    /**
     Creates a `Where` clause from a `CoreStoreObject.Value` property.
     ```
     let person = CoreStore.fetchOne(From<Person>(), Person.where { $0.age > 20 })
     ```
     */
    @inline(__always)
    public static func > (_ attribute: ValueContainer<O>.Required<V>, _ value: V) -> Where {
        
        return Where("%K > %@", attribute.keyPath, value)
    }
    
    /**
     Creates a `Where` clause from a `CoreStoreObject.Value` property.
     ```
     let person = CoreStore.fetchOne(From<Person>(), Person.where { $0.age <= 20 })
     ```
     */
    @inline(__always)
    public static func <= (_ attribute: ValueContainer<O>.Required<V>, _ value: V) -> Where {
        
        return Where("%K <= %@", attribute.keyPath, value)
    }
    
    /**
     Creates a `Where` clause from a `CoreStoreObject.Value` property.
     ```
     let person = CoreStore.fetchOne(From<Person>(), Person.where { $0.age >= 20 })
     ```
     */
    @inline(__always)
    public static func >= (_ attribute: ValueContainer<O>.Required<V>, _ value: V) -> Where {
        
        return Where("%K >= %@", attribute.keyPath, value)
    }
    
    /**
     Creates a `Where` clause from a `CoreStoreObject.Value` property.
     ```
     let dog = CoreStore.fetchOne(From<Dog>(), Dog.where { ["Pluto", "Snoopy", "Scooby"] ~= $0.nickname })
     ```
     */
    @inline(__always)
    public static func ~= <S: Sequence>(_ sequence: S, _ attribute: ValueContainer<O>.Required<V>) -> Where where S.Iterator.Element == V {
        
        return Where(attribute.keyPath, isMemberOf: sequence)
    }
}


// MARK: - ValueContainer.Optional

public extension ValueContainer.Optional {
    
    /**
     Creates a `Where` clause from a `CoreStoreObject.Value` property.
     ```
     let person = CoreStore.fetchOne(From<Person>(), Person.where { $0.nickname == "John" })
     ```
     */
    @inline(__always)
    public static func == (_ attribute: ValueContainer<O>.Optional<V>, _ value: V?) -> Where {
        
        return Where(attribute.keyPath, isEqualTo: value)
    }
    
    /**
     Creates a `Where` clause from a `CoreStoreObject.Value` property.
     ```
     let person = CoreStore.fetchOne(From<Person>(), Person.where { $0.nickname != "John" })
     ```
     */
    @inline(__always)
    public static func != (_ attribute: ValueContainer<O>.Optional<V>, _ value: V?) -> Where {
        
        return !Where(attribute.keyPath, isEqualTo: value)
    }
    
    /**
     Creates a `Where` clause from a `CoreStoreObject.Value` property.
     ```
     let person = CoreStore.fetchOne(From<Person>(), Person.where { $0.age < 20 })
     ```
     */
    @inline(__always)
    public static func < (_ attribute: ValueContainer<O>.Optional<V>, _ value: V?) -> Where {
        
        if let value = value {
            
            return Where("%K < %@", attribute.keyPath, value)
        }
        else {
            
            return Where("%K < nil", attribute.keyPath)
        }
    }
    
    /**
     Creates a `Where` clause from a `CoreStoreObject.Value` property.
     ```
     let person = CoreStore.fetchOne(From<Person>(), Person.where { $0.age > 20 })
     ```
     */
    @inline(__always)
    public static func > (_ attribute: ValueContainer<O>.Optional<V>, _ value: V?) -> Where {
        
        if let value = value {
            
            return Where("%K > %@", attribute.keyPath, value)
        }
        else {
            
            return Where("%K > nil", attribute.keyPath)
        }
    }
    
    /**
     Creates a `Where` clause from a `CoreStoreObject.Value` property.
     ```
     let person = CoreStore.fetchOne(From<Person>(), Person.where { $0.age <= 20 })
     ```
     */
    @inline(__always)
    public static func <= (_ attribute: ValueContainer<O>.Optional<V>, _ value: V?) -> Where {
        
        if let value = value {
            
            return Where("%K <= %@", attribute.keyPath, value)
        }
        else {
            
            return Where("%K <= nil", attribute.keyPath)
        }
    }
    
    /**
     Creates a `Where` clause from a `CoreStoreObject.Value` property.
     ```
     let person = CoreStore.fetchOne(From<Person>(), Person.where { $0.age >= 20 })
     ```
     */
    @inline(__always)
    public static func >= (_ attribute: ValueContainer<O>.Optional<V>, _ value: V?) -> Where {
        
        if let value = value {
            
            return Where("%K >= %@", attribute.keyPath, value)
        }
        else {
            
            return Where("%K >= nil", attribute.keyPath)
        }
    }
    
    /**
     Creates a `Where` clause from a `CoreStoreObject.Value` property.
     ```
     let dog = CoreStore.fetchOne(From<Dog>(), Dog.where { ["Pluto", "Snoopy", "Scooby"] ~= $0.nickname })
     ```
     */
    @inline(__always)
    public static func ~= <S: Sequence>(_ sequence: S, _ attribute: ValueContainer<O>.Optional<V>) -> Where where S.Iterator.Element == V {
        
        return Where(attribute.keyPath, isMemberOf: sequence)
    }
}


// MARK: - RelationshipContainer.ToOne

public extension RelationshipContainer.ToOne {
    
    /**
     Creates a `Where` clause from a `CoreStoreObject.Relationship` property.
     ```
     let dog = CoreStore.fetchOne(From<Dog>(), Dog.where { $0.master == me })
     ```
     */
    @inline(__always)
    public static func == (_ relationship: RelationshipContainer<O>.ToOne<D>, _ object: D?) -> Where {
        
        return Where(relationship.keyPath, isEqualTo: object)
    }
    
    /**
     Creates a `Where` clause from a `CoreStoreObject.Relationship` property.
     ```
     let dog = CoreStore.fetchOne(From<Dog>(), Dog.where { $0.master != me })
     ```
     */
    @inline(__always)
    public static func != (_ relationship: RelationshipContainer<O>.ToOne<D>, _ object: D?) -> Where {
        
        return !Where(relationship.keyPath, isEqualTo: object)
    }
    
    /**
     Creates a `Where` clause from a `CoreStoreObject.Relationship` property.
     ```
     let dog = CoreStore.fetchOne(From<Dog>(), Dog.where { $0.master ~= me })
     ```
     */
    @inline(__always)
    public static func ~= (_ relationship: RelationshipContainer<O>.ToOne<D>, _ object: D?) -> Where {
        
        return Where(relationship.keyPath, isEqualTo: object)
    }
    
    /**
     Creates a `Where` clause from a `CoreStoreObject.Relationship` property.
     ```
     let dog = CoreStore.fetchOne(From<Dog>(), Dog.where { [john, joe, bob] ~= $0.master })
     ```
     */
    @inline(__always)
    public static func ~= <S: Sequence>(_ sequence: S, _ relationship: RelationshipContainer<O>.ToOne<D>) -> Where where S.Iterator.Element == D {
        
        return Where(relationship.keyPath, isMemberOf: sequence)
    }
}