requireInjector(getfenv(1))

local class = require('class')
local Config = require('config')
local Event = require('event')
local FileUI = require('ui.fileui')
local NFT = require('nft')
local SHA1 = require('sha1')
local Tween = require('ui.tween')
local UI = require('ui')
local Util = require('util')

local REGISTRY_DIR = 'usr/.registry'
local TEMPLATE = [[
local env = { }
for k,v in pairs(getfenv(1)) do
  env[k] = v 
end
setmetatable(env, { __index = _G })

local s, m = os.run(env, 'sys/apps/appRun.lua',  %s, ...)
if not s then
  error(m)
end
]]

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

local config = {
  Recent = { },
  currentCategory = 'Apps',
}
Config.load('Overview', config)

local applications = { }

local function loadApplications()

  local requirements = {
    turtle = function() return turtle end,
    advancedTurtle = function() return turtle and term.isColor() end,
    pocket = function() return pocket end,
    advancedPocket = function() return pocket and term.isColor() end,
    advancedComputer = function() return not turtle and not pocket and term.isColor() end,
  }

  applications = Util.readTable('sys/etc/app.db')

  if fs.exists(REGISTRY_DIR) then
    local files = fs.list(REGISTRY_DIR)
    for _,file in pairs(files) do
      local app = Util.readTable(fs.combine(REGISTRY_DIR, file))
      if app and app.key then
        app.filename = fs.combine(REGISTRY_DIR, file)
        applications[app.key] = app
      end
    end
  end

  Util.each(applications, function(v, k) v.key = k end)
  applications = Util.filter(applications, function(_, a)
    if a.disabled then
      return false
    end

    if a.requires then
      local fn = requirements[a.requires]
      if fn and not fn() then
        return false
      end
    end

    return Util.startsWidth(a.run, 'http') or shell.resolveProgram(a.run)
  end)
end

loadApplications()

local defaultIcon = NFT.parse([[
8071180
8007180
7180071]])

local sx, sy = term.current().getSize()
local maxRecent = math.ceil(sx * sy / 62)

local function elipse(s, len)
  if #s > len then
    s = s:sub(1, len - 2) .. '..'
  end
  return s
end

local buttons = { }
local categories = { }
table.insert(buttons, { text = 'Recent', event = 'category' })
for _,f in pairs(applications) do
  if not categories[f.category] then
    categories[f.category] = true
    table.insert(buttons, { text = f.category, event = 'category' })
  end
end
table.insert(buttons, { text = '+', event = 'new' })

local function parseIcon(iconText)
  local icon

  local s, m = pcall(function()
    icon = NFT.parse(iconText)
    if icon then
      if icon.height > 3 or icon.width > 8 then
        error('Invalid size')
      end
    end
    return icon
  end)

  if s then
    return icon
  end

  return s, m
end

local page = UI.Page {
  tabBar = UI.TabBar {
    buttons = buttons,
  },
  container = UI.ViewportWindow {
    y = 2,
  },
  notification = UI.Notification(),
  accelerators = {
    r = 'refresh',
    e = 'edit',
    s = 'shell',
    l = 'lua',
    [ 'control-l' ] = 'refresh',
    [ 'control-n' ] = 'new',
    delete = 'delete',
  },
}

function page:draw()
  self.tabBar:draw()
  self.container:draw()
end

UI.Icon = class(UI.Window)
function UI.Icon:init(args)
  local defaults = {
    UIElement = 'Icon',
    width = 14,
    height = 4,
  }
  UI.setProperties(defaults, args)
  UI.Window.init(self, defaults)
end

function UI.Icon:eventHandler(event)
  if event.type == 'mouse_click' then
    self:setFocus(self.button)
    return true
  elseif event.type == 'mouse_doubleclick' then
    self:emit({ type = self.button.event, button = self.button })
  elseif event.type == 'mouse_rightclick' then
    self:setFocus(self.button)
    self:emit({ type = 'edit', button = self.button })
  end
  return UI.Window.eventHandler(self, event)
end

function page.container:setCategory(categoryName)

  -- reset the viewport window
  self.children = { }
  self.offy = 0

  local function filter(it, f)
    local ot = { }
    for _,v in pairs(it) do
      if f(v) then
        table.insert(ot, v)
      end
    end
    return ot
  end

  local filtered

  if categoryName == 'Recent' then
    filtered = { }

    for _,v in ipairs(config.Recent) do
      local app = Util.find(applications, 'key', v)
      if app then -- and fs.exists(app.run) then
        table.insert(filtered, app)
      end
    end

  else
    filtered = filter(applications, function(a) 
      return a.category == categoryName -- and fs.exists(a.run)
    end)
    table.sort(filtered, function(a, b) return a.title < b.title end)
  end

  for _,program in ipairs(filtered) do

    local icon
    if program.icon then
      icon = parseIcon(program.icon)
    end
    if not icon then
      icon = defaultIcon
    end

    local title = elipse(program.title, 8)

    local width = math.max(icon.width + 2, #title + 2)
    table.insert(self.children, UI.Icon({
      width = width,
      image = UI.NftImage({
        x = math.floor((width - icon.width) / 2) + 1,
        image = icon,
        width = 5,
        height = 3,
      }),
      button = UI.Button({
        x = math.floor((width - #title - 2) / 2) + 1,
        y = 4,
        text = title,
        backgroundColor = self.backgroundColor,
        width = #title + 2,
        event = 'button',
        app = program,
      }),
    }))
  end

  local gutter = 2
  if UI.term.width <= 26 then
    gutter = 1
  end
  local col, row = gutter, 2
  local count = #self.children

  -- reposition all children
  for k,child in ipairs(self.children) do
    child.x = -10
    child.y = math.floor(self.height)
    child.tween = Tween.new(6, child, { x = col, y = row }, 'outSine')

    if k < count then
      col = col + child.width
      if col + self.children[k + 1].width + gutter - 2 > UI.term.width then
        col = gutter
        row = row + 5
      end
    end
  end

  self:initChildren()
  self.animate = true
end

function page.container:draw()
  if self.animate then
    self.animate = false
    for i = 1, 6 do
      for _,child in ipairs(self.children) do
        child.tween:update(1)
        child.x = math.floor(child.x)
        child.y = math.floor(child.y)
      end
      UI.ViewportWindow.draw(self)
      self:sync()
      os.sleep()
    end
  else
    UI.ViewportWindow.draw(self)
  end
end

function page:refresh()
  local pos = self.container.offy
  self:focusFirst(self)
  self.container:setCategory(config.currentCategory)    
  self.container:setScrollPosition(pos)
end

function page:resize()
  self:refresh()
  UI.Page.resize(self)
end

function page:eventHandler(event)

  if event.type == 'category' then
    self.tabBar:selectTab(event.button.text)
    self.container:setCategory(event.button.text)
    self.container:draw()
    self:sync()

    config.currentCategory = event.button.text
    Config.update('Overview', config)

  elseif event.type == 'button' then
    for k,v in ipairs(config.Recent) do
      if v == event.button.app.key then
        table.remove(config.Recent, k)
        break
      end
    end
    table.insert(config.Recent, 1, event.button.app.key)
    if #config.Recent > maxRecent then
      table.remove(config.Recent, maxRecent + 1)
    end
    Config.update('Overview', config)
    multishell.openTab({
      title = event.button.app.title,
      path = 'sys/apps/shell',
      args = { event.button.app.run },
      focused = true,
    })

  elseif event.type == 'shell' then
    multishell.openTab({
      path = 'sys/apps/shell',
      focused = true,
    })

  elseif event.type == 'lua' then
    multishell.openTab({
      path = 'sys/apps/Lua.lua',
      focused = true,
    })

  elseif event.type == 'focus_change' then
    if event.focused.parent.UIElement == 'Icon' then
      event.focused.parent:scrollIntoView()
    end

  elseif event.type == 'tab_change' then
    if event.current > event.last then
      --self.container:setTransition(UI.effect.slideLeft)
    else
      --self.container:setTransition(UI.effect.slideRight)
    end

  elseif event.type == 'refresh' then
    loadApplications()
    self:refresh()
    self:draw()
    self.notification:success('Refreshed')

  elseif event.type == 'delete' then
    local focused = page:getFocused()
    if focused.app then
      focused.app.disabled = true
      local filename = focused.app.filename or fs.combine(REGISTRY_DIR, focused.app.key)
      Util.writeTable(filename, focused.app)
      loadApplications()
      page:refresh()
      page:draw()
      self.notification:success('Removed')
    end

  elseif event.type == 'new' then
    local category = 'Apps'
    if config.currentCategory ~= 'Recent' then
      category = config.currentCategory or 'Apps'
    end
    UI:setPage('editor', { category = category })

  elseif event.type == 'edit' then
    local focused = page:getFocused()
    if focused.app then
      UI:setPage('editor', focused.app)
    end

  else
    UI.Page.eventHandler(self, event)
  end
  return true
end

local formWidth = math.max(UI.term.width - 14, 26)

local editor = UI.Dialog {
  height = 11,
  width = formWidth,
  title = 'Edit application',
  form = UI.Form {
    y = 2,
    height = 9,
    title = UI.TextEntry {
      formLabel = 'Title', formKey = 'title', limit = 11, help = 'Application title',
      required = true,
    },
    run = UI.TextEntry {
      formLabel = 'Run', formKey = 'run', limit = 100, help = 'Full path to application',
      required = true,
    },
    category = UI.TextEntry {
      formLabel = 'Category', formKey = 'category', limit = 11, help = 'Category of application',
      required = true,
    },
    loadIcon = UI.Button {
      x = 11, y = 6, 
      text = 'Icon', event = 'loadIcon', help = 'Select icon'
    },
    image = UI.NftImage {
      y = 6,
      x = 2,
      height = 3,
      width = 8,
    },
  },
  statusBar = UI.StatusBar(),
  iconFile = '',
}

function editor:enable(app)
  if app then
    self.form:setValues(app)

    local icon
    if app.icon then
      icon = parseIcon(app.icon)
    end
    self.form.image:setImage(icon)
  end
  UI.Page.enable(self)
  self:focusFirst()
end

function editor.form.image:draw()
  self:clear()
  UI.NftImage.draw(self)
end

function editor:updateApplications(app)
  if not app.key then
    app.key = SHA1.sha1(app.title)
  end
  local filename = app.filename or fs.combine(REGISTRY_DIR, app.key)
  Util.writeTable(filename, app)
  loadApplications()
end

function editor:eventHandler(event)

  if event.type == 'form_cancel' or event.type == 'cancel' then
    UI:setPreviousPage()

  elseif event.type == 'focus_change' then
    self.statusBar:setStatus(event.focused.help or '')
    self.statusBar:draw()

  elseif event.type == 'loadIcon' then
    local fileui = FileUI({
      x = self.x,
      y = self.y,
      z = 2,
      width = self.width,
      height = self.height,
    })
    --fileui:setTransition(UI.effect.explode)
    UI:setPage(fileui, fs.getDir(self.iconFile), function(fileName)
      if fileName then
        self.iconFile = fileName
        local s, m = pcall(function()
          local iconLines = Util.readFile(fileName)
          if not iconLines then
            error('Unable to load file')
          end
          local icon, m = parseIcon(iconLines)
          if not icon then
            error(m)
          end
          self.form.values.icon = iconLines
          self.form.image:setImage(icon)
          self.form.image:draw()
        end)
        if not s and m then
          local msg = m:gsub('.*: (.*)', '%1')
          page.notification:error(msg)
        end
      end
    end)

  elseif event.type == 'form_invalid' then
    page.notification:error(event.message)

  elseif event.type == 'form_complete' then
    local values = self.form.values
    UI:setPreviousPage()
    self:updateApplications(values)
    page:refresh()
    page:draw()
  else
    return UI.Page.eventHandler(self, event)
  end
  return true
end

UI:setPages({
  editor = editor,
  main = page,
})

Event.on('os_register_app', function()
  loadApplications()
  page:refresh()
  page:draw()
  page:sync()
end)

page.tabBar:selectTab(config.currentCategory or 'Apps')
page.container:setCategory(config.currentCategory or 'Apps')
UI:setPage(page)

Event.pullEvents()
UI.term:reset()