[OpenComputers] IRC client + terminal glasses

local component = require("component")
local computer = require("computer")
if not component.isAvailable("internet") then
  io.stderr:write("OpenIRC requires an Internet Card to run!\n")
  return
end

local glasses = component.openperipheral_bridge
local event = require("event")
local internet = require("internet")
local shell = require("shell")
local term = require("term")
local text = require("text")
local w, h = component.gpu.getResolution()
local chat, text_color, acolor = {}, 0x00ff00

local args, options = shell.parse(...)
if #args < 1 then
  print("Usage: irc <nickname> [server[:port]]")
  return
end

local nick = args[1]
local host = args[2] or "irc.esper.net:6667"
local size = 1

if not host:find(":") then
  host = host .. ":6667"
end

local sock, reason = internet.open(host)
if not sock then
  io.stderr:write(reason .. "\n")
  return
end

local function draw()
  glasses.clear()
  glasses.addBox(1, 1, 312*size*2, #chat*size*10+(size*3), 0x000000, 0.5)
  for i = 1, #chat do
    if chat[i]:find(nick) then
      acolor = 0xffff00
    end
    glasses.addText(3, size*10*i-(size*6), chat[i], acolor).setScale(size)
    acolor = text_color
  end
  glasses.sync()
end

local function print(message, overwrite)
  for i = 1, #message, 104 do
    table.insert(chat, message:sub(i, i+104))
  end
  if #chat > 10 then
    table.remove(chat, 1)
  end
  draw()
  local line
  repeat
    line, message = text.wrap(text.trim(message), w, w)
    if not overwrite then
      component.gpu.copy(1, 1, w, h - 1, 0, -1)
    end
    overwrite = false
    component.gpu.fill(1, h-1, w, 1, " ")
    component.gpu.set(1, h-1, line)
  until not message or message == ""
end

function autocreate(table, key)
  table[key] = {}
  return table[key]
end

local function name(identity)
  return identity and identity:match("^[^!]+") or identity or "Anonymous"
end

local callback = nil
local whois = setmetatable({}, {__index=autocreate})
local names = setmetatable({}, {__index=autocreate})
local timer

local ignore = {
  [213]=true, [214]=true, [215]=true, [216]=true, [217]=true,
  [218]=true, [231]=true, [232]=true, [233]=true, [240]=true,
  [241]=true, [244]=true, [244]=true, [246]=true, [247]=true,
  [250]=true, [300]=true, [316]=true, [361]=true, [362]=true,
  [363]=true, [373]=true, [384]=true, [492]=true,
  [265]=true, [266]=true, [330]=true
}

local commands = {
--Replys
  RPL_WELCOME = "001",
  RPL_YOURHOST = "002",
  RPL_CREATED = "003",
  RPL_MYINFO = "004",
  RPL_BOUNCE = "005",
  RPL_LUSERCLIENT = "251",
  RPL_LUSEROP = "252",
  RPL_LUSERUNKNOWN = "253",
  RPL_LUSERCHANNELS = "254",
  RPL_LUSERME = "255",
  RPL_AWAY = "301",
  RPL_UNAWAY = "305",
  RPL_NOWAWAY = "306",
  RPL_WHOISUSER = "311",
  RPL_WHOISSERVER = "312",
  RPL_WHOISOPERATOR = "313",
  RPL_WHOISIDLE = "317",
  RPL_ENDOFWHOIS = "318",
  RPL_WHOISCHANNELS = "319",
  RPL_CHANNELMODEIS = "324",
  RPL_NOTOPIC = "331",
  RPL_TOPIC = "332",
  RPL_NAMREPLY = "353",
  RPL_ENDOFNAMES = "366",
  RPL_MOTDSTART = "375",
  RPL_MOTD = "372",
  RPL_ENDOFMOTD = "376",
  RPL_WHOISSECURE = "671",
  RPL_HELPSTART = "704",
  RPL_HELPTXT = "705",
  RPL_ENDOFHELP = "706",
  RPL_UMODEGMSG = "718",

--Errors
  ERR_BANLISTFULL = "478",
  ERR_CHANNELISFULL = "471",
  ERR_UNKNOWNMODE = "472",
  ERR_INVITEONLYCHAN = "473",
  ERR_BANNEDFROMCHAN = "474",
  ERR_CHANOPRIVSNEEDED = "482",
  ERR_UNIQOPRIVSNEEDED = "485",
  ERR_USERNOTINCHANNEL = "441",
  ERR_NOTONCHANNEL = "442",
  ERR_NICKCOLLISION = "436",
  ERR_NICKNAMEINUSE = "433",
  ERR_ERRONEUSNICKNAME = "432",
  ERR_WASNOSUCHNICK = "406",
  ERR_TOOMANYCHANNELS = "405",
  ERR_CANNOTSENDTOCHAN = "404",
  ERR_NOSUCHCHANNEL = "403",
  ERR_NOSUCHNICK = "401",
  ERR_MODELOCK = "742"
}

local function handleCommand(prefix, command, args, message)
  if command == "PING" then
    sock:write(string.format("PONG :%s\r\n", message))
    sock:flush()
  elseif command == "NICK" then
    local oldNick, newNick = name(prefix), tostring(args[1] or message)
    if oldNick == nick then
      nick = newNick
    end
    print(oldNick .. " is now known as " .. newNick .. ".")
  elseif command == "MODE" then
    if #args == 2 then
      print("[" .. args[1] .. "] " .. name(prefix) .. " set mode".. ( #args[2] > 2 and "s" or "" ) .. " " .. tostring(args[2] or message) .. ".")
    else
      local setmode = {}
      local cumode = "+"
      args[2]:gsub(".", function(char)
        if char == "-" or char == "+" then
          cumode = char
        else
          table.insert(setmode, {cumode, char})
        end
      end)
      local d = {}
      local users = {}
      for i = 3, #args do
        users[i-2] = args[i]
      end
      users[#users+1] = message
      local last
      local ctxt = ""
      for c = 1, #users do
        if not setmode[c] then
          break
        end
        local mode = setmode[c][2]
        local pfx = setmode[c][1]=="+"
        local key = mode == "o" and (pfx and "opped" or "deoped") or
          mode == "v" and (pfx and "voiced" or "devoiced") or
          mode == "q" and (pfx and "quieted" or "unquieted") or
          mode == "b" and (pfx and "banned" or "unbanned") or
          "set " .. setmode[c][1] .. mode .. " on"
        if last ~= key then
          if last then
            print(ctxt)
          end
          ctxt = "[" .. args[1] .. "] " .. name(prefix) .. " " .. key
          last = key
        end
        ctxt = ctxt .. " " .. users[c]
      end
      if #ctxt > 0 then
        print(ctxt)
      end
    end
  elseif command == "QUIT" then
    print(name(prefix) .. " quit (" .. (message or "Quit") .. ").")
  elseif command == "JOIN" then
    print("[" .. args[1] .. "] " .. name(prefix) .. " entered the room.")
  elseif command == "PART" then
    print("[" .. args[1] .. "] " .. name(prefix) .. " has left the room (quit: " .. (message or "Quit") .. ").")
  elseif command == "TOPIC" then
    print("[" .. args[1] .. "] " .. name(prefix) .. " has changed the topic to: " .. message)
  elseif command == "KICK" then
    print("[" .. args[1] .. "] " .. name(prefix) .. " kicked " .. args[2])
  elseif command == "PRIVMSG" then
    local ctcp = message:match("^\1(.-)\1$")
    if ctcp then
      print("[" .. name(prefix) .. "] CTCP " .. ctcp)
      local ctcp, param = ctcp:match("^(%S+) ?(.-)$")
      ctcp = ctcp:upper()
      if ctcp == "TIME" then
        sock:write("NOTICE " .. name(prefix) .. " :\001TIME " .. os.date() .. "\001\r\n")
        sock:flush()
      elseif ctcp == "VERSION" then
        sock:write("NOTICE " .. name(prefix) .. " :\001VERSION Minecraft/OpenComputers Lua 5.2\001\r\n")
        sock:flush()
      elseif ctcp == "PING" then
        sock:write("NOTICE " .. name(prefix) .. " :\001PING " .. param .. "\001\r\n")
        sock:flush()
      end
    else
      if string.find(message, nick) then
        computer.beep()
      end
      if string.find(message, "\001ACTION") then
        print("[" .. args[1] .. "] " .. name(prefix) .. string.gsub(string.gsub(message, "\001ACTION", ""), "\001", ""))
      else
        print("[" .. args[1] .. "] " .. name(prefix) .. ": " .. message)
      end
    end
  elseif command == "NOTICE" then
    print("[NOTICE] " .. message)
  elseif command == "ERROR" then
    print("[ERROR] " .. message)
  elseif tonumber(command) and ignore[tonumber(command)] then
  elseif command == commands.RPL_WELCOME then
    print(message)
  elseif command == commands.RPL_LUSERCLIENT then
    print(message)
  elseif command == commands.RPL_LUSERME then
    print(message)
  elseif command == commands.RPL_AWAY then
    print(string.format("%s is away: %s", name(args[1]), message))
  elseif command == commands.RPL_UNAWAY or command == commands.RPL_NOWAWAY then
    print(message)
  elseif command == commands.RPL_WHOISUSER then
    local nick = args[2]:lower()
    whois[nick].nick = args[2]
    whois[nick].user = args[3]
    whois[nick].host = args[4]
    whois[nick].realName = message
  elseif command == commands.RPL_WHOISSERVER then
    local nick = args[2]:lower()
    whois[nick].server = args[3]
    whois[nick].serverInfo = message
  elseif command == commands.RPL_WHOISOPERATOR then
    local nick = args[2]:lower()
    whois[nick].isOperator = true
  elseif command == commands.RPL_WHOISIDLE then
    local nick = args[2]:lower()
    whois[nick].idle = tonumber(args[3])
  elseif command == commands.RPL_WHOISSECURE then
    local nick = args[2]:lower()
    whois[nick].secureconn = "Is using a secure connection"
  elseif command == commands.RPL_ENDOFWHOIS then
    local nick = args[2]:lower()
    local info = whois[nick]
    if info.nick then print("Nick: " .. info.nick) end
    if info.user then print("User name: " .. info.user) end
    if info.realName then print("Real name: " .. info.realName) end
    if info.host then print("Host: " .. info.host) end
    if info.server then print("Server: " .. info.server .. (info.serverInfo and (" (" .. info.serverInfo .. ")") or "")) end
    if info.secureconn then print(info.secureconn) end
    if info.channels then print("Channels: " .. info.channels) end
    if info.idle then print("Idle for: " .. info.idle) end
    whois[nick] = nil
  elseif command == commands.RPL_WHOISCHANNELS then
    local nick = args[2]:lower()
    whois[nick].channels = message
  elseif command == commands.RPL_CHANNELMODEIS then
    print("Channel mode for " .. args[1] .. ": " .. args[2] .. " (" .. args[3] .. ")")
  elseif command == commands.RPL_NOTOPIC then
    print("No topic is set for " .. args[1] .. ".")
  elseif command == commands.RPL_TOPIC then
    print("Topic for " .. args[1] .. ": " .. message)
  elseif command == commands.RPL_NAMREPLY then
    local channel = args[3]
    table.insert(names[channel], message)
  elseif command == commands.RPL_ENDOFNAMES then
    local channel = args[2]
    print("Users on " .. channel .. ": " .. (#names[channel] > 0 and table.concat(names[channel], " ") or "none"))
    names[channel] = nil
  elseif command == commands.RPL_MOTDSTART then
    if options.motd then
      print(message .. args[1])
    end
  elseif command == commands.RPL_MOTD then
    if options.motd then
      print(message)
    end
  elseif command == commands.RPL_ENDOFMOTD then
  elseif command == commands.RPL_HELPSTART or
  command == commands.RPL_HELPTXT or
  command == commands.RPL_ENDOFHELP then
    print(message)
  elseif command == commands.ERR_BANLISTFULL or
  command == commands.ERR_BANNEDFROMCHAN or
  command == commands.ERR_CANNOTSENDTOCHAN or
  command == commands.ERR_CHANNELISFULL or
  command == commands.ERR_CHANOPRIVSNEEDED or
  command == commands.ERR_ERRONEUSNICKNAME or
  command == commands.ERR_INVITEONLYCHAN or
  command == commands.ERR_NICKCOLLISION or
  command == commands.ERR_NOSUCHNICK or
  command == commands.ERR_NOTONCHANNEL or
  command == commands.ERR_UNIQOPRIVSNEEDED or
  command == commands.ERR_UNKNOWNMODE or
  command == commands.ERR_USERNOTINCHANNEL or
  command == commands.ERR_WASNOSUCHNICK or
  command == commands.ERR_MODELOCK then
    print("[ERROR]: " .. message)
  elseif tonumber(command) and (tonumber(command) >= 200 and tonumber(command) < 400) then
    print("[Response " .. command .. "] " .. table.concat(args, ", ") .. ": " .. message)
  elseif tonumber(command) and (tonumber(command) >= 400 and tonumber(command) < 600) then
    print("[Error] " .. table.concat(args, ", ") .. ": " .. message)
  else
    print("Unhandled command: " .. command .. ": " .. message)
  end
end

local function check(line)
  if sock and line and line ~= "" then
    line = text.trim(line)
    if line:lower():sub(1,4) == "/me " then
      print("[" .. (target or "?") .. "] " .. nick .. " " .. line:sub(5), true)
    elseif line~="" then
      print("[" .. (target or "?") .. "] " .. nick .. ": " .. line, true)
    end
    if line:lower():sub(1, 5) == "/msg " then
      local user, message = line:sub(6):match("^(%S+) (.+)$")
      if message then
        message = text.trim(message)
      end
      if not user or not message or message == "" then
        print("Invalid use of /msg. Usage: /msg nick|channel message.")
        line = ""
      else
        target = user
        line = "PRIVMSG " .. target .. " :" .. message
      end
    elseif line:lower():sub(1, 6) == "/size " then
      local ssize = tonumber(line:sub(7))
      if type(ssize) == "number" then
        size = ssize
      end
    elseif line:lower():sub(1, 6) == "/join " then
      local channel = text.trim(line:sub(7))
      if not channel or channel == "" then
        print("Invalid use of /join. Usage: /join channel.")
        line = ""
      else
        target = channel
        line = "JOIN " .. channel
      end
    elseif line:lower():sub(1, 5) == "/lua " then
      local script = text.trim(line:sub(6))
      local result, reason = load(script, "=stdin", nil, setmetatable({print=print, socket=sock, nick=nick}, {__index=_G}))
      if not result then
        result, reason = load("return " .. script, "=stdin", nil, setmetatable({print=print, socket=sock, nick=nick}, {__index=_G}))
      end
      line = ""
      if not result then
        print("Error: " .. tostring(reason))
      else
        result, reason = pcall(result)
        if not result then
          print("Error: " .. tostring(reason))
        elseif type(reason) == "function" then
          callback = reason
        elseif reason then
          line = tostring(reason)
        end
      end
    elseif line:lower():sub(1,4) == "/me " then
      if not target then
        print("No default target set. Use /msg or /join to set one.")
        line = ""
      else
        line = "PRIVMSG " .. target .. " :\001ACTION " .. line:sub(5) .. "\001"
      end
    elseif line:sub(1, 1) == "/" then
      line = line:sub(2)
    elseif line ~= "" then
      if not target then
        print("No default target set. Use /msg or /join to set one.")
        line = ""
      else
        line = "PRIVMSG " .. target .. " :" .. line
      end
    end
    if line and line ~= "" then
      sock:write(line .. "\r\n")
      sock:flush()
    end
  end
end

local function keyboard(...)
  local data = {...}
  check(data[5])
end

event.listen('glasses_chat_command', keyboard)

local result, reason = pcall(function()
  term.clear()
  print("Welcome to OpenIRC!")
  sock:setTimeout(0.05)
  sock:write(string.format("NICK %s\r\n", nick))
  sock:write(string.format("USER %s 0 * :%s [OpenComputers]\r\n", nick:lower(), nick))
  sock:flush()
  timer = event.timer(0.5, function()
    if not sock then
      return false
    end
    repeat
      local ok, line = pcall(sock.read, sock)
      if ok then
        if not line then
          print("Connection lost.")
          sock:close()
          sock = nil
          return false
        end
        line = text.trim(line)
        local match, prefix = line:match("^(:(%S+) )")
        if match then line = line:sub(#match + 1) end
        local match, command = line:match("^(([^:]%S*))")
        if match then line = line:sub(#match + 1) end
        local args = {}
        repeat
          local match, arg = line:match("^( ([^:]%S*))")
          if match then
            line = line:sub(#match + 1)
            table.insert(args, arg)
          end
        until not match
        local message = line:match("^ :(.*)$")

        if callback then
          local result, reason = pcall(callback, prefix, command, args, message)
          if not result then
            print("Error in callback: " .. tostring(reason))
          end
        end
        handleCommand(prefix, command, args, message)
      end
    until not ok
  end, math.huge)
  local target = nil
  local history = {}
  repeat
    term.setCursor(1, h)
    term.write((target or "#") .. "> ")
    local line = term.read(history)
    check(line)
  until not sock or not line
end)

if sock then
  sock:write("QUIT\r\n")
  sock:close()
end

event.ignore('glasses_chat_command', keyboard)

if timer then
  event.cancel(timer)
end

if not result then
  error(reason, 0)
end
return reason