tippero/tipbot/command_manager.py

217 lines
6.1 KiB
Python
Raw Normal View History

#!/bin/python
#
# Cryptonote tipbot - commands
2015-01-01 17:43:01 +00:00
# Copyright 2014,2015 moneromooo
#
# The Cryptonote tipbot is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License as published
# by the Free Software Foundation; either version 2, or (at your option)
# any later version.
#
import tipbot.config as config
2014-12-31 10:30:07 +00:00
from tipbot.utils import *
modules = dict()
commands = dict()
calltable=dict()
idles = []
2015-01-01 10:06:09 +00:00
cleanup = dict()
2015-01-13 12:28:05 +00:00
def SendToProxy(link,msg):
link.send(msg)
2015-01-13 12:28:05 +00:00
def RunRegisteredCommand(link,ifyes,yesdata,ifno,nodata):
if link.identity() not in calltable:
calltable[link.identity()] = []
calltable[link.identity()].append([link,ifyes,yesdata,ifno,nodata])
if link.network.is_identified(link):
RunNextCommand(link,True)
else:
2015-01-13 12:28:05 +00:00
link.network.identify(link)
2015-01-13 12:28:05 +00:00
def IsAdmin(link):
return link.identity() in config.admins
2015-01-13 12:28:05 +00:00
def RunAdminCommand(link,ifyes,yesdata,ifno,nodata):
if not IsAdmin(link):
log_warn('RunAdminCommand: %s is not admin, cannot call %s with %s' % (str(link.identity()),str(ifyes),str(yesdata)))
link.send("Access denied")
return
2015-01-13 12:28:05 +00:00
RunRegisteredCommand(link,ifyes,yesdata,ifno,nodata)
2015-01-13 12:28:05 +00:00
def RunNextCommand(link,registered):
identity = link.identity()
if identity not in calltable:
log_error('Nothing in queue for %s' % identity)
return
try:
2015-01-13 12:28:05 +00:00
link=calltable[identity][0][0]
if registered:
2015-01-13 12:28:05 +00:00
calltable[identity][0][1](link,calltable[identity][0][2])
else:
2015-01-13 12:28:05 +00:00
calltable[identity][0][3](link,calltable[identity][0][4])
del calltable[identity][0]
except Exception, e:
log_error('RunNextCommand: Exception in action, continuing: %s' % str(e))
2015-01-13 12:28:05 +00:00
del calltable[identity][0]
2015-01-13 12:28:05 +00:00
def Commands(link,cmd):
if IsAdmin(link):
all = True
else:
all = False
module_name = GetParam(cmd,1)
if module_name:
2015-01-13 12:28:05 +00:00
link.send_private("Commands for %s's %s module:" % (config.tipbot_name,module_name))
else:
2015-01-13 12:28:05 +00:00
link.send_private("Commands for %s (use !commands <modulename> for help about the module's commands):" % config.tipbot_name)
msgs = dict()
for command_name in commands:
for c in commands[command_name]:
if 'admin' in c and c['admin'] and not all:
continue
module = c['module']
if module_name:
if module_name != module:
continue
synopsis = c['name']
if 'parms' in c:
synopsis = synopsis + " " + c['parms']
2015-01-13 12:28:05 +00:00
link.send_private("%s - %s" % (synopsis, c['help']))
else:
if module in msgs:
msgs[module] = msgs[module] +(", ")
else:
msgs[module] = module + " module: "
msgs[module] = msgs[module] +(c['name'])
if not module_name:
for msg in msgs:
2015-01-13 12:28:05 +00:00
link.send_private("%s" % msgs[msg])
def RegisterModule(module):
if module['name'] in modules:
log_error('a module named %s is already registered' % module['name'])
return
modules[module['name']] = module
2015-01-11 09:47:24 +00:00
def GetModuleNameList(admin):
if admin:
all = True
else:
all = False
module_names = []
for command_name in commands:
for c in commands[command_name]:
if 'admin' in c and c['admin'] and not all:
continue
module = c['module']
if not module in module_names:
module_names.append(module)
return module_names
def RegisterCommand(command):
2015-01-01 17:43:01 +00:00
if command['name'] in commands:
log_warn('module %s redefined function %s from module %s' % (command['module'],command['name'],commands[command['name']][0]['module']))
else:
commands[command['name']] = []
commands[command['name']].append(command)
2015-01-01 10:06:09 +00:00
def RegisterIdleFunction(module,function):
idles.append((module,function))
def RegisterCleanupFunction(module,function):
cleanup.append((module,function))
2015-01-13 12:28:05 +00:00
def OnCommand(link,cmd,check_admin,check_registered):
cmdparts = cmd[0].split(':')
log_log('cmdparts: %s' % str(cmdparts))
if len(cmdparts) == 2:
modulename = cmdparts[0]
cmdname = cmdparts[1]
elif len(cmdparts) == 1:
modulename = None
cmdname = cmdparts[0]
else:
2015-01-13 12:28:05 +00:00
link.send("Invalid command, try !help")
return
log_log('modulename: %s, cmdname: %s' % (str(modulename),str(cmdname)))
if cmdname in commands:
log_log('%s found in commands' % (str(cmd[0])))
if len(commands[cmdname]) > 1:
if not modulename:
msg = ""
for c in commands[cmdname]:
if msg != "":
msg = msg + ", "
msg = msg + c['module'] + ":" + cmd[0]
2015-01-13 12:28:05 +00:00
link.send("Ambiguous command, try one of: %s" % msg)
return
c = None
for command in commands[cmdname]:
if command['module'] == modulename:
c = command
break
if not c:
2015-01-13 12:28:05 +00:00
link.send("Invalid command, try !help")
return
else:
c = commands[cmdname][0]
if 'admin' in c and c['admin']:
2015-01-13 12:28:05 +00:00
check_admin(link,c['function'],cmd,SendToProxy,"You must be admin")
elif 'registered' in c and c['registered']:
2015-01-13 12:28:05 +00:00
check_registered(link,c['function'],cmd,SendToProxy,"You must be registered with Freenode")
else:
2015-01-13 12:28:05 +00:00
c['function'](link,cmd)
else:
2015-01-13 12:28:05 +00:00
link.send("Invalid command, try !help")
2015-01-13 12:28:05 +00:00
def RunIdleFunctions(param=None):
for f in idles:
try:
2015-01-01 10:06:09 +00:00
f[1](param)
except Exception,e:
2015-01-01 10:06:09 +00:00
log_error("Exception running idle function %s from module %s: %s" % (str(f[1]),str(f[2]),str(e)))
2015-01-13 12:28:05 +00:00
def RunModuleHelpFunction(module,link):
if module in modules:
2015-01-01 14:23:34 +00:00
try:
2015-01-13 12:28:05 +00:00
modules[module]['help'](link)
2015-01-01 14:23:34 +00:00
except Exception,e:
log_error("Exception running help function %s from module %s: %s" % (str(modules[module]['help']),str(module),str(e)))
else:
2015-01-13 12:28:05 +00:00
link.send_private('No help found for module %s' % module)
2015-01-01 14:23:34 +00:00
def UnregisterModule(module):
2015-01-01 10:06:09 +00:00
global commands
global idles
2015-01-01 10:06:09 +00:00
if module in cleanup:
cleanup[module]()
del cleanup[module]
if module in modules:
del modules[module]
2015-01-01 10:06:09 +00:00
new_idles = []
for f in idles:
if f[0] != module:
new_idles.append(f)
idles = new_idles
new_commands = dict()
for cmd in commands:
newlist = []
for c in commands[cmd]:
if c['module'] != module:
newlist.append(c)
if len(newlist) > 0:
new_commands[cmd] = newlist
2015-01-01 10:06:09 +00:00
commands = new_commands