bitbot-3.11-fork/modules/soundcloud.py

#--depends-on commands
#--require-config soundcloud-api-key

import json, re, time
from src import ModuleManager, utils

URL_SOUNDCLOUD_TRACK = "http://api.soundcloud.com/tracks"
URL_SOUNDCLOUD_RESOLVE = "http://api.soundcloud.com/resolve"
REGEX_SOUNDCLOUD = "https?://soundcloud.com/([^/]+)/([^/]+)"

class Module(ModuleManager.BaseModule):
    _name = "SoundCloud"

    @utils.hook("received.command.sc", alias_of="soundcloud")
    @utils.hook("received.command.soundcloud")
    def soundcloud(self, event):
        """
        :help: Search SoundCloud
        :usage: <term>
        """
        query = None
        url = None

        if event["args"]:
            match = re.search(REGEX_SOUNDCLOUD, event["args"])
            if match:
                url = match.string
            else:
                query = event["args"]
        else:
            last_soundcloud = event["target"].buffer.find(REGEX_SOUNDCLOUD)
            if last_soundcloud:
                url = last_soundcloud.match

        if not query and not url:
            raise utils.EventError("no search phrase provided")
        has_query = not query == None
        get_params = {"limit": 1,
            "client_id": self.bot.config["soundcloud-api-key"]}

        if query:
            get_params["q"] = query
        else:
            get_params["url"] = url

        page = utils.http.request(
            URL_SOUNDCLOUD_TRACK if has_query else URL_SOUNDCLOUD_RESOLVE,
            get_params=get_params, json=True)

        if page:
            if len(page.data):
                page = page.data[0] if has_query else page
                title = page["title"]
                user = page["user"]["username"]
                duration = time.strftime("%H:%M:%S", time.gmtime(page[
                    "duration"]/1000))
                if duration.startswith("00:"):
                    duration = duration[3:]
                link = page["permalink_url"]
                event["stdout"].write("%s [%s] (posted by %s) %s" % (title,
                    duration, user, link))
            else:
                event["stderr"].write("No results found")
        else:
            raise utils.EventResultsError()