Support using docstrings as command help

1 files changed, 6 insertions, 2 deletions

diff --git a/modules/commands.py b/modules/commands.py
index 02787791..74f70e1d 100644
--- a/modules/commands.py
+++ b/modules/commands.py
@@ -179,8 +179,12 @@ class Module(object):
             if command in self.events.on("received").on(
                     "command").get_children():
                 hooks = self.events.on("received.command").on(command).get_hooks()
-                if hooks and "help" in hooks[0].kwargs:
-                    event["stdout"].write("%s: %s" % (command, hooks[0].kwargs["help"]))
+                kwargs = hooks[0].kwargs
+                help = hooks[0].kwargs.get("help", None
+                    ) or hooks[0].function.__doc__
+
+                if help:
+                    event["stdout"].write("%s: %s" % (command, help.strip()))
                 else:
                     event["stderr"].write("No help available for %s" % command)
             else: