local socket = require "socket" local error = error local setmetatable = setmetatable local rawget = rawget local unpack = unpack local pairs = pairs local assert = assert local require = require local tonumber = tonumber local type = type local pcall = pcall module "irc" local meta = {} meta.__index = meta _META = meta require "util" require "asyncoperations" require "handlers" local meta_preconnect = {} function meta_preconnect.__index(o, k) local v = rawget(meta_preconnect, k) if not v and meta[k] then error(("field '%s' is not accessible before connecting"):format(k), 2) end return v end function new(data) local o = { nick = assert(data.nick, "Field 'nick' is required"); username = data.username or "lua"; realname = data.realname or "Lua owns"; nickGenerator = data.nickGenerator or defaultNickGenerator; hooks = {}; track_users = true; } assert(checkNick(o.nick), "Erroneous nickname passed to irc.new") return setmetatable(o, meta_preconnect) end function meta:hook(name, id, f) f = f or id self.hooks[name] = self.hooks[name] or {} self.hooks[name][id] = f return id or f end meta_preconnect.hook = meta.hook function meta:unhook(name, id) local hooks = self.hooks[name] assert(hooks, "no hooks exist for this event") assert(hooks[id], "hook ID not found") hooks[id] = nil end meta_preconnect.unhook = meta.unhook function meta:invoke(name, ...) local hooks = self.hooks[name] if hooks then for id,f in pairs(hooks) do if f(...) then return true end end end end function meta_preconnect:connect(_host, _port) local host, port, password, secure, timeout if type(_host) == "table" then host = _host.host port = _host.port timeout = _host.timeout password = _host.password secure = _host.secure else host = _host port = _port end host = host or error("host name required to connect", 2) port = port or 6667 local s = socket.tcp() s:settimeout(timeout or 30) s:settimeout(90) assert(s:connect(host, port)) if secure then local work, ssl = pcall(require, "ssl") if not work then error("LuaSec required for secure connections", 2) end local params if type(secure) == "table" then params = secure else params = {mode = "client", protocol = "tlsv1"} end s = ssl.wrap(s, params) success, errmsg = s:dohandshake() if not success then error(("could not make secure connection: %s"):format(errmsg), 2) end end self.socket = s setmetatable(self, meta) self:send("CAP REQ multi-prefix") self:invoke("PreRegister", self) self:send("CAP END") if password then self:send("PASS %s", password) end self:send("NICK %s", self.nick) self:send("USER %s 0 * :%s", self.username, self.realname) self.channels = {} s:settimeout(0) repeat self:think() socket.select(nil, nil, 0.1) -- Sleep so that we don't eat CPU until self.authed end function meta:disconnect(message) message = message or "Bye!" self:invoke("OnDisconnect", message, false) self:send("QUIT :%s", message) self:shutdown() end function meta:shutdown() self.socket:close() setmetatable(self, nil) end local function getline(self, errlevel) local line, err = self.socket:receive("*l") if not line and err ~= "timeout" and err ~= "wantread" then self:invoke("OnDisconnect", err, true) self:shutdown() error(err, errlevel) end return line end function meta:think() while true do local line = getline(self, 3) if line and #line > 0 then if not self:invoke("OnRaw", line) then self:handle(parse(line)) end else break end end end local handlers = handlers function meta:handle(prefix, cmd, params) local handler = handlers[cmd] if handler then return handler(self, prefix, unpack(params)) end end local whoisHandlers = { ["311"] = "userinfo"; ["312"] = "node"; ["319"] = "channels"; ["330"] = "account"; -- Freenode ["307"] = "registered"; -- Unreal } function meta:whois(nick) self:send("WHOIS %s", nick) local result = {} while true do local line = getline(self, 3) if line then local prefix, cmd, args = parse(line) local handler = whoisHandlers[cmd] if handler then result[handler] = args elseif cmd == "318" then break else self:handle(prefix, cmd, args) end end end if result.account then result.account = result.account[3] elseif result.registered then result.account = result.registered[2] end return result end function meta:topic(channel) self:send("TOPIC %s", channel) end