steve's initial framework

git-svn-id: svn://svn.zzq.org/warmachine/trunk@2 3ede8657-8418-0410-873f-eb3fb5a66eab

24 files changed, 339 insertions, 0 deletions

diff --git a/actions/ActionMap.py b/actions/ActionMap.py
new file mode 100644
index 0000000..4c8431f
--- /dev/null
+++ b/actions/ActionMap.py
@@ -0,0 +1,16 @@
+import say, op, deop, quit, part, voice, devoice
+import join, kick, pythondoc, settopic, coin
+
+actions = {}
+actions['say']       = say.say()
+actions['op']        = op.op()
+actions['deop']      = deop.deop()
+actions['quit']      = quit.quit()
+actions['part']      = part.part()
+actions['voice']     = voice.voice()
+actions['devoice']   = devoice.devoice()
+actions['kick']      = kick.kick()
+actions['join']      = join.join()
+actions['pythondoc'] = pythondoc.pythondoc()
+actions['settopic']  = settopic.settopic()
+actions['coin']      = coin.coin()
diff --git a/actions/__init__.py b/actions/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/actions/__init__.py
diff --git a/actions/coin.py b/actions/coin.py
new file mode 100644
index 0000000..05dfbda
--- /dev/null
+++ b/actions/coin.py
@@ -0,0 +1,17 @@
+import random
+
+class coin:
+
+    def __init__(self):
+        self = []
+
+    def getAction(self, data):
+        ret = int(random.random()*10)
+        print ret
+        thecoin = 'The coin lands on it\'s SIDE!!!'
+        if ret >= 6:
+           thecoin = "The coin lands on HEAD."
+        else:
+           thecoin = "The coin lands on TAILS."
+        input = data.split(' ')
+        return 'PRIVMSG ' + input[4] + ' :' + thecoin
diff --git a/actions/deop.py b/actions/deop.py
new file mode 100644
index 0000000..d9c61eb
--- /dev/null
+++ b/actions/deop.py
@@ -0,0 +1,8 @@
+class deop:
+
+    def __init__(self):
+        self = []
+
+    def getAction(self, data):
+        input = data.split(' ')
+        return 'MODE ' + input[5] + ' -o ' + input[6]
diff --git a/actions/devoice.py b/actions/devoice.py
new file mode 100644
index 0000000..9539634
--- /dev/null
+++ b/actions/devoice.py
@@ -0,0 +1,8 @@
+class devoice:
+
+    def __init__(self):
+        self = []
+
+    def getAction(self, data):
+        input = data.split(' ')
+        return 'MODE ' + input[5] + ' -v ' + input[6]
diff --git a/actions/join.py b/actions/join.py
new file mode 100644
index 0000000..7a36947
--- /dev/null
+++ b/actions/join.py
@@ -0,0 +1,7 @@
+class join:
+
+    def __init__(self):
+        self = []
+
+    def getAction(self, data):
+        return 'JOIN ' + data[data.index('join')+5:]
diff --git a/actions/kick.py b/actions/kick.py
new file mode 100644
index 0000000..86416fa
--- /dev/null
+++ b/actions/kick.py
@@ -0,0 +1,9 @@
+class kick:
+
+    def __init__(self):
+        self = []
+
+    def getAction(self, data):
+        input = data.split(' ')
+        return 'KICK ' + input[5] + ' ' + input[6]
+
diff --git a/actions/op.py b/actions/op.py
new file mode 100644
index 0000000..60102c4
--- /dev/null
+++ b/actions/op.py
@@ -0,0 +1,8 @@
+class op:
+
+    def __init__(self):
+        self = []
+
+    def getAction(self, data):
+        input = data.split(' ')
+        return 'MODE ' + input[5] + ' +o ' + input[6]
diff --git a/actions/part.py b/actions/part.py
new file mode 100644
index 0000000..f55e63f
--- /dev/null
+++ b/actions/part.py
@@ -0,0 +1,7 @@
+class part:
+
+    def __init__(self):
+        self = []
+
+    def getAction(self, data):
+        return 'PART ' + data[data.index('part')+5:]
diff --git a/actions/pythondoc.py b/actions/pythondoc.py
new file mode 100644
index 0000000..e420723
--- /dev/null
+++ b/actions/pythondoc.py
@@ -0,0 +1,22 @@
+import os
+
+class pythondoc:
+
+    def __init__(self):
+        self = []
+
+    def getAction(self, data):
+        try:
+            input   = data.split(' ')
+            module  = input[5].replace(' ', '')
+            try:
+                module2 = input[6][0:-2] 
+                execstring = "python -c \"from " + module + " import " + module2 + "\nprint " + module2 + ".__doc__\n\" > /tmp/wmpycmd" 
+            except:
+                module = input[5][0:-2]
+                execstring = "python -c \"import " + module + "\nprint " + module + ".__doc__\n\" > /tmp/wmpycmd" 
+            os.system(execstring)
+            doc = file('/tmp/wmpycmd').read().replace("\n", " ")
+        except:
+            doc = "I don't know nuthn about that."
+        return 'PRIVMSG ' + input[2] + ' :' + doc       
diff --git a/actions/quit.py b/actions/quit.py
new file mode 100644
index 0000000..c945f5d
--- /dev/null
+++ b/actions/quit.py
@@ -0,0 +1,7 @@
+class quit:
+
+    def __init__(self):
+        self = []
+
+    def getAction(self, data):
+        return 'QUIT\r\n'
diff --git a/actions/say.py b/actions/say.py
new file mode 100644
index 0000000..7c7273f
--- /dev/null
+++ b/actions/say.py
@@ -0,0 +1,11 @@
+class say:
+
+    def __init__(self):
+       self = []
+
+    def getAction(self, data):
+        input = data.split(' ')
+        words = ""
+        for word in input[6:]:
+            words += ' ' + word
+        return 'PRIVMSG ' + input[5] + ' :' + words[1:]
diff --git a/actions/settopic.py b/actions/settopic.py
new file mode 100644
index 0000000..8fa574a
--- /dev/null
+++ b/actions/settopic.py
@@ -0,0 +1,11 @@
+class settopic:
+
+    def __init__(self):
+        self = []
+
+    def getAction(self, data):
+        input = data.split(' ')
+        words = ""
+        for word in input[6:]:
+            words += ' ' + word
+        return 'TOPIC ' + input[5] + ' :' + words[1:]
diff --git a/actions/voice.py b/actions/voice.py
new file mode 100644
index 0000000..fe42d05
--- /dev/null
+++ b/actions/voice.py
@@ -0,0 +1,8 @@
+class voice:
+
+    def __init__(self):
+        self = []
+
+    def getAction(self, data):
+        input = data.split(' ')
+        return 'MODE ' + input[5] + ' +v ' + input[6]
diff --git a/conf/__init__.py b/conf/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/conf/__init__.py
diff --git a/conf/users.py b/conf/users.py
new file mode 100644
index 0000000..1b7ffc7
--- /dev/null
+++ b/conf/users.py
@@ -0,0 +1,4 @@
+user = []
+user.append('ashcrow')
+user.append('beav')
+user.append('wes')
diff --git a/passiveactions/PassiveActionMap.py b/passiveactions/PassiveActionMap.py
new file mode 100644
index 0000000..087754c
--- /dev/null
+++ b/passiveactions/PassiveActionMap.py
@@ -0,0 +1,8 @@
+import pong, spy, rootcheck, autoop, logtoconsole
+
+passiveactions = {}
+passiveactions['pong']      = pong.pong()
+passiveactions['spy']       = spy.spy()
+passiveactions['rootcheck'] = rootcheck.rootcheck()
+passiveactions['autoop']    = autoop.autoop()
+passiveactions['logtoconsole']    = logtoconsole.logtoconsole()
diff --git a/passiveactions/__init__.py b/passiveactions/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/passiveactions/__init__.py
diff --git a/passiveactions/autoop.py b/passiveactions/autoop.py
new file mode 100644
index 0000000..45ae9e4
--- /dev/null
+++ b/passiveactions/autoop.py
@@ -0,0 +1,15 @@
+class autoop:
+
+    def __init__(self):
+        self = []
+
+    def getAction(self, data, users):
+        try:
+            input = data.split()
+            if 'JOIN' in input[1]:
+                user = data[1:data.index('!')]
+                if user in users:
+                    print "autoop"
+                    return 'MODE ' + input[2][1:] + ' +o ' + user
+        except:
+            return False
diff --git a/passiveactions/logtoconsole.py b/passiveactions/logtoconsole.py
new file mode 100644
index 0000000..533e082
--- /dev/null
+++ b/passiveactions/logtoconsole.py
@@ -0,0 +1,11 @@
+class logtoconsole:
+
+    def __init__(self):
+        self = []
+
+    def getAction(self, data, users):
+        try:
+            print data
+            return False
+        except:
+            return False
diff --git a/passiveactions/pong.py b/passiveactions/pong.py
new file mode 100644
index 0000000..04f663b
--- /dev/null
+++ b/passiveactions/pong.py
@@ -0,0 +1,13 @@
+class pong:
+
+    def __init__(self):
+        self = []
+
+    def getAction(self, data, users):
+        try:
+            if data.find ( 'PING' ) != -1:
+                print 'Ponged ' +  data.split()[1]
+                return 'PONG ' + data.split()[1]
+        except:
+            return False
+       
diff --git a/passiveactions/rootcheck.py b/passiveactions/rootcheck.py
new file mode 100644
index 0000000..c5cc7a7
--- /dev/null
+++ b/passiveactions/rootcheck.py
@@ -0,0 +1,17 @@
+class rootcheck:
+
+    def __init__(self):
+        self = []
+
+    def getAction(self, data, users):
+        try:
+            user = data[1:data.index('!')]
+            input = data.split()
+            if 'JOIN' in input[1]:
+                if 'root' in input[0]:
+                    cmd = "PRIVMSG " + input[2][1:] + " :" + user +": Root?!?!?!!??!"
+                    cmd += "\r\nKICK " + input[2][1:] + " " + user + " :Root + irc = kick"
+                    return cmd
+        except:
+            return False
+       
diff --git a/passiveactions/spy.py b/passiveactions/spy.py
new file mode 100644
index 0000000..c1fc1f6
--- /dev/null
+++ b/passiveactions/spy.py
@@ -0,0 +1,23 @@
+spyusers = []
+spyusers.append('stevetest')
+
+spymaster = ""
+
+class spy:
+
+    def __init__(self):
+        self = []
+
+    def getAction(self, data, users):
+        try:
+            user = data[1:data.index('!')]
+            if user in spyusers:
+                if user in data:
+                    input = data.split()
+                    words = ""
+                    for word in input[3:]:
+                        words += ' ' + word
+                    return 'PRIVMSG ' + spymaster + ' :' + user + ' in ' + input[2] + ' said "' + words[2:] + '"'
+        except:
+            return False
+       
diff --git a/warmachine.py b/warmachine.py
new file mode 100644
index 0000000..b5ab333
--- /dev/null
+++ b/warmachine.py
@@ -0,0 +1,109 @@
+import socket
+import sys
+from actions.ActionMap import *
+from passiveactions.PassiveActionMap import *
+from conf.users import *
+
+class irc:
+
+    def __init__(self, server=None, nick=None, name=None, port=6667):
+        """
+        IRC connection library that needs at least server, nick and name
+        """
+        self.setServer(server)
+        self.setNick(nick)
+        self.setName(name)
+        self.setPort(port)
+
+    def setServer(self, server):
+        """
+        Sets the servername.
+        """
+        self.server = server
+
+    def setNick(self, nick):
+        """
+        Sets the nickname.
+        """
+        self.nick = nick
+
+    def setName(self, name):
+        """
+        Sets the name.
+        """
+        self.name = name
+
+    def setPort(self, port):
+        """
+        Sets the port number to connect to.
+        """
+        self.port = port
+
+    def locaActions(self):
+        """
+        Loads the actions internally.
+        """
+        
+
+    def connect(self):
+        """
+        Connects to the irc server.
+        """
+        self.irc = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+        self.irc.connect((self.server, self.port))
+        self.log(self.irc.recv(4096))
+        self.irc.send('NICK ' + self.nick + '\r\n')
+        self.irc.send('USER ' + self.name + ' 8 * :Warmachine\r\n')
+
+    def join(self, chan):
+        """
+        Joins a channel
+        """
+        self.send('JOIN ' + chan)
+
+    def log(self, string):
+        """
+        Takes care of log formating.
+        """
+        print 'log: ' + string
+
+    def send(self, command):
+        """
+        Sends commands straight to the irc server.
+        """
+        self.irc.send(command + '\r\n')
+
+    def MainLoop(self):
+        """
+        Main Event Loop that parses generic commands
+        """
+        while True:
+           data = self.irc.recv(4096)
+           # Passive Actions
+           try:
+               for key in passiveactions.keys():
+                   pa = passiveactions[key].getAction(data, user)
+                   if pa:
+                       self.send(pa)
+               # Direct Actions
+               if data.find(self.nick + ':') != -1:
+                   curuser = data[1:data.index('!')]
+                   if curuser in user:
+                       input = data.split()
+                       for key in actions.keys():
+                           if data.find(key) != -1:
+                               self.send(actions[key].getAction(data))
+                               break
+                   else:
+                       input = data.split()
+                       self.send('PRIVMSG ' + input[2] + ' :' + curuser + ': stop bothering me jerk.')
+           except Exception,e:
+               print "Action failed"
+               print e
+
+
+if __name__ == '__main__':
+    i = irc('SERVER', 'NICK', 'USER')
+    i.connect()
+    i.join('#CHANNEL')
+    i.MainLoop()