install.bash

#!/usr/bin/env bash

DOTFILES_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"

main() {
    set -eu
    install_basics
    install_homebrew
    install_homebrew_packages
    trust_github
    setup_bashrc
    setup_gitconfig
    setup_tmux
    unminimize_ubuntu
    mise_install
    stow_dotfiles
    install_neovim_plugins
    install_go_packages
    install_npm_packages
    install_pi_packages
    setup_mcporter
    symlink_go
    exec bash
}

install_basics() {
    local commands=(curl git make ping clang)
    if is_macos; then
        for command in "${commands[@]}"; do
            if ! command_exists "$command"; then
                echo "command $command not found, please install it manually"
                return 1
            fi
        done
    elif ! command_exists "${commands[@]}"; then
        echo "-> install basics"
        sudo apt-get -y update
        sudo apt-get -y install \
            build-essential \
            clang\
            curl \
            file \
            git \
            iproute2 \
            iputils-ping \
            man-db
    fi
}

install_homebrew() {
    if is_macos; then
        export HOMEBREW_PREFIX="/opt/homebrew"
    else
        export HOMEBREW_PREFIX="/home/linuxbrew/.linuxbrew"
    fi

    if [ ! -x "$HOMEBREW_PREFIX/bin/brew" ]; then
        echo "-> install homebrew"
        NONINTERACTIVE=1 /bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)"
    fi
    eval "$($HOMEBREW_PREFIX/bin/brew shellenv)"
}

install_homebrew_packages() {
    echo "-> install homebrew packages"
    local brew_packages=(
	    protobuf
        atlassian-labs/acli/acli
        bash-completion@2
        bat
        black
        btop
        claude-code
        cloc
        codex
        curl
        direnv
        fd
        fzf
        gawk
        gdb
        gh
        git
        git-delta
        glow
        golangci-lint
        grpcurl
        htop
        hyperfine
        imagemagick
        isort
        jj
        jjui
        jq
        kubectx
        less
        llm
        magic-wormhole
        mise
        neovim
        nono
        pyright
        ripgrep
        rsync
        socat
        stow
        tmux
        trafilatura
        tree
        tree-sitter-cli
        unzip
        uv
        vhs
        watchman
        yq
        zoxide
    )
    brew install -q "${brew_packages[@]}"
    if is_macos; then
        brew install -q pngpaste
        brew install -q --cask keycastr snowflake-snowsql
    else
        brew install -q bubblewrap
    fi
    if is_datadog; then
        brew install -q datadog/tap/dd-auth datadog/tap/ddoc
        if is_macos; then
            brew install -q --cask datadog/tap/datadog-workspaces
        fi
    fi
}

trust_github() {
    echo "-> trust github"
    mkdir -pm700 ~/.ssh
    ssh-keyscan -t rsa,ecdsa,ed25519 github.com >> ~/.ssh/known_hosts
}

is_macos() {
    if [[ "$OSTYPE" =~ ^darwin ]]; then
        return 0
    else
        return 1
    fi
}

is_datadog() {
    [ -e "$HOME/dd" ]
}

quiet_run() {
    local output
    if ! output=$("$@" 2>&1); then
        echo "$output"
        return 1
    fi
}

command_exists() {
    local commands=("$@")
    local missing=0
    for command in "${commands[@]}"; do
        if ! command -v "$command" > /dev/null 2>&1; then
            missing=1
            break
        fi
    done
    return $missing
}

setup_bashrc() {
    if ! grep "$DOTFILES_DIR" "$HOME/.bashrc" > /dev/null; then
        echo "-> setup bashrc"
        echo "export DOTFILES_DIR=\"$DOTFILES_DIR\"" >> "$HOME/.bashrc"
        echo ". $DOTFILES_DIR/bash/.bashrc" >> "$HOME/.bashrc"
    fi
}

setup_gitconfig() {
    # Ensure .gitconfig exists to avoid grep warnings
    touch "$HOME/.gitconfig"

    if ! grep "$DOTFILES_DIR" "$HOME/.gitconfig" > /dev/null; then
        echo "-> add .gitconfig include to $HOME/.gitconfig"
        echo "[include]" >> "$HOME/.gitconfig"
        echo "  path = \"$DOTFILES_DIR/git/.gitconfig\"" >> "$HOME/.gitconfig"
    fi
    if ! grep -E 'includeIf.+DataDog' "$HOME/.gitconfig" > /dev/null; then
        echo "-> add .gitconfig.datadog include to $HOME/.gitconfig for DataDog"
        echo "[includeIf \"gitdir:~/go/src/github.com/DataDog/**\"]" >> "$HOME/.gitconfig"
        echo "  path = \"$DOTFILES_DIR/git/.gitconfig.datadog\"" >> "$HOME/.gitconfig"
    fi
    if ! grep -E 'includeIf.+open-telemetry' "$HOME/.gitconfig" > /dev/null; then
        echo "-> add .gitconfig.open-telemetry include to $HOME/.gitconfig for OpenTelemetry"
        echo "[includeIf \"gitdir:~/go/src/github.com/open-telemetry/**\"]" >> "$HOME/.gitconfig"
        echo "  path = \"$DOTFILES_DIR/git/.gitconfig.datadog\"" >> "$HOME/.gitconfig"
    fi
    if [[ "${USER:-}" == "bits" ]] && ! grep '# workspace config' "$HOME/.gitconfig" > /dev/null; then
        echo "-> add workspace config to $HOME/.gitconfig"
        echo "# workspace config" >> "$HOME/.gitconfig"
        echo "[delta]" >> "$HOME/.gitconfig"
        echo "  hyperlinks-file-link-format = "cursor://vscode-remote/ssh-remote+workspace-${HOSTNAME}/{path}:{line}"
" >> "$HOME/.gitconfig"
    fi
}

setup_tmux() {
    # Ensure .tmux.conf exists to avoid grep warnings
    touch "$HOME/.tmux.conf"

    if ! grep "$DOTFILES_DIR" "$HOME/.tmux.conf" > /dev/null; then
        echo "-> setup tmux config"
        echo "source-file \"$DOTFILES_DIR/tmux/.tmux.conf\"" >> "$HOME/.tmux.conf"
    fi
}

unminimize_ubuntu() {
    if grep "minimized" "$(which man)" > /dev/null; then
        echo "-> unminimizing ubuntu install"
        yes | DEBIAN_FRONTEND=noninteractive unminimize
    fi
}

stow_dotfiles() {
    echo "-> stow dotfiles"
    mkdir -p "$HOME/.claude" "$HOME/.pi/agent" "$HOME/.mcporter/daemon"
    stow_file_adopt git/.gitignore
    stow --adopt -t "$HOME" jj jjui neovim mise claude codex idea kitty pi
}

stow_file_adopt() {
    local rel_path="$1"
    local src="$DOTFILES_DIR/$rel_path"
    local dst="$HOME/${rel_path#*/}"

    if [ -L "$dst" ]; then
        return
    fi

    mkdir -p "$(dirname "$dst")"
    if [ -e "$dst" ]; then
        cp "$dst" "$src"
    fi
    ln -sf "$src" "$dst"
}

install_neovim_plugins() {
    echo "-> install neovim plugins"
    quiet_run nvim --headless  "+Lazy! restore" "+Lazy! clean" "+Lazy! install" +qa
    quiet_run nvim --headless "+MasonToolsUpdateSync" +qa
}

mise_install() {
    echo "-> install mise tools"
    # Avoid GitHub API rate limits for mise's aqua backend
    if [ -z "${GITHUB_TOKEN:-}" ] && command -v gh &>/dev/null; then
        export GITHUB_TOKEN=$(gh auth token 2>/dev/null)
    fi
    mise trust -q
    mise install -q
    mise upgrade -q
    eval "$(mise activate bash)"
}

install_go_packages() {
    echo "-> install go packages"
    export GOPROXY="https://proxy.golang.org,direct"
    local go_packages=(
        github.com/bokwoon95/wgo@latest
        github.com/go-delve/delve/cmd/dlv@latest
        github.com/maruel/panicparse/v2/cmd/pp@latest
        golang.org/x/perf/cmd/benchstat@latest
        golang.org/x/pkgsite/cmd/pkgsite@latest
        golang.org/x/tools/cmd/stress@latest
        honnef.co/go/tools/cmd/staticcheck@latest
    )
    for package in "${go_packages[@]}"; do
        go install "$package" &
    done
    wait
}

install_npm_packages() {
    echo "-> install npm packages"
    quiet_run npm install -g markserv @mariozechner/pi-coding-agent mcporter @mermaid-js/mermaid-cli
}

install_pi_packages() {
    echo "-> install pi packages"
    pi update
}

setup_mcporter() {
    echo "-> setup mcporter config"
    mkdir -p "$HOME/.mcporter"

    # Remove legacy claude mcp servers (migrated to mcporter)
    for name in datadog datadog-staging chrome-devtools snowflake; do
        claude mcp remove "$name" &>/dev/null || true
    done

    local snowflake_block=""
    if [ -n "${SNOWFLAKE_ACCOUNT:-}" ] && [ -f "$HOME/.config/mcp/snowflake-config.yaml" ]; then
        local user_email
        user_email="$(whoami)@datadoghq.com"
        snowflake_block=$(cat <<SNOWFLAKE
,
    "snowflake": {
      "command": "uvx",
      "args": [
        "--python", "3.12",
        "--python-preference=managed",
        "--from", "git+https://github.com/Snowflake-Labs/mcp",
        "mcp-server-snowflake",
        "--service-config-file", "~/.config/mcp/snowflake-config.yaml",
        "--authenticator", "externalbrowser",
        "--transport", "stdio",
        "--user", "${user_email}"
      ],
      "env": {
        "SNOWFLAKE_ACCOUNT": "${SNOWFLAKE_ACCOUNT}",
        "SNOWFLAKE_USER": "${user_email}",
        "SNOWFLAKE_DATABASE": "REPORTING"
      }
    }
SNOWFLAKE
)
    else
        echo "   skipping snowflake mcp (see Confluence: 'Snowflake for MCP')"
    fi

    cat > "$HOME/.mcporter/mcporter.json" <<EOF
{
  "mcpServers": {
    "atlassian": {
      "baseUrl": "https://mcp.atlassian.com/v1/mcp",
      "auth": "oauth"
    },
    "datadog": {
      "baseUrl": "https://mcp.datadoghq.com/api/unstable/mcp-server/mcp?toolsets=core,profiling",
      "auth": "oauth"
    },
    "datadog-staging": {
      "baseUrl": "https://mcp.datad0g.com/api/unstable/mcp-server/mcp?toolsets=core,profiling",
      "auth": "oauth"
    },
    "chrome-devtools": {
      "command": "npx",
      "args": ["-y", "chrome-devtools-mcp@latest"]
    },
    "google-workspace": {
      "command": "npx",
      "args": ["-y", "mcp-remote", "https://google-workspace-mcp-server-834963730936.us-central1.run.app/mcp"]
    }${snowflake_block}
  },
  "imports": []
}
EOF
}

fetch_repositories() {
    echo "-> fetch jj repositories in ~/go/src"

    local repos=()
    while IFS= read -r repo; do
        repos+=("$repo")
    done < <(find "$HOME/go/src" -mindepth 4 -maxdepth 4 -type d -name .jj 2>/dev/null | sed 's|/\.jj$||')

    if [ ${#repos[@]} -eq 0 ]; then
        return
    fi

    # Parallel but bounded. GIT_SSH_COMMAND disables SSH ControlMaster to avoid
    # `mux_client_request_session: Session open refused by peer` races when many
    # fetches hit github.com at once; scoped to xargs, so no restore needed.
    printf '%s\n' "${repos[@]}" \
        | GIT_SSH_COMMAND='ssh -o ControlMaster=no -o ControlPath=none' \
          xargs -n1 -P 16 -I {} bash -c '
            repo="$1"
            cd "$repo" || exit 0
            # Skip repos with no git remotes.
            if [ -z "$(jj git remote list 2>/dev/null)" ]; then
                exit 0
            fi
            # --ignore-working-copy bypasses the snapshot, so large files in
            # the working copy do not abort the fetch.
            if output=$(jj --ignore-working-copy git fetch 2>&1); then
                :
            else
                echo "   failed: $repo"
                echo "$output" | sed "s/^/      /"
            fi
        ' _ {}
}

symlink_go() {
    # Hack: Make sure the VS Code Go extension can always find the go binary
    echo "-> symlink go to /usr/local/bin"
    local target="/usr/local/bin/go"
    local go_bin
    go_bin="$(mise which go)"
    if [ "$(readlink "$target" 2>/dev/null)" = "$go_bin" ]; then
        return
    fi
    sudo ln -sf "$go_bin" "$target"
}

# Only run main when executed directly, not when sourced.
if [[ "${BASH_SOURCE[0]}" == "$0" ]]; then
    main "$@"
fi