summaryrefslogtreecommitdiffstats
path: root/Reaktor/IRC/bot2.py
diff options
context:
space:
mode:
authorChinaman <root@chinaman>2011-09-06 19:47:58 +0200
committerChinaman <root@chinaman>2011-09-06 19:47:58 +0200
commit108f3616e3f4958752d881192ef29e5fc4c2b045 (patch)
tree3c67478c852265219b72e6e1b05467d7065b7ba8 /Reaktor/IRC/bot2.py
parentb2d65500160bcdf7abb2bf985f7da582b810e25c (diff)
parentc3bc5a6d16868c121aca780f3109155797b51d76 (diff)
Merge branch 'master' of github.com:krebscode/painload
Diffstat (limited to 'Reaktor/IRC/bot2.py')
-rwxr-xr-xReaktor/IRC/bot2.py129
1 files changed, 129 insertions, 0 deletions
diff --git a/Reaktor/IRC/bot2.py b/Reaktor/IRC/bot2.py
new file mode 100755
index 00000000..0279c0f6
--- /dev/null
+++ b/Reaktor/IRC/bot2.py
@@ -0,0 +1,129 @@
+#! /usr/bin/env python
+#
+# //Reaktor/IRC/bot2.py
+#
+
+from __future__ import print_function
+from irclib import SimpleIRCClient, ServerConnectionError, is_channel
+from sys import exit
+from os import environ as env
+import re
+
+class IRCBot(SimpleIRCClient):
+ def __init__(self, target):
+ SimpleIRCClient.__init__(self)
+ self.target = target
+
+ def on_pubmsg(self, connection, event):
+
+ def PRIVMSG(target, text):
+ self.connection.privmsg(target, text)
+
+ def ME(target, text):
+ PRIVMSG(target, 'ACTION ' + text + '')
+
+ def is_executable(x):
+ import os
+ return os.path.exists(x) and os.access(x, os.X_OK)
+
+ _nickname = connection.get_nickname()
+ _source = event.source()
+ _from = _source.split('!', 1)[0]
+ _target = event.target()
+
+ try:
+ _, _handle, _command, _argument, _ = re.split(
+ '^(\w+):\s*(\w+)(?:\s+(.*))?$', event.arguments()[0])
+ except ValueError, error:
+ if re.search(_nickname, event.arguments()[0]):
+ PRIVMSG(self.target, 'I\'m so famous')
+ return # ignore
+
+ if _handle == _nickname or _handle == 'ALL':
+
+ from os.path import realpath, dirname, join
+ from subprocess import Popen as popen, PIPE
+
+ Reaktor_dir = dirname(realpath(dirname(__file__)))
+ public_commands = join(Reaktor_dir, 'public_commands')
+ command = join(public_commands, _command)
+
+ if is_executable(command):
+
+ env = {}
+ if _argument != None:
+ env['argument'] = _argument
+
+ try:
+ p = popen([command], stdin=PIPE, stdout=PIPE, stderr=PIPE, env=env)
+ except OSError, error:
+ ME(self.target, 'is made of stupid')
+ print('OSError@%s: %s' % (command, error))
+ return
+
+ stdout, stderr = [ x[:len(x)-1] for x in
+ [ x.split('\n') for x in p.communicate()]]
+ code = p.returncode
+ pid = p.pid
+
+ print('command: %s -> %s' % (command, code))
+ [print('%s stdout: %s' % (pid, x)) for x in stdout]
+ [print('%s stderr: %s' % (pid, x)) for x in stderr]
+
+ if code == 0:
+ [PRIVMSG(self.target, x) for x in stdout]
+ [PRIVMSG(_source, x) for x in stderr]
+ else:
+ ME(self.target, 'mimimi')
+
+ else:
+ PRIVMSG(self.target, _from + ': you are made of stupid')
+
+ def on_welcome(self, connection, event):
+ print('I\'m welcome! :D joining to %s now...' % (self.target))
+ if is_channel(self.target):
+ connection.join(self.target)
+ else:
+ self.connection.privmsg(self.target, 'lol')
+ self.connection.quit('Pong timeout: 423 seconds')
+
+ def on_join(self, connection, event):
+ print('Es passiert in %s' % (self.target))
+
+ def on_disconnect(self, connection, event):
+ # TODO reconnect
+ exit(0)
+
+# retrieve the value of a [singleton] variable from a tinc.conf(5)-like file
+def getconf1(x, path):
+ from re import findall
+ pattern = '(?:^|\n)\s*' + x + '\s*=\s*(.*\w)\s*(?:\n|$)'
+ y = findall(pattern, open(path, 'r').read())
+ if len(y) < 1:
+ raise AttributeError("len(getconf1('%s', '%s') < 1)" % (x, path))
+ if len(y) > 1:
+ y = ' '.join(y)
+ raise AttributeError("len(getconf1('%s', '%s') > 1)\n ====> %s"
+ % (x, path, y))
+ return y[0]
+
+def main():
+ name = getconf1('Name', '/etc/tinc/retiolum/tinc.conf')
+ nick = str(env.get('nick', name))
+ host = str(env.get('host', 'supernode'))
+ port = int(env.get('port', 6667))
+ target = str(env.get('target', '#retiolum'))
+ print('====> irc://%s@%s:%s/%s' % (nick, host, port, target))
+
+ client = IRCBot(target)
+ try:
+ from getpass import getuser
+ client.connect(host, port, nick, username=getuser(),
+ ircname='//Reaktor running at %s.retiolum' % (name))
+ except ServerConnectionError, error:
+ print(error)
+ exit(1)
+ client.start()
+
+if __name__ == "__main__":
+ main()