require = requireInjector(getfenv(1))
local Event = require('event')
local UI = require('ui')
local Socket = require('socket')

multishell.setTitle(multishell.getCurrent(), 'Network')
UI:configure('Network', ...)

local gridColumns = {
  { heading = 'Label',  key = 'label'     },
  { heading = 'Dist',   key = 'distance'  },
  { heading = 'Status', key = 'status'    },
}

if UI.term.width >= 30 then
  table.insert(gridColumns, { heading = 'Fuel',   key = 'fuel' })
  table.insert(gridColumns, { heading = 'Uptime', key = 'uptime' })
end

local page = UI.Page({
  menuBar = UI.MenuBar({
    buttons = {
      { text = 'Telnet', event = 'telnet' },
      { text = 'VNC',    event = 'vnc'    },
      { text = 'Reboot', event = 'reboot' },
    },
  }),
  grid = UI.ScrollingGrid({
    y = 2,
    values = network,
    columns = gridColumns,
    sortColumn = 'label',
    autospace = true,
  }),
  notification = UI.Notification(),
  accelerators = {
    q = 'quit',
    c = 'clear',
  },
})

function sendCommand(host, command)

  if not device.wireless_modem then
    page.notification:error('Wireless modem not present')
    return
  end

  page.notification:info('Connecting')
  page:sync()

  local socket = Socket.connect(host, 161)
  if socket then
    socket:write({ type = command })
    socket:close()
    page.notification:success('Command sent')
  else
    page.notification:error('Failed to connect')
  end
end

function page:eventHandler(event)
  local t = self.grid.selected
  if t then
    if event.type == 'telnet' or event.type == 'grid_select' then
      multishell.openTab({
        path = '/apps/telnet.lua',
        focused = true,
        args = { t.id },
        title = t.label,
      })
    elseif event.type == 'vnc' then
      multishell.openTab({
        path = '/apps/vnc.lua',
        focused = true,
        args = { t.id },
        title = t.label,
      })
    elseif event.type == 'reboot' then
      sendCommand(t.id, 'reboot')
    elseif event.type == 'shutdown' then
      sendCommand(t.id, 'shutdown')
    end
  end
  if event.type == 'quit' then
    Event.exitPullEvents()
  end
  UI.Page.eventHandler(self, event)
end

function page.grid:getRowTextColor(row, selected)
  if not row.active then
    return colors.orange
  end
  return UI.Grid.getRowTextColor(self, row, selected)
end

function page.grid:getDisplayValues(row)
  row = Util.shallowCopy(row)
  if row.uptime then
    if row.uptime < 60 then
      row.uptime = string.format("%ds", math.floor(row.uptime))
    else
      row.uptime = string.format("%sm", math.floor(row.uptime/6)/10)
    end
  end
  if row.fuel then
    row.fuel = Util.toBytes(row.fuel)
  end
  if row.distance then
    row.distance = Util.round(row.distance, 1)
  end
  return row
end

function page.grid:draw()
  self:adjustWidth()
  UI.Grid.draw(self)
  if page.notification.enabled then
    page.notification:draw()
  end
end

function updateComputers()
  while true do
    page.grid:update()
    page.grid:draw()
    page:sync()
    os.sleep(1)
  end
end

Event.addHandler('device_attach', function(h, deviceName)
  if deviceName == 'wireless_modem' then
    page.notification:success('Modem connected')
    page:sync()
  end
end)

Event.addHandler('device_detach', function(h, deviceName)
  if deviceName == 'wireless_modem' then
    page.notification:error('Wireless modem not attached')
    page:sync()
  end
end)

if not device.wireless_modem then
  page.notification:error('Wireless modem not attached')
end

UI:setPage(page)
Event.pullEvents(updateComputers)
UI.term:reset()