diff options
| -rw-r--r-- | modules/info.py | 29 |
1 files changed, 29 insertions, 0 deletions
diff --git a/modules/info.py b/modules/info.py new file mode 100644 index 00000000..ee67c3e4 --- /dev/null +++ b/modules/info.py @@ -0,0 +1,29 @@ +import os +from src import IRCBot, ModuleManager, utils + +class Module(ModuleManager.BaseModule): + @utils.hook("received.command.version") + def version(self, event): + commit_hash = None + git_dir = os.path.join(self.bot.directory, ".git") + head_filepath = os.path.join(git_dir, "HEAD") + print(head_filepath) + if os.path.isfile(head_filepath): + ref = None + with open(head_filepath, "r") as head_file: + ref = head_file.readline().split(" ", 1)[1].strip() + + ref_filepath = os.path.join(git_dir, ref) + print(ref_filepath) + if os.path.isfile(ref_filepath): + with open(ref_filepath, "r") as ref_file: + commit_hash = ref_file.readline().strip() + + out = "Version: BitBot %s" % IRCBot.VERSION + if not commit_hash == None: + out = "%s (%s)" % (out, commit_hash[:8]) + event["stdout"].write(out) + + @utils.hook("received.command.source") + def source(self, event): + event["stdout"].write("Source: %s" % IRCBot.SOURCE) |