40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
import re
|
|
from src import ModuleManager, utils
|
|
|
|
REGEX_URL = re.compile("https?://\S+", re.I)
|
|
|
|
class Module(ModuleManager.BaseModule):
|
|
@utils.hook("received.command.t", alias_of="title")
|
|
@utils.hook("received.command.title", usage="[URL]")
|
|
def title(self, event):
|
|
"""
|
|
:help: Get the title of a URL
|
|
:usage: [URL]
|
|
"""
|
|
url = None
|
|
if len(event["args"]) > 0:
|
|
url = event["args_split"][0]
|
|
else:
|
|
url = event["target"].buffer.find(REGEX_URL)
|
|
if url:
|
|
url = re.search(REGEX_URL, url.message).group(0)
|
|
if not url:
|
|
raise utils.EventError("No URL provided/found.")
|
|
|
|
try:
|
|
page = utils.http.request(url, soup=True)
|
|
except Exception as e:
|
|
self.log.error("failed to get URL title", exc_info=True)
|
|
page = None
|
|
pass
|
|
|
|
if not page:
|
|
raise utils.EventError("Failed to get URL.")
|
|
|
|
title = page.data.title
|
|
if title:
|
|
title = title.text.replace("\n", " ").replace("\r", ""
|
|
).replace(" ", " ").strip()
|
|
event["stdout"].write(title)
|
|
else:
|
|
event["stderr"].write("No title found.")
|