Add auto-github-timeout to prevent duplicate auto-githubs

1 files changed, 37 insertions, 17 deletions

diff --git a/modules/github/__init__.py b/modules/github/__init__.py
index b38c652c..c7150478 100644
--- a/modules/github/__init__.py
+++ b/modules/github/__init__.py
@@ -113,6 +113,9 @@ CHECK_RUN_FAILURES = ["failure", "cancelled", "timed_out", "action_required"]
 @utils.export("channelset", {"setting": "auto-github",
     "help": "Enable/disable automatically getting github issue/PR info",
     "validate": utils.bool_or_none, "example": "on"})
+@utils.export("channelset", {"setting": "auto-github-timeout",
+    "help": "Set amount of seconds between auto-github duplicates",
+    "validate": utils.int_or_none, "example": "300"})
 class Module(ModuleManager.BaseModule):
     def _parse_ref(self, channel, ref):
         repo, _, number = ref.rpartition("#")
@@ -235,6 +238,20 @@ class Module(ModuleManager.BaseModule):
         else:
             event["stderr"].write("Issue/PR not found")
 
+    def _cache_ref(self, ref):
+        return "auto-github-%s" % ref.lower()
+    def _auto_github_timeout(self, channel, ref):
+        timeout = channel.get_setting("auto-github-timeout", None)
+        if not timeout == None:
+            cache = self._cache_ref(ref)
+            if not self.bot.cache.has_item(cache):
+                self.bot.cache.temporary_cache(cache, timeout)
+                return True
+            else:
+                return False
+        else:
+            return True
+
     @utils.hook("command.regex")
     def url_regex(self, event):
         """
@@ -245,14 +262,15 @@ class Module(ModuleManager.BaseModule):
             event.eat()
             ref = "%s/%s#%s" % (event["match"].group(1),
                 event["match"].group(2), event["match"].group(4))
-            try:
-                result = self._get_info(event["target"], ref)
-            except utils.EventError:
-                return
-            if result:
-                if event["target"].get_setting("github-hide-prefix", False):
-                    event["stdout"].hide_prefix()
-                event["stdout"].write(result)
+            if self._auto_github_timeout(event["target"], ref):
+                try:
+                    result = self._get_info(event["target"], ref)
+                except utils.EventError:
+                    return
+                if result:
+                    if event["target"].get_setting("github-hide-prefix", False):
+                        event["stdout"].hide_prefix()
+                    event["stdout"].write(result)
 
     @utils.hook("command.regex")
     def ref_regex(self, event):
@@ -262,15 +280,17 @@ class Module(ModuleManager.BaseModule):
         """
         if event["target"].get_setting("auto-github", False):
             event.eat()
-            try:
-                result = self._get_info(event["target"],
-                    event["match"].group(0))
-            except utils.EventError:
-                return
-            if result:
-                if event["target"].get_setting("github-hide-prefix", False):
-                   event["stdout"].hide_prefix()
-                event["stdout"].write(result)
+            ref = event["match"].group(0)
+            if self._auto_github_timeout(event["target"], ref):
+                try:
+                    result = self._get_info(event["target"],
+                        event["match"].group(0))
+                except utils.EventError:
+                    return
+                if result:
+                    if event["target"].get_setting("github-hide-prefix", False):
+                       event["stdout"].hide_prefix()
+                    event["stdout"].write(result)
 
     @utils.hook("received.command.ghwebhook", min_args=1, channel_only=True)
     def github_webhook(self, event):