add all of the current files

1 files changed, 331 insertions, 0 deletions

diff --git a/src/irc/channel.lua b/src/irc/channel.lua
new file mode 100644
index 0000000..4e77ae9
--- /dev/null
+++ b/src/irc/channel.lua
@@ -0,0 +1,331 @@
+-- initialization {{{
+local base =   _G
+local irc =    require 'irc'
+local misc =   require 'irc.misc'
+local socket = require 'socket'
+local table =  require 'table'
+-- }}}
+
+module 'irc.channel'
+
+-- object metatable {{{
+local mt = {
+    -- __index() {{{
+    __index =    function(self, key)
+                     if key == "name" then
+                         return self._name
+                     elseif key == "topic" then
+                         return self._topic
+                     elseif key == "chanmode" then
+                         return self._chanmode
+                     else
+                         return _M[key]
+                     end
+                 end,
+    -- }}}
+    -- __newindex() {{{
+    __newindex = function(self, key, value)
+                     if key == "name" then
+                         return
+                     elseif key == "topic" then
+                         irc.send("TOPIC", self._name, value)
+                     elseif key == "chanmode" then
+                         return
+                     else
+                         base.rawset(self, key, value)
+                     end
+                 end,
+    -- }}}
+    -- __concat() {{{
+    __concat =   function(first, second)
+                     local first_str, second_str
+
+                     if base.type(first) == "table" then
+                         first_str = first._name
+                     else
+                         first_str = first
+                     end
+                     if base.type(second) == "table" then
+                         second_str = second._name
+                     else
+                         second_str = second
+                     end
+
+                     return first_str .. second_str
+                 end,
+    -- }}}
+    -- __tostring() {{{
+    __tostring = function(self)
+                     return self._name
+                 end
+    -- }}}
+}
+-- }}}
+
+-- private methods {{{
+-- set_basic_mode() - sets a no-arg mode on a channel {{{
+local function set_basic_mode(self, set, letter)
+    if set then
+        irc.send("MODE", self.name, "+" .. letter)
+    else
+        irc.send("MODE", self.name, "-" .. letter)
+    end
+end
+-- }}}
+-- }}}
+
+-- constructor {{{
+function new(chan)
+    return base.setmetatable({_name = chan, _topic = {}, _chanmode = "",
+                              _members = {}}, mt)
+end
+-- }}}
+
+-- public methods {{{
+-- iterators {{{
+-- each_op() {{{
+function each_op(self)
+    return function(state, arg)
+               return misc.value_iter(state, arg,
+                                      function(v)
+                                          return v:sub(1, 1) == "@"
+                                      end)
+           end,
+           self._members,
+           nil
+end
+-- }}}
+
+-- each_voice() {{{
+function each_voice(self)
+    return function(state, arg)
+               return misc.value_iter(state, arg,
+                                      function(v)
+                                          return v:sub(1, 1) == "+"
+                                      end)
+           end,
+           self._members,
+           nil
+end
+-- }}}
+
+-- each_user() {{{
+function each_user(self)
+    return function(state, arg)
+               return misc.value_iter(state, arg,
+                                      function(v)
+                                          return v:sub(1, 1) ~= "@" and
+                                                 v:sub(1, 1) ~= "+"
+                                      end)
+           end,
+           self._members,
+           nil
+end
+-- }}}
+
+-- each_member() {{{
+function each_member(self)
+    return misc.value_iter, self._members, nil
+end
+-- }}}
+-- }}}
+
+-- return tables of users {{{
+-- ops() {{{
+function ops(self)
+    local ret = {}
+    for nick in self:each_op() do
+        table.insert(ret, nick)
+    end
+    return ret
+end
+-- }}}
+
+-- voices() {{{
+function voices(self)
+    local ret = {}
+    for nick in self:each_voice() do
+        table.insert(ret, nick)
+    end
+    return ret
+end
+-- }}}
+
+-- users() {{{
+function users(self)
+    local ret = {}
+    for nick in self:each_user() do
+        table.insert(ret, nick)
+    end
+    return ret
+end
+-- }}}
+
+-- members() {{{
+function members(self)
+    local ret = {}
+    -- not just returning self._members, since the return value shouldn't be
+    -- modifiable
+    for nick in self:each_member() do
+        table.insert(ret, nick)
+    end
+    return ret
+end
+-- }}}
+-- }}}
+
+-- setting modes {{{
+-- ban() - ban a user from a channel {{{
+-- TODO: hmmm, this probably needs an appropriate mask, rather than a nick
+function ban(self, name)
+    irc.send("MODE", self.name, "+b", name)
+end
+-- }}}
+
+-- unban() - remove a ban on a user {{{
+-- TODO: same here
+function unban(self, name)
+    irc.send("MODE", self.name, "-b", name)
+end
+-- }}}
+
+-- voice() - give a user voice on a channel {{{
+function voice(self, name)
+    irc.send("MODE", self.name, "+v", name)
+end
+-- }}}
+
+-- devoice() - remove voice from a user {{{
+function devoice(self, name)
+    irc.send("MODE", self.name, "-v", name)
+end
+-- }}}
+
+-- op() - give a user ops on a channel {{{
+function op(self, name)
+    irc.send("MODE", self.name, "+o", name)
+end
+-- }}}
+
+-- deop() - remove ops from a user {{{
+function deop(self, name)
+    irc.send("MODE", self.name, "-o", name)
+end
+-- }}}
+
+-- set_limit() - set a channel limit {{{
+function set_limit(self, new_limit)
+    if new_limit then
+        irc.send("MODE", self.name, "+l", new_limit)
+    else
+        irc.send("MODE", self.name, "-l")
+    end
+end
+-- }}}
+
+-- set_key() - set a channel password {{{
+function set_key(self, key)
+    if key then
+        irc.send("MODE", self.name, "+k", key)
+    else
+        irc.send("MODE", self.name, "-k")
+    end
+end
+-- }}}
+
+-- set_private() - set the private state of a channel {{{
+function set_private(self, set)
+    set_basic_mode(self, set, "p")
+end
+-- }}}
+
+-- set_secret() - set the secret state of a channel {{{
+function set_secret(self, set)
+    set_basic_mode(self, set, "s")
+end
+-- }}}
+
+-- set_invite_only() - set whether joining the channel requires an invite {{{
+function set_invite_only(self, set)
+    set_basic_mode(self, set, "i")
+end
+-- }}}
+
+-- set_topic_lock() - if true, the topic can only be changed by an op {{{
+function set_topic_lock(self, set)
+    set_basic_mode(self, set, "t")
+end
+-- }}}
+
+-- set_no_outside_messages() - if true, users must be in the channel to send messages to it {{{
+function set_no_outside_messages(self, set)
+    set_basic_mode(self, set, "n")
+end
+-- }}}
+
+-- set moderated() - set whether voice is required to speak {{{
+function set_moderated(self, set)
+    set_basic_mode(self, set, "m")
+end
+-- }}}
+-- }}}
+
+-- accessors {{{
+-- add_user() {{{
+function add_user(self, user, mode)
+    mode = mode or ''
+    self._members[user] = mode .. user
+end
+-- }}}
+
+-- remove_user() {{{
+function remove_user(self, user)
+    self._members[user] = nil
+end
+-- }}}
+
+-- change_status() {{{
+function change_status(self, user, on, mode)
+    if on then
+        if mode == 'o' then
+            self._members[user] = '@' .. user
+        elseif mode == 'v' then
+            self._members[user] = '+' .. user
+        end
+    else
+        if (mode == 'o' and self._members[user]:sub(1, 1) == '@') or
+           (mode == 'v' and self._members[user]:sub(1, 1) == '+') then
+            self._members[user] = user
+        end
+    end
+end
+-- }}}
+
+-- contains() {{{
+function contains(self, nick)
+    for member in self:each_member() do
+        local member_nick = member:gsub('@+', '')
+        if member_nick == nick then
+            return true
+        end
+    end
+    return false
+end
+-- }}}
+
+-- change_nick {{{
+function change_nick(self, old_nick, new_nick)
+    for member in self:each_member() do
+        local member_nick = member:gsub('@+', '')
+        if member_nick == old_nick then
+            local mode = self._members[old_nick]:sub(1, 1)
+            if mode ~= '@' and mode ~= '+' then mode = "" end
+            self._members[old_nick] = nil
+            self._members[new_nick] = mode .. new_nick
+            break
+        end
+    end
+end
+-- }}}
+-- }}}
+-- }}}