omnibot/lib/config.ex

defmodule Omnibot.Config do
  @moduledoc """
  The full configuration for an instance of Omnibot.

  This configuration structure contains both the IRC server connection information, along with all
  plugins and their configuration.

  ## Configuration keys

    - server: the IRC server to connect to. **(required)**
    - nick: the IRC nickname to use for the bot. **(optional, default "omnibot")**
    - user: the IRC username to use for the bot. **(optional, default "omnibot")**
    - real: the IRC realname to use for the bot. **(optional, default "omnibot")**
    - port: the port to connect to the server on. **(optional, default 6667)**
    - ssl: whether to use SSL or not. Note that IRC usually uses port 6697 for
           SSL. **(optional, default false)**
    - plugins: a list of plugins to load and their configuration. **(optional, default [])**
    - plugin_paths: a list of locations to look for additional plugins. **(optional, default [])**

  ## Plugins configuration

  Inside the `plugins` key listed above, a list of plugins is expected. Plugins are either a single
  atom, or a tuple of the plugin and a keyword list of configuration.

  ### Plugin channels

  All plugins have a `channels` configuration value, determining which channels this plugin is
  active in. This value may also be set to the atom value `:all`, which indicates that it is active
  on all channels.

  At least one plugin in the configuration list must contain a list of channels, so that the bot
  knows which channels to join.

  ## Example configuration

  iex> %Omnibot.Config {
  ...>   nick: "omnibot_testing",
  ...>   server: "irc.bonerjamz.us",
  ...>   port: 6667,
  ...>   ssl: false,
  ...>   plugins: [
  ...>     {Omnibot.Contrib.OnConnect, commands: [
  ...>       ["privmsg", "nickserv", "register", "password123", "omnibot@omni.bot"],
  ...>       ["privmsg", "nickserv", "identify", "password123"]
  ...>     ]},
  ...>     {Omnibot.Contrib.Linkbot, channels: :all},
  ...>     {Omnibot.Contrib.Fortune, channels: :all},
  ...>     {Omnibot.Contrib.Wordbot, channels: ["#idleville"], ignore: ["username"]},
  ...>   ],
  ...>   plugin_paths: [{"plugins", recurse: true}]
  ...> }

  """

  alias Omnibot.Irc.Msg

  @enforce_keys [:server]
  defstruct [
    :server,
    nick: "omnibot",
    user: "omnibot",
    real: "omnibot",
    port: 6667,
    ssl: false,
    plugins: [],
    plugin_paths: []
  ]

  @doc """
  Gets all channels that the bot should join via its plugins.

  ## Parameters

      - cfg: the configuration value to operate on.

  ## Examples

  iex> cfg = %Omnibot.Config {
  ...>     server: "irc.example.com",
  ...>     plugins: [
  ...>       {ExamplePlugin, channels: ["#omnibot", "#example"]},
  ...>       {OtherPlugin, channels: ["#example"]}
  ...>     ]
  ...> }
  iex> Omnibot.Config.all_channels(cfg)
  ["#example", "#omnibot"]
  """
  def all_channels(cfg) do
    Enum.flat_map(cfg.plugins, fn
      {_, cfg} -> if cfg[:channels] in [nil, :all],
          do: [],
          else: cfg[:channels]
    end)
    |> MapSet.new()
    |> MapSet.to_list()
  end

  @doc """
  Creates an `Omnibot.Irc.Msg.Prefix` from this configuration to be sent to a server.

  ## Parameters

      - cfg: the configuration value to operate on.

  ## Examples

  iex> cfg = %Omnibot.Config { server: "irc.example.com" }
  iex> Omnibot.Config.msg_prefix(cfg)
  %Omnibot.Irc.Msg.Prefix {:nick => "omnibot", :user => "omnibot", :host => nil}
  """
  def msg_prefix(cfg) do
    %Msg.Prefix{
      nick: cfg.nick,
      user: cfg.user
    }
  end

  @doc """
  Make a new message with the given command and parameters using the given
  configuration to build the prefix.

  ## Parameters

      - cfg: the configuration value to operate on.
      - command: the IRC command to send
      - params: a list of parameters to pass to the IRC command.

  ## Examples

  iex> cfg = %Omnibot.Config { server: "irc.example.com" }
  iex> Omnibot.Config.msg(cfg, "PRIVMSG", ["#testing", "this is a test message"])
  %Omnibot.Irc.Msg{
    prefix: %Omnibot.Irc.Msg.Prefix {:nick => "omnibot", :user => "omnibot", :host => nil},
    command: "PRIVMSG",
    params: ["#testing", "this is a test message"],
  }
  """
  def msg(cfg, command, params \\ []) do
    %Msg{
      prefix: msg_prefix(cfg),
      command: command,
      params: params
    }
  end

  def channel_plugins(cfg, channel) do
    cfg.plugins
    |> Enum.filter(fn {_plug, cf} ->
      cf[:channels] == :all or channel in Keyword.get(cf, :channels, [])
    end)
  end

  def load(path) do
    with {_, bindings} <- Code.eval_file(path) do
      cfg = bindings[:config] 
      plugins = cfg.plugins
                |> Enum.map(fn
                  plug when is_atom(plug) -> {plug, plug.default_config()}
                  {plug, cfg}-> {plug, cfg ++ plug.default_config()}
                end)
         %Omnibot.Config{cfg | plugins: plugins}
    end
  end
end