From 909150606eab44979eb9595145796f5bcb067955 Mon Sep 17 00:00:00 2001
From: tobdob <tobias.dubois@gmail.com>
Date: Tue, 18 Oct 2022 10:56:36 +0200
Subject: [PATCH] add boolean config options for common patterns (#14)

---
 README.md            | 16 ++++++----------
 lua/trim/config.lua  |  8 ++++----
 lua/trim/init.lua    | 13 +++++++++++++
 lua/trim/trimmer.lua |  2 +-
 4 files changed, 24 insertions(+), 15 deletions(-)

diff --git a/README.md b/README.md
index d77a1f4..801adad 100644
--- a/README.md
+++ b/README.md
@@ -21,12 +21,10 @@ Plug 'cappyzawa/trim.nvim'
 -- default config
 local config = {
   disable = {},
-  patterns = {
-    [[%s/\s\+$//e]],           -- remove unwanted spaces
-    [[%s/\($\n\s*\)\+\%$//]],  -- trim last line
-    [[%s/\%^\n\+//]],          -- trim first line
-    [[%s/\(\n\n\)\n\+/\1/]],   -- replace multiple blank lines with a single line
-  },
+  patterns = {},
+  trim_trailing = true,
+  trim_last_line = true,
+  trim_first_line = true,
 }
 ```
 
@@ -37,11 +35,9 @@ lua <<EOF
     -- you can specify filetypes.
     disable = {"markdown"},
 
-    -- if you want to ignore space of top
+    -- if you want to remove multiple blank lines
     patterns = {
-      [[%s/\s\+$//e]],
-      [[%s/\($\n\s*\)\+\%$//]],
-      [[%s/\(\n\n\)\n\+/\1/]],
+      [[%s/\(\n\n\)\n\+/\1/]],   -- replace multiple blank lines with a single line
     },
   })
 EOF
diff --git a/lua/trim/config.lua b/lua/trim/config.lua
index f7c8732..52417ca 100644
--- a/lua/trim/config.lua
+++ b/lua/trim/config.lua
@@ -1,9 +1,9 @@
 local config = {
   disable = {},
-  patterns = {
-    [[%s/\s\+$//e]], [[%s/\($\n\s*\)\+\%$//]], [[%s/\%^\n\+//]],
-    [[%s/\(\n\n\)\n\+/\1/]]
-  }
+  patterns = {},
+  trim_trailing = true,
+  trim_last_line = true,
+  trim_first_line = true,
 }
 
 return config
diff --git a/lua/trim/init.lua b/lua/trim/init.lua
index 4376398..1417684 100644
--- a/lua/trim/init.lua
+++ b/lua/trim/init.lua
@@ -17,6 +17,19 @@ M.setup = function(cfg)
   cfg = cfg or {}
   if not cfg.disable then cfg.disable = config.disable end
   if not cfg.patterns then cfg.patterns = config.patterns end
+  if cfg.trim_trailing == nil then cfg.trim_trailing = config.trim_trailing end
+  if cfg.trim_first_line == nil then cfg.trim_first_line = config.trim_first_line end
+  if cfg.trim_last_line == nil then cfg.trim_last_line = config.trim_last_line end
+
+  if cfg.trim_first_line then
+    table.insert(cfg.patterns, [[%s/\%^\n\+//]])
+  end
+  if cfg.trim_last_line then
+    table.insert(cfg.patterns, [[%s/\($\n\s*\)\+\%$//]])
+  end
+  if cfg.trim_trailing then
+    table.insert(cfg.patterns, 1, [[%s/\s\+$//e]])
+  end
 
   if not vim.api.nvim_create_autocmd then
     vim.notify_once('trim.nvim requires nvim 0.7.0+.', vim.log.levels.ERROR)
diff --git a/lua/trim/trimmer.lua b/lua/trim/trimmer.lua
index 9a03f7f..9a10224 100644
--- a/lua/trim/trimmer.lua
+++ b/lua/trim/trimmer.lua
@@ -5,7 +5,7 @@ local trimmer = {}
 
 trimmer.trim = function(patterns)
 	local save = vim.fn.winsaveview()
-	for _, v in pairs(patterns) do
+	for _, v in ipairs(patterns) do
 		api.nvim_exec(string.format("keepjumps keeppatterns silent! %s", v), false)
 	end
 	vim.fn.winrestview(save)