first commit.

1 files changed, 105 insertions, 0 deletions

diff --git a/modules/youtube.py b/modules/youtube.py
new file mode 100644
index 00000000..b5066e6f
--- /dev/null
+++ b/modules/youtube.py
@@ -0,0 +1,105 @@
+import re
+import Utils
+
+REGEX_YOUTUBE = re.compile(
+    "https?://(?:www.)?(?:youtu.be/|youtube.com/watch\?[^\S]*v=)(\w{11})",
+    re.I)
+REGEX_ISO8601 = re.compile("PT(\d+H)?(\d+M)?(\d+S)?", re.I)
+
+URL_YOUTUBESEARCH = "https://www.googleapis.com/youtube/v3/search"
+URL_YOUTUBEVIDEO = "https://www.googleapis.com/youtube/v3/videos"
+
+URL_YOUTUBESHORT = "https://youtu.be/%s"
+
+ARROW_UP = "▲"
+ARROW_DOWN = "▼"
+
+class Module(object):
+    def __init__(self, bot):
+        self.bot = bot
+        bot.events.on("received").on("command").on("yt", "youtube"
+            ).hook(self.yt,
+            help="Find a video on youtube")
+        bot.events.on("received").on("message").on("channel").hook(
+            self.channel_message)
+        bot.events.on("boot").on("done").hook(self.boot_done)
+
+    def validate_setchannel(self, s):
+        return s.lower() == "true"
+    def boot_done(self, event):
+        self.bot.events.on("postboot").on("configure").on(
+            "channelset").call(setting="autoyoutube",
+            help="Disable/Enable automatically getting info from youtube URLs",
+            validate=self.validate_setchannel)
+
+    def get_video_page(self, video_id, part):
+        return Utils.get_url(URL_YOUTUBEVIDEO, get_params={"part": part,
+            "id": video_id, "key": self.bot.config["google-api-key"]},
+            json=True)
+    def video_details(self, video_id):
+        snippet = self.get_video_page(video_id, "snippet")
+        if snippet["items"]:
+            snippet = snippet["items"][0]["snippet"]
+            statistics = self.get_video_page(video_id, "statistics")[
+                "items"][0]["statistics"]
+            content = self.get_video_page(video_id, "contentDetails")[
+                "items"][0]["contentDetails"]
+            video_uploader = snippet["channelTitle"]
+            video_title = snippet["title"]
+            video_views = statistics["viewCount"]
+            video_likes = statistics["likeCount"]
+            video_dislikes = statistics["dislikeCount"]
+            video_duration = content["duration"]
+
+            match = re.match(REGEX_ISO8601, video_duration)
+            video_duration = ""
+            video_duration += "%s:" % match.group(1)[:-1].zfill(2
+                ) if match.group(1) else ""
+            video_duration += "%s:" % match.group(2)[:-1].zfill(2
+                ) if match.group(2) else ""
+            video_duration += "%s" % match.group(3)[:-1].zfill(2
+                ) if match.group(3) else ""
+            return "%s (%s) uploaded by %s, %s views (%s%s%s%s) %s" % (
+                video_title, video_duration, video_uploader, "{:,}".format(
+                int(video_views)), video_likes, ARROW_UP, ARROW_DOWN, video_dislikes,
+                URL_YOUTUBESHORT % video_id)
+
+    def yt(self, event):
+        video_id = None
+        search = None
+        if event["args"]:
+            search = event["args"]
+        else:
+            last_youtube = event["channel"].log.find(REGEX_YOUTUBE)
+            if last_youtube:
+                video_id = re.search(REGEX_YOUTUBE, last_youtube.message).group(1)
+        if search or video_id:
+            if not video_id:
+                search_page = Utils.get_url(URL_YOUTUBESEARCH,
+                    get_params={"q": search, "part": "snippet",
+                    "maxResults": "1", "type": "video",
+                    "key": self.bot.config["google-api-key"]},
+                    json=True)
+                if search_page:
+                    if search_page["pageInfo"]["totalResults"] > 0:
+                        video_id = search_page["items"][0]["id"]["videoId"]
+                    else:
+                        event["stderr"].write("No videos found")
+                else:
+                    event["stderr"].write("Failed to load results")
+            if video_id:
+                event["stdout"].write(self.video_details(video_id))
+            else:
+                event["stderr"].write("No search phrase provided")
+        else:
+           event["stderr"].write("No search phrase provided")
+
+    def channel_message(self, event):
+        match = re.search(REGEX_YOUTUBE, event["message"])
+        if match and event["channel"].get_setting("autoyoutube", False):
+            youtube_id = match.group(1)
+            video_details = self.video_details(youtube_id)
+            if video_details:
+                self.bot.events.on("send").on("stdout").call(target=event[
+                    "channel"], message=video_details, module_name="Youtube",
+                    server=event["server"])