Mikubot-2/otouto/plugins/pokemon-go.lua

-- 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)
    :t('pogo', true):t('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
  for _, set in pairs(db.membership) do
  	setmetatable(set, utilities.set_meta)
  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