Source

webcandy / src / webmstn_sup.erl

The default branch has multiple heads

%% @author sunneach http://bitbuckets.com/sunneach
%% @copyright 2009 sunneach

%% @doc Supervisor for the webmstn application.

-module(webmstn_sup).
-author('sunneach http://bitbuckets.com/sunneach').

-behaviour(supervisor).

%% External exports
-export([start_link/0, upgrade/0]).

%% supervisor callbacks
-export([init/1]).

%% @spec start_link() -> ServerRet
%% @doc API for starting the supervisor.
start_link() ->
    supervisor:start_link({local, ?MODULE}, ?MODULE, []).

%% @spec upgrade() -> ok
%% @doc Add processes if necessary.
upgrade() ->
    {ok, {_, Specs}} = init([]),

    Old = sets:from_list(
	    [Name || {Name, _, _, _} <- supervisor:which_children(?MODULE)]),
    New = sets:from_list([Name || {Name, _, _, _, _, _} <- Specs]),
    Kill = sets:subtract(Old, New),

    sets:fold(fun (Id, ok) ->
		      supervisor:terminate_child(?MODULE, Id),
		      supervisor:delete_child(?MODULE, Id),
		      ok
	      end, ok, Kill),

    [supervisor:start_child(?MODULE, Spec) || Spec <- Specs],
    ok.

%% @spec prep_dispatch(DispatchTemplate) -> Dispatch
%% @doc replace 'PATHTOWWW' -> absolute path to "./priv/www" in the Dispatch
%% @doc Example: {['*',webmstn_static_resource,['PATHTOWWW']}
%% @doc becomes {['*',webmstn_static_resource,["c:/tmp/webcandy/priv/www"]}
%% @doc if the app was cloned into c:/tmp/webcandy
prep_dispatch(DispatchTemplate) -> 
	WWWPath = filename:absname("priv/www"),
	lists:map(fun({Pattern,Resource,Paths})->
				{Pattern, Resource,
				        lists:map(fun(Code) ->
									if 
										Code =:= 'PATHTOWWW' -> WWWPath;
										true                 -> Code
									end
								end, Paths)} 
				end, DispatchTemplate ).



%% @spec init([]) -> SupervisorTree
%% @doc supervisor callback.
init([]) ->
    Ip = case os:getenv("WEBMACHINE_IP") of false -> "0.0.0.0"; Any -> Any end,   
	{ok, DispatchTemplate} = file:consult("priv/dispatch.conf"),
    io:format("************************************DispatchTmpl: ~p~n",[DispatchTemplate]),
	Dispatch = prep_dispatch(DispatchTemplate),
    io:format("Dispatch: ~p~n",[Dispatch]),
    WebConfig = [
		 {ip, Ip},
		 {port, 8000},
                 {log_dir, "priv/log"},
		 {dispatch, Dispatch
}],
    io:format("WebConfig: ~p~n",[WebConfig]),
    Web = {webmachine_mochiweb,
	   {webmachine_mochiweb, start, [WebConfig]},
	   permanent, 5000, worker, dynamic},
    Processes = [Web],
    io:format("Processes: ~p~n",[Processes]),
    {ok, {{one_for_one, 10, 10}, Processes}}.