1
#!/usr/bin/env bash
2

3
# Extract VS Code version from code-server
4
get_vscode_version() {
5
    code-server --version | head -n1
6
}
7

8
# Get user-data-dir from running code-server process
9
get_user_data_dir() {
10
    # Use ps with POSIX-compliant options
11
    local process_info
12
    if command -v ps >/dev/null 2>&1; then
13
        # Try BSD-style first (macOS), fallback to POSIX
14
        process_info=$(ps aux 2>/dev/null | grep -v grep | grep "code-server" | head -n 1) ||
15
        process_info=$(ps -ef 2>/dev/null | grep -v grep | grep "code-server" | head -n 1)
16
    fi
17

18
    if [ -n "$process_info" ]; then
19
        echo "$process_info" | grep -o -- '--user-data-dir=[^ ]*' | sed 's/--user-data-dir=//'
20
    fi
21
}
22

23
# Find compatible extension version
24
find_compatible_version() {
25
    local extension_id="$1"
26
    local vscode_version="$2"
27

28
    local response
29
    response=$(curl -s -X POST "https://marketplace.visualstudio.com/_apis/public/gallery/extensionquery" \
30
        -H "Content-Type: application/json" \
31
        -H "Accept: application/json;api-version=3.0-preview.1" \
32
        -d "{
33
            \"filters\": [{
34
                \"criteria\": [
35
                    {\"filterType\": 7, \"value\": \"$extension_id\"},
36
                    {\"filterType\": 12, \"value\": \"4096\"}
37
                ],
38
                \"pageSize\": 50
39
            }],
40
            \"flags\": 4112
41
        }")
42

43
    echo "$response" | jq -r --arg vscode_version "$vscode_version" '
44
        .results[0].extensions[0].versions[] |
45
        select(.version | test("^[0-9]+\\.[0-9]+\\.[0-9]*$")) |
46
        select(.version | length < 8) |
47
        {
48
            version: .version,
49
            engine: (.properties[] | select(.key == "Microsoft.VisualStudio.Code.Engine") | .value)
50
        } |
51
        select(.engine | ltrimstr("^") | split(".") |
52
            map(split("-")[0] | tonumber?) as $engine_parts |
53
            ($vscode_version | split(".") | map(tonumber)) as $vscode_parts |
54
            (
55
                ($engine_parts[0] // 0) < $vscode_parts[0] or
56
                (($engine_parts[0] // 0) == $vscode_parts[0] and ($engine_parts[1] // 0) < $vscode_parts[1]) or
57
                (($engine_parts[0] // 0) == $vscode_parts[0] and ($engine_parts[1] // 0) == $vscode_parts[1] and ($engine_parts[2] // 0) <= $vscode_parts[2])
58
            )
59
        ) |
60
        .version' | head -n 1
61
}
62

63
# Install extension
64
install_extension() {
65
    local extension_id="$1"
66
    local version="$2"
67
    local user_data_dir="$3"
68
    local extension_name
69
    extension_name=$(echo "$extension_id" | cut -d'.' -f2)
70
    local temp_dir="/tmp/code-extensions"
71

72
    echo "Installing $extension_id v$version..."
73

74
    # Create temp directory
75
    mkdir -p "$temp_dir"
76

77
    # Download
78
    echo "  Downloading..."
79
    # Use curl with portable options (--progress-bar not available everywhere)
80
    curl -L "https://marketplace.visualstudio.com/_apis/public/gallery/publishers/GitHub/vsextensions/$extension_name/$version/vspackage" \
81
        -o "$temp_dir/$extension_name.vsix.gz"
82

83
    if [ ! -f "$temp_dir/$extension_name.vsix.gz" ]; then
84
        echo "  ✗ Download failed for $extension_id"
85
        return 1
86
    fi
87

88
    # Decompress (handle both gunzip and gzip -d)
89
    if command -v gunzip >/dev/null 2>&1; then
90
        gunzip -f "$temp_dir/$extension_name.vsix.gz"
91
    else
92
        gzip -df "$temp_dir/$extension_name.vsix.gz"
93
    fi
94

95
    # Install with user-data-dir if provided
96
    if [ -n "$user_data_dir" ]; then
97
        code-server --user-data-dir="$user_data_dir" --force --install-extension "$temp_dir/$extension_name.vsix"
98
    else
99
        code-server --force --install-extension "$temp_dir/$extension_name.vsix"
100
    fi
101

102
    # Clean up
103
    rm -f "$temp_dir/$extension_name.vsix"
104

105
    echo "  ✓ $extension_id installed successfully!"
106
    return 0
107
}
108

109
# Check for required dependencies
110
check_dependencies() {
111
    local missing_deps=()
112

113
    # Check for required commands
114
    for cmd in curl jq code-server; do
115
        if ! command -v "$cmd" >/dev/null 2>&1; then
116
            missing_deps+=("$cmd")
117
        fi
118
    done
119

120
    # Check for either gunzip or gzip
121
    if ! command -v gunzip >/dev/null 2>&1 && ! command -v gzip >/dev/null 2>&1; then
122
        missing_deps+=("gunzip/gzip")
123
    fi
124

125
    if [ "${#missing_deps[@]}" -gt 0 ]; then
126
        echo "Error: Missing required dependencies: ${missing_deps[*]}"
127
        echo "Please install the missing dependencies and try again."
128
        exit 1
129
    fi
130
}
131

132
# Main script
133
echo "GitHub Copilot Extensions Installer"
134
echo "===================================="
135
echo ""
136

137
# Check dependencies
138
check_dependencies
139

140
# Get VS Code version
141
VSCODE_VERSION="$(get_vscode_version)"
142

143
if [ -z "$VSCODE_VERSION" ]; then
144
    echo "Error: Could not extract VS Code version from code-server"
145
    exit 1
146
fi
147

148
echo "Detected VS Code version: $VSCODE_VERSION"
149

150
# Check for user-data-dir in running code-server
151
USER_DATA_DIR="$(get_user_data_dir)"
152
if [ -n "$USER_DATA_DIR" ]; then
153
    echo "Detected user-data-dir: $USER_DATA_DIR"
154
fi
155
echo ""
156

157
# Extensions to install
158
# Use portable array declaration
159
EXTENSIONS="GitHub.copilot GitHub.copilot-chat"
160
FAILED=0
161

162
# Iterate through space-separated list for portability
163
for ext in $EXTENSIONS; do
164
    echo "Processing $ext..."
165

166
    # Find compatible version
167
    version="$(find_compatible_version "$ext" "$VSCODE_VERSION")"
168

169
    if [ -z "$version" ]; then
170
        echo "  ✗ No compatible version found for $ext"
171
        FAILED="$((FAILED + 1))"
172
    else
173
        echo "  Found compatible version: $version"
174
        if ! install_extension "$ext" "$version" "$USER_DATA_DIR"; then
175
            FAILED="$((FAILED + 1))"
176
        fi
177
    fi
178
    echo ""
179
done
180

181
# Summary
182
echo "===================================="
183
if [ $FAILED -eq 0 ]; then
184
    echo "✓ All extensions installed successfully!"
185
    # Clean up temp directory on success
186
    rm -rf /tmp/code-extensions
187
else
188
    echo "⚠ Completed with $FAILED error(s)"
189
    exit 1
190
fi