local setmetatable = setmetatable
local pairs,ipairs = pairs, ipairs
local type,string  = type,string
local print,unpack = print, unpack
local table        = table
local beautiful    = require( "beautiful"               )
local util         = require( "awful.util"              )
local aw_key       = require( "awful.key"               )
local object       = require( "radical.object"          )
local vertical     = require( "radical.layout.vertical" )
local theme        = require( "radical.theme"           )
local item_mod     = require( "radical.item"            )
local common       = require( "radical.common"          )

local capi = { mouse = mouse, screen = screen , keygrabber = keygrabber, root=root, }

local module = {
  arrow_type = {
    NONE     = 0,
    PRETTY   = 1,
    CENTERED = 2,
  },
  event      ={
    NEVER    = 0,
    BUTTON1  = 1,
    BUTTON2  = 2,
    BUTTON3  = 3,
    SELECTED = 100,
    HOVER    = 1000,
    LEAVE    = 1001,
  },
  item_flags     = {
    NONE      = 999999 ,
    DISABLED  = 1 , -- Cannot be interacted with
    URGENT    = 2 , -- Need attention
    SELECTED  = 3 , -- Single item selected [[FOCUS]]
    PRESSED   = 4 , -- Mouse pressed
    HOVERED   = -1 , -- Mouse hover
    CHANGED   = 6 , -- The item changed, need attention
    USED      = 7 , -- Common flag
    CHECKED   = 8 , -- When checkbox isn't enough
    ALTERNATE = 9 ,
    HIGHLIGHT = 10 ,
    HEADER    = 11,

    -- Implementation defined flags
    USR1     = 101,
    USR2     = 102,
    USR3     = 103,
    USR4     = 104,
    USR5     = 105,
    USR6     = 106,
    USR7     = 107,
    USR8     = 108,
    USR9     = 109,
    USR10    = 110,
  },
  colors_by_id = theme.colors_by_id
}

theme.register_color(module.item_flags.DISABLED  , "disabled"  , "disabled"  , true )
theme.register_color(module.item_flags.URGENT    , "urgent"    , "urgent"    , true )
theme.register_color(module.item_flags.SELECTED  , "focus"     , "focus"     , true )
theme.register_color(module.item_flags.PRESSED   , "pressed"   , "pressed"   , true )
theme.register_color(module.item_flags.HOVERED   , "hover"     , "hover"     , true )
theme.register_color(module.item_flags.CHANGED   , "changed"   , "changed"   , true )
theme.register_color(module.item_flags.USED      , "used"      , "used"      , true )
theme.register_color(module.item_flags.CHECKED   , "checked"   , "checked"   , true )
theme.register_color(module.item_flags.ALTERNATE , "alternate" , "alternate" , true )
theme.register_color(module.item_flags.HIGHLIGHT , "highlight" , "highlight" , true )
--     register_color(item_flags.HEADER    , ""
--     register_color(item_flags.USR1      , ""
--     register_color(item_flags.USR2      , ""
--     register_color(item_flags.USR3      , ""
--     register_color(item_flags.USR4      , ""
--     register_color(item_flags.USR5      , ""
--     register_color(item_flags.USR6      , ""
--     register_color(item_flags.USR7      , ""
--     register_color(item_flags.USR8      , ""
--     register_color(item_flags.USR9      , ""
--     register_color(item_flags.USR10     , ""


local function filter(data)
  if not data.filter == false then
    local fs,visible_counter = data.filter_string:lower(),0
    data._internal.visible_item_count = 0
    for k,v in pairs(data.items) do
      local tmp = v._filter_out
      v._filter_out = (v.text:lower():find(fs) == nil)-- or (fs ~= "")
      if tmp ~= v._filter_out then
        v.widget:emit_signal("widget::updated")
      end
      if (not v._filter_out) and (not v.widget.visible == false) then
        visible_counter = visible_counter + v.height
        data._internal.visible_item_count = data._internal.visible_item_count +1
        v.f_key = data._internal.visible_item_count
      end
    end
    data._total_item_height = visible_counter

    -- Make sure to select an item
    if data._current_item and data._current_item._filter_out then
      local n = data.next_item
      if n then
        n.selected = true
      end
    end

  end
end

-- Get the number of visible rows
local function get_visible_row_count(data)
  local visblerow = data.filter_string == "" and data.rowcount or data._internal.visible_item_count
  if data.max_items and data.max_items < data.rowcount then
    visblerow = data.max_items
    if data.filter_string ~= "" then
      local cur,vis = (data._start_at or 1),0
      while (data._internal.items[cur] and data._internal.items[cur]) and cur < data.max_items + (data._start_at or 1) do
        vis = vis + (data._internal.items[cur]._filter_out and 0 or 1)
        cur = cur +1
      end
      visblerow = vis
    end
  end
  return visblerow
end

------------------------------------KEYBOARD HANDLING-----------------------------------
local function activateKeyboard(data)
  if not data then return end
  if not data or grabKeyboard == true then return end
  if (not (data._internal.private_data.enable_keyboard == false)) and data.visible == true then
    capi.keygrabber.run(function(mod, key, event)
      for k,v in pairs(data._internal.filter_hooks or {}) do --TODO modkeys
        if (k.key == "Mod4" or k.key == "Mod1") and (key == "End" or key == "Super_L" or key == "Alt_L") then
          local found = false
          for k3,v3 in ipairs(mod) do
            for k4,v4 in ipairs({"Mod4","Mod1"})do
              if v3 == v4 and event == k.event then
                local retval,self = v(data,mod)
                if self and type(self) == "table" then
                  data = self
                end
              end
            end
          end
        end
        if k.key == key and k.event == event then
          local retval, self = v(data,mod)
          if self and type(self) == "table" then
            data = self
          end
          if retval == false then
            data.visible = false
            capi.keygrabber.stop()
          end
          return retval
        end
      end
      if event == "release" then
          return true
      end

      if (key == 'Return') and data._current_item and data._current_item.button1 then
          data._current_item.button1(data,data._current_item)
          if data.sub_menu_on ~= module.event.BUTTON1 then
            data.visible = false
          end
      elseif key == 'Escape' or (key == 'Tab' and data.filter_string == "") then
        data.visible = false
        capi.keygrabber.stop()
      elseif (key == 'BackSpace') and data.filter_string ~= "" and data.filter == true then
        data.filter_string = data.filter_string:sub(1,-2)
        filter(data)
      elseif data.filter == true and key:len() == 1 then
        data.filter_string = data.filter_string .. key:lower()
        filter(data)
      else
        data.visible = false
        capi.keygrabber.stop()
      end
      return true
    end)
  end
end

---------------------------------ITEM HANDLING----------------------------------
local function add_item(data,args)
  local item = item_mod(data,args)
  data._internal.setup_item(data,item,args)
  if args.selected == true then
    item.selected = true
  end

  -- Show sub-menu
  if item._private_data.sub_menu_f  or item._private_data.sub_menu_m then
    if data.sub_menu_on == module.event.SELECTED then
      item.widget:set_menu(
        function()
          local sub_menu = item._private_data.sub_menu_m

          if not sub_menu and item._private_data.sub_menu_f then
            sub_menu = item._private_data.sub_menu_f(item.widget)
          end

          if sub_menu and (item._private_data.sub_menu_f or sub_menu.rowcount > 0) then
            sub_menu.arrow_type = module.arrow_type.NONE
            sub_menu.parent_item = item
            item._tmp_menu = sub_menu
            data._tmp_menu = sub_menu
          end

          return sub_menu
        end,
        "mouse::enter"
      )
    end
  end

  item.index = data.rowcount
  data:emit_signal("item::added",item)
  return item
end

local function add_items(data,items)
  local ret = {}
  for k,item in ipairs(items) do
    ret[k] = data:add_item(item)
  end
  return ret
end


local function add_widget(data,widget,args)
  args = args or {}
  data._internal.has_widget = true
  widget._fit = widget.fit
  widget.fit = function(self,context,width,height)
    local w,h = widget._fit(self, context, width or 1, height or 1)
    return args.width or w,args.height or h
  end

  local item,private_data = object({
    private_data = {
      widget = widget,
      selected = false,
    },
    force_private = {
      visible = true,
      selected = true,
    },
    autogen_getmap  = true,
    autogen_setmap  = true,
    autogen_signals = true,
  })
  item._private_data = private_data
  item._internal = {}

  data._internal.widgets[#data._internal.widgets+1] = item
  data._internal.items[#data._internal.items+1] = item
  data:emit_signal("widget::added",item,widget)
end

local function add_widgets(data,widgets)
  for k,item in ipairs(widgets) do
    data:add_widget(item)
  end
end

local function add_prefix_widget(data,widget,args)
  data:emit_signal("prefix_widget::added",widget,args)
end

local function add_suffix_widget(data,widget,args)
  data:emit_signal("suffix_widget::added",widget,args)
end

-- Sum all widgets height and width
local function get_widget_fit_sum(data)
  local h,w = 0,0
  -- TODO query this from the layout itself
  for k,v in ipairs(data._internal.widgets) do
    local fw,fh = v.widget:get_preferred_size()
    w,h = w + fw,h + fh
  end
  return w,h
end

local function get_widget_fit_width_sum(data)
  -- TODO query this from the layout itself
  return get_widget_fit_sum(data)
end

local function get_widget_fit_height_sum(data)
  -- TODO query this from the layout itself
  local w,h = get_widget_fit_sum(data)
  return h
end

local function add_embeded_menu(data,menu)
  add_widget(data,menu._internal.layout)
  menu._embeded_parent = data
end

local function add_colors_namespace(data,namespace)
  theme.add_colors_from_namespace(data,namespace)
end

local function add_key_binding(data,mod,key,func)
  capi.root.keys(util.table.join(capi.root.keys(),aw_key(mod or {}, key, func and func() or function ()
      data.visible = not data.visible
  end)))
end


---------------------------------MENU HANDLING----------------------------------
local function new(args)
  local args = args or {}
  local internal = args.internal or {}
  if not internal.items then internal.items = {} end
  if not internal.widgets then internal.widgets = {} end

  -- All the magic in the universe
  local data,private_data = object({
    private_data = {
      -- Default settings
      bg              = args.bg or beautiful.menu_bg_normal or beautiful.bg_normal or "#000000",
      fg              = args.fg or beautiful.menu_fg_normal or beautiful.fg_normal or "#ffffff",
      bg_header       = args.bg_header    or beautiful.menu_bg_header or beautiful.fg_normal,
      bg_prefix       = args.bg_prefix    or nil,
      border_color    = args.border_color or beautiful.menu_border_color or beautiful.border_color or "#333333",
      border_width    = args.border_width or beautiful.menu_border_width or beautiful.border_width or 3,
      separator_color = args.separator_color or beautiful.menu_separator_color or args.border_color or beautiful.menu_border_color or beautiful.border_color or "#333333",
      item_height     = args.item_height  or beautiful.menu_height or 30,
      item_width      = args.item_width or nil,
      width           = args.width or args.menu_width or beautiful.menu_width or 130,
      default_width   = args.width or args.menu_width or beautiful.menu_width or 130,
      icon_size       = args.icon_size or nil,
      auto_resize     = args.auto_resize or true,
      parent_geometry = args.parent or nil,
      arrow_type      = args.arrow_type or beautiful.menu_arrow_type or module.arrow_type.PRETTY,
      visible         = args.visible or false,
      direction       = args.direction or "top",
      has_changed     = false,
      row             = args.row or nil,
      column          = args.column or nil,
      layout          = args.layout or nil,
      screen          = args.screen or nil,
      style           = args.style  or nil,
      item_style      = args.item_style or beautiful.menu_default_item_style or require("radical.item.style.basic"),
      item_layout     = args.item_layout or nil,
      filter          = args.filter ~= false,
      show_filter     = args.show_filter or false,
      filter_string   = args.filter_string or "",
      suffix_widget   = args.suffix_widget or nil,
      prefix_widget   = args.prefix_widget or nil,
      fkeys_prefix    = args.fkeys_prefix or false,
      overlay_draw    = args.overlay_draw or nil,
      filter_underlay = args.filter_underlay or nil,
      filter_prefix   = args.filter_prefix or "Filter:",
      enable_keyboard = (args.enable_keyboard ~= false),
      max_items       = args.max_items or nil,
      disable_markup  = args.disable_markup or false,
      x               = args.x or 0,
      y               = args.y or 0,
      sub_menu_on     = args.sub_menu_on or module.event.SELECTED,
      select_on       = args.select_on or module.event.HOVER,
      opacity         = args.opacity or beautiful.menu_opacity or 1,
      spacing         = args.spacing or nil,
      default_margins = args.default_margins or beautiful.menu_default_margins or {},
      icon_per_state  = args.icon_per_state or false,
      default_item_margins  = args.default_item_margins or {},
      icon_transformation   = args.icon_transformation or nil,
      filter_underlay_style = args.filter_underlay_style or nil,
      filter_underlay_color = args.filter_underlay_color,
      filter_placeholder    = args.filter_placeholder or "",
      disable_submenu_icon  = args.disable_submenu_icon or false,
      item_border_color     = args.item_border_color or beautiful.menu_item_border_color
        or args.border_color or beautiful.menu_border_color or beautiful.border_color or nil,
      item_border_width     = args.item_border_width or beautiful.menu_item_border_width or nil,
    },
    force_private = {
      parent  = true,
      visible = true,
    },
    always_handle = {
      width = true,
      height = true,
    },
    autogen_getmap  = true,
    autogen_setmap  = true,
    autogen_signals = true,
  })
  internal.private_data = private_data
  
  -- Methods
  data.add_item,data.add_widget,data.add_embeded_menu,data._internal,data.add_key_binding = add_item,add_widget,add_embeded_menu,internal,add_key_binding
  data.add_prefix_widget,data.add_suffix_widget,data.add_items,data.add_widgets=add_prefix_widget,add_suffix_widget,add_items,add_widgets
  data.add_colors_namespace = add_colors_namespace
  
  -- Load colors
  theme.setup_colors(data,args)

  -- Getters
  data.get_is_menu               = function(_) return true end
  data.get_margin                = function(_) return {left=0,bottom=0,right=0,left=0} end
  data.get_items                 = function(_) return internal.items end
  data.get_rowcount              = function(_) return #internal.items end
  data.get_visible_row_count     = get_visible_row_count
  data.get_widget_fit_height_sum = get_widget_fit_height_sum
  data.get_widget_fit_width_sum  = get_widget_fit_width_sum

  -- Setters
  data.set_auto_resize  = function(_,val) private_data[""] = val end
  data.set_parent_geometry = function(_,value)
    private_data.parent_geometry = value --TODO delete
  end

  data.set_visible = function(_,value)
    private_data.visible = value
    if data._tmp_menu and data._current_item then
      data._current_item._tmp_menu = nil
      data.item_style(data._current_item,{})
    end
    if internal.has_changed and data.style then
      data.style(data,{arrow_x=20,margin=internal.margin})
    end
    if internal.set_position then
      internal.set_position(data)
    end
    if internal.set_visible then
      internal:set_visible(value)
    end
    if value and not capi.keygrabber.isrunning() then
      activateKeyboard(data)
    elseif data.parent_geometry and not data.parent_geometry.is_menu and data.enable_keyboard then
      capi.keygrabber.stop()
    end

    -- Hide the sub menus when hiding
    if data._tmp_menu and not value then
      data._tmp_menu.visible = false
    end
  end

  data.add_colors_group = function(data,section)
    theme.add_section(data,section,args)
  end

  data.set_layout = function(_,value)
    if value then
      local f = value.setup_key_hooks or common.setup_key_hooks
      f(value, data)
    end
    private_data.layout = value
  end

--   set_map.auto_resize = function(value)
--     for k,v in ipairs(internal.items) do
--       TODO check all items size, ajustthe fit and global width
--     end
--   end

  data.get_current_index = function(_)
    if data._current_item then
      for k,v in ipairs(internal.items) do --rows
        if data._current_item == v then
          return k
        end
      end
    end
  end

  data.get_previous_item = function(_)
    local candidate,idx = internal.items[(data.current_index or 0)-1],(data.current_index or 0)-1
    while candidate and (candidate.widget.visible == false or candidate._filter_out) and idx > 0 do
      candidate,idx = internal.items[idx - 1],idx-1
    end
    return (candidate or internal.items[data.rowcount])
  end
  data.get_next_item     = function(_)
    local candidate,idx = internal.items[(data.current_index or 0)+1],(data.current_index or 0)+1
    while candidate and (candidate.widget.visible == false or candidate._filter_out) and idx <= data.rowcount do
      candidate,idx = internal.items[idx + 1],idx+1
    end
    return (candidate or internal.items[1])
  end

  --Repaint when appearance properties change
  for k,v in ipairs({"bg","fg","border_color","border_width","item_height","width","arrow_type"}) do
    data:connect_signal(v.."::changed",function()
      if data.visible and data.style then
--         data.style(data,{arrow_x=20,margin=internal.margin})
      else
        data.has_changed = true
      end
    end)
  end

  function data:add_key_hook(mod, key, event, func)
    if key and event and func then
        internal.filter_hooks = internal.filter_hooks or {}
        internal.filter_hooks[{key = key, event = event, mod = mod}] = func
    end
  end

  function data:remove_key_hook(key) --TODO broken?
      for k,v in pairs(internal.filter_hooks or {}) do
          if k.key == key then
              internal.filter_hooks[k] = nil
              break
          end
      end
  end

  function data:clear()
    internal.items = {}
    data:emit_signal("clear::menu")
  end

  function data:swap(item1,item2)
    if not item1 or not item2 then return end
    if not item1 or not item2 and item1 ~= item2 then return end
    local idx1,idx2
    for k,v in ipairs(internal.items) do --rows
      for k2,v2 in ipairs(v) do --columns
        if item2 == v2 then
          idx2 = k
        end
        if item1 == v2 then
          idx1 = k
        end
      end
      if idx1 and idx2 then
        break
      end
    end
    if idx1 and idx2 then
      internal.items[idx1],internal.items[idx2] = internal.items[idx2],internal.items[idx1]
      item1.index,item2.index = idx2,idx1
      data:emit_signal("item::swapped",item1,item2,idx1,idx2)
    end
  end

  function data:move(item,idx)
    if not item or not idx then return end
    local idx1 = nil
    for k,v in ipairs(internal.items) do --rows
      if item == v then
        idx1 = k
        break
      end
    end
    if idx1 then
--       if idx < idx1 then
--         idx = idx + 1
--       end
      if idx1 > #internal.items + 1 then
        idx1 = #internal.items + 1
      end
      if idx ~= idx1 then
        table.insert(internal.items,idx1,table.remove(internal.items,idx))
        item.index = idx
        data:emit_signal("item::moved",item,idx,idx1)
        for i=idx,idx1 do
          internal.items[i].index = i
        end
      end
    end
  end

  function data:remove(item)
    if not item then return end
    local idx1 = nil
    for k,v in ipairs(internal.items) do --rows
      if item == v then
        idx1 = k
        break
      end
    end
    if idx1 then
      table.remove(internal.items,idx1)
      data:emit_signal("item::removed",item,idx1)
      for i=idx1,#internal.items do
        internal.items[i].index = i
      end
    end
  end

  function data:append(item)
    if not item then return end
    internal.items[#internal.items + 1] = item
    data:emit_signal("item::appended",item)
  end

  function data:scroll_up()
    if data.max_items ~= nil and data.rowcount >= data.max_items and (data._start_at or 1) > 1 then
      local current_item = data._current_item
      if current_item then
        current_item:set_selected(false,true)
      end
      data._start_at  = (data._start_at or 1) - 1
      internal.items[data._start_at].widget:set_visible(false)
      data:emit_signal("_hidden::changed",internal.items[data._start_at])
      internal.items[data._start_at+data.max_items].widget:set_visible(true)
      data:emit_signal("_hidden::changed",internal.items[data._start_at+data.max_items])
      filter(data)

      --HACK upstream bug
      internal.items[data._start_at].widget:emit_signal("widget::layout_changed")
      internal.items[data._start_at+data.max_items].widget:emit_signal("widget::layout_changed")
      data._internal.layout:emit_signal("widget::redraw_needed")
      data._internal.layout:emit_signal("widget::layout_changed")
    end
  end

  function data:scroll_down()
    if data.max_items ~= nil and data.rowcount >= data.max_items and (data._start_at or 1)+data.max_items <= data.rowcount then
      local current_item = data._current_item
      if current_item then
        current_item:set_selected(false,true)
      end
      data._start_at  = (data._start_at or 1) + 1
      internal.items[data._start_at-1].widget:set_visible(true)
      data:emit_signal("_hidden::changed",internal.items[data._start_at-1])
      internal.items[data._start_at-1+data.max_items].widget:set_visible(false)
      data:emit_signal("_hidden::changed",internal.items[data._start_at-1+data.max_items])
      filter(data)

      --HACK upstream bug
      internal.items[data._start_at-1].widget:emit_signal("widget::layout_changed")
      internal.items[data._start_at-1+data.max_items].widget:emit_signal("widget::layout_changed")
      data._internal.layout:emit_signal("widget::redraw_needed")
      data._internal.layout:emit_signal("widget::layout_changed")
    end
  end

  function data:hide()
    data.visible = false

    if data.parent_geometry and data.parent_geometry.parent_menu then
      local parent = data.parent_geometry.parent_menu
      while parent do
        parent.visible = false
        parent = parent.parent_geometry and parent.parent_geometry.is_menu and parent.parent_geometry
      end
    end
  end

  if private_data.layout then
    local f = private_data.layout.setup_key_hooks or common.setup_key_hooks
    f(value, data)
  end

  data._internal.setup_drawable(data)

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