From 697f5fb577d1895b2a216f532ed20da64d6c8922 Mon Sep 17 00:00:00 2001
From: Stefano Pigozzi <ste.pigozzi@gmail.com>
Date: Sat, 9 Nov 2019 16:33:15 +0100
Subject: [PATCH] Add peertube command

---
 royalnet/commands/__init__.py             |  3 +-
 royalnet/commands/commanderrors.py        |  4 ++
 royalnet/packs/royal/commands/__init__.py |  2 +
 royalnet/packs/royal/commands/peertube.py | 81 +++++++++++++++++++++++
 4 files changed, 89 insertions(+), 1 deletion(-)
 create mode 100644 royalnet/packs/royal/commands/peertube.py

diff --git a/royalnet/commands/__init__.py b/royalnet/commands/__init__.py
index edc24479..b4fc1cc3 100644
--- a/royalnet/commands/__init__.py
+++ b/royalnet/commands/__init__.py
@@ -2,7 +2,7 @@ from .commandinterface import CommandInterface
 from .command import Command
 from .commanddata import CommandData
 from .commandargs import CommandArgs
-from .commanderrors import CommandError, InvalidInputError, UnsupportedError, KeyboardExpiredError
+from .commanderrors import CommandError, InvalidInputError, UnsupportedError, KeyboardExpiredError, ConfigurationError
 
 __all__ = [
     "CommandInterface",
@@ -13,4 +13,5 @@ __all__ = [
     "InvalidInputError",
     "UnsupportedError",
     "KeyboardExpiredError",
+    "ConfigurationError",
 ]
diff --git a/royalnet/commands/commanderrors.py b/royalnet/commands/commanderrors.py
index c083ced8..fbd5725b 100644
--- a/royalnet/commands/commanderrors.py
+++ b/royalnet/commands/commanderrors.py
@@ -23,3 +23,7 @@ class UnsupportedError(CommandError):
 
 class KeyboardExpiredError(CommandError):
     """A special type of exception that can be raised in keyboard handlers to mark a specific keyboard as expired."""
+
+
+class ConfigurationError(CommandError):
+    """The command is misconfigured and cannot work."""
diff --git a/royalnet/packs/royal/commands/__init__.py b/royalnet/packs/royal/commands/__init__.py
index 016d8db9..01eeda13 100644
--- a/royalnet/packs/royal/commands/__init__.py
+++ b/royalnet/packs/royal/commands/__init__.py
@@ -23,6 +23,7 @@ from .emojify import EmojifyCommand
 from .leagueoflegends import LeagueoflegendsCommand
 from .diarioquote import DiarioquoteCommand
 from .mp3 import Mp3Command
+from .peertube import PeertubeCommand
 
 # Enter the commands of your Pack here!
 available_commands = [
@@ -50,6 +51,7 @@ available_commands = [
     LeagueoflegendsCommand,
     DiarioquoteCommand,
     Mp3Command,
+    PeertubeCommand,
 ]
 
 # Don't change this, it should automatically generate __all__
diff --git a/royalnet/packs/royal/commands/peertube.py b/royalnet/packs/royal/commands/peertube.py
new file mode 100644
index 00000000..f0c60a45
--- /dev/null
+++ b/royalnet/packs/royal/commands/peertube.py
@@ -0,0 +1,81 @@
+import aiohttp
+import asyncio
+import datetime
+import logging
+import dateparser
+from royalnet.commands import *
+from royalnet.utils import telegram_escape
+
+
+log = logging.getLogger(__name__)
+
+
+class PeertubeCommand(Command):
+    name: str = "peertube"
+
+    description: str = "Guarda quando è uscito l'ultimo video su RoyalTube."
+
+    _url = r"https://pt.steffo.eu/feeds/videos.json?sort=-publishedAt&filter=local"
+
+    _ready = asyncio.Event()
+
+    _timeout = 300
+
+    _latest_date: datetime.datetime = None
+
+    _telegram_group_id = -1001153723135
+
+    def __init__(self, interface: CommandInterface):
+        super().__init__(interface)
+        if self.interface.name == "telegram":
+            self.loop.create_task(self._ready_up())
+            self.loop.create_task(self._update())
+
+    async def _get_json(self):
+        log.debug("Getting jsonfeed")
+        async with aiohttp.ClientSession() as session:
+            async with session.get(self._url) as response:
+                log.debug("Parsing jsonfeed")
+                j = await response.json()
+                log.debug("Jsonfeed parsed successfully")
+        return j
+
+    async def _send(self, message):
+        client = self.interface.bot.client
+        await self.interface.bot.safe_api_call(client.send_message,
+                                               chat_id=self._telegram_group_id,
+                                               text=telegram_escape(message),
+                                               parse_mode="HTML",
+                                               disable_webpage_preview=True)
+
+    async def _ready_up(self):
+        j = await self._get_json()
+        if j["version"] != "https://jsonfeed.org/version/1":
+            raise ConfigurationError("_url is not a jsonfeed")
+        videos = j["items"]
+        for video in reversed(videos):
+            date_modified = dateparser.parse(video["date_modified"])
+            if self._latest_date is None or date_modified > self._latest_date:
+                log.debug(f"Found newer video: {date_modified}")
+                self._latest_date = date_modified
+        self._ready.set()
+
+    async def _update(self):
+        await self._ready.wait()
+        while True:
+            j = await self._get_json()
+            videos = j["items"]
+            for video in reversed(videos):
+                date_modified = dateparser.parse(video["date_modified"])
+                if date_modified > self._latest_date:
+                    log.debug(f"Found newer video: {date_modified}")
+                    self._latest_date = date_modified
+                    await self._send(f"🆕 Nuovo video su RoyalTube!\n"
+                                     f"[b]{video['title']}[/b]\n"
+                                     f"{video['url']}")
+            await asyncio.sleep(self._timeout)
+
+    async def run(self, args: CommandArgs, data: CommandData) -> None:
+        if self.interface.name != "telegram":
+            raise UnsupportedError()
+        await data.reply(f"ℹ️ Ultimo video caricato il: [b]{self._latest_date.isoformat()}[/b]")