#!/usr/bin/env bash

# !!! This script may overwrite your current config. Don't blindly execute !!!

# Script to automate Claude Code after installation
# install via `npm install -g @anthropic-ai/claude-code`

# Key features:
# - gets the onboarding wizard out of the way
# - configure API keys
# - trust current directory
# - configure MCP and permission
#  -set timeouts and opt out on things

##################################
# Clean past claude configuration
# Not enabled by default
##################################
# rm -rf ~/.claude
mkdir -p ~/.claude

##################################
# Setting up the API access :
# - claude code has two modes : either via Claude Max or using an API
# - claude max opens an oauth flow, which we can't use
# - we can just use the ANTHROPIC_API_KEY , but it will stil ask for approval
# - we found another way of providing the API by using a shell script that provides that key
# - this is useful for example if you want to fetch from somewhere else like 1password or so
# idea via https://www.reddit.com/r/ClaudeAI/comments/1jwvssa/claude_code_with_api_key/
##################################
# Setup API helper
##################################
echo 'echo ${ANTHROPIC_API_KEY}' > /home/node/.claude/anthropic_key_helper.sh
chmod +x /home/node/.claude/anthropic_key_helper.sh

##################################
# Configuring claude code
##################################
# - on the one hand claude code wants us to use claude config
# - but for example I was not able to configure the theme using it:
# the command executed with no errors, but kept asking for the theme
# also see https://github.com/anthropics/claude-code/issues/434
# and https://github.com/anthropics/claude-code/issues/441

# 
# - https://docs.anthropic.com/en/docs/claude-code/settings#settings-files
# - on the other hand the docs mentions it'd deprecating that command
# - we resort to creating a skeleton json file
#
# - shiftEnterKeyBindingInstalled configures the ask for terminal install
# - hasCompletedOnboarding indicates configuration is done
# - set the theme to dark here
#
# for the API key it's not enough to setup the apiKeyHelper
# - you have to mark it as approved
# - for refer to that key is uses the last 20 chars of the key it seems
# - here we assume you have ANTHROPIC_API_KEY configured as env var
##################################
ANTHROPIC_API_KEY_LAST_20_CHARS=${ANTHROPIC_API_KEY: -20}

# We write the global config to ~/.claude.json
# Warning this overwrites your existing
cat <<EOM > ~/.claude.json
{
    "customApiKeyResponses": {
        "approved": [ "$ANTHROPIC_API_KEY_LAST_20_CHARS"],
        "rejected": [  ]
    },
    "shiftEnterKeyBindingInstalled": true,
    "theme": "dark" ,
    "hasCompletedOnboarding": true
}
EOM

# to configure the API helper goes into the .claude/settings.json file
claude config set --global apiKeyHelper ~/.claude/anthropic_key_helper.sh

##################################
# Trust the current dir/project
# - when you enter a new directoy, claude asks it you trust it
# - we use the claude config to trust it
##################################
claude config set hasTrustDialogAccepted true
claude config set hasCompletedProjectOnboarding true

#################################
# Configuring MCP servers
# - it has an claude mcp subcommand
# - I prefer to add it through the json command at is has more options
# - also beware that the env vars you set are visible using the claude mcp 
# - ideally you limit the read access to the .claude.json or ~/.claude with umask or similar
#################################

# claude code install doesn't give much output during install
# therefore I pull the container typically seperate
docker pull mcp/puppeteer

# example that generates the config into the MCP_JSON var
read -r -d '' MCP_JSON <<'EOF'
{
      "command": "docker",
      "args": ["run", "-i", "--rm", "--init", "-e", "DOCKER_CONTAINER=true", "mcp/puppeteer"]
}
EOF

# then add it
claude mcp add-json puppeteer "$MCP_JSON"
# listing the server
claude mcp list

#################################
# Setting up permissions
# - we can config it using claude config add
# - though it says that's soon deprecated
# - we use add instead of set because it's an array
# - it writes the settings to ~/.claude/settings.json
# - mcp servers use the mcp__ prefix in the settings
# https://docs.anthropic.com/en/docs/claude-code/settings
#################################
claude config add allowedTools "Edit,Bash"

#################################
# IDE extension:
# - Claude code comes with a VSCODE/Cursor extension
# - this extension is not available through the marketplace
# - it is part of the npm package
# - Note: this only works in an IDE terminal , not in postcreate commands , I set it in my .bashrc
#################################
# set the IDE code for `vscode` , `cursor` for cursor
IDE_CMD=cursor

# Do this in a tempdir
tempdir=$(mktemp -d)
cd $tempdir

# downloads the package
npm pack @anthropic-ai/claude-code
tar -xzvf anthropic-ai-claude-code-*.tgz
# Install the extension
# requires a reload of the editor
$IDE_CMD --install-extension package/vendor/claude-code.vsix

#################################
# Interesting (undocumented settings)
#################################
# found via https://github.com/Helmi/claude-simone?tab=readme-ov-file#enabling-parallel-task-execution
claude config set --global  parallelTasksCount 3

#################################
# Handy env vars
#################################

# enable debug logging
# export ANTHROPIC_LOG=debug

# disable non-essential traffic and disable telemetry
# - Equivalent of setting DISABLE_AUTOUPDATER, DISABLE_BUG_COMMAND, DISABLE_ERROR_REPORTING, and DISABLE_TELEMETRY
# export CLAUDE_CODE_DISABLE_NONESSENTIAL_TRAFFIC=true

# define frequency of api helper refresh
# CLAUDE_CODE_API_KEY_HELPER_TTL_MS

# set/increase the bash timeout for longer commands
# BASH_DEFAULT_TIMEOUT_MS

# set the mcp timeout
# MCP_TIMEOUT