package com.mongodb.mongosh.service

import com.mongodb.CursorType
import com.mongodb.ReadConcern
import com.mongodb.ReadConcernLevel
import com.mongodb.ReadPreference
import com.mongodb.client.AggregateIterable
import com.mongodb.client.FindIterable
import com.mongodb.client.MongoDatabase
import com.mongodb.client.MongoIterable
import com.mongodb.client.model.*
import com.mongodb.mongosh.result.CommandException
import org.bson.Document
import java.util.concurrent.TimeUnit


internal fun <T> convert(o: T,
                         converters: Map<String, (T, Any?) -> Either<T>>,
                         defaultConverter: (T, String, Any?) -> Either<T>,
                         map: Map<*, *>?): Either<T> {
    if (map == null) return Right(o)
    var accumulator = o
    for (key in map.keys) {
        if (key !is String) continue
        val value = map[key]
        val converter = converters[key]
        val res = if (converter != null) converter(accumulator, value) else defaultConverter(accumulator, key, value)
        when (res) {
            is Right -> accumulator = res.value
            is Left -> return res
        }
    }
    return Right(accumulator)
}

internal val writeConcernConverters: Map<String, (MongoDatabase, Any?) -> Either<MongoDatabase>> = mapOf(
        "w" to { db, value ->
            when (value) {
                is Number -> Right(db.withWriteConcern(db.writeConcern.withW(value.toInt())))
                is String -> Right(db.withWriteConcern(db.writeConcern.withW(value)))
                else -> Left(CommandException("w has to be a number or a string", "FailedToParse"))
            }
        },
        "j" to { db, value ->
            when (value) {
                is Boolean -> Right(db.withWriteConcern(db.writeConcern.withJournal(value)))
                is Number -> Right(db.withWriteConcern(db.writeConcern.withJournal(value != 0)))
                else -> Left(CommandException("j must be numeric or a boolean value", "FailedToParse"))
            }
        },
        "wtimeout" to { db, value ->
            when (value) {
                is Number -> Right(db.withWriteConcern(db.writeConcern.withWTimeout(value.toLong(), TimeUnit.MILLISECONDS)))
                else -> Right(db.withWriteConcern(db.writeConcern.withWTimeout(0, TimeUnit.MILLISECONDS)))
            }
        }
)

internal val writeConcernDefaultConverter = unrecognizedField<MongoDatabase>("write concern")


internal val readConcernConverter: (MongoDatabase, Any?) -> Either<MongoDatabase> = { db, value ->
    if (value is Map<*, *>) convert(db, readConcernConverters, readConcernDefaultConverter, value)
    else Left(CommandException("invalid parameter: expected an object (readConcern)", "FailedToParse"))
}

internal val readConcernConverters: Map<String, (MongoDatabase, Any?) -> Either<MongoDatabase>> = mapOf(
        typed("level", String::class.java) { db, value ->
            db.withReadConcern(ReadConcern(ReadConcernLevel.fromString(value)))
        }
)

internal val readConcernDefaultConverter = unrecognizedField<MongoDatabase>("read concern")


internal val dbConverters: Map<String, (MongoDatabase, Any?) -> Either<MongoDatabase>> = mapOf(
        typed("writeConcern", Map::class.java) { db, value ->
            convert(db, writeConcernConverters, writeConcernDefaultConverter, value).getOrThrow()
        },
        "readConcern" to readConcernConverter,
        "readPreference" to { db, value ->
            if (value is Map<*, *>) convert(db, readPreferenceConverters, readPreferenceDefaultConverter, value)
            else Left(CommandException("invalid parameter: expected an object (readPreference)", "FailedToParse"))
        }
)

internal val dbDefaultConverter = unrecognizedField<MongoDatabase>("db options")

internal fun <T> unrecognizedField(objectName: String): (T, String, Any?) -> Either<T> = { _, key, _ ->
    Left(CommandException("unrecognized $objectName field: $key", "FailedToParse"))
}

internal val readPreferenceConverters: Map<String, (MongoDatabase, Any?) -> Either<MongoDatabase>> = mapOf(
        "mode" to { db, value ->
            when (value) {
                is String -> {
                    val pref = when (value) {
                        "primary" -> ReadPreference.primary()
                        "primaryPreferred" -> ReadPreference.primaryPreferred()
                        "secondary" -> ReadPreference.secondary()
                        "secondaryPreferred" -> ReadPreference.secondaryPreferred()
                        "nearest" -> ReadPreference.nearest()
                        else -> null
                    }
                    if (pref == null) Left(IllegalArgumentException("Unknown read preference mode: $value"))
                    else Right(db.withReadPreference(pref))
                }
                else -> Left(CommandException("mode has to be a string", "FailedToParse"))
            }
        }
)

internal val readPreferenceDefaultConverter = unrecognizedField<MongoDatabase>("read preference")

internal val collationConverters: Map<String, (Collation.Builder, Any?) -> Either<Collation.Builder>> = mapOf(
        typed("locale", String::class.java) { collation, value ->
            collation.locale(value)
        },
        typed("caseLevel", Boolean::class.java) { collation, value ->
            collation.caseLevel(value)
        },
        typed("caseFirst", String::class.java) { collation, value ->
            collation.collationCaseFirst(CollationCaseFirst.fromString(value))
        },
        typed("strength", Int::class.java) { collation, value ->
            collation.collationStrength(CollationStrength.fromInt(value))
        },
        typed("numericOrdering", Boolean::class.java) { collation, value ->
            collation.numericOrdering(value)
        },
        typed("alternate", String::class.java) { collation, value ->
            collation.collationAlternate(CollationAlternate.fromString(value))
        },
        typed("maxVariable", String::class.java) { collation, value ->
            collation.collationMaxVariable(CollationMaxVariable.fromString(value))
        },
        typed("backwards", Boolean::class.java) { collation, value ->
            collation.backwards(value)
        }
)

internal val collationDefaultConverter = unrecognizedField<Collation.Builder>("collation")

internal val iterableConverters: Map<String, (MongoIterable<*>, Any?) -> Either<MongoIterable<*>>> = mapOf(
        typed("batchSize", Number::class.java) { iterable, value ->
            iterable.batchSize(value.toInt())
        },
        typed("cursor", Document::class.java) { iterable, value ->
            convert(iterable, cursorConverters, cursorDefaultConverter, value).getOrThrow() as AggregateIterable<*>
        }
)

internal val aggregateConverters: Map<String, (AggregateIterable<*>, Any?) -> Either<AggregateIterable<*>>> = iterableConverters + mapOf(
        typed("collation", Document::class.java) { iterable, value ->
            val collation = convert(Collation.builder(), collationConverters, collationDefaultConverter, value)
                    .getOrThrow()
                    .build()
            iterable.collation(collation)
        },
        typed("allowDiskUse", Boolean::class.java) { iterable, value ->
            iterable.allowDiskUse(value)
        },
        typed("maxTimeMS", Number::class.java) { iterable, value ->
            iterable.maxTime(value.toLong(), TimeUnit.MILLISECONDS)
        },
        typed("bypassDocumentValidation", Boolean::class.java) { iterable, value ->
            iterable.bypassDocumentValidation(value)
        },
        "hint" to { iterable, value ->
            when (value) {
                is String -> Right(iterable.hint(Document(value, 1)))
                is Document -> Right(iterable.hint(value))
                else -> Left(CommandException("hint must be string or object value", "TypeMismatch"))
            }
        },
        typed("comment", String::class.java) { iterable, value ->
            iterable.comment(value)
        }
)

internal val aggregateDefaultConverter = unrecognizedField<AggregateIterable<*>>("aggregate options")

operator fun <K, V, V1> Map<K, V>.plus(map: Map<K, V1>): Map<K, V1> {
    return (this.keys.asSequence() + map.keys.asSequence()).associateWith { (this[it] ?: map[it]) as V1 }
}

internal val findConverters: Map<String, (FindIterable<*>, Any?) -> Either<FindIterable<*>>> = iterableConverters + mapOf(
        typed("allowPartialResults", Boolean::class.java) { iterable, value ->
            iterable.partial(value)
        },
        typed("collation", Document::class.java) { iterable, value ->
            val collation = convert(Collation.builder(), collationConverters, collationDefaultConverter, value)
                    .getOrThrow()
                    .build()
            iterable.collation(collation)
        },
        typed("comment", String::class.java) { iterable, value ->
            iterable.comment(value)
        },
        "hint" to { iterable, value ->
            when (value) {
                is String -> Right(iterable.hint(Document(value, 1)))
                is Document -> Right(iterable.hint(value))
                else -> Left(CommandException("hint must be string or object value", "TypeMismatch"))
            }
        },
        typed("limit", Number::class.java) { iterable, value ->
            iterable.limit(value.toInt())
        },
        typed("max", Document::class.java) { iterable, value ->
            iterable.max(value)
        },
        typed("maxTimeMS", Number::class.java) { iterable, value ->
            iterable.maxTime(value.toLong(), TimeUnit.MILLISECONDS)
        },
        typed("min", Document::class.java) { iterable, value ->
            iterable.min(value)
        },
        typed("noCursorTimeout", Boolean::class.java) { iterable, value ->
            iterable.noCursorTimeout(value)
        },
        typed("oplogReplay", Boolean::class.java) { iterable, value ->
            iterable.oplogReplay(value)
        },
        typed("projection", Document::class.java) { iterable, value ->
            iterable.projection(value)
        },
        typed("returnKey", Boolean::class.java) { iterable, value ->
            iterable.returnKey(value)
        },
        typed("sort", Document::class.java) { iterable, value ->
            iterable.sort(value)
        },
        typed("skip", Number::class.java) { iterable, value ->
            iterable.skip(value.toInt())
        },
        typed("tailable", String::class.java) { iterable, value ->
            iterable.cursorType(CursorType.valueOf(value))
        }
)

internal val findDefaultConverter = unrecognizedField<FindIterable<*>>("find options")

internal val cursorConverters: Map<String, (MongoIterable<*>, Any?) -> Either<MongoIterable<*>>> = mapOf(
        typed("batchSize", Int::class.java) { iterable, v ->
            iterable.batchSize(v)
        }
)

internal val cursorDefaultConverter = unrecognizedField<MongoIterable<*>>("cursor")

internal val countOptionsConverters: Map<String, (CountOptions, Any?) -> Either<CountOptions>> = mapOf(
        typed("limit", Number::class.java) { opt, value ->
            opt.limit(value.toInt())
        },
        typed("skip", Number::class.java) { opt, value ->
            opt.skip(value.toInt())
        },
        "hint" to { opt, value ->
            when (value) {
                is String -> Right(opt.hint(Document(value, 1)))
                is Map<*, *> -> Right(opt.hint(Document(value as Map<String, Any?>)))
                else -> Left(CommandException("hint must be string or object value", "TypeMismatch"))
            }
        },
        typed("maxTimeMS", Number::class.java) { opt, value ->
            opt.maxTime(value.toLong(), TimeUnit.MILLISECONDS)
        },
        typed("collation", Map::class.java) { opt, value ->
            val collation = convert(Collation.builder(), collationConverters, collationDefaultConverter, value)
                    .getOrThrow()
                    .build()
            opt.collation(collation)
        }
)

internal val countOptionsDefaultConverter = unrecognizedField<CountOptions>("count options")

internal val estimatedCountOptionsConverters: Map<String, (EstimatedDocumentCountOptions, Any?) -> Either<EstimatedDocumentCountOptions>> = mapOf(
        typed("maxTimeMS", Number::class.java) { opt, value ->
            opt.maxTime(value.toLong(), TimeUnit.MILLISECONDS)
        }
)

internal val estimatedCountOptionsDefaultConverter = unrecognizedField<EstimatedDocumentCountOptions>("estimate count options")

internal val replaceOptionsConverters: Map<String, (ReplaceOptions, Any?) -> Either<ReplaceOptions>> = mapOf(
        typed("upsert", Boolean::class.java) { opt, value ->
            opt.upsert(value)
        },
        typed("collation", Map::class.java) { opt, value ->
            val collation = convert(Collation.builder(), collationConverters, collationDefaultConverter, value)
                    .getOrThrow()
                    .build()
            opt.collation(collation)
        },
        typed("bypassDocumentValidation", Boolean::class.java) { opt, value ->
            opt.bypassDocumentValidation(value)
        },
        "filter" to { opt, _ -> Right(opt) },
        "replacement" to { opt, _ -> Right(opt) }
)

internal val replaceOptionsDefaultConverter = unrecognizedField<ReplaceOptions>("replace options")

internal val findOneAndReplaceOptionsConverters: Map<String, (FindOneAndReplaceOptions, Any?) -> Either<FindOneAndReplaceOptions>> = mapOf(
        typed("projection", Document::class.java) { opt, value ->
            opt.projection(value)
        },
        typed("sort", Document::class.java) { opt, value ->
            opt.sort(value)
        },
        typed("maxTimeMS", Number::class.java) { opt, value ->
            opt.maxTime(value.toLong(), TimeUnit.MILLISECONDS)
        },
        typed("upsert", Boolean::class.java) { opt, value ->
            opt.upsert(value)
        },
        typed("returnOriginal", Boolean::class.java) { opt, value ->
            opt.returnDocument(if (value) ReturnDocument.BEFORE else ReturnDocument.AFTER)
        },
        typed("collation", Map::class.java) { opt, value ->
            val collation = convert(Collation.builder(), collationConverters, collationDefaultConverter, value)
                    .getOrThrow()
                    .build()
            opt.collation(collation)
        }
)

internal val findOneAndReplaceOptionsDefaultConverters = unrecognizedField<FindOneAndReplaceOptions>("find and replace options")

internal val bulkWriteOptionsConverters: Map<String, (BulkWriteOptions, Any?) -> Either<BulkWriteOptions>> = mapOf(
        typed("ordered", Boolean::class.java) { opt, value ->
            opt.ordered(value)
        },
        typed("bypassDocumentValidation", Boolean::class.java) { opt, value ->
            opt.bypassDocumentValidation(value)
        }
)

internal val bulkWriteOptionsDefaultConverter = unrecognizedField<BulkWriteOptions>("bulk write options")


internal val deleteConverters: Map<String, (DeleteOptions, Any?) -> Either<DeleteOptions>> = mapOf(
        typed("collation", Map::class.java) { opt, value ->
            val collation = convert(Collation.builder(), collationConverters, collationDefaultConverter, value)
                    .getOrThrow()
                    .build()
            opt.collation(collation)
        }
)

internal val deleteDefaultConverter = unrecognizedField<DeleteOptions>("delete options")

internal val findOneAndUpdateConverters: Map<String, (FindOneAndUpdateOptions, Any?) -> Either<FindOneAndUpdateOptions>> = mapOf(
        typed("collation", Map::class.java) { opt, value ->
            val collation = convert(Collation.builder(), collationConverters, collationDefaultConverter, value)
                    .getOrThrow()
                    .build()
            opt.collation(collation)
        },
        typed("projection", Document::class.java) { opt, value ->
            opt.projection(value)
        },
        typed("sort", Document::class.java) { opt, value ->
            opt.projection(value)
        },
        typed("maxTimeMS", Number::class.java) { opt, value ->
            opt.maxTime(value.toLong(), TimeUnit.MILLISECONDS)
        },
        typed("upsert", Boolean::class.java) { opt, value ->
            opt.upsert(value)
        },
        typed("returnOriginal", Boolean::class.java) { opt, value ->
            opt.returnDocument(if (value) ReturnDocument.BEFORE else ReturnDocument.AFTER)
        },
        typed("arrayFilters", List::class.java) { opt, value ->
            if (value.any { it !is Document }) {
                throw IllegalArgumentException("arrayFilters must be a list of objects: $value")
            }
            opt.arrayFilters(value.filterIsInstance<Document>())
        }
)

internal val findOneAndUpdateDefaultConverter = unrecognizedField<FindOneAndUpdateOptions>("find one and update options")

internal val findOneAndDeleteConverters: Map<String, (FindOneAndDeleteOptions, Any?) -> Either<FindOneAndDeleteOptions>> = mapOf(
        typed("collation", Map::class.java) { opt, value ->
            val collation = convert(Collation.builder(), collationConverters, collationDefaultConverter, value)
                    .getOrThrow()
                    .build()
            opt.collation(collation)
        },
        typed("projection", Document::class.java) { opt, value ->
            opt.projection(value)
        },
        typed("sort", Document::class.java) { opt, value ->
            opt.projection(value)
        },
        typed("maxTimeMS", Number::class.java) { opt, value ->
            opt.maxTime(value.toLong(), TimeUnit.MILLISECONDS)
        }
)

internal val findOneAndDeleteDefaultConverter = unrecognizedField<FindOneAndDeleteOptions>("find one and delete options")

internal val updateConverters: Map<String, (UpdateOptions, Any?) -> Either<UpdateOptions>> = mapOf(
        typed("collation", Map::class.java) { opt, value ->
            val collation = convert(Collation.builder(), collationConverters, collationDefaultConverter, value)
                    .getOrThrow()
                    .build()
            opt.collation(collation)
        },
        typed("upsert", Boolean::class.java) { opt, value ->
            opt.upsert(value)
        },
        typed("arrayFilters", List::class.java) { opt, value ->
            if (value.any { it !is Document }) {
                throw IllegalArgumentException("arrayFilters must be a list of objects: $value")
            }
            opt.arrayFilters(value.filterIsInstance<Document>())
        },
        typed("bypassDocumentValidation", Boolean::class.java) { opt, value ->
            opt.bypassDocumentValidation(value)
        }
)

internal val updateDefaultConverter = unrecognizedField<UpdateOptions>("update options")

internal val indexModelConverters: Map<String, (IndexModel, Any?) -> Either<IndexModel>> = mapOf(
        typed("collation", Map::class.java) { model, value ->
            val collation = convert(Collation.builder(), collationConverters, collationDefaultConverter, value)
                    .getOrThrow()
                    .build()
            IndexModel(model.keys, model.options.collation(collation))
        },
        typed("key", Map::class.java) { model, value ->
            IndexModel(value as Document, model.options)
        },
        typed("background", Boolean::class.java) { model, value ->
            IndexModel(model.keys, model.options.background(value))
        },
        typed("unique", Boolean::class.java) { model, value ->
            IndexModel(model.keys, model.options.unique(value))
        },
        typed("name", String::class.java) { model, value ->
            IndexModel(model.keys, model.options.name(value))
        },
        typed("partialFilterExpression", Map::class.java) { model, value ->
            IndexModel(model.keys, model.options.partialFilterExpression(value as Document))
        },
        typed("sparse", Boolean::class.java) { model, value ->
            IndexModel(model.keys, model.options.sparse(value))
        },
        typed("expireAfterSeconds", Number::class.java) { model, value ->
            IndexModel(model.keys, model.options.expireAfter(value.toLong(), TimeUnit.SECONDS))
        },
        typed("hidden", Boolean::class.java) { model, value ->
            IndexModel(model.keys, model.options.hidden(value))
        },
        typed("storageEngine", Map::class.java) { model, value ->
            IndexModel(model.keys, model.options.storageEngine(value as Document))
        },
        typed("weights", Map::class.java) { model, value ->
            IndexModel(model.keys, model.options.weights(value as Document))
        },
        typed("default_language", String::class.java) { model, value ->
            IndexModel(model.keys, model.options.defaultLanguage(value))
        },
        typed("language_override", String::class.java) { model, value ->
            IndexModel(model.keys, model.options.languageOverride(value))
        },
        typed("textIndexVersion", Int::class.java) { model, value ->
            IndexModel(model.keys, model.options.textVersion(value))
        },
        typed("2dsphereIndexVersion", Int::class.java) { model, value ->
            IndexModel(model.keys, model.options.sphereVersion(value))
        },
        typed("bits", Int::class.java) { model, value ->
            IndexModel(model.keys, model.options.bits(value))
        },
        typed("min", Double::class.java) { model, value ->
            IndexModel(model.keys, model.options.min(value))
        },
        typed("max", Double::class.java) { model, value ->
            IndexModel(model.keys, model.options.max(value))
        },
        typed("bucketSize", Double::class.java) { model, value ->
            IndexModel(model.keys, model.options.bucketSize(value))
        },
        typed("wildcardProjection", Map::class.java) { model, value ->
            IndexModel(model.keys, model.options.wildcardProjection(value as Document))
        }
)

internal val indexModelDefaultConverter = unrecognizedField<IndexModel>("index model")

internal val createCollectionOptionsConverters: Map<String, (CreateCollectionOptions, Any?) -> Either<CreateCollectionOptions>> = mapOf(
        typed("capped", Boolean::class.java) { opt, value ->
            opt.capped(value)
        },
        typed("autoIndexId", Boolean::class.java) { _, _ ->
            throw IllegalArgumentException("autoIndexId was deprecated and removed")
        },
        typed("size", Number::class.java) { opt, value ->
            opt.sizeInBytes(value.toLong())
        },
        typed("max", Number::class.java) { opt, value ->
            opt.maxDocuments(value.toLong())
        },
        typed("storageEngine", Document::class.java) { opt, value ->
            opt.storageEngineOptions(value)
        },
        typed("validator", Document::class.java) { opt, value ->
            opt.validationOptions.validator(value)
            opt
        },
        typed("validationLevel", String::class.java) { opt, value ->
            opt.validationOptions.validationLevel(ValidationLevel.fromString(value))
            opt
        },
        typed("validationAction", String::class.java) { opt, value ->
            opt.validationOptions.validationAction(ValidationAction.fromString(value))
            opt
        },
        typed("indexOptionDefaults", Document::class.java) { opt, value ->
            opt.indexOptionDefaults(IndexOptionDefaults().storageEngine(value))
        },
        typed("collation", Document::class.java) { opt, value ->
            val collation = convert(Collation.builder(), collationConverters, collationDefaultConverter, value)
                    .getOrThrow()
                    .build()
            opt.collation(collation)
        }
)

internal val createCollectionOptionsConverter = unrecognizedField<CreateCollectionOptions>("create collection options")

internal val createViewOptionsConverters: Map<String, (CreateViewOptions, Any?) -> Either<CreateViewOptions>> = mapOf(
        typed("collation", Document::class.java) { opt, value ->
            val collation = convert(Collation.builder(), collationConverters, collationDefaultConverter, value)
                    .getOrThrow()
                    .build()
            opt.collation(collation)
        },
        "viewOn" to { opt, _ -> Right(opt) },
        "pipeline" to { opt, _ -> Right(opt) }
)

internal val createViewOptionsConverter = unrecognizedField<CreateViewOptions>("create view options")


internal val updateOptionsConverters: Map<String, (UpdateOptions, Any?) -> Either<UpdateOptions>> = mapOf(
        typed("collation", Document::class.java) { opt, value ->
            val collation = convert(Collation.builder(), collationConverters, collationDefaultConverter, value)
                    .getOrThrow()
                    .build()
            opt.collation(collation)
        },
        typed("upsert", Boolean::class.java) { opt, value ->
            opt.upsert(value)
        },
        typed("bypassDocumentValidation", Boolean::class.java) { opt, value ->
            opt.bypassDocumentValidation(value)
        },
        typed("arrayFilters", List::class.java) { opt, value ->
            opt.arrayFilters(value.filterIsInstance<Document>())
        },
        "filter" to { opt, _ -> Right(opt) },
        "update" to { opt, _ -> Right(opt) }
)

internal val updateOptionsDefaultConverter = unrecognizedField<UpdateOptions>("update options")

internal val insertManyConverters: Map<String, (InsertManyOptions, Any?) -> Either<InsertManyOptions>> = mapOf(
        typed("ordered", Boolean::class.java) { opt, value ->
            opt.ordered(value)
        },
        typed("bypassDocumentValidation", Boolean::class.java) { opt, value ->
            opt.bypassDocumentValidation(value)
        }
)

internal val insertManyDefaultConverter = unrecognizedField<InsertManyOptions>("insert many options")

internal fun <T, C> typed(name: String, clazz: Class<C>, apply: (T, C) -> T): Pair<String, (T, Any?) -> Either<T>> =
        name to { o, value ->
            val casted = value as? C
            if (casted != null) {
                try {
                    Right(apply(o, casted))
                } catch (t: Throwable) {
                    Left<T>(t)
                }
            } else Left(CommandException("$name has to be a ${clazz.simpleName}", "TypeMismatch"))
        }