forked from Stan/limnoria-wfg
96 lines
3.7 KiB
Python
96 lines
3.7 KiB
Python
|
###
|
||
|
# Copyright (c) 2014, scythetwirler
|
||
|
# Copyright (c) 2024, Stanislas Daniel Claude Dolcini
|
||
|
# All rights reserved.
|
||
|
#
|
||
|
# Redistribution and use in source and binary forms, with or without
|
||
|
# modification, are permitted provided that the following conditions are met:
|
||
|
#
|
||
|
# * Redistributions of source code must retain the above copyright notice,
|
||
|
# this list of conditions, and the following disclaimer.
|
||
|
# * Redistributions in binary form must reproduce the above copyright notice,
|
||
|
# this list of conditions, and the following disclaimer in the
|
||
|
# documentation and/or other materials provided with the distribution.
|
||
|
# * Neither the name of the author of this software nor the name of
|
||
|
# contributors to this software may be used to endorse or promote products
|
||
|
# derived from this software without specific prior written consent.
|
||
|
#
|
||
|
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||
|
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||
|
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||
|
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
|
||
|
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||
|
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||
|
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||
|
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||
|
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||
|
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||
|
# POSSIBILITY OF SUCH DAMAGE.
|
||
|
|
||
|
###
|
||
|
|
||
|
import re
|
||
|
import urllib.request
|
||
|
from html.parser import HTMLParser
|
||
|
import supybot.utils as utils
|
||
|
import supybot.plugins as plugins
|
||
|
import supybot.ircutils as ircutils
|
||
|
import supybot.callbacks as callbacks
|
||
|
import supybot.conf as conf
|
||
|
from supybot import ircmsgs
|
||
|
|
||
|
class TitleParser(HTMLParser):
|
||
|
def __init__(self):
|
||
|
super().__init__()
|
||
|
self.in_title = False
|
||
|
self.title = None
|
||
|
|
||
|
def handle_starttag(self, tag, attrs):
|
||
|
if tag.lower() == "title":
|
||
|
self.in_title = True
|
||
|
|
||
|
def handle_endtag(self, tag):
|
||
|
if tag.lower() == "title":
|
||
|
self.in_title = False
|
||
|
|
||
|
def handle_data(self, data):
|
||
|
if self.in_title:
|
||
|
self.title = data.strip()
|
||
|
|
||
|
class wfg(callbacks.Plugin):
|
||
|
"""This plugin contains random, vaguely WFG-related commands of questionable utility."""
|
||
|
def __init__(self, irc):
|
||
|
self.__parent = super(wfg, self)
|
||
|
self.__parent.__init__(irc)
|
||
|
|
||
|
def doPrivmsg(self, irc, msg):
|
||
|
nick = msg.nick
|
||
|
channel = msg.args[0]
|
||
|
payload = msg.args[1]
|
||
|
|
||
|
channels = self.registryValue('channels')
|
||
|
if channels and channel not in channels:
|
||
|
return
|
||
|
matchobj = re.search(r'#\d+', payload)
|
||
|
if matchobj is not None:
|
||
|
ticket_number = matchobj.group(0).lstrip('#')
|
||
|
ticket_url = self.registryValue('ticketUrl')
|
||
|
url = f'{ticket_url.rstrip("/")}/ticket/{ticket_number}'
|
||
|
|
||
|
# Fetch the title of the ticket page
|
||
|
try:
|
||
|
with urllib.request.urlopen(url) as response:
|
||
|
html = response.read().decode('utf-8')
|
||
|
parser = TitleParser()
|
||
|
parser.feed(html)
|
||
|
title = parser.title if parser.title else "No title found"
|
||
|
# Clean up the title if necessary
|
||
|
title = re.sub(r'\s*–\s*Wildfire Games\s*$', '', title)
|
||
|
except Exception as e:
|
||
|
return
|
||
|
|
||
|
response = f'{title} – {url}'
|
||
|
irc.queueMsg(ircmsgs.privmsg(channel, response))
|
||
|
|
||
|
Class = wfg
|