--
-- Button Widget.
-- @copyright Jefferson Gonzalez
-- @license MIT
--

local style = require "core.style"
local Widget = require "libraries.widget"

---@class widget.button.icon
---@field public code string | nil
---@field public color renderer.color | nil
---@field public hover_color renderer.color | nil
local ButtonIcon = {}

---@class widget.button : widget
---@field public padding widget.position
---@field public icon widget.button.icon
---@field public expanded boolean
local Button = Widget:extend()

---Constructor
---@param parent widget
---@param label string
function Button:new(parent, label)
  Button.super.new(self, parent)

  self.type_name = "widget.button"

  self.icon = {
    code = nil, color = nil, hover_color = nil
  }

  self.padding = {
    x = style.padding.x,
    y = style.padding.y
  }

  self.expanded = false

  self:set_label(label or "")
end

---When set to true the button width will be the same as parent
---@param expand? boolean | nil
function Button:toggle_expand(expand)
  if type(expand) == "boolean" then
    self.expanded = expand
  else
    self.expanded = not self.expanded
  end
end

---Set the icon drawn alongside the button text.
---@param code? string
---@param color? renderer.color
---@param hover_color? renderer.color
function Button:set_icon(code, color, hover_color)
  self.icon.code = code
  self.icon.color = color
  self.icon.hover_color = hover_color

  self:set_label(self.label)
end

---Set the button text and recalculates the widget size.
---@param text string
function Button:set_label(text)
  Button.super.set_label(self, text)

  local font = self:get_font()
  local border = self.border.width * 2

  if self.expanded and self.parent then
    self.size.x = self.parent.size.x - self.position.rx - border
  else
    self.size.x = font:get_width(self.label) + (self.padding.x * 2) - border
  end

  self.size.y = font:get_height() + (self.padding.y * 2) - border

  if self.icon.code then
    local icon_w = style.icon_font:get_width(self.icon.code)

    if self.label ~= "" then
      icon_w = icon_w + (self.padding.x / 2)
    end

    local icon_h = style.icon_font:get_height() + (self.padding.y * 2) - border

    self.size.x = self.size.x + icon_w
    self.size.y = math.max(self.size.y, icon_h)
  end
end

function Button:on_mouse_enter(...)
  Button.super.on_mouse_enter(self, ...)
  self.hover_text = style.accent
  self.hover_back = style.line_highlight
end

function Button:on_mouse_leave(...)
  Button.super.on_mouse_leave(self, ...)
  self.hover_text = nil
  self.hover_back = nil
end

function Button:on_scale_change(new_scale, prev_scale)
  Button.super.on_scale_change(self, new_scale, prev_scale)
  self.padding.x = self.padding.x * (new_scale / prev_scale)
  self.padding.y = self.padding.y * (new_scale / prev_scale)
end

function Button:update()
  if not Button.super.update(self) then return false end

  -- update size
  self:set_label(self.label)

  return true
end

function Button:draw()
  self.background_color = self.hover_back or style.background

  if not Button.super.draw(self) then return false end

  local font = self:get_font()

  local offsetx = self.position.x + self.padding.x
  local offsety = self.position.y
  local h = self:get_height()
  local ih, th = style.icon_font:get_height(), font:get_height()

  if self.icon.code then
    local normal = self.icon.color or style.text
    local hover = self.icon.hover_color or style.accent
    renderer.draw_text(
      style.icon_font,
      self.icon.code,
      offsetx,
      th > ih and (offsety + (h / 2)) - (ih/2) or (offsety + self.padding.y),
      self.hover_text and hover or normal
    )
    offsetx = offsetx + style.icon_font:get_width(self.icon.code) + (style.padding.x / 2)
  end

  if self.label ~= "" then
    renderer.draw_text(
      font,
      self.label,
      offsetx,
      ih > th and (offsety + (h / 2)) - (th/2) or (offsety + self.padding.y),
      self.hover_text or self.foreground_color or style.text
    )
  end

  return true
end


return Button