%%
%% %CopyrightBegin%
%% 
%% Copyright Ericsson AB 1997-2009. All Rights Reserved.
%% 
%% The contents of this file are subject to the Erlang Public License,
%% Version 1.1, (the "License"); you may not use this file except in
%% compliance with the License. You should have received a copy of the
%% Erlang Public License along with this software. If not, it can be
%% retrieved online at http://www.erlang.org/.
%% 
%% Software distributed under the License is distributed on an "AS IS"
%% basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See
%% the License for the specific language governing rights and limitations
%% under the License.
%% 
%% %CopyrightEnd%
%%
%%
-module(asn1_db).
%-compile(export_all).
-export([dbnew/1,dbsave/2,dbload/1,dbput/3,dbget/2,dbget_all/1]).
-export([dbget_all_mod/1,dbstop/0,dbclear/0,dberase_module/1,dbstart/1,stop_server/1]).
%% internal exports
-export([dbloop0/1,dbloop/2]).

%% Db stuff
dbstart(Includes) ->    
    start_server(asn1db, asn1_db, dbloop0, [Includes]).

dbloop0(Includes) ->
    dbloop(Includes, ets:new(asn1, [set,named_table])).
	
opentab(Tab,Mod,[]) ->
    opentab(Tab,Mod,["."]);
opentab(Tab,Mod,Includes) ->
    Base = lists:concat([Mod,".asn1db"]),
    opentab2(Tab,Base,Mod,Includes,ok).

opentab2(_Tab,_Base,_Mod,[],Error) ->
    Error;
opentab2(Tab,Base,Mod,[Ih|It],_Error) ->
    File = filename:join(Ih,Base),
    case ets:file2tab(File) of
	{ok,Modtab} ->
	    ets:insert(Tab,{Mod, Modtab}),
	    {ok,Modtab};
	NewErr -> 
	    opentab2(Tab,Base,Mod,It,NewErr)
    end.


dbloop(Includes, Tab) ->
    receive
	{From,{set, Mod, K2, V}} ->
	    [{_,Modtab}] = ets:lookup(Tab,Mod),
	    ets:insert(Modtab,{K2, V}),
	    From ! {asn1db, ok},
	    dbloop(Includes, Tab);
	{From, {get, Mod, K2}} ->
	    Result = case ets:lookup(Tab,Mod) of
			 [] -> 
			     opentab(Tab,Mod,Includes);
			 [{_,Modtab}] -> {ok,Modtab}
		     end,
	    case Result of
		{ok,Newtab} ->
		    From ! {asn1db, lookup(Newtab, K2)};
		_Error ->
		    From ! {asn1db, undefined}
	    end,
	    dbloop(Includes, Tab);
	{From, {all_mod, Mod}} ->
	    [{_,Modtab}] = ets:lookup(Tab,Mod),
	    From ! {asn1db, ets:tab2list(Modtab)},
	    dbloop(Includes, Tab);
	{From, {delete_mod, Mod}} ->
	    [{_,Modtab}] = ets:lookup(Tab,Mod),
	    ets:delete(Modtab),
	    ets:delete(Tab,Mod),
	    From ! {asn1db, ok},
	    dbloop(Includes, Tab);
	{From, {save, OutFile,Mod}} ->
	    [{_,Mtab}] = ets:lookup(Tab,Mod),
	    From ! {asn1db, ets:tab2file(Mtab,OutFile)},
	    dbloop(Includes,Tab);
	{From, {load, Mod}} ->
	    Result = case ets:lookup(Tab,Mod) of
			 [] -> 
			     opentab(Tab,Mod,Includes);
			 [{_,Modtab}] -> {ok,Modtab}
		     end,
	    From ! {asn1db,Result},
	    dbloop(Includes,Tab);
	{From, {new, Mod}} ->
	    case ets:lookup(Tab,Mod) of
		[{_,Modtab}] -> 
		    ets:delete(Modtab);
		_  ->
		    true
	    end,
	    Tabname = list_to_atom(lists:concat(["asn1_",Mod])),
	    ets:new(Tabname, [set,named_table]),
	    ets:insert(Tab,{Mod,Tabname}),
	    From ! {asn1db, ok},
	    dbloop(Includes,Tab);
	{From, stop} ->
		    From ! {asn1db, ok};  %% nothing to store
	{From, clear} ->
	    ModTabList = [Mt||{_,Mt} <- ets:tab2list(Tab)],
	    lists:foreach(fun(T) -> ets:delete(T) end,ModTabList),
	    ets:delete(Tab),
	    From ! {asn1db, cleared},
	    dbloop(Includes, ets:new(asn1, [set]));
	{From,{new_includes,[NewIncludes]}} ->
	    From ! {asn1db,done},
	    dbloop(NewIncludes,Tab)
    end.


%%all(Tab, K) ->
%%    pickup(K, ets:match(Tab, {{K, '$1'}, '$2'})).
%%pickup(K, []) -> [];
%%pickup(K, [[V1,V2] |T]) ->
%%    [{{K,V1},V2} | pickup(K, T)].

lookup(Tab, K) ->
    case ets:lookup(Tab, K) of
	[] -> undefined;
	[{K,V}] -> V
    end.


dbnew(Module) -> req({new,Module}).
dbsave(OutFile,Module) -> req({save,OutFile,Module}).
dbload(Module) -> req({load,Module}).
    
dbput(Module,K,V) -> req({set, Module, K, V}).
dbget(Module,K) ->   req({get, Module, K}).
dbget_all(K) ->   req({get_all, K}).
dbget_all_mod(Mod) -> req({all_mod,Mod}).
dbstop() ->       stop_server(asn1db).
dbclear() ->      req(clear).
dberase_module({module,M})->  
    req({delete_mod, M}).

req(R) ->
    asn1db ! {self(), R},
    receive {asn1db, Reply} -> Reply end.

stop_server(Name) ->
    stop_server(Name, whereis(Name)).
stop_server(_, undefined) -> stopped;
stop_server(Name, _Pid) ->
    Name  ! {self(), stop},
    receive {Name, _} -> stopped end.


start_server(Name,Mod,Fun,Args) ->	
    case whereis(Name) of
	undefined ->
	    register(Name, spawn(Mod,Fun, Args));
	_Pid ->
	    req({new_includes,Args})
    end.