111 lines
2.9 KiB
Python
Executable file
111 lines
2.9 KiB
Python
Executable file
#!/usr/bin/python
|
|
""" check_supybot_plugin -- ensure that a plugin is loaded by supybot.
|
|
|
|
Run like:
|
|
|
|
check_supybot_plugin --target fedmsg
|
|
check_supybot_plugin --target koji --debug
|
|
|
|
"""
|
|
from __future__ import print_function
|
|
|
|
from builtins import str
|
|
from builtins import map
|
|
import argparse
|
|
import sys
|
|
import socket
|
|
import string
|
|
import uuid
|
|
|
|
|
|
def process_args():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument(
|
|
'-t', '--target', default=None, dest='target',
|
|
help="Required. The plugin we're looking for."
|
|
)
|
|
parser.add_argument(
|
|
'-n', '--nick', default=None, dest='nick',
|
|
help="NICK to use when connecting to freenode.",
|
|
)
|
|
parser.add_argument(
|
|
'-d', '--debug', default=False, action='store_true',
|
|
help='Print out debug information.', dest='debug',
|
|
)
|
|
parser.add_argument(
|
|
'-H', '--host', default='irc.freenode.net',
|
|
help='Host to connect to.', dest='host',
|
|
)
|
|
parser.add_argument(
|
|
'-p', '--port', default=6667, type=int,
|
|
help='Host to connect to.', dest='port',
|
|
)
|
|
return parser.parse_args()
|
|
|
|
args = process_args()
|
|
|
|
# Use a random nick so people can't mess with us
|
|
if not args.nick:
|
|
args.nick = 'nrpe-' + str(uuid.uuid4()).split('-')[0]
|
|
|
|
name = "NRPE Bot"
|
|
readbuffer = ""
|
|
|
|
if not args.target:
|
|
print("UNKNOWN: No 'target' specified.")
|
|
sys.exit(3)
|
|
|
|
args.target = args.target.lower()
|
|
|
|
if args.debug:
|
|
print("connecting to %s/%i" % (args.host, args.port))
|
|
|
|
try:
|
|
s = socket.socket()
|
|
s.connect((args.host, args.port))
|
|
|
|
if args.debug:
|
|
print("as %s/%s (%s)" % (args.nick, args.nick, name))
|
|
|
|
s.send(("nick %s\r\n" % args.nick).encode())
|
|
s.send(("USER %s %s bla :%s\r\n" % (args.nick, args.host, name)).encode())
|
|
|
|
while 1:
|
|
readbuffer = readbuffer+s.recv(1024).decode()
|
|
temp = readbuffer.split("\n")
|
|
readbuffer = temp.pop()
|
|
|
|
for line in temp:
|
|
line = line.rstrip()
|
|
|
|
if args.debug:
|
|
print(" * ", line)
|
|
|
|
line = line.split()
|
|
|
|
if line[1] == 'MODE':
|
|
msg = "privmsg zodbot :list\r\n".encode()
|
|
if args.debug:
|
|
print("sending:")
|
|
print(" ->", msg)
|
|
s.send(msg)
|
|
|
|
if line[0] == ':zodbot!supybot@fedora/bot/zodbot':
|
|
if args.debug:
|
|
print("Got our response..")
|
|
|
|
plugins = list(map(str.lower, ' '.join(line[3:][1:]).split(', ')))
|
|
|
|
if args.target in plugins:
|
|
print("OK")
|
|
s.send("QUIT".encode())
|
|
sys.exit(0)
|
|
else:
|
|
print("CRITICAL: %r not loaded by supybot" % args.target)
|
|
s.send("QUIT".encode())
|
|
sys.exit(2)
|
|
except Exception as e:
|
|
print("UNKNOWN: ", str(e))
|
|
if args.debug:
|
|
raise
|
|
sys.exit(3)
|