[ycc] Dusting module, now named tinyurl

modules/tinyurl.py

@@ -0,0 +1,119 @@
+"""URL reducer module"""
+
+# PYTHON STUFFS #######################################################
+
+import re
+from urllib.parse import urlparse
+from urllib.parse import quote
+
+from nemubot.exception import IRCException
+from nemubot.hooks import hook
+from nemubot.message import Text
+from nemubot.tools import web
+
+
+# MODULE VARIABLES ####################################################
+
+PROVIDERS = {
+    "tinyurl": "http://tinyurl.com/api-create.php?url=",
+    "ycc": "http://ycc.fr/redirection/create/",
+}
+DEFAULT_PROVIDER = "ycc"
+
+
+# LOADING #############################################################
+
+def load(context):
+    global DEFAULT_PROVIDER
+
+    if "provider" in context.config:
+        if context.config["provider"] == "custom":
+            PROVIDERS["custom"] = context.config["provider_url"]
+        DEFAULT_PROVIDER = context.config["provider"]
+
+
+# MODULE CORE #########################################################
+
+def reduce(url):
+    """Ask YCC website to reduce given URL
+
+    Argument:
+    url -- the URL to reduce
+    """
+
+    snd_url = PROVIDERS[DEFAULT_PROVIDER] + quote(url, "/:%@&=?")
+    return web.getURLContent(snd_url)
+
+
+def gen_response(res, msg, srv):
+    if res is None:
+        raise IRCException("bad URL : %s" % srv)
+    else:
+        return Text("URL for %s: %s" % (srv, res), server=None,
+                    to=msg.to_response)
+
+
+## URL stack
+
+LAST_URLS = dict()
+
+
+@hook("msg_default")
+def parselisten(msg):
+    parseresponse(msg)
+    return None
+
+
+@hook("all_post")
+def parseresponse(msg):
+    global LAST_URLS
+    try:
+        urls = re.findall("([a-zA-Z0-9+.-]+:(?://)?[^ ]+)", msg.text)
+        for url in urls:
+            o = urlparse(url)
+            if o.scheme != "":
+                if o.netloc == "ycc.fr" or o.netloc == "tinyurl.com" or (
+                        o.netloc == "" and len(o.path) < 10):
+                    continue
+                for recv in msg.receivers:
+                    if recv not in LAST_URLS:
+                        LAST_URLS[recv] = list()
+                    LAST_URLS[recv].append(url)
+    except:
+        pass
+    return msg
+
+
+# MODULE INTERFACE ####################################################
+
+@hook("cmd_hook", "tinyurl",
+      help="Reduce any given URL",
+      help_usage={None: "Reduce the last URL said on the channel",
+                  "URL [URL ...]": "Reduce the given URL(s)"})
+def cmd_reduceurl(msg):
+    minify = list()
+
+    if not len(msg.args):
+        global LAST_URLS
+        if msg.channel in LAST_URLS and len(LAST_URLS[msg.channel]) > 0:
+            minify.append(LAST_URLS[msg.channel].pop())
+        else:
+            raise IRCException("I have no more URL to reduce.")
+
+    if len(msg.args) > 4:
+        raise IRCException("I cannot reduce as much URL at once.")
+    else:
+        minify += msg.args
+
+    res = list()
+    for url in minify:
+        o = urlparse(url, "http")
+        if o.scheme != "":
+            minief_url = reduce(url)
+            if o.netloc == "":
+                res.append(gen_response(minief_url, msg, o.scheme))
+            else:
+                res.append(gen_response(minief_url, msg, o.netloc))
+        else:
+            res.append(gen_response(None, msg, url))
+    return res

modules/ycc.py

@@ -1,96 +0,0 @@
-# coding=utf-8
-
-"""URL reducer module"""
-
-import re
-from urllib.parse import urlparse
-from urllib.parse import quote
-
-from nemubot.exception import IRCException
-from nemubot.hooks import hook
-from nemubot.message import Text
-from nemubot.tools import web
-
-nemubotversion = 3.4
-
-
-def help_full():
-    return ("!ycc [<url>]: with an argument, reduce the given <url> thanks to "
-            "ycc.fr; without argument, reduce the last URL said on the current"
-            " channel.")
-
-LAST_URLS = dict()
-
-
-def reduce(url):
-    """Ask YCC website to reduce given URL
-
-    Argument:
-    url -- the URL to reduce
-    """
-
-    snd_url = "http://ycc.fr/redirection/create/" + quote(url, "/:%@&=?")
-    return web.getURLContent(snd_url)
-
-
-def gen_response(res, msg, srv):
-    if res is None:
-        raise IRCException("mauvaise URL : %s" % srv)
-    else:
-        return Text("URL pour %s : %s" % (srv, res), server=None,
-                    to=msg.to_response)
-
-
-@hook("cmd_hook", "ycc")
-def cmd_ycc(msg):
-    minify = list()
-
-    if not len(msg.args):
-        global LAST_URLS
-        if msg.channel in LAST_URLS and len(LAST_URLS[msg.channel]) > 0:
-            minify.append(LAST_URLS[msg.channel].pop())
-        else:
-            raise IRCException("je n'ai pas d'autre URL à réduire.")
-
-    if len(msg.args) > 4:
-        raise IRCException("je ne peux pas réduire autant d'URL d'un seul coup.")
-    else:
-        minify += msg.args
-
-    res = list()
-    for url in minify:
-        o = urlparse(url, "http")
-        if o.scheme != "":
-            minief_url = reduce(url)
-            if o.netloc == "":
-                res.append(gen_response(minief_url, msg, o.scheme))
-            else:
-                res.append(gen_response(minief_url, msg, o.netloc))
-        else:
-            res.append(gen_response(None, msg, url))
-    return res
-
-
-@hook("msg_default")
-def parselisten(msg):
-    parseresponse(msg)
-    return None
-
-
-@hook("all_post")
-def parseresponse(msg):
-    global LAST_URLS
-    try:
-        urls = re.findall("([a-zA-Z0-9+.-]+:(?://)?[^ ]+)", msg.text)
-        for url in urls:
-            o = urlparse(url)
-            if o.scheme != "":
-                if o.netloc == "ycc.fr" or (o.netloc == "" and
-                                            len(o.path) < 10):
-                    continue
-                if msg.channel not in LAST_URLS:
-                    LAST_URLS[msg.channel] = list()
-                LAST_URLS[msg.channel].append(url)
-    except:
-        pass
-    return msg