From 92938c95a5f070971d4097e8205a44b073b03fad Mon Sep 17 00:00:00 2001
From: AZ Huang <aitjcize@gmail.com>
Date: Thu, 20 Feb 2014 19:27:49 +0800
Subject: [PATCH] Make SyncBot remember things.

---
 tox-irc-sync.py | 29 +++++++++++++++++++++++------
 1 file changed, 23 insertions(+), 6 deletions(-)

diff --git a/tox-irc-sync.py b/tox-irc-sync.py
index a295530..5c53179 100644
--- a/tox-irc-sync.py
+++ b/tox-irc-sync.py
@@ -17,6 +17,7 @@ IRC_PORT = 6667
 NAME = NICK = IDENT = REALNAME = "SyncBot"
 
 CHANNEL = '#tox-ontopic'
+MEMORY_DB = 'memory.pickle'
 
 class SyncBot(Tox):
     def __init__(self):
@@ -33,6 +34,10 @@ class SyncBot(Tox):
         self.tox_group_id = None
 
         self.irc_init()
+        self.memory = {}
+
+        if exists(MEMORY_DB):
+            self.memory = pickle.load(MEMORY_DB)
 
     def irc_init(self):
         self.irc = socket.socket()
@@ -172,13 +177,25 @@ class SyncBot(Tox):
         else:
             self.ensure_exe(self.send_message, (friendid, message))
 
+    def send_both(self, content):
+        self.sent = content
+        self.ensure_exe(self.group_message_send, (self.tox_group_id, content))
+        self.irc_send('PRIVMSG %s :%s\r\n' % (CHANNEL, content))
+
     def handle_command(self, cmd):
-        if cmd[1:] in ['syncbot', 'echobot']:
-            self.ensure_exe(self.group_message_send,
-                    (self.tox_group_id, self.get_address()))
-            self.sent = self.get_address()
-            self.irc_send('PRIVMSG %s :%s\r\n' %
-                    (CHANNEL, self.get_address()))
+        cmd = cmd[1:]
+        if cmd in ['syncbot', 'echobot']:
+            self.send_both(self.get_address())
+        elif cmd.startswith('remember '):
+            args = cmd[9:].split(' ')
+            subject = args[0]
+            desc = ' '.join(args[1:])
+            self.memory[subject] = desc
+            pickle.dump(self.memory, MEMORY_DB)
+            self.send_both('Remembering ^%s: %s' % (subject, desc))
+        elif self.memory.has_key(cmd):
+            self.send_both(self.memory[cmd])
+
 
 t = SyncBot()
 t.loop()