---------------------------------------------------------------------------
-- @author Emmanuel Lepage Vallee <elv1313@gmail.com>
-- @copyright 2014 Emmanuel Lepage Vallee
-- @release devel
-- @note Based on the original tasklist.lua
-- @license GPLv2+ because of copy paste from the old tasklist.lua
---------------------------------------------------------------------------

local capi = {client = client}
local rawset = rawset
local radical   = require( "radical"      )
local tag       = require( "awful.tag"    )
local beautiful = require( "beautiful"    )
local client = require( "awful.client" )
local wibox     = require( "wibox"        )
local client_menu =  require("radical.impl.tasklist.client_menu")

local sticky,urgent,instances,module = {},{},{},{}

-- Default button implementation
module.buttons = {
  [1] = function (c)
    if c == capi.client.focus then
      c.minimized = true
    else
      -- Without this, the following
      -- :isvisible() makes no sense
      c.minimized = false
      if not c:isvisible() then
          tag.viewonly(c:tags()[1])
      end
      -- This will also un-minimize
      -- the client, if needed
      capi.client.focus = c
      c:raise()
    end
  end,
  [3] = function(c)
    client_menu.client = c
    local menu = client_menu()
    menu.visible = not menu.visible
  end,
  [4] = function ()
    client.focus.byidx(1)
    if client.focus then client.focus:raise() end
  end,
  [5] = function ()
    client.focus.byidx(-1)
    if capi.client.focus then client.focus:raise() end
  end
}

local function sticky_callback(c)
  local val = c.sticky
  sticky[c] = val and true or nil
  for k,v in ipairs(instances) do
    if val then
      v.menu:append(v.cache[c])
    else
      v.menu:remove(v.cache[c])
    end
  end
end

local function urgent_callback(c)
  local val = c.urgent
  urgent[c] = val and true or nil
  local item = instances[c.screen].cache[c]
  if item then
    item.state[radical.base.item_flags.URGENT] = val or nil
  end
end

local function unmanage_callback(c)
  sticky[c] = nil
  urgent[c] = nil
  for k,v in ipairs(instances) do
    v.menu:remove(v.cache[c])
    v.cache[c] = nil
  end
end

-- Reload <float> <ontop> and <sticky> labels
local function reload_underlay(c)
  local cache = instances[c.screen].cache
  local udl,item = {},cache[c]
  if item then
    if c.ontop then
      udl[#udl+1] = "ontop"
    end
    if client.floating.get(c) then
      udl[#udl+1] = "float"
    end
    if c.sticky then
      udl[#udl+1] = "sticky"
    end
    item.underlay = udl
    item.widget:emit_signal("widget::updated")
  end
end

-- Reload title and icon
local function reload_content(c,b,a)
  local cache = instances[c.screen].cache
  local item = cache[c]
  if item then
    item.text = c.name or "N/A"
    item.icon = c.icon
  end
end

local function create_client_item(c,screen)
  local cache = instances[screen].cache
  local menu = instances[screen].menu
  -- If it already exist, don't waste time creating a copy
  if cache[c] then
    return menu:append(cache[c])
  end

  -- Too bad, let's create a new one
  local item = menu:add_item{text=c.name,icon=c.icon}
  item.client = c
  cache[c] = item
  return item
end

local function add_client(c,screen)
  if not (c.skip_taskbar or c.hidden or c.type == "splash" or c.type == "dock" or c.type == "desktop") and c.screen == screen then
    local ret = create_client_item(c,screen)
    reload_underlay(c)
    if c.focus == c then
      ret.selected = true
    end
  end
end

-- Unselect the old focussed client
local function unfocus(c)
  local cache = instances[c.screen].cache
  local item = cache[c]
  if item and item.selected then
    item.selected = false
  end
end

-- Select the newly focussed client
local function focus(c)
  local cache = instances[c.screen].cache
  local item = cache[c]
  if item then
    item.selected = true
  end
end

local function new(screen)
  local cache,menu = setmetatable({}, { __mode = 'k' }),radical.flexbar {
    select_on=radical.base.event.NEVER,
    fg       = beautiful.fg_normal,
    bg_focus = beautiful.taglist_bg_image_selected2,
    bg_urgent = beautiful.taglist_bg_image_urgent2,
    bg_hover   = beautiful.menu_bg_focus,
    disable_markup = true,
    underlay_style = radical.widgets.underlay.draw_arrow,
    overlay = function(data,item,cd,w,h)
--       print("foo!")
    end,
    icon_transformation = beautiful.tasklist_icon_transformation
  }

  -- Clear the menu and repopulate it
  local function load_clients(t)
    if not t then return end
    if t.selected and tag.getscreen(t) == screen then
      menu:clear()
      for k, c in ipairs(t:clients()) do
        if not c.sticky then
          add_client(c,screen)
        end
      end
      for c,_ in pairs(sticky) do
        add_client(c,screen)
      end
    end
  end

  -- Add and remove clients from the tasklist
  local function tagged(c,t)
    if t.selected and not c.sticky and tag.getscreen(t) == screen then
      add_client(c,screen)
    end
  end
  local function untagged(c,t)
    local item = cache[c]
    if t.selected and tag.getscreen(t) == screen then
      menu:remove(item)
    end
  end

  -- Connect to a bunch of signals
  tag.attached_connect_signal(screen, "property::selected" , load_clients)
  tag.attached_connect_signal(screen, "property::activated", load_clients)
  capi.client.connect_signal("tagged"            , tagged            )
  capi.client.connect_signal("untagged"          , untagged          )

  instances[screen] = {menu = menu, cache = cache }

  load_clients(tag.selected(screen))

  -- Try to make awesome think radical.bar is a real widget
  -- Use "menu._internal.layout" directly in :add to avoid
  -- the proxy overhead, for now it doesn't event work on 3.5.2
--   rawset(menu,"fit",function(self,...)
--     return menu._internal.layout.fit(menu._internal.layout,...)
--   end)
--   rawset(menu,"draw",function(self,...)
--     return menu._internal.layout.draw(menu._internal.layout,...)
--   end)
--   rawset(menu,"add_signal",function(self,...)
--     return menu._internal.layout.add_signal(menu._internal.layout,...)
--   end)
--   rawset(menu,"disconnect_signal",function(a,...)
--     return menu._internal.layout.disconnect_signal(menu._internal.layout,...)
--   end)
--   rawset(menu,"connect_signal",function(a,...)
--     return menu._internal.layout.disconnect_signal(menu._internal.layout,...)
--   end)
--   rawset(menu,"emit_signal",function(a,...)
--     return menu._internal.layout.emit_signal(menu._internal.layout,...)
--   end)
  
  menu:connect_signal("button::press",function(menu,item,button_id,mod)
    if module.buttons and module.buttons[button_id] then
      module.buttons[button_id](item.client,menu,item,button_id,mod)
    end
  end)

  return menu,menu._internal.layout
end

-- Global callbacks
capi.client.connect_signal("property::sticky"  , sticky_callback   )
capi.client.connect_signal("property::urgent"  , urgent_callback   )
capi.client.connect_signal("unmanage"          , unmanage_callback )
capi.client.connect_signal("focus"             , focus             )
capi.client.connect_signal("unfocus"           , unfocus           )
capi.client.connect_signal("property::sticky"  , reload_underlay   )
capi.client.connect_signal("property::ontop"   , reload_underlay   )
capi.client.connect_signal("property::floating", reload_underlay   )
capi.client.connect_signal("property::name"    , reload_content    )
capi.client.connect_signal("property::icon"    , reload_content    )

return setmetatable(module, { __call = function(_, ...) return new(...) end })
-- kate: space-indent on; indent-width 2; replace-tabs on;