starter/lua/mj/ruby_utils.lua

local M = {}

M.snake_case_to_ruby_class = function(path)
  -- Remove .rb suffix if present
  path = path:gsub("%.rb$", "")

  -- Split by slash and convert each part to CamelCase
  local parts = vim.split(path, "/")
  for i, part in ipairs(parts) do
    parts[i] = part
      :gsub("_(%a)", function(letter)
        return letter:upper()
      end)
      :gsub("^%a", string.upper)
  end

  -- Join parts with "::" to form the Ruby class name
  local ruby_class = table.concat(parts, "::")
  return ruby_class
end

M.replace_selected_text_with_ruby_class = function()
  -- Get the current visual selection
  local start_pos = vim.fn.getpos("'<")
  local end_pos = vim.fn.getpos("'>")

  -- Get the lines in the selected region
  local lines = vim.fn.getline(start_pos[2], end_pos[2])

  -- If there's more than one line, we join them into one string
  local selected_text = table.concat(lines, "\n")

  -- Get the exact selection from the start column to the end column
  selected_text = string.sub(selected_text, start_pos[3], end_pos[3])

  -- Convert the selected text to Ruby class format
  local ruby_class = M.snake_case_to_ruby_class(selected_text)

  -- Replace the selected text with the converted Ruby class
  vim.fn.setline(start_pos[2], ruby_class)
end

M.insert_ruby_class_based_on_file_name = function()
  -- Get the current file name (without path)
  local filename = vim.fn.expand("%:t")

  -- Remove the .rb suffix if present
  filename = filename:gsub("%.rb$", "")

  -- Convert the file name to Ruby class format
  local ruby_class = M.snake_case_to_ruby_class(filename)

  -- Insert the class name at the current cursor position
  vim.api.nvim_put({ ruby_class }, "c", true, true)
end

-- Keybindings (adjust as necessary)
vim.api.nvim_set_keymap(
  "v",
  "<leader>r",
  [[:lua require'module_name'.replace_selected_text_with_ruby_class()<CR>]],
  { noremap = true, silent = true }
)
vim.api.nvim_set_keymap(
  "n",
  "<leader>i",
  [[:lua require'module_name'.insert_ruby_class_based_on_file_name()<CR>]],
  { noremap = true, silent = true }
)

return M