From 4a05dd168333ce781962f2eabd974ba0bb504a17 Mon Sep 17 00:00:00 2001
From: Guldoman <giulio.lettieri@gmail.com>
Date: Tue, 7 Dec 2021 21:44:20 +0100
Subject: [PATCH] Split `TreeView:draw` into multiple functions

This allows plugins to override each aspect of TreeView item drawing.
---
 data/plugins/treeview.lua | 85 ++++++++++++++++++++++++++-------------
 1 file changed, 56 insertions(+), 29 deletions(-)

diff --git a/data/plugins/treeview.lua b/data/plugins/treeview.lua
index 5dcb035d..18d622e1 100644
--- a/data/plugins/treeview.lua
+++ b/data/plugins/treeview.lua
@@ -43,6 +43,9 @@ function TreeView:new()
   self.cache = {}
   self.tooltip = { x = 0, y = 0, begin = 0, alpha = 0 }
 
+  self.item_icon_width = 0
+  self.item_text_spacing = 0
+
   local on_dirmonitor_modify = core.on_dirmonitor_modify
   function core.on_dirmonitor_modify(dir, filepath)
     if self.cache[dir.name] then
@@ -260,6 +263,9 @@ function TreeView:update()
     self.tooltip.alpha = 0
   end
 
+  self.item_icon_width = style.icon_font:get_width("D")
+  self.item_text_spacing = style.icon_font:get_width("f") / 2
+
   TreeView.super.update(self)
 end
 
@@ -312,42 +318,63 @@ function TreeView:get_item_text(item, active, hovered)
 end
 
 
+function TreeView:draw_item_text(item, active, hovered, x, y, w, h)
+  local item_text, item_font, item_color = self:get_item_text(item, active, hovered)
+  common.draw_text(item_font, item_color, item_text, nil, x, y, 0, h)
+end
+
+
+function TreeView:draw_item_icon(item, active, hovered, x, y, w, h)
+  local icon_char, icon_font, icon_color = self:get_item_icon(item, active, hovered)
+  common.draw_text(icon_font, icon_color, icon_char, nil, x, y, 0, h)
+  return self.item_icon_width + self.item_text_spacing
+end
+
+
+function TreeView:draw_item_body(item, active, hovered, x, y, w, h)
+    x = x + self:draw_item_icon(item, active, hovered, x, y, w, h)
+    self:draw_item_text(item, active, hovered, x, y, w, h)
+end
+
+
+function TreeView:draw_item_chevron(item, active, hovered, x, y, w, h)
+  if item.type == "dir" then
+    local chevron_icon = item.expanded and "-" or "+"
+    local chevron_color = hovered and style.accent or style.text
+    common.draw_text(style.icon_font, chevron_color, chevron_icon, nil, x, y, 0, h)
+  end
+  return style.padding.x
+end
+
+
+function TreeView:draw_item_background(item, active, hovered, x, y, w, h)
+  if hovered then
+    renderer.draw_rect(x, y, w, h, style.line_highlight)
+  end
+end
+
+
+function TreeView:draw_item(item, active, hovered, x, y, w, h)
+  self:draw_item_background(item, active, hovered, x, y, w, h)
+
+  x = x + item.depth * style.padding.x + style.padding.x
+  x = x + self:draw_item_chevron(item, active, hovered, x, y, w, h)
+
+  self:draw_item_body(item, active, hovered, x, y, w, h)
+end
+
+
 function TreeView:draw()
   self:draw_background(style.background2)
 
-  local icon_width = style.icon_font:get_width("D")
-  local spacing = style.icon_font:get_width("f") / 2
-
   local doc = core.active_view.doc
   local active_filename = doc and system.absolute_path(doc.filename or "")
 
   for item, x,y,w,h in self:each_item() do
-    local icon_char, icon_font, icon_color =
-      self:get_item_icon(item, item.abs_filename == active_filename,
-                         item == self.hovered_item)
-    local item_text, item_font, item_color =
-      self:get_item_text(item, item.abs_filename == active_filename,
-                         item == self.hovered_item)
-
-    -- hovered item background
-    if item == self.hovered_item then
-      renderer.draw_rect(x, y, w, h, style.line_highlight)
-    end
-
-    -- icons
-    x = x + item.depth * style.padding.x + style.padding.x
-    if item.type == "dir" then
-      local expand_icon = item.expanded and "-" or "+"
-      local expand_color = item == self.hovered_item and style.accent or style.text
-      common.draw_text(style.icon_font, expand_color, expand_icon, nil, x, y, 0, h)
-    end
-    x = x + style.padding.x
-    common.draw_text(icon_font, icon_color, icon_char, nil, x, y, 0, h)
-    x = x + icon_width
-
-    -- text
-    x = x + spacing
-    x = common.draw_text(item_font, item_color, item_text, nil, x, y, 0, h)
+    self:draw_item(item,
+      item.abs_filename == active_filename,
+      item == self.hovered_item,
+      x, y, w, h)
   end
 
   self:draw_scrollbar()