bitbot-3.11-fork/modules/yourls.py

import urllib.parse
from src import ModuleManager, utils

def _parse(s):
    parsed = urllib.parse.urlparse(s)
    return urllib.parse.urljoin(s, parsed.path), parsed.query

SETTING = utils.FunctionSetting(_parse, "yourls",
    "Set YOURLS server (and token) to use for URL shortening",
    example="https://bitbot.dev/yourls-api.php?1002a612b4",
    format=utils.sensitive_format)

@utils.export("botset", SETTING)
@utils.export("serverset", SETTING)
@utils.export("channelset", SETTING)
class Module(ModuleManager.BaseModule):
    @utils.export("shorturl-x-yourls")
    def _shorturl(self, server, context, url):
        setting = server.get_setting("yourls",
            self.bot.get_setting("yourls", None))
        if context:
            setting = context.get_setting("yourls", setting)

        if not setting == None:
            shortener_url, token = setting

            page = utils.http.request(shortener_url, get_params={
                "signature": token,
                "action": "shorturl",
                "url": url,
                "format": "json"}).json()
            if page:
                return page["shorturl"]
            return None