From 222a35646d3cecac905c248ff7786b3c4dff764e Mon Sep 17 00:00:00 2001
From: Tim Keller <tjkeller.xyz>
Date: Wed, 27 Nov 2024 23:30:04 -0600
Subject: modulenames

---
 lua/minitab/init.lua | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

(limited to 'lua/minitab')

diff --git a/lua/minitab/init.lua b/lua/minitab/init.lua
index 0504d42..1ef909c 100644
--- a/lua/minitab/init.lua
+++ b/lua/minitab/init.lua
@@ -1,14 +1,15 @@
-local M = {}
+local M = {
+	moduleNames = { 'init.lua', '__init__.py', }
+}
 
 M.fileTitle = function(file)
 	file = vim.fn.fnamemodify(file, ':p:~')
 	local tail = vim.fn.fnamemodify(file, ':t')
 
-	if not tail then
-		return vim.fn.fnamemodify(file, ':p:~:h') .. '/' -- e.g. ~/dir/
-	elseif tail == 'init.lua'
-	    or tail == '__init__.py' then
-		return vim.fn.fnamemodify(vim.fn.fnamemodify(file, ':h'), ':t') .. '/' .. tail -- e.g. luatab/init.lua
+	if tail == '' then
+		return vim.fn.fnamemodify(file, ':p:~') -- netrw; e.g. ~/dir/
+	elseif vim.tbl_contains(M.moduleNames, tail) then
+		return vim.fn.fnamemodify(vim.fn.fnamemodify(file, ':h'), ':t') .. '/' .. tail -- e.g. minitab/init.lua
 	else
 		return tail
 	end
-- 
cgit v1.2.3