feat: add filename_generator key in config to allow user to customize filename

Author: A7Lavinraj

lua/assistant/config.lua

@@ -1,6 +1,7 @@
 ---@class Assistant.Config.Core
 ---@field process_budget integer
 ---@field port integer
+---@field filename_generator fun(str: string): string
 
 ---@class Assistant.Config.UI
 ---@field border string

lua/assistant/core/tcp.lua

@@ -139,8 +139,10 @@ local function start_server()
             table.insert(sources, key)
           end
 
+          local generator = config.values.core.filename_generator or utils.to_snake_case
+
           picker:pick(sources, { prompt = 'source' }, function(source)
-            local testcase_class_snake = utils.to_snake_case(data.languages.java.taskClass)
+            local testcase_class_snake = generator(data.languages.java.taskClass)
             local filepath = string.format('%s/.ast/%s.json', fs.find_root() or fs.make_root(), testcase_class_snake)
             fs.write(filepath, chunk)
 

lua/assistant/utils.lua

@@ -31,16 +31,17 @@ end
 ---@param str string
 function utils.to_snake_case(str)
   return str
-    :gsub('(%s)', '_')
-    :gsub('(%u%l)', '_%1')
-    :gsub('(%l)(%u)', '%1_%2')
-    :gsub('^_', '')
     :gsub('^%s*(.-)%s*$', '%1')
-    :gsub('%s+', '_')
+    :gsub('[^%w%s_]', '')
+    :gsub('(%u+)(%u%l)', '%1_%2')
     :gsub('(%l)(%u)', '%1_%2')
-    :gsub('(%l)(%d)', '%1_%2')
-    :gsub('(%d)(%l)', '%1_%2')
-    :gsub('(%d)(%u)', '%1_%2')
+    :gsub('(%d)(%a)', '%1_%2')
+    :gsub('(%a)(%d)', '%1_%2')
+    :gsub('%s+', '_')
+    :gsub('_+', '_')
+    :gsub('^_', '')
+    :gsub('_$', '')
+    :lower()
 end
 
 ---@param str string