from ..adapters.couchdb import CouchDB
from .base import NoSQLDialect
from . import dialects


@dialects.register_for(CouchDB)
class CouchDBDialect(NoSQLDialect):
    def _and(self, first, second, query_env={}):
        return "(%s && %s)" % (
            self.expand(first, query_env=query_env),
            self.expand(second, query_env=query_env),
        )

    def _or(self, first, second, query_env={}):
        return "(%s || %s)" % (
            self.expand(first, query_env=query_env),
            self.expand(second, query_env=query_env),
        )

    def eq(self, first, second=None, query_env={}):
        if second is None:
            return "(%s == null)" % self.expand(first, query_env=query_env)
        return "(%s == %s)" % (
            self.expand(first, query_env=query_env),
            self.expand(second, first.type, query_env=query_env),
        )

    def ne(self, first, second=None, query_env={}):
        if second is None:
            return "(%s != null)" % self.expand(first, query_env=query_env)
        return "(%s != %s)" % (
            self.expand(first, query_env=query_env),
            self.expand(second, first.type, query_env=query_env),
        )

    def comma(self, first, second, query_env={}):
        return "%s + %s" % (
            self.expand(first, query_env=query_env),
            self.expand(second, query_env=query_env),
        )