X-Git-Url: https://jfr.im/git/erebus.git/blobdiff_plain/d1ea05b06625b91b6d4fc9580fff461b47b3570c..662003f2c6a877b3628d3e95f1d9f385cbb27107:/modules/eval.py

diff --git a/modules/eval.py b/modules/eval.py
index 04f08fa..77f6a1a 100644
--- a/modules/eval.py
+++ b/modules/eval.py
@@ -1,22 +1,71 @@
+# Erebus IRC bot - Author: Erebus Team
+# vim: fileencoding=utf-8
+# !EVAL and !EXEC commands
+# This file is released into the public domain; see http://unlicense.org/
+
+# module info
+modinfo = {
+	'author': 'Erebus Team',
+	'license': 'public domain',
+	'compatible': [0],
+	'depends': [],
+	'softdeps': ['help'],
+}
+
 # preamble
 import modlib
 lib = modlib.modlib(__name__)
 modstart = lib.modstart
 modstop = lib.modstop
 
-#module code
+# module code
 import sys
+import ctlmod
+
+
+def module(name):
+	return lib.mod(name)
 
+@lib.hook('eval', needchan=False, wantchan=True, glevel=lib.OWNER)
+@lib.help("<python>", "eval")
+@lib.argsGE(1)
+def cmd_eval(bot, user, chan, realtarget, *args):
+	if chan is not None: replyto = chan
+	else: replyto = user
 
-@lib.hook('eval')
-def cmd_eval(bot, user, chan, *args):
 	try: ret = eval(' '.join(args))
-	except: bot.msg(chan, "Error (%s): %s" % (sys.exc_info()[0], sys.exc_info()[1]))
-	else: bot.msg(chan, "Done: %r" % (ret))
+	except Exception: bot.msg(replyto, "Error: %s %s" % (sys.exc_info()[0], sys.exc_info()[1]))
+	else: bot.msg(replyto, "Done: %r" % (ret,))
+
+
+@lib.hook('exec', needchan=False, wantchan=True, glevel=lib.OWNER)
+@lib.help("<python>", "exec")
+@lib.argsGE(1)
+def cmd_exec(bot, user, chan, realtarget, *args):
+	if chan is not None: replyto = chan
+	else: replyto = user
+
+	try: exec(' '.join(args))
+	except Exception: bot.msg(replyto, "Error: %s %s" % (sys.exc_info()[0], sys.exc_info()[1]))
+	else: bot.msg(replyto, "Done.")
+
+@lib.hook('exception', needchan=False, glevel=lib.OWNER)
+@lib.help(None, "cause an exception")
+def cmd_exception(*args, **kwargs):
+	raise Exception()
 
+@lib.hook(needchan=False, wantchan=True, glevel=lib.ADMIN)
+@lib.help("<nick> <message>", "inject a line as though it came from <nick>", "note that this injects lines, not commands", "ex: INJECT DimeCadmium !WHOAMI")
+def inject(bot, user, chan, realtarget, *args):
+	targetuser = bot.parent.user(args[0], create=False)
+	if targetuser is None:
+		bot.msg(user, "User is unknown.")
+		return
+	if targetuser.glevel > user.glevel:
+		bot.msg(user, "That user has a higher access level than you.")
+		return
 
-@lib.hook('exec')
-def cmd_exec(bot, user, chan, *args):
-	try: exec ' '.join(args)
-	except: bot.msg(chan, "Error: %s %s" % (sys.exc_info()[0], sys.exc_info()[1]))
-	else: bot.msg(chan, "Done.")
+	if chan is not None:
+		bot.parsemsg(bot.parent.user(args[0], create=False), str(chan), ' '.join(args[1:]))
+	else:
+		bot.parsemsg(bot.parent.user(args[0], create=False), str(bot), ' '.join(args[1:]))