macula_mdns

%% Copyright (c) 2012-2016 Peter Morgan <peter.james.morgan@gmail.com>
%%
%% Licensed under the Apache License, Version 2.0 (the "License");
%% you may not use this file except in compliance with the License.
%% You may obtain a copy of the License at
%%
%% http://www.apache.org/licenses/LICENSE-2.0
%%
%% Unless required by applicable law or agreed to in writing, software
%% distributed under the License is distributed on an "AS IS" BASIS,
%% WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
%% See the License for the specific language governing permissions and
%% limitations under the License.

-module(mdns_sup).
-behaviour(supervisor).

-export([init/1]).
-export([start_link/0]).
-export([worker/1]).

start_link() ->
    supervisor:start_link({local, ?MODULE}, ?MODULE, []).

init([]) ->
    Procs = [supervisor(mdns_discover_sup), supervisor(mdns_advertise_sup)] ++ manage() ++ mesh(),
    {ok, {#{}, Procs}}.


manage() ->
    [worker(mdns_manage) || mdns_config:can(discover)].

mesh() ->
    [worker(mdns_erlang_tcp_mesh) || mdns_config:can(mesh)].


supervisor(Module) ->
    supervisor(Module, permanent).

supervisor(Module, Restart) ->
    supervisor(Module, Restart, []).

supervisor(Module, Restart, Parameters) ->
    supervisor(Module, Module, Restart, Parameters).

supervisor(Id, Module, Restart, Parameters) ->
    #{id => Id,
      start => {Module, start_link, Parameters},
      restart => Restart,
      type => supervisor,
      shutdown => infinity}.

worker(Module) ->
    worker(Module, transient).

worker(Module, Restart) ->
    worker(Module, Restart, []).

worker(Module, Restart, Parameters) ->
    worker(Module, Module, Restart, Parameters).

worker(Id, Module, Restart, Parameters) ->
    #{id => Id,
      start => {Module, start_link, Parameters},
      restart => Restart,
      shutdown => 5000}.