local type = type
local ipairs = ipairs
local color = require("gears.color")
local gsurface = require("gears.surface")
local cairo = require("lgi").cairo
local pango = require("lgi").Pango
local pangocairo = require("lgi").PangoCairo
local beautiful = require("beautiful")

local module = {}

local pango_l,pango_crx = {},{}

local function draw_item(cr,x,y,width,height,padding,args)
  cr:save()
  cr:rectangle(x,y,width+padding,height+padding)
  cr:clip()
  cr:set_source(color(args.bg or beautiful.bg_alternate))
  cr:arc(x + (height-padding)/2 + 2, y + (height-padding)/2 + padding/4 + (args.margins or 0), (height-padding)/2+(args.padding or 0)/2,0,2*math.pi)
  cr:fill()
  cr:arc(x + width - (height-padding)/2 - 2, y + (height-padding)/2 + padding/4 + (args.margins or 0), (height-padding)/2+(args.padding or 0)/2,0,2*math.pi)
  cr:rectangle(x + (height-padding)/2+2, y + padding/4 + (args.margins or 0)-(args.padding or 0)/2,width - (height),(height-padding)+(args.padding or 0))
  cr:fill()
  cr:set_source(color(args.fg or beautiful.bg_normal))
  cr:set_operator(cairo.Operator.CLEAR)
  cr:move_to(x+height/2 + 2,y+padding/4 + (args.margins or 0)-(args.padding or 0)/2)
  cr:show_layout(pango_l[height])
  cr:restore()
end

function module.draw_arrow(cr,x,y,width,height,padding,args)
  cr:save()
  cr:rectangle(x,y,width+padding,height+padding)
  cr:clip()
  padding=padding/2
  local mid = (height-2*padding)/2
  cr:move_to(x+mid      ,padding)
  cr:line_to(x+width-mid,padding)
  cr:line_to(x+width         ,mid+padding)
  cr:line_to(x+width-mid,height-padding)
  cr:line_to(x+mid      ,height-padding)
  cr:line_to(x+0             ,mid+padding)
  cr:line_to(x+mid      ,padding)
  cr:close_path()
  cr:set_source(color(args.bg or beautiful.bg_alternate))
  cr:fill()
  cr:set_source(color(args.fg or beautiful.bg_normal))
  cr:set_operator(cairo.Operator.CLEAR)
  cr:move_to(x+height/2+2,y+padding/2)
  cr:show_layout(pango_l[height])
  cr:restore()
end

function module.fit(text,args)
  local args = args or {}
  local height = args.height or (beautiful.menu_height)
  local padding = height/4--beautiful.default_height/3

  -- Get text height
  if not pango_l[height] then
    local pango_crx = pangocairo.font_map_get_default():create_context()
    pango_l[height] = pango.Layout.new(pango_crx)
    local desc = pango.FontDescription()
    desc:set_family("Verdana")
    desc:set_weight(pango.Weight.BOLD)
    desc:set_size((height-padding*2) * pango.SCALE)
--     desc:set_variant(pango.Variant.SMALL_CAPS)
    pango_l[height]:set_font_description(desc)
  end

  -- Compute full width
  local ret,full_width = {},0
  for k,v in ipairs(type(text) == "table" and text or {text}) do
    pango_l[height].text = v
    ret[k] = pango_l[height]:get_pixel_extents().width + height + padding
    full_width = full_width + ret[k]
  end
  return full_width,ret
end

function module.draw(text,args)
  local args = args or {}
  local height = args.height or (beautiful.menu_height)
  local padding = height/4--beautiful.default_height/3
  local full_width,ret = module.fit(text,args)

  local img = cairo.ImageSurface.create(cairo.Format.ARGB32, full_width+(args.padding_right or 0), height+padding)
  cr = cairo.Context(img)

  -- Draw every item
  local x = 0
  for k,v in ipairs(type(text) == "table" and text or {text}) do
    pango_l[height].text = v
    local draw_f = args.style or draw_item
    draw_f(cr,x,0,ret[k],height,padding,args)
    x =  x + ret[k]
  end
  return img
end

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