From 8c8f3c928dd2c441bf5aa973af0d08dcf7e95ea7 Mon Sep 17 00:00:00 2001
From: Alek Ratzloff <alekratz@gmail.com>
Date: Sat, 13 Jun 2020 20:59:09 -0400
Subject: [PATCH] Add commands() for modules, which scoops up all commands
 defined.

Signed-off-by: Alek Ratzloff <alekratz@gmail.com>
---
 lib/module.ex | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/lib/module.ex b/lib/module.ex
index 20f032d..9caf556 100644
--- a/lib/module.ex
+++ b/lib/module.ex
@@ -19,6 +19,8 @@ defmodule Omnibot.Module do
 
         @impl true
         def on_init(_cfg), do: nil
+
+        def commands(), do: MapSet.to_list(@commands)
       end
     end
   end
@@ -69,7 +71,9 @@ defmodule Omnibot.Module do
             line = Enum.join(params, " ")
 
             case String.split(line, " ") do
-              [cmd | params] -> on_channel_msg(irc, channel, nick, cmd, params)
+              [cmd | params] -> if Enum.member?(commands(), cmd),
+                  do: on_channel_msg(irc, channel, nick, cmd, params),
+                  else: on_channel_msg(irc, channel, nick, line)
               _ -> on_channel_msg(irc, channel, nick, line)
             end
 
@@ -92,6 +96,7 @@ defmodule Omnibot.Module do
 
       defoverridable Module
 
+      @commands MapSet.new()
       @before_compile Omnibot.Module.Hooks
     end
   end
@@ -112,6 +117,7 @@ defmodule Omnibot.Module do
 
   defmacro command(cmd, opts) do
     quote generated: true do
+      @commands MapSet.put(@commands, unquote(cmd))
       @impl Omnibot.Module
       def on_channel_msg(var!(irc), var!(channel), var!(nick), unquote(cmd), var!(params)) do
         unquote(opts[:do])
@@ -132,6 +138,7 @@ defmodule Omnibot.Module do
       )
 
     quote generated: true do
+      @commands MapSet.put(@commands, unquote(cmd))
       @impl Omnibot.Module
       def on_channel_msg(var!(irc), var!(channel), var!(nick), unquote(cmd), unquote(params)) do
         unquote(opts[:do])