weechat/scripts/python/autoauth.py
2006-09-16 20:48:42 +00:00

246 lines
8.6 KiB
Python

# -*- coding: iso-8859-1 -*-
# =============================================================================
# autoauth.py (c) October 2005 by kolter <kolter+dev@openics.org>
# Python script for WeeChat.
#
# Licence : GPL v2
# Description : Permits to auto-authenticate when changing nick
# Syntax : try /auth help to get help on this script
#
#
# ### changelog ###
#
# * version 0.5
# - fix bug when script script is run for first time
# - rewrite half script to improve access to settings
# - add a feature to permit to run command(s) when identified
# - add completion for commands
# * version 0.4
# - use set_plugin_config and get_plugin_config to read ans save settings
# - remove deprecated import
# * version 0.3
# - add return codes
# * version 0.2
# - correct weechatdir with weechat_dir while using weechat.get_info
# * version 0.1 :
# - first release
#
# =============================================================================
VERSION="0.5"
NAME="autoauth"
import weechat
weechat.register (NAME, VERSION, "", "Auto authentification while changing nick")
weechat.add_message_handler("NOTICE", "auth_notice_check")
weechat.add_command_handler(
"auth",
"auth_command",
"Auto authentification while changing nick",
"{ add $nick $pass [$server=current] | del $nick [$server=current] | list | cmd [$command [$server=current]] }",
" add : add authorization for $nick with password $pass for $server\n"
" del : del authorization for $nick for $server\n"
" list : list all authorization settings\n"
" cmd : command(s) (separated by '|') to run when identified for $server\n"
" %n will be replaced by current nick in each command",
"add|del|list|cmd %- %S %S"
)
def auth_cmdlist():
cmd = ''
cmds = weechat.get_plugin_config("commands")
if cmds == '':
weechat.prnt("[%s] commands (empty)" % (NAME))
else:
weechat.prnt("[%s] commands (list)" % (NAME))
for c in cmds.split("####"):
weechat.prnt(" --> %s : '%s' " % (c.split(":::")[0], c.split(":::")[1]))
def auth_cmdget(server):
cmd = ''
cmds = weechat.get_plugin_config("commands")
if cmds != '':
for c in cmds.split("####"):
if c.find(":::") != -1:
if c.split(":::")[0] == server:
cmd = ":::".join(c.split(":::")[1:])
break
return cmd
def auth_cmdset(server, command):
cmds = weechat.get_plugin_config("commands")
found = False
conf = []
if cmds != '':
for c in cmds.split("####"):
if c.find(":::") != -1:
if c.split(":::")[0] == server:
found = True
conf.append("%s:::%s" % (server, command))
else:
conf.append(c)
if not found:
conf.append("%s:::%s" % (server, command))
weechat.set_plugin_config("commands", "####".join(conf))
weechat.prnt("[%s] command '%s' successfully added for server %s" % (NAME, command, server))
def auth_cmdunset(server):
cmds = weechat.get_plugin_config("commands")
found = False
conf = []
if cmds != '':
for c in cmds.split("####"):
if c.find(":::") != -1:
if c.split(":::")[0] != server:
conf.append(c)
else:
found = True
if found:
weechat.prnt("[%s] command for server '%s' successfully removed" % (NAME, server))
weechat.set_plugin_config("commands", "####".join(conf))
def auth_cmd(args, server):
if server == '':
if args == '':
auth_cmdlist()
else:
weechat.prnt("[%s] error while setting command, can't find a server" % (NAME))
else:
if args == '':
auth_cmdunset(server)
else:
auth_cmdset(server, args)
def auth_list():
data = weechat.get_plugin_config("data")
if data == "":
weechat.prnt("[%s] accounts (empty)" % (NAME))
else:
weechat.prnt("[%s] accounts (list)" % (NAME))
for e in data.split(","):
if e.find("=") == -1:
continue
(serv_nick, passwd) = e.split("=")
(server, nick) = serv_nick.split(".")
weechat.prnt(" --> %s@%s " % (nick, server))
def auth_notice_check(server, args):
if args.find("If this is your nickname, type /msg NickServ") != -1 or args.find("This nickname is registered and protected.") != -1 :
passwd = auth_get(weechat.get_info("nick"), server)
if passwd != None:
weechat.command("/quote nickserv identify %s" % (passwd), "", server)
commands = auth_cmdget(server)
if commands != '':
for c in commands.split("|"):
weechat.command(c.strip().replace("%n", weechat.get_info('nick')))
return weechat.PLUGIN_RC_OK
def auth_del(the_nick, the_server):
data = weechat.get_plugin_config("data")
found = False
conf = []
for e in data.split(","):
if e.find("=") == -1:
continue
(serv_nick, passwd) = e.split("=")
(server, nick) = serv_nick.split(".")
if the_nick == nick and the_server == server:
found = True
else:
conf.append("%s.%s=%s" % (server, nick, passwd))
if found:
weechat.set_plugin_config("data", ",".join(conf))
weechat.prnt("[%s] nick '%s@%s' successfully remove" % (NAME, the_nick, the_server))
else:
weechat.prnt("[%s] an error occured while removing nick '%s@%s' (not found)" % (NAME, the_nick, the_server))
def auth_add(the_nick, the_passwd, the_server):
data = weechat.get_plugin_config("data")
found = False
conf = []
for e in data.split(","):
if e.find("=") == -1:
continue
(serv_nick, passwd) = e.split("=")
(server, nick) = serv_nick.split(".")
if the_nick == nick and the_server == server:
passwd = the_passwd
found = True
conf.append("%s.%s=%s" % (server, nick, passwd))
if not found:
conf.append("%s.%s=%s" % (the_server, the_nick, the_passwd))
weechat.set_plugin_config("data", ",".join(conf))
weechat.prnt("[%s] nick '%s@%s' successfully added" % (NAME, the_nick, the_server))
def auth_get(the_nick, the_server):
data = weechat.get_plugin_config("data")
for e in data.split(","):
if e.find("=") == -1:
continue
(serv_nick, passwd) = e.split("=")
(server, nick) = serv_nick.split(".")
if the_nick == nick and the_server == server:
return passwd
return None
def auth_command(server, args):
list_args = args.split(" ")
#strip spaces
while '' in list_args:
list_args.remove('')
while ' ' in list_args:
list_args.remove(' ')
if len(list_args) == 0:
weechat.command("/help auth")
elif list_args[0] not in ["add", "del", "list", "cmd"]:
weechat.prnt("[%s] bad option while using /auth command, try '/help auth' for more info" % (NAME))
elif list_args[0] == "cmd":
if len(list_args[1:]) == 1 and list_args[1] in weechat.get_server_info().keys():
auth_cmd("", list_args[1])
elif len(list_args[1:]) == 1:
auth_cmd(list_args[1], weechat.get_info('server'))
elif len(list_args[1:]) >= 2:
if list_args[-1] in weechat.get_server_info().keys():
auth_cmd(" ".join(list_args[1:-1]), list_args[-1])
else:
auth_cmd(" ".join(list_args[1:]), weechat.get_info('server'))
else:
auth_cmd(" ".join(list_args[1:]), weechat.get_info(server))
elif list_args[0] == "list":
auth_list()
elif list_args[0] == "add":
if len(list_args) < 3 or (len(list_args) == 3 and weechat.get_info("server") == ''):
weechat.prnt("[%s] bad option while using /auth command, try '/help auth' for more info" % (NAME))
else:
if len(list_args) == 3:
auth_add(list_args[1], list_args[2], weechat.get_info("server"))
else:
auth_add(list_args[1], list_args[2], list_args[3])
elif list_args[0] == "del":
if len(list_args) < 2:
weechat.prnt("[%s] bad option while using /auth command, try '/help auth' for more info" % (NAME))
else:
if len(list_args) == 2:
auth_del(list_args[1], weechat.get_info("server"))
else:
auth_del(list_args[1], list_args[2])
else:
pass
return weechat.PLUGIN_RC_OK