nim-datastore/datastore/sql.nim

import std/times
import std/options

import pkg/chronos
import pkg/questionable
import pkg/questionable/results
import pkg/sqlite3_abi
from pkg/stew/results as stewResults import isErr
import pkg/upraises

import std/sequtils
import ../datastore
import ./threads/backend
import ./threads/sqlbackend

import pkg/chronos
import pkg/taskpools

import ./threads/sqlbackend
import ./threads/threadproxyds
import ./datastore

export datastore, Taskpool

push: {.upraises: [].}

type
  SQLiteDatastore* = ref object of Datastore
    db: SQLiteBackend[KeyId, DataBuffer]

proc path*(self: SQLiteDatastore): string =
  self.db.path()

proc readOnly*(self: SQLiteDatastore): bool =
  self.db.readOnly()

method has*(self: SQLiteDatastore,
            key: Key): Future[?!bool] {.async.} =
  return self.db.has(KeyId.new key.id())

method delete*(self: SQLiteDatastore,
               key: Key): Future[?!void] {.async.} =
  return self.db.delete(KeyId.new key.id())

method delete*(self: SQLiteDatastore,
               keys: seq[Key]): Future[?!void] {.async.} =
  let dkeys = keys.mapIt(KeyId.new it.id())
  return self.db.delete(dkeys)

method get*(self: SQLiteDatastore,
            key: Key): Future[?!seq[byte]] {.async.} =
  self.db.get(KeyId.new key.id()).map() do(d: DataBuffer) -> seq[byte]:
    d.toSeq()

method put*(self: SQLiteDatastore,
            key: Key,
            data: seq[byte]): Future[?!void] {.async.} =
  self.db.put(KeyId.new key.id(), DataBuffer.new data)

method put*(self: SQLiteDatastore,
            batch: seq[BatchEntry]): Future[?!void] {.async.} =
  var dbatch: seq[tuple[key: KeyId, data: DataBuffer]]
  for entry in batch:
    dbatch.add((KeyId.new entry.key.id(), DataBuffer.new entry.data))
  self.db.put(dbatch)

method close*(self: SQLiteDatastore): Future[?!void] {.async.} =
  self.db.close()

method queryIter*(self: SQLiteDatastore,
                  query: Query
                 ): ?!(iterator(): ?!QueryResponse) =


proc new*(
  T: type SQLiteDatastore,
  path: string,
  readOnly = false): ?!SQLiteDatastore =

  success SQLiteDatastore(
    db: ? newSQLiteBackend[KeyId, DataBuffer](path, readOnly))

proc new*(
  T: type SQLiteDatastore,
  db: SQLiteBackend[KeyId, DataBuffer]): ?!T =

  success T(
    db: db,
    readOnly: db.readOnly)