-module(aarondb_ets_ffi).
-export([init_table/2, insert/2, lookup/2, delete/2, prune_eavt/3, prune_aevt/3, get_raw_binary/2]).

init_table(Name, Type) ->
    TableName = binary_to_atom(Name, utf8),
    EtsType = case Type of
        set -> set;
        ordered_set -> ordered_set;
        bag -> bag;
        duplicate_bag -> duplicate_bag
    end,
    case ets:whereis(TableName) of
        undefined ->
            ets:new(TableName, [EtsType, public, named_table, {read_concurrency, true}]);
        _ ->
            ok
    end,
    ok.

insert(Table, Object) ->
    TableName = binary_to_atom(Table, utf8),
    ets:insert(TableName, Object),
    ok.

lookup(Table, Key) ->
    TableName = binary_to_atom(Table, utf8),
    ets:lookup(TableName, Key).

delete(Table, Key) ->
    TableName = binary_to_atom(Table, utf8),
    ets:delete(TableName, Key),
    ok.

prune_eavt(Table, Key, Attr) ->
    T = binary_to_atom(Table, utf8),
    Pattern = {Key, {datom, Key, Attr, '_', '_', '_'}},
    ets:match_delete(T, Pattern),
    ok.

prune_aevt(Table, Attr, Key) ->
    T = binary_to_atom(Table, utf8),
    Pattern = {Attr, {datom, Key, Attr, '_', '_', '_'}},
    ets:match_delete(T, Pattern),
    ok.

get_raw_binary(Table, Key) ->
    TableName = binary_to_atom(Table, utf8),
    Objects = ets:lookup(TableName, Key),
    {ok, term_to_binary(Objects)}.