Crackbot/irc/init.lua at master · astravexton/Crackbot · GitHub

History

225 lines (179 loc) · 4.11 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

171

172

173

174

175

176

177

178

179

180

181

182

183

184

185

186

187

188

189

190

191

192

193

194

195

196

197

198

199

200

201

202

203

204

205

206

207

208

209

210

211

212

213

214

215

216

217

218

219

220

221

222

223

224

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 "irc.util"

require "irc.asyncoperations"

require "irc.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 '"..k.."' is not accessible before connecting", 2)

end

return v

end

function new(user)

local o = {

nick = assert(user.nick, "Field 'nick' is required");

username = user.username or "lua";

realname = user.realname or "Lua owns";

hooks = {};

track_users = true;

}

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

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)

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

self.socket = s

setmetatable(self, meta)

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()

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

local handlers = handlers

function meta:handle(prefix, cmd, params)

local handler = handlers[cmd]

if handler then

return handler(self, prefix, unpack(params))

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

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

Crackbot/irc/init.lua at master · astravexton/Crackbot · GitHub

Expand file tree

Breadcrumbs

init.lua

Latest commit

History

Breadcrumbs

init.lua

File metadata and controls