1 files changed, 25 insertions, 10 deletions
diff --git a/scripts/lua/mason-scripts/utils.lua b/scripts/lua/mason-scripts/utils.lua
index 771d419..49a20df 100644
--- a/scripts/lua/mason-scripts/utils.lua
+++ b/scripts/lua/mason-scripts/utils.lua
@@ -1,21 +1,36 @@
 local fs = require "mason-core.fs"
+local _ = require "mason-core.functional"
 
 local M = {}
 
 ---@async
 ---@param path string
 ---@param contents string
----@param flags string
+---@param flags string?
 function M.write_file(path, contents, flags)
-    fs.async.write_file(
-        path,
-        table.concat({
-            "-- THIS FILE IS GENERATED. DO NOT EDIT MANUALLY.",
-            "-- stylua: ignore start",
-            contents,
-        }, "\n"),
-        flags
-    )
+    local header = _.cond({
+        {
+            _.matches "%.lua$",
+            _.always {
+                "-- THIS FILE IS GENERATED. DO NOT EDIT MANUALLY.",
+                "-- stylua: ignore start",
+            },
+        },
+        {
+            _.matches "%.md$",
+            _.always {
+                "<!--- THIS FILE IS GENERATED. DO NOT EDIT MANUALLY. -->",
+            },
+        },
+        {
+            _.T,
+            _.always {
+                "# THIS FILE IS GENERATED. DO NOT EDIT MANUALLY.",
+            },
+        },
+    }, path)
+
+    fs.async.write_file(path, table.concat(_.concat(header, { contents }), "\n"), flags)
 end
 
 return M