summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authortv <tv@xso>2011-09-06 13:36:28 +0200
committertv <tv@xso>2011-09-06 13:36:28 +0200
commitff45fed42361e4046cf01e9290b4ce74fff48a56 (patch)
tree3e76c04f85f253ca6c4ccfca06d3e52489d6109c
parenta5790c48a3ecb1d109770caead316013197562d3 (diff)
//Synapse: implement something like IRC-RPC
-rwxr-xr-xSynapse/bot2.py98
1 files changed, 65 insertions, 33 deletions
diff --git a/Synapse/bot2.py b/Synapse/bot2.py
index 66621ed1..0583d329 100755
--- a/Synapse/bot2.py
+++ b/Synapse/bot2.py
@@ -1,21 +1,76 @@
#! /usr/bin/env python
+from __future__ import print_function
from irclib import SimpleIRCClient, ServerConnectionError, is_channel
from sys import exit
from os import environ as env
-import tokenize
-import cStringIO
-import pprint
import re
-pp = pprint.PrettyPrinter(indent=2)
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:
+ 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
+
+ public_commands = join(realpath(dirname(__file__)), 'public_commands')
+ command = join(public_commands, _command)
+
+ if is_executable(command):
+ try:
+ p = popen([command], stdin=PIPE, stdout=PIPE, stderr=PIPE)
+ except OSError, error:
+ ME(self.target, 'I am made of stupid')
+ print('OSError@%s: %s' % (argv, 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, _from + ': ' + x) for x in stdout]
+ [PRIVMSG(_source, x) for x in stderr]
+ else:
+ ME(self.target, 'mimimi')
+
+ else:
+ ME(self.target, 'believes that ' + _from + ' is made of stupid')
+
def on_welcome(self, connection, event):
- print 'I\'m welcome! :D joining to %s now...' % (self.target)
+ print('I\'m welcome! :D joining to %s now...' % (self.target))
if is_channel(self.target):
connection.join(self.target)
else:
@@ -23,49 +78,26 @@ class IRCBot(SimpleIRCClient):
self.connection.quit('Pong timeout: 423 seconds')
def on_join(self, connection, event):
- print 'Es passiert in %s' % (self.target)
+ print('Es passiert in %s' % (self.target))
def on_disconnect(self, connection, event):
+ # TODO reconnect
exit(0)
-
- def on_pubmsg(self, connection, event):
- arguments = ''.join(event.arguments())
- nickname = connection.get_nickname()
- server_name = connection.get_server_name()
- is_connected = connection.is_connected()
- #readline = cStringIO.StringIO(''.join(event.arguments())).readline
- #self.connection.privmsg(self.target, 'nickname: ' + nickname)
- #self.connection.privmsg(self.target, 'server_name: ' + server_name)
-
- arguments = re.split(':\s+', arguments, 1)
- if len(arguments) != 2:
- return
-
- target, arguments = arguments
-
- arguments = re.split('\s+', arguments, 1)
-
- command, arguments = arguments if len(arguments) == 2 else [arguments[0],[]]
-
- self.connection.privmsg(self.target, '- target: ' + target)
- self.connection.privmsg(self.target, '- command: ' + command)
- self.connection.privmsg(self.target, '- arguments: ' + pp.pformat(arguments))
-
def main():
host = str(env.get('host', 'irc.freenode.org'))
port = int(env.get('port', 6667))
nick = str(env.get('nick', 'crabspasm'))
target = str(env.get('target', '#tincspasm'))
- print '====> irc://%s@%s:%s/%s' % (nick, host, port, target)
+ print('====> irc://%s@%s:%s/%s' % (nick, host, port, target))
client = IRCBot(target)
try:
client.connect(host, port, nick)
except ServerConnectionError, error:
- print error
+ print(error)
exit(1)
client.start()
if __name__ == "__main__":
- main()
+ main()