feat: make model selection cofigurable

Author: TheLazyLemur

lua/claudecode/config.lua

@@ -15,6 +15,12 @@ M.defaults = {
     vertical_split = true,
     open_in_current_tab = true, -- Use current tab instead of creating new tab
   },
+  models = {
+    { name = "Claude Opus 4 (Latest)", value = "claude-opus-4-20250514" },
+    { name = "Claude Sonnet 4 (Latest)", value = "claude-sonnet-4-20250514" },
+    { name = "Claude 3.7 Sonnet (Latest)", value = "claude-3-7-sonnet-latest" },
+    { name = "Claude 3.5 Haiku (Latest)", value = "claude-3-5-haiku-latest" },
+  },
 }
 
 --- Validates the provided configuration table.
@@ -61,6 +67,16 @@ function M.validate(config)
   assert(type(config.diff_opts.vertical_split) == "boolean", "diff_opts.vertical_split must be a boolean")
   assert(type(config.diff_opts.open_in_current_tab) == "boolean", "diff_opts.open_in_current_tab must be a boolean")
 
+  -- Validate models
+  assert(type(config.models) == "table", "models must be a table")
+  assert(#config.models > 0, "models must not be empty")
+  
+  for i, model in ipairs(config.models) do
+    assert(type(model) == "table", "models[" .. i .. "] must be a table")
+    assert(type(model.name) == "string" and model.name ~= "", "models[" .. i .. "].name must be a non-empty string")
+    assert(type(model.value) == "string" and model.value ~= "", "models[" .. i .. "].value must be a non-empty string")
+  end
+
   return true
 end
 

lua/claudecode/init.lua

@@ -332,12 +332,7 @@ function M._create_commands()
 end
 
 M.open_with_model = function()
-  local models = {
-    { name = "Claude Opus 4 (Latest)", value = "claude-opus-4-20250514" },
-    { name = "Claude Sonnet 4 (Latest)", value = "claude-sonnet-4-20250514" },
-    { name = "Claude 3.7 Sonnet (Latest)", value = "claude-3-7-sonnet-latest" },
-    { name = "Claude 3.5 Haiku (Latest)", value = "claude-3-5-haiku-latest" },
-  }
+  local models = M.state.config.models
 
   vim.ui.select(models, {
     prompt = "Select Claude model:",

tests/unit/config_spec.lua

@@ -22,6 +22,7 @@ describe("Configuration", function()
     expect(config.defaults).to_have_key("auto_start")
     expect(config.defaults).to_have_key("log_level")
     expect(config.defaults).to_have_key("track_selection")
+    expect(config.defaults).to_have_key("models")
   end)
 
   it("should validate valid configuration", function()
@@ -38,6 +39,9 @@ describe("Configuration", function()
         vertical_split = true,
         open_in_current_tab = true,
       },
+      models = {
+        { name = "Test Model", value = "test-model" },
+      },
     }
 
     local success = config.validate(valid_config)
@@ -74,6 +78,54 @@ describe("Configuration", function()
     expect(success).to_be_false()
   end)
 
+  it("should reject invalid models configuration", function()
+    local invalid_config = {
+      port_range = { min = 10000, max = 65535 },
+      auto_start = true,
+      log_level = "debug",
+      track_selection = false,
+      visual_demotion_delay_ms = 50,
+      diff_opts = {
+        auto_close_on_accept = true,
+        show_diff_stats = true,
+        vertical_split = true,
+        open_in_current_tab = true,
+      },
+      models = {}, -- Empty models array should be rejected
+    }
+
+    local success, _ = pcall(function()
+      config.validate(invalid_config)
+    end)
+
+    expect(success).to_be_false()
+  end)
+
+  it("should reject models with invalid structure", function()
+    local invalid_config = {
+      port_range = { min = 10000, max = 65535 },
+      auto_start = true,
+      log_level = "debug",
+      track_selection = false,
+      visual_demotion_delay_ms = 50,
+      diff_opts = {
+        auto_close_on_accept = true,
+        show_diff_stats = true,
+        vertical_split = true,
+        open_in_current_tab = true,
+      },
+      models = {
+        { name = "Test Model" }, -- Missing value field
+      },
+    }
+
+    local success, _ = pcall(function()
+      config.validate(invalid_config)
+    end)
+
+    expect(success).to_be_false()
+  end)
+
   it("should merge user config with defaults", function()
     local user_config = {
       auto_start = true,
@@ -86,6 +138,7 @@ describe("Configuration", function()
     expect(merged_config.log_level).to_be("debug")
     expect(merged_config.port_range.min).to_be(config.defaults.port_range.min)
     expect(merged_config.track_selection).to_be(config.defaults.track_selection)
+    expect(merged_config.models).to_be_table()
   end)
 
   teardown()