feat: support icons_by_file_prefix

Author: BBboy01

lua/nvim-web-devicons.lua

@@ -1,7 +1,7 @@
 local M = {}
 
 -- When adding new icons, remember to add an entry to the `filetypes` table, if applicable.
-local icons, icons_by_filename, icons_by_file_extension, icons_by_operating_system
+local icons, icons_by_filename, icons_by_file_prefix, icons_by_file_extension, icons_by_operating_system
 
 function M.get_icons()
   return icons
@@ -17,9 +17,10 @@ local function refresh_icons()
   end
 
   icons_by_filename = theme.icons_by_filename
+  icons_by_file_prefix = theme.icons_by_file_prefix
   icons_by_file_extension = theme.icons_by_file_extension
   icons_by_operating_system = theme.icons_by_operating_system
-  icons = vim.tbl_extend("keep", {}, icons_by_filename, icons_by_file_extension, icons_by_operating_system)
+  icons = vim.tbl_extend("keep", {}, icons_by_filename, icons_by_file_prefix, icons_by_file_extension, icons_by_operating_system)
 end
 
 -- Map of filetypes -> icon names
@@ -332,14 +333,18 @@ function M.setup(opts)
   end
 
   local user_filename_icons = user_icons.override_by_filename
+  local user_file_pre_icons = user_icons.override_by_prefix
   local user_file_ext_icons = user_icons.override_by_extension
 
   icons =
-    vim.tbl_extend("force", icons, user_icons.override or {}, user_filename_icons or {}, user_file_ext_icons or {})
+    vim.tbl_extend("force", icons, user_icons.override or {}, user_filename_icons or {}, user_file_pre_icons or {}, user_file_ext_icons or {})
 
   if user_filename_icons then
     icons_by_filename = vim.tbl_extend("force", icons_by_filename, user_filename_icons)
   end
+  if user_file_pre_icons then
+    icons_by_file_prefix = vim.tbl_extend("force", icons_by_file_prefix, user_file_pre_icons)
+  end
   if user_file_ext_icons then
     icons_by_file_extension = vim.tbl_extend("force", icons_by_file_extension, user_file_ext_icons)
   end
@@ -359,6 +364,24 @@ function M.get_default_icon()
   return default_icon
 end
 
+-- recursively iterate over each segment separated by '.' to parse prefix with multiple dots in filename
+local function iterate_multi_dotted_prefix(name, icon_table)
+  if name == nil then
+    return nil
+  end
+
+  local compound_pre, rest = name:match "(.?[^.]*)%.(.*)"
+  local icon = icon_table[compound_pre]
+  if icon then
+    return icon
+  end
+  if not rest then
+    return nil
+  end
+
+  return iterate_multi_dotted_prefix(rest, icon_table)
+end
+
 -- recursively iterate over each segment separated by '.' to parse extension with multiple dots in filename
 local function iterate_multi_dotted_extension(name, icon_table)
   if name == nil then
@@ -374,6 +397,17 @@ local function iterate_multi_dotted_extension(name, icon_table)
   return iterate_multi_dotted_extension(compound_ext, icon_table)
 end
 
+local function get_icon_by_prefix(name, prefix, opts)
+  local is_strict = if_nil(opts and opts.strict, global_opts.strict)
+  local icon_table = is_strict and icons_by_file_prefix or icons
+
+  if prefix ~= nil then
+    return icon_table[prefix]
+  end
+
+  return iterate_multi_dotted_prefix(name, icon_table)
+end
+
 local function get_icon_by_extension(name, ext, opts)
   local is_strict = if_nil(opts and opts.strict, global_opts.strict)
   local icon_table = is_strict and icons_by_file_extension or icons
@@ -385,7 +419,7 @@ local function get_icon_by_extension(name, ext, opts)
   return iterate_multi_dotted_extension(name, icon_table)
 end
 
-function M.get_icon(name, ext, opts)
+function M.get_icon(name, pre_or_ext, opts)
   if type(name) == "string" then
     name = name:lower()
   end
@@ -398,9 +432,9 @@ function M.get_icon(name, ext, opts)
   local is_strict = if_nil(opts and opts.strict, global_opts.strict)
   local icon_data
   if is_strict then
-    icon_data = icons_by_filename[name] or get_icon_by_extension(name, ext, opts) or (has_default and default_icon)
+    icon_data = icons_by_filename[name] or get_icon_by_prefix(name, pre_or_ext, opts) or get_icon_by_extension(name, pre_or_ext, opts) or (has_default and default_icon)
   else
-    icon_data = icons[name] or get_icon_by_extension(name, ext, opts) or (has_default and default_icon)
+    icon_data = icons[name] or get_icon_by_prefix(name, pre_or_ext, opts) or get_icon_by_extension(name, pre_or_ext, opts) or (has_default and default_icon)
   end
 
   if icon_data then
@@ -419,7 +453,7 @@ function M.get_icon_by_filetype(ft, opts)
   return M.get_icon(name or "", nil, opts)
 end
 
-function M.get_icon_colors(name, ext, opts)
+function M.get_icon_colors(name, pre_or_ext, opts)
   if not loaded then
     M.setup()
   end
@@ -428,9 +462,9 @@ function M.get_icon_colors(name, ext, opts)
   local is_strict = if_nil(opts and opts.strict, global_opts.strict)
   local icon_data
   if is_strict then
-    icon_data = icons_by_filename[name] or get_icon_by_extension(name, ext, opts) or (has_default and default_icon)
+    icon_data = icons_by_filename[name] or get_icon_by_prefix(name, pre_or_ext, opts) or get_icon_by_extension(name, pre_or_ext, opts) or (has_default and default_icon)
   else
-    icon_data = icons[name] or get_icon_by_extension(name, ext, opts) or (has_default and default_icon)
+    icon_data = icons[name] or get_icon_by_prefix(name, pre_or_ext, opts) or get_icon_by_extension(name, pre_or_ext, opts) or (has_default and default_icon)
   end
 
   if icon_data then
@@ -449,8 +483,8 @@ function M.get_icon_colors_by_filetype(ft, opts)
   return M.get_icon_colors(name or "", nil, opts)
 end
 
-function M.get_icon_color(name, ext, opts)
-  local data = { M.get_icon_colors(name, ext, opts) }
+function M.get_icon_color(name, pre_or_ext, opts)
+  local data = { M.get_icon_colors(name, pre_or_ext, opts) }
   return data[1], data[2]
 end
 
@@ -461,8 +495,8 @@ function M.get_icon_color_by_filetype(ft, opts)
   return M.get_icon_color(name or "", nil, opts)
 end
 
-function M.get_icon_cterm_color(name, ext, opts)
-  local data = { M.get_icon_colors(name, ext, opts) }
+function M.get_icon_cterm_color(name, pre_or_ext, opts)
+  local data = { M.get_icon_colors(name, pre_or_ext, opts) }
   return data[1], data[3]
 end
 

lua/nvim-web-devicons/icons-default.lua

@@ -227,12 +227,6 @@ local icons_by_filename = {
     cterm_color = "68",
     name = "Dockerfile",
   },
-  ["dockerfile"] = {
-    icon = "󰡨",
-    color = "#458ee6",
-    cterm_color = "68",
-    name = "Dockerfile",
-  },
   ["favicon.ico"] = {
     icon = "",
     color = "#cbcb41",
@@ -367,13 +361,10 @@ local icons_by_filename = {
   },
 }
 
+local icons_by_file_prefix = {
+}
+
 local icons_by_file_extension = {
-  ["Dockerfile"] = {
-    icon = "󰡨",
-    color = "#458ee6",
-    cterm_color = "68",
-    name = "Dockerfile",
-  },
   ["R"] = {
     icon = "󰟔",
     color = "#2266ba",
@@ -2139,6 +2130,7 @@ local icons_by_operating_system = {
 
 return {
   icons_by_filename = icons_by_filename,
+  icons_by_file_prefix = icons_by_file_prefix,
   icons_by_file_extension = icons_by_file_extension,
   icons_by_operating_system = icons_by_operating_system,
 }

lua/nvim-web-devicons/icons-light.lua

@@ -227,12 +227,6 @@ local icons_by_filename = {
     cterm_color = "25",
     name = "Dockerfile",
   },
-  ["dockerfile"] = {
-    icon = "󰡨",
-    color = "#2e5f99",
-    cterm_color = "25",
-    name = "Dockerfile",
-  },
   ["favicon.ico"] = {
     icon = "",
     color = "#666620",
@@ -367,13 +361,10 @@ local icons_by_filename = {
   },
 }
 
+local icons_by_file_prefix = {
+}
+
 local icons_by_file_extension = {
-  ["Dockerfile"] = {
-    icon = "󰡨",
-    color = "#2e5f99",
-    cterm_color = "25",
-    name = "Dockerfile",
-  },
   ["R"] = {
     icon = "󰟔",
     color = "#1a4c8c",
@@ -2139,6 +2130,7 @@ local icons_by_operating_system = {
 
 return {
   icons_by_filename = icons_by_filename,
+  icons_by_file_prefix = icons_by_file_prefix,
   icons_by_file_extension = icons_by_file_extension,
   icons_by_operating_system = icons_by_operating_system,
 }

scripts/generate_colors.lua

@@ -136,27 +136,35 @@ end
 local lines = generate_lines()
 
 -- second table
-if fn.search("^local icons_by_file_extension", "c") == 0 then
-  error_exit("Table 'icons_by_file_extension' not found in lua/nvim-web-devicons/icons-default.lua", 1)
+if fn.search("^local icons_by_file_prefix", "c") == 0 then
+  error_exit("Table 'icons_by_file_prefix' not found in lua/nvim-web-devicons/icons-default.lua", 1)
 end
 local lines2 = generate_lines()
 
 -- third table
+if fn.search("^local icons_by_file_extension", "c") == 0 then
+  error_exit("Table 'icons_by_file_extension' not found in lua/nvim-web-devicons/icons-default.lua", 1)
+end
+local lines3 = generate_lines()
+
+-- fourth table
 if fn.search("^local icons_by_operating_system", "c") == 0 then
   error_exit("Table 'icons_by_operating_system' not found in lua/nvim-web-devicons/icons-default.lua", 1)
 end
-local lines3 = generate_lines()
+local lines4 = generate_lines()
 
-table.insert(lines3, "return {")
-table.insert(lines3, "  icons_by_filename = icons_by_filename,")
-table.insert(lines3, "  icons_by_file_extension = icons_by_file_extension,")
-table.insert(lines3, "  icons_by_operating_system = icons_by_operating_system,")
-table.insert(lines3, "}")
+table.insert(lines4, "return {")
+table.insert(lines4, "  icons_by_filename = icons_by_filename,")
+table.insert(lines4, "  icons_by_file_prefix = icons_by_file_prefix,")
+table.insert(lines4, "  icons_by_file_extension = icons_by_file_extension,")
+table.insert(lines4, "  icons_by_operating_system = icons_by_operating_system,")
+table.insert(lines4, "}")
 
 -- write both tables to file
 fn.writefile(lines, "lua/nvim-web-devicons/icons-light.lua")
 fn.writefile(lines2, "lua/nvim-web-devicons/icons-light.lua", "a")
 fn.writefile(lines3, "lua/nvim-web-devicons/icons-light.lua", "a")
+fn.writefile(lines4, "lua/nvim-web-devicons/icons-light.lua", "a")
 
 print "Finished creating new file!"