65 lines
2.1 KiB
Python
65 lines
2.1 KiB
Python
#--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()
|