mirror of
https://github.com/krislamo/Flea
synced 2024-09-19 18:20:34 +00:00
Kris Lamoureux
037052d1c7
Added logging of all sent/received data and errors. Passwords are censored from the terminal and log file. Fixed case-sensitivity of the Control nick and channel.
223 lines
7.5 KiB
Python
223 lines
7.5 KiB
Python
# An IRC bot named Flea
|
|
# Copyright (C) 2016 Kris Lamoureux
|
|
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>
|
|
|
|
import re
|
|
|
|
# Init
|
|
controlchan = ''
|
|
chans = {}
|
|
access = []
|
|
queue = []
|
|
|
|
|
|
def chkcontrol(irc):
|
|
global access
|
|
global queue
|
|
|
|
nick = irc.pack["nick"].lower()
|
|
|
|
for person in access:
|
|
if nick == person:
|
|
return True
|
|
|
|
if queue.count(nick) == 0:
|
|
queue.append(nick)
|
|
irc.Whois(nick)
|
|
|
|
def noaccess(irc):
|
|
msg = "You don't have access to that."
|
|
irc.Notice(msg, irc.pack["nick"])
|
|
|
|
def main(irc):
|
|
global controlchan
|
|
global access
|
|
global chans
|
|
global queue
|
|
|
|
# Set control channel
|
|
controlchan = irc.config["channel"]
|
|
controlusr = irc.config["control"]
|
|
|
|
# NOTICEs are used to "control" Flea
|
|
if irc.pack["cmd"] == "NOTICE":
|
|
|
|
# Nick must match the nick in settings
|
|
# e.g. control = Kris
|
|
if irc.pack["nick"].lower() == controlusr.lower():
|
|
|
|
# Split message by spaces, for functions with
|
|
# required parameters. e.g. "<cmd> <param1> <param2>"
|
|
if irc.pack["text"].find(' ') != -1:
|
|
command = irc.pack["text"].lower().split(' ')
|
|
|
|
# Command Flea to join a channel
|
|
if command[0] == "join":
|
|
if chkcontrol(irc):
|
|
irc.Join(command[1])
|
|
else:
|
|
noaccess(irc)
|
|
|
|
# Command Flea to part
|
|
elif command[0] == "part":
|
|
if chkcontrol(irc):
|
|
irc.Part(command[1],
|
|
"Fleabot https://github.com/Kris619/Flea")
|
|
else:
|
|
noaccess(irc)
|
|
|
|
# Single word commands
|
|
else:
|
|
if irc.pack["text"] == "acc":
|
|
if chkcontrol(irc):
|
|
irc.Notice("You already have access to Control.",
|
|
irc.pack["nick"])
|
|
|
|
if irc.pack["text"] == "quit":
|
|
if chkcontrol(irc):
|
|
return "QUIT"
|
|
else:
|
|
noaccess(irc)
|
|
|
|
# Check if commanding Nick is identified
|
|
elif irc.pack["cmd"] == "307":
|
|
if irc.pack["text"] == "is identified for this nick":
|
|
|
|
usernick = irc.pack["params"][1]
|
|
|
|
# Check if Nick is in the queue for access list
|
|
if queue.count(usernick) == 1:
|
|
|
|
# Remove Nick from queue
|
|
queue.remove(usernick)
|
|
|
|
# Check if user is set to control in settings
|
|
# e.g. control = Kris
|
|
if usernick.lower() == controlusr.lower():
|
|
|
|
|
|
# Is control user inside the control channel?
|
|
if chans[controlchan.lower()].count(usernick) == 1:
|
|
|
|
# Grant access
|
|
access.append(usernick)
|
|
|
|
print usernick+" added to Access"
|
|
|
|
irc.Notice(
|
|
"You have been given access to Control.",
|
|
usernick)
|
|
else:
|
|
irc.Notice(
|
|
"You are not in the Control channel: "+controlchan,
|
|
usernick)
|
|
|
|
# Keep track of users in every channel
|
|
# "353" lists users to a channel
|
|
elif irc.pack["cmd"] == "353":
|
|
|
|
# If channel is in dictionary of channels
|
|
if chans.keys().count(irc.pack["params"][2]) == 1:
|
|
|
|
# Split list of channel users by space
|
|
# "+Kris @Flea" > ["+Kris", "@Flea"]
|
|
if irc.pack["text"].find(' ') != -1:
|
|
chan_nicks = irc.pack["text"].split(' ')
|
|
|
|
for aNick in chan_nicks:
|
|
if aNick != '':
|
|
|
|
# Remove prefix; e.g. "+Kris" > "Kris"
|
|
if re.search("^[~&@%+]", aNick):
|
|
aNick = aNick[1:]
|
|
|
|
# Add nick to it's channel
|
|
chans[irc.pack["params"][2]].append(aNick)
|
|
|
|
# Track users joining channels
|
|
elif irc.pack["cmd"] == "JOIN":
|
|
|
|
# Check to see if it is the bot that joined
|
|
if irc.pack["nick"] == irc.config["nick"]:
|
|
# Create channel user list if it doesn't exist
|
|
if chans.keys().count(irc.pack["text"]) == 0:
|
|
chans[irc.pack["text"].lower()] = [irc.pack["text"]]
|
|
|
|
# Another user joined
|
|
else:
|
|
# Add user to channel; chans["#Channel"].append("Kris")
|
|
chans[irc.pack["text"]].append(irc.pack["nick"])
|
|
|
|
# Track users parting from the channel
|
|
elif irc.pack["cmd"] == "PART":
|
|
# Remove nick from channel they parted from
|
|
chans[irc.pack["params"][0]].remove(irc.pack["nick"])
|
|
|
|
# If someone parted the control channel
|
|
# check if it was someone on the access list
|
|
if irc.pack["params"][0] == controlchan:
|
|
for usr in access:
|
|
if usr == irc.pack["nick"]:
|
|
# Someone on the access list left.
|
|
access.remove(irc.pack["nick"])
|
|
print irc.pack["nick"]+" removed from Access"
|
|
|
|
# Track users parting from the channel
|
|
elif irc.pack["cmd"] == "QUIT":
|
|
|
|
# Remove user from channels
|
|
for channel in chans.keys():
|
|
if chans[channel].count(irc.pack["nick"]) == 1:
|
|
chans[channel].remove(irc.pack["nick"])
|
|
|
|
# Remove access if user quitting had it
|
|
for usr in access:
|
|
if usr == irc.pack["nick"]:
|
|
access.remove(irc.pack["nick"])
|
|
print irc.pack["nick"]+" removed from Access"
|
|
|
|
# Track users being kicked from the channel
|
|
elif irc.pack["cmd"] == "KICK":
|
|
|
|
# Remove nick from channel they were kicked from
|
|
chans[irc.pack["params"][0]].remove(irc.pack["params"][1])
|
|
|
|
# If someone was kicked from the control channel
|
|
# check if it was someone on the access list
|
|
if irc.pack["params"][0] == controlchan:
|
|
if irc.pack["params"][1] == irc.config["nick"]:
|
|
print "Access list is erased, control channel is gone"
|
|
access = []
|
|
else:
|
|
for usr in access:
|
|
if usr == irc.pack["params"][1]:
|
|
# Someone on the access list was kicked.
|
|
access.remove(irc.pack["params"][1])
|
|
print irc.pack["params"][1]+" removed from Access"
|
|
|
|
# Track users changing their nick
|
|
elif irc.pack["cmd"] == "NICK":
|
|
|
|
for channel in chans.keys():
|
|
if channel.count(irc.pack["nick"]) == 1:
|
|
channel.remove(irc.pack["nick"])
|
|
channel.append(irc.pack["text"])
|
|
|
|
# Remove access to users changing their nick
|
|
for usr in access:
|
|
if access.count(irc.pack["nick"]) == 1:
|
|
access.remove(irc.pack["nick"])
|
|
print irc.pack["nick"]+" removed from Access"
|