feat: format_on_save functions can return a callback as the second value

1 files changed, 14 insertions, 8 deletions

diff --git a/lua/conform/init.lua b/lua/conform/init.lua
index e2ba40a..ff30686 100644
--- a/lua/conform/init.lua
+++ b/lua/conform/init.lua
@@ -84,15 +84,18 @@ M.setup = function(opts)
       pattern = "*",
       group = aug,
       callback = function(args)
-        local format_args = opts.format_on_save
+        local format_args, callback = opts.format_on_save, nil
         if type(format_args) == "function" then
-          format_args = format_args(args.buf)
+          format_args, callback = format_args(args.buf)
         end
         if format_args then
-          M.format(vim.tbl_deep_extend("force", format_args, {
-            buf = args.buf,
-            async = false,
-          }))
+          M.format(
+            vim.tbl_deep_extend("force", format_args, {
+              buf = args.buf,
+              async = false,
+            }),
+            callback
+          )
         end
       end,
     })
@@ -109,9 +112,9 @@ M.setup = function(opts)
         if vim.b[args.buf].conform_applying_formatting then
           return
         end
-        local format_args = opts.format_after_save
+        local format_args, callback = opts.format_after_save, nil
         if type(format_args) == "function" then
-          format_args = format_args(args.buf)
+          format_args, callback = format_args(args.buf)
         end
         if format_args then
           M.format(
@@ -127,6 +130,9 @@ M.setup = function(opts)
                   vim.b[args.buf].conform_applying_formatting = false
                 end)
               end
+              if callback then
+                callback(err)
+              end
             end
           )
         end