initial commit

2 files changed, 170 insertions, 1 deletions

diff --git a/README.md b/README.md
index 86abb8e..93458ed 100644
--- a/README.md
+++ b/README.md
@@ -1,3 +1,8 @@
 # tracer
 
-irc tracery bot
\ No newline at end of file
+irc tracery bot
+
+deps:
+* https://github.com/aparrish/pytracery
+
+edit channels and botnick to test somewhere in irc
diff --git a/tracer.py b/tracer.py
new file mode 100755
index 0000000..070d4fb
--- /dev/null
+++ b/tracer.py
@@ -0,0 +1,164 @@
+import socket
+import re
+from random import randint, choice
+import sys, os
+import time
+import subprocess
+import tracery
+from tracery.modifiers import base_english
+import json
+import random
+
+DB = {}
+
+def grammar(rules):
+    try:
+        res = tracery.Grammar(rules)
+        res.add_modifiers(base_english)
+        return res
+    except Exception as e:
+        print(e)
+
+def load_rules(path):
+    try:
+        with open(path) as f:
+            return json.loads(f.read())
+    except Exception as e:
+        print(e)
+
+def populate():
+    global DB
+    DB = {}
+    for subdir in os.listdir("/home"):
+        d = "/home/"+subdir+"/.tracery"
+        if os.path.isdir(d):
+            for p in os.listdir(d):
+                rule = d+"/"+p
+                name = p.replace(".json", "")
+                #print(p, rule, name)
+                if p in DB:
+                    DB[name].append(grammar(load_rules(rule)))
+                else:
+                    DB[name] = [grammar(load_rules(rule))]
+
+populate()
+#print(DB)
+
+def generate(rule):
+    populate()
+    if rule in DB:
+        g = random.choice(DB[rule])
+        return g.flatten("#origin#")
+
+
+def listify(col):
+    if type(col) == type([]):
+        return col
+    else:
+        return [col]
+
+def shuffle(col):
+    a = random.choice(col.keys())
+    b = random.choice(col.keys())
+    if 'origin' in [a,b]:
+        print("origin discard")
+        return col
+    temp = col[a]
+    col[a] = col[b]
+    col[b] = temp
+    return col
+
+def fuse(argv):
+    populate()
+    raw = {}
+    for gk in argv:
+        if gk in DB:
+            g = random.choice(DB[gk]).raw
+            for k in g:
+                if k in raw:
+                    raw[k] = listify(raw[k])+listify(g[k])
+                else:
+                    raw[k] = g[k]
+    for i in range(20):
+        raw = shuffle(raw)
+    return grammar(raw).flatten("#origin#")
+
+
+server = "127.0.0.1"
+channels = []
+if len(sys.argv) > 1:
+    for c in sys.argv[1:]:
+        channels.append("#"+c)
+botnick = "tracer"
+
+def ping():
+    ircsock.send("PONG :Pong\n")
+
+def joinchan(chan):
+    ircsock.send("JOIN "+ chan +"\n")
+
+def sendmsg(chan , msg): # This is the send message function, it simply sends messages to the channel.
+      ircsock.send("PRIVMSG "+ chan +" :"+ msg.encode("utf8") +"\n")
+
+def wexec(msg):
+    ircsock.send("EXEC -msg "+channel+" "+msg)
+
+if __name__ == "__main__":
+    ircsock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+    ircsock.connect((server, 6667)) # Here we connect to the server using port 6667
+    ircsock.send("USER "+ botnick +" "+ botnick +" "+ botnick +" :tracery bot.\n") # user authentication
+    ircsock.send("NICK "+ botnick +"\n") # here we actually assign the nick to the bot
+
+    for c in channels:
+        joinchan(c)
+
+
+
+def send(channel, s):
+    sendmsg("#"+channel, s)
+
+def think(chan, nick, msg):
+    words = re.split('[ \t\s]+', msg)
+    if len(words) > 0 and nick != "tracer":
+        if words[0] == "!!list":
+            res = ""
+            for k in DB:
+                res += k+" "
+            send(chan, res.encode("utf8")[:475])
+        elif words[0] == "!!fuse":
+            if "|" in words:
+                res = fuse(words[1:words.index("|")])
+                if res:
+                    send(chan, " ".join(words[words.index("|")+1:])+" "+res)
+            else:
+                res = fuse(words[1:])
+                if res:
+                    send(chan,res[0:475])
+        elif words[0][0:2] == "!!":
+            print(words)
+            res = generate(words[0][2:])
+            if res:
+                if len(words) >= 3:
+                    if words[1] == "|":
+                        send(chan, " ".join(words[2:]) + " " + res)
+                    else:
+                        send(chan, res)
+                else:
+                    send(chan, res)
+
+if __name__ == "__main__":
+    while 1:
+        ircmsg = ircsock.recv(2048)
+        ircmsg = ircmsg.strip('\n\r')
+
+        if ircmsg.find("PING") != -1:
+            ping()
+
+        m = re.match(':(?P<nick>[^ ]+)!.*PRIVMSG #(?P<chan>\w+) :(?P<msg>.*)', ircmsg)
+        if m and m.groupdict():
+            m = m.groupdict()
+            try:
+                think(m["chan"], m["nick"], m["msg"])
+            except Exception as e:
+                print "ERROR" + str(m)
+                print(e)