-- mod_storage_lmdb

-- Copyright (C) 2015 Kim Alvefur

--

-- This file is MIT/X11 licensed.

-- 

-- Depends on lightningdbm

-- https://github.com/shmul/lightningdbm

--

-- luacheck: globals prosody open

local lmdb = require"lightningmdb";

local lfs = require"lfs";

local path = require"util.paths";

local serialization = require"util.serialization";

local serialize = serialization.serialize;

local deserialize = serialization.deserialize;

local keyval = {};

local keyval_mt = { __index = keyval, flags = lmdb.MDB_CREATE };

function keyval:set(user, value)

	local t = self.env:txn_begin(nil, 0);

	if type(value) == "table" and next(value) == nil then

		value = nil;

	end

	if value ~= nil then

		value = serialize(value);

	end

	local ok, err;

	if value ~= nil then

		ok, err = t:put(self.db, user, value, 0);

	else

		ok, err = t:del(self.db, user, value);

	end

	if not ok then

		t:abort();

		return nil, err;

	end

	return t:commit();

end

function keyval:get(user)

	local t = self.env:txn_begin(nil, 0);

	local data, err = t:get(self.db, user, 0);

	if not data then

		t:abort();

		return nil, err;

	end

	t:commit();

	return deserialize(data);

end

local drivers = {

	keyval = keyval_mt;

}

	typ = typ or "keyval";

	local driver_mt = drivers[typ];

	if not driver_mt then

		return nil, "unsupported-store";

	end

	local t = env:txn_begin(nil, 0);

	local db = t:dbi_open(store.."_"..typ, driver_mt.flags);

	return setmetatable({ env = env, store = store, type = typ, db = db }, driver_mt);

end

end