From cb5f939ab27b2c2ef2e1d4ac6fe16c5ba6332f39 Mon Sep 17 00:00:00 2001
From: Steven Arcangeli <stevearc@stevearc.com>
Date: Fri, 15 Sep 2023 17:43:14 -0700
Subject: feat: utility function to extend the built-in formatter args (#50)

---
 lua/conform/util.lua | 42 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)

(limited to 'lua/conform/util.lua')

diff --git a/lua/conform/util.lua b/lua/conform/util.lua
index ea13e2d..5813736 100644
--- a/lua/conform/util.lua
+++ b/lua/conform/util.lua
@@ -72,4 +72,46 @@ M.wrap_callback = function(cb, wrapper)
   end
 end
 
+---Helper function to add to the default args of a formatter.
+---@param args string|string[]|fun(ctx: conform.Context): string|string[]
+---@param extra_args string|string[]|fun(ctx: conform.Context): string|string[]
+---@param opts? { append?: boolean }
+---@example
+--- local util = require("conform.util")
+--- local prettier = require("conform.formatters.prettier")
+--- require("conform").formatters.prettier = vim.tbl_deep_extend("force", prettier, {
+---   args = util.extend_args(prettier.args, { "--tab", "--indent", "2" }),
+---   range_args = util.extend_args(prettier.range_args, { "--tab", "--indent", "2" }),
+--- })
+M.extend_args = function(args, extra_args, opts)
+  opts = opts or {}
+  return function(ctx)
+    if type(args) == "function" then
+      args = args(ctx)
+    end
+    if type(extra_args) == "function" then
+      extra_args = extra_args(ctx)
+    end
+    if type(args) == "string" then
+      if type(extra_args) ~= "string" then
+        extra_args = table.concat(extra_args, " ")
+      end
+      if opts.append then
+        return args .. " " .. extra_args
+      else
+        return extra_args .. " " .. args
+      end
+    else
+      if type(extra_args) == "string" then
+        error("extra_args must be a table when args is a table")
+      end
+      if opts.append then
+        return vim.tbl_flatten({ args, extra_args })
+      else
+        return vim.tbl_flatten({ extra_args, args })
+      end
+    end
+  end
+end
+
 return M
-- 
cgit v1.2.3-70-g09d2