requireInjector(getfenv(1))

local Event    = require('event')
local Socket   = require('socket')
local Terminal = require('terminal')
local Util     = require('util')

local remoteId
local args = { ... }
if #args == 1 then
  remoteId = tonumber(args[1])
else
  print('Enter host ID')
  remoteId = tonumber(read())
end

if not remoteId then
  error('Syntax: vnc <host ID>')
end

multishell.setTitle(multishell.getCurrent(), 'VNC-' .. remoteId)

print('connecting...')
local socket = Socket.connect(remoteId, 5900)

if not socket then
  error('Unable to connect to ' .. remoteId .. ' on port 5900')
end

local w, h = term.getSize()
socket:write({
  type = 'termInfo',
  width = w,
  height = h,
  isColor = term.isColor(),
})

local ct = Util.shallowCopy(term.current())

if not ct.isColor() then
  Terminal.toGrayscale(ct)
end

Event.addRoutine(function()
  while true do
    local data = socket:read()
    if not data then
      break
    end
    for _,v in ipairs(data) do
      ct[v.f](unpack(v.args))
    end
  end
end)

ct.clear()
ct.setCursorPos(1, 1)

local filter = Util.transpose({
  'char', 'paste', 'key', 'key_up', 
  'mouse_scroll', 'mouse_click', 'mouse_drag', 'mouse_up',
})

while true do
  local e = Event.pullEvent()
  local event = e[1]

  if not socket.connected then
    print()
    print('Connection lost')
    print('Press enter to exit')
    read()
    break
  end

  if filter[event] then
    socket:write({
      type = 'shellRemote',
      event = e,
    })
  elseif event == 'terminate' then
    socket:close()
    ct.setBackgroundColor(colors.black)
    ct.clear()
    ct.setCursorPos(1, 1)
    break
  end
end