-- Track which mode we're in (work or normal)
vim.g.claude_mode = "normal"

local function get_claude_command()
  if vim.g.claude_mode == "work" then
    return {
      bin = "sh",
      bin_args = {
        "-c",
        "CLAUDE_CONFIG_DIR=~/.claude-work claude"
      },
    }
  else
    return {
      bin = "claude",
    }
  end
end

require("claudecode").setup(get_claude_command())