Added Peertube (pt) module
This commit is contained in:
parent
8b655663b5
commit
eff9c4153c
12
README.md
12
README.md
|
@ -500,6 +500,18 @@ For example, if a wood lathe is broken, create issue with labels
|
||||||
Repository name must be in format TampereHacklab/Inventaario - you can
|
Repository name must be in format TampereHacklab/Inventaario - you can
|
||||||
use this as a example to see how the labels work.
|
use this as a example to see how the labels work.
|
||||||
|
|
||||||
|
### PeerTube search
|
||||||
|
|
||||||
|
Searches PeerTube instances for videos. Uses Sepia Serch at https://sepiasearch.org/
|
||||||
|
by default, but you can set any single instance to search on.
|
||||||
|
|
||||||
|
#### Usage
|
||||||
|
|
||||||
|
* !pt [query] - Search for exactly one video matching query
|
||||||
|
* !ptall [query] - Search up to 15 videos matching query
|
||||||
|
* !pt setinstance [url] - Set instance url, must end with / (example: https://peertube.cpy.re/)
|
||||||
|
* !pt showinstance - Print which instance is used
|
||||||
|
|
||||||
## Bot setup
|
## Bot setup
|
||||||
|
|
||||||
* Create a Matrix user
|
* Create a Matrix user
|
||||||
|
|
|
@ -0,0 +1,78 @@
|
||||||
|
from typing import Text
|
||||||
|
import urllib
|
||||||
|
import urllib.request
|
||||||
|
from urllib.parse import urlencode, quote_plus
|
||||||
|
import json
|
||||||
|
import time
|
||||||
|
|
||||||
|
from modules.common.module import BotModule
|
||||||
|
|
||||||
|
class PeerTubeClient:
|
||||||
|
def __init__(self):
|
||||||
|
self.instance_url = 'https://sepiasearch.org/'
|
||||||
|
|
||||||
|
def search(self, search_string, count=0):
|
||||||
|
if count == 0:
|
||||||
|
count = 15 # Pt default, could also remove from params..
|
||||||
|
params = urlencode({'search': search_string, 'count': count}, quote_via=quote_plus)
|
||||||
|
search_url = self.instance_url + 'api/v1/search/videos?' + params
|
||||||
|
response = urllib.request.urlopen(search_url)
|
||||||
|
data = json.loads(response.read().decode("utf-8"))
|
||||||
|
return data
|
||||||
|
|
||||||
|
class MatrixModule(BotModule):
|
||||||
|
def matrix_start(self, bot):
|
||||||
|
super().matrix_start(bot)
|
||||||
|
self.add_module_aliases(bot, ['ptall'])
|
||||||
|
if not self.instance_url:
|
||||||
|
instance_url = 'https://sepiasearch.org/'
|
||||||
|
|
||||||
|
async def matrix_message(self, bot, room, event):
|
||||||
|
args = event.body.split()
|
||||||
|
if len(args) == 3:
|
||||||
|
if args[1] == "setinstance":
|
||||||
|
bot.must_be_owner(event)
|
||||||
|
self.instance_url = args[2]
|
||||||
|
bot.save_settings()
|
||||||
|
await bot.send_text(room, 'Instance url set to ' + self.instance_url, bot_ignore=True)
|
||||||
|
return
|
||||||
|
|
||||||
|
if len(args) == 2:
|
||||||
|
if args[1] == "showinstance":
|
||||||
|
await bot.send_text(room, 'Using instance at ' + self.instance_url, bot_ignore=True)
|
||||||
|
return
|
||||||
|
|
||||||
|
if len(args) > 1:
|
||||||
|
query = event.body[len(args[0])+1:]
|
||||||
|
p = PeerTubeClient()
|
||||||
|
p.instance_url = self.instance_url
|
||||||
|
count = 1
|
||||||
|
if args[0] == '!ptall':
|
||||||
|
count = 0
|
||||||
|
data = p.search(query, count)
|
||||||
|
if len(data['data']) > 0:
|
||||||
|
for video in data['data']:
|
||||||
|
video_url = video["url"] #self.instance_url + 'videos/watch/' + video["uuid"]
|
||||||
|
duration = time.strftime('%H:%M:%S', time.gmtime(video["duration"]))
|
||||||
|
instancedata = video["account"]["host"]
|
||||||
|
html = f'<a href="{video_url}">{video["name"]}</a> {video["description"] or ""} [{duration}] @ {instancedata}'
|
||||||
|
text = f'{video_url} : {video["name"]} {video.get("description") or ""} [{duration}]'
|
||||||
|
await bot.send_html(room, html, text, bot_ignore=True)
|
||||||
|
else:
|
||||||
|
await bot.send_text(room, 'Sorry, no videos found found.', bot_ignore=True)
|
||||||
|
|
||||||
|
else:
|
||||||
|
await bot.send_text(room, 'Usage: !pt <query> or !ptall <query> to return all results')
|
||||||
|
|
||||||
|
def get_settings(self):
|
||||||
|
data = super().get_settings()
|
||||||
|
data['instance_url'] = self.instance_url
|
||||||
|
return data
|
||||||
|
|
||||||
|
def set_settings(self, data):
|
||||||
|
super().set_settings(data)
|
||||||
|
if data.get("instance_url"):
|
||||||
|
self.instance_url = data["instance_url"]
|
||||||
|
|
||||||
|
def help(self):
|
||||||
|
return ('PeerTube search')
|
Loading…
Reference in New Issue