From 82fe1a8dc6abba089c21cc7f5aaa863327295a29 Mon Sep 17 00:00:00 2001
From: Brayden Banks <bb010g@gmail.com>
Date: Wed, 13 Jul 2016 17:00:58 -0700
Subject: [PATCH] =?UTF-8?q?Add=20Pok=C3=A9mon=20Go=20plugin.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 otouto/plugins/pokemon-go.lua | 82 +++++++++++++++++++++++++++++++++++
 otouto/utilities.lua          | 31 +++++++++++++
 2 files changed, 113 insertions(+)
 create mode 100644 otouto/plugins/pokemon-go.lua

diff --git a/otouto/plugins/pokemon-go.lua b/otouto/plugins/pokemon-go.lua
new file mode 100644
index 0000000..c2b0a8b
--- /dev/null
+++ b/otouto/plugins/pokemon-go.lua
@@ -0,0 +1,82 @@
+-- For some reason you can't have an é in variable names. :(
+local pokemon_go = {}
+
+local utilities = require('otouto.utilities')
+
+pokemon_go.command = 'pokego <team>'
+
+function pokemon_go:init(config)
+  pokemon_go.triggers = utilities.triggers(self.info.username, config.cmd_pat)
+    :t('pokego', true):t('pokégo', true)
+    :t('pokemongo', true):t('pokémongo', true).table
+  pokemon_go.doc = [[```
+]]..config.cmd_pat..[[pokego <team>
+Set your Pokémon Go team for statistical purposes. The team must be valid, and can be referred to by name or color (or the first letter of either). Giving no team name will show statistics.
+```]]
+  local db = self.database.pokemon_go
+  if not db then
+    self.database.pokemon_go = {}
+    db = self.database.pokemon_go
+  end
+  if not db.membership then
+    db.membership = {}
+  end
+end
+
+local team_ref = {
+  mystic = "Mystic",
+  m = "Mystic",
+  valor = "Valor",
+  v = "Valor",
+  instinct = "Instinct",
+  i = "Instinct",
+  blue = "Mystic",
+  b = "Mystic",
+  red = "Valor",
+  r = "Valor",
+  yellow = "Instinct",
+  y = "Instinct"
+}
+
+function pokemon_go:action(msg, config)
+  local output
+  local input = utilities.input(msg.text_lower)
+
+  if input then
+    local team = team_ref[input]
+    if not team then
+      output = 'Invalid team.'
+    else
+      local id_str = tostring(msg.from.id)
+      local db = self.database.pokemon_go
+      local db_membership = db.membership
+      if not db_membership[team] then
+        db_membership[team] = utilities.new_set()
+      end
+
+      for t, set in pairs(db_membership) do
+        if t ~= team then
+          set:remove(id_str)
+        else
+          set:add(id_str)
+        end
+      end
+
+      output = 'Your team is now '..team..'.'
+    end
+  else
+    local db = self.database.pokemon_go
+    local db_membership = db.membership
+
+    local output_temp = {'Membership:'}
+    for t, set in pairs(db_membership) do
+      table.insert(output_temp, t..': '..#set)
+    end
+
+    output = table.concat(output_temp, '\n')
+  end
+
+  utilities.send_reply(self, msg, output)
+end
+
+return pokemon_go
diff --git a/otouto/utilities.lua b/otouto/utilities.lua
index 4fa6fba..79016e1 100755
--- a/otouto/utilities.lua
+++ b/otouto/utilities.lua
@@ -323,4 +323,35 @@ utilities.char = {
 	utf_8 = '([%z\1-\127\194-\244][\128-\191]*)',
 }
 
+utilities.set_meta = {}
+utilities.set_meta.__index = utilities.set_meta
+function utilities.new_set()
+  return setmetatable({__count = 0}, utilities.set_meta)
+end
+function utilities.set_meta:add(x)
+  if x == "__count" then
+    return false
+  else
+    if not self[x] then
+      self[x] = true
+      self.__count = self.__count + 1
+    end
+    return true
+  end
+end
+function utilities.set_meta:remove(x)
+  if x == "__count" then
+    return false
+  else
+    if self[x] then
+      self[x] = nil
+      self.__count = self.__count - 1
+    end
+    return true
+  end
+end
+function utilities.set_meta:__len()
+  return self.__count
+end
+
 return utilities