import Foundation
import SwiftProtobuf

public class QueryBuilder {
    private var inner: Exocore_Store_EntityQuery

    init() {
        self.inner = Exocore_Store_EntityQuery()
    }

    public static func withIds(_ ids: [String]) -> QueryBuilder {
        let builder = QueryBuilder()
        var idsPredicate = Exocore_Store_IdsPredicate()
        idsPredicate.ids = ids
        builder.inner.ids = idsPredicate
        return builder
    }

    public static func withId(_ id: String) -> QueryBuilder {
        self.withIds([id])
    }

    public static func withTrait<M: Message>(_ message: M.Type) -> QueryBuilder {
        let builder = QueryBuilder()
        var traitPredicate = Exocore_Store_TraitPredicate()
        traitPredicate.traitName = M.protoMessageName
        builder.inner.trait = traitPredicate
        return builder
    }

    public static func withTrait<M: Message>(_ message: M.Type, query: Exocore_Store_TraitQuery) -> QueryBuilder {
        let builder = QueryBuilder()
        var traitPredicate = Exocore_Store_TraitPredicate()
        traitPredicate.traitName = M.protoMessageName
        traitPredicate.query = query
        builder.inner.trait = traitPredicate
        return builder
    }

    public static func matching(query: String) -> QueryBuilder {
        let builder = QueryBuilder()
        var queryPredicate = Exocore_Store_MatchPredicate()
        queryPredicate.query = query
        builder.inner.match = queryPredicate
        return builder
    }

    public static func all() -> QueryBuilder {
        let builder = QueryBuilder()
        builder.inner.all = Exocore_Store_AllPredicate()
        return builder
    }

    public static func fromQueryString(query: String) -> QueryBuilder {
        let builder = QueryBuilder()
        var qs = Exocore_Store_QueryStringPredicate()
        qs.query = query
        builder.inner.queryString = qs
        return builder
    }

    public func count(_ count: Int) -> QueryBuilder {
        var paging = Exocore_Store_Paging()
        paging.count = UInt32(count)
        self.inner.paging = paging

        return self
    }

    public func orderByField(_ field: String, ascending: Bool = false) -> QueryBuilder {
        var ordering = Exocore_Store_Ordering()
        ordering.field = field
        ordering.ascending = ascending
        self.inner.ordering = ordering

        return self
    }

    public func orderByOperationIds(ascending: Bool = false) -> QueryBuilder {
        var ordering = Exocore_Store_Ordering()
        ordering.ascending = ascending
        ordering.operationID = true
        self.inner.ordering = ordering

        return self
    }

    public func project(withProjection projection: Exocore_Store_Projection) -> QueryBuilder {
        self.inner.projections.append(projection)

        return self
    }

    public func project(withProjections projections: [Exocore_Store_Projection]) -> QueryBuilder {
        self.inner.projections.append(contentsOf: projections)

        return self
    }

    public func includeDeleted() -> QueryBuilder {
        self.inner.includeDeleted = true

        return self
    }

    public func build() -> Exocore_Store_EntityQuery {
        self.inner
    }
}

public class TraitQueryBuilder {
    private var inner: Exocore_Store_TraitQuery

    init() {
        self.inner = Exocore_Store_TraitQuery()
    }

    public static func refersTo(field: String, entityId: String, traitId: String = "") -> TraitQueryBuilder {
        let builder = TraitQueryBuilder()
        var predicate = Exocore_Store_TraitFieldReferencePredicate()
        predicate.field = field
        predicate.reference = Exocore_Store_ReferencePredicate()
        predicate.reference.entityID = entityId
        predicate.reference.traitID = traitId
        builder.inner.reference = predicate

        return builder
    }

    public static func matching(query: String) -> TraitQueryBuilder {
        let builder = TraitQueryBuilder()
        var predicate = Exocore_Store_MatchPredicate()
        predicate.query = query
        builder.inner.match = predicate

        return builder
    }

    public static func fromQueryString(query: String) -> TraitQueryBuilder {
        let builder = TraitQueryBuilder()
        var qs = Exocore_Store_QueryStringPredicate()
        qs.query = query
        builder.inner.queryString = qs
        
        return builder
    }

    public func build() -> Exocore_Store_TraitQuery {
        self.inner
    }
}