From 09d1e3a6078ca8ae71724bd38cad6c379e8b3525 Mon Sep 17 00:00:00 2001
From: Simon Hauser <Simon-Hauser@outlook.de>
Date: Fri, 12 Feb 2021 22:30:55 +0100
Subject: [PATCH] feat: buffer previewers ls is now colorized (#505)

Requires up to date plenary version
---
 lua/telescope/previewers/buffer_previewer.lua | 56 +++++++++++++++++--
 plugin/telescope.vim                          | 17 ++++++
 2 files changed, 69 insertions(+), 4 deletions(-)

diff --git a/lua/telescope/previewers/buffer_previewer.lua b/lua/telescope/previewers/buffer_previewer.lua
index 352bd46..5560333 100644
--- a/lua/telescope/previewers/buffer_previewer.lua
+++ b/lua/telescope/previewers/buffer_previewer.lua
@@ -15,6 +15,50 @@ local previewers = {}
 
 local ns_previewer = vim.api.nvim_create_namespace('telescope.previewers')
 
+local color_hash = {
+  ['p'] = 'TelescopePreviewPipe',
+  ['c'] = 'TelescopePreviewCharDev',
+  ['d'] = 'TelescopePreviewDirectory',
+  ['b'] = 'TelescopePreviewBlock',
+  ['l'] = 'TelescopePreviewLink',
+  ['s'] = 'TelescopePreviewSocket',
+  ['.'] = 'TelescopePreviewNormal',
+  ['r'] = 'TelescopePreviewRead',
+  ['w'] = 'TelescopePreviewWrite',
+  ['x'] = 'TelescopePreviewExecute',
+  ['-'] = 'TelescopePreviewHyphen',
+  ['T'] = 'TelescopePreviewSticky',
+  ['S'] = 'TelescopePreviewSticky',
+  [2]   = 'TelescopePreviewSize',
+  [3]   = 'TelescopePreviewUser',
+  [4]   = 'TelescopePreviewGroup',
+  [5]   = 'TelescopePreviewDate',
+}
+color_hash[6]   = function(line)
+  return color_hash[line:sub(1, 1)]
+end
+
+local colorize_ls = function(bufnr, data, sections)
+  local windows_add = path.separator == '\\' and 2 or 0
+  for lnum, line in ipairs(data) do
+    local section = sections[lnum]
+    for i = 1, section[1].end_index - 1 do -- Highlight permissions
+      local c = line:sub(i, i)
+      vim.api.nvim_buf_add_highlight(bufnr, ns_previewer, color_hash[c], lnum - 1, i - 1, i)
+    end
+    for i = 2, #section do -- highlights size, (user, group), date and name
+      local hl_group = color_hash[i + (i ~= 2 and windows_add or 0)]
+      vim.api.nvim_buf_add_highlight(bufnr,
+        ns_previewer,
+        type(hl_group) == "function" and hl_group(line) or hl_group,
+        lnum - 1,
+        section[i].start_index - 1,
+        section[i].end_index - 1
+      )
+    end
+  end
+end
+
 previewers.file_maker = function(filepath, bufnr, opts)
   opts = opts or {}
   if opts.use_ft_detect == nil then opts.use_ft_detect = true end
@@ -24,9 +68,13 @@ previewers.file_maker = function(filepath, bufnr, opts)
     filepath = vim.fn.expand(filepath)
     local stat = vim.loop.fs_stat(filepath) or {}
     if stat.type == 'directory' then
-      pscan.ls_async(filepath, { hidden = true, on_exit = vim.schedule_wrap(function(data)
-        vim.api.nvim_buf_set_lines(bufnr, 0, -1, false, data)
-        if opts.callback then opts.callback(bufnr) end
+      pscan.ls_async(filepath, {
+        hidden = true,
+        group_directories_first = true,
+        on_exit = vim.schedule_wrap(function(data, sections)
+          vim.api.nvim_buf_set_lines(bufnr, 0, -1, false, data)
+          colorize_ls(bufnr, data, sections)
+          if opts.callback then opts.callback(bufnr) end
       end)})
     else
       path.read_file_async(filepath, vim.schedule_wrap(function(data)
@@ -438,7 +486,7 @@ previewers.git_branch_log = defaulter(function(opts)
           if current_remote <= table.getn(remotes) then
             local value = 'remotes/' .. remotes[current_remote] .. '/' .. entry.value
             current_remote = current_remote + 1
-            putils.job_maker(gen_cmd(value), bufnr, { callback = handle_results })
+            putils.job_maker(gen_cmd(value), bufnr, { cwd = opts.cwd, callback = handle_results })
           else
             vim.api.nvim_buf_set_lines(bufnr, 0, -1, false, { "No log found for branch: " .. entry.value })
           end
diff --git a/plugin/telescope.vim b/plugin/telescope.vim
index b3668d9..0fd0b60 100644
--- a/plugin/telescope.vim
+++ b/plugin/telescope.vim
@@ -27,6 +27,23 @@ highlight default link TelescopePromptPrefix Identifier
 highlight default link TelescopePreviewLine Visual
 highlight default link TelescopePreviewMatch Search
 
+highlight default link TelescopePreviewPipe Constant
+highlight default link TelescopePreviewCharDev Constant
+highlight default link TelescopePreviewDirectory Directory
+highlight default link TelescopePreviewBlock Constant
+highlight default link TelescopePreviewLink Special
+highlight default link TelescopePreviewSocket Statement
+highlight default link TelescopePreviewNormal Normal
+highlight default link TelescopePreviewRead Constant
+highlight default link TelescopePreviewWrite Statement
+highlight default link TelescopePreviewExecute String
+highlight default link TelescopePreviewHyphen NonText
+highlight default link TelescopePreviewSticky Keyword
+highlight default link TelescopePreviewSize String
+highlight default link TelescopePreviewUser Constant
+highlight default link TelescopePreviewGroup Constant
+highlight default link TelescopePreviewDate Directory
+
 " Used for Picker specific Results highlighting
 highlight default link TelescopeResultsClass Function
 highlight default link TelescopeResultsConstant Constant