mirror of
https://github.com/nikdoof/kosbot.git
synced 2025-12-11 17:42:15 +00:00
85 lines
2.6 KiB
Python
85 lines
2.6 KiB
Python
import os
|
|
import logging
|
|
import sleekxmpp
|
|
import requests
|
|
try:
|
|
from urlparse import urljoin
|
|
except ImportError:
|
|
from urllib.parse import urljoin
|
|
|
|
class KOSCheckBot(sleekxmpp.ClientXMPP):
|
|
|
|
KOS_API_URL = 'http://kos.cva-eve.org/api/'
|
|
|
|
def __init__(self, jid, password, rooms, nickname):
|
|
sleekxmpp.ClientXMPP.__init__(self, jid, password)
|
|
self.rooms = rooms
|
|
self.nickname = nickname
|
|
|
|
self.add_event_handler('session_start', self.session_start)
|
|
self.add_event_handler('groupchat_message', self.muc_message)
|
|
|
|
def session_start(self, event):
|
|
for room in self.rooms:
|
|
self.plugin['xep_0045'].joinMUC(room, self.nickname, wait=True)
|
|
|
|
def muc_message(self, msg):
|
|
if msg['mucnick'] == self.nickname or msg['body'][0] != '!':
|
|
return
|
|
|
|
cmd = msg['body'].split(' ')[0][1:]
|
|
args = msg['body'].split(' ')[1:]
|
|
|
|
if cmd == 'kos':
|
|
result = self.cmd_kos(msg, cmd, args)
|
|
|
|
msg.reply(result).send()
|
|
|
|
def cmd_kos(self, msg, cmd, args):
|
|
arg = ' '.join(args)
|
|
resp = requests.get(self.KOS_API_URL, params={
|
|
'c': 'json',
|
|
'q': arg,
|
|
'type': 'unit',
|
|
'details': None
|
|
})
|
|
if resp.status_code != requests.codes.ok:
|
|
return "Something went wrong (Error %s)" % resp.status_code
|
|
try:
|
|
data = resp.json()
|
|
except:
|
|
return "KOS API returned invalid data."
|
|
if data['message'] != 'OK':
|
|
return "KOS API returned an error."
|
|
if data['total'] == 0:
|
|
return "KOS returned no results (Not on KOS)"
|
|
|
|
results = []
|
|
for result in data['results']:
|
|
text = '{} ({}) - {}'.format(
|
|
result['label'],
|
|
result['type'],
|
|
'KOS' if result['kos'] else 'Not KOS'
|
|
)
|
|
results.append(text)
|
|
return '\n'.join(results)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
logging.basicConfig(level=logging.DEBUG)
|
|
|
|
jid = os.environ.get('KOSBOT_JID')
|
|
password = os.environ.get('KOSBOT_PASSWORD')
|
|
nickname = os.environ.get('KOSBOT_NICKNAME', 'KOSBot')
|
|
rooms = os.environ.get('KOSBOT_ROOMS')
|
|
|
|
rooms = [x.strip() for x in rooms.split(',')]
|
|
|
|
bot = KOSCheckBot(jid, password, rooms, nickname)
|
|
bot.register_plugin('xep_0030') # Service Discovery
|
|
bot.register_plugin('xep_0045') # Multi-User Chat
|
|
bot.register_plugin('xep_0199') # XMPP Ping
|
|
|
|
if bot.connect():
|
|
bot.process(block=True)
|