idea.indexes.sh

#!/usr/bin/env zsh
set -euo pipefail

# https://www.jetbrains.com/help/idea/shared-indexes.html

# USAGE:
# idea.indexes.sh /path/to/project/dir

CDN_ROOT="$HOME/_idea_cdn"
mkdir -p "$CDN_ROOT"

TMP_OUTPUT_DIR="$(mktemp -d)"
function cleanup() {
  rm -rf "$TMP_OUTPUT_DIR"
}
trap cleanup SIGINT
#trap cleanup ERR

#IDEA_BIN="open -na 'Intellij IDEA' --args"
IDEA_BIN="/Applications/IntelliJ IDEA.app/Contents/MacOS/idea"

CDN_LAYOUT_TOOL="cdn-layout-tool"

PROJECT_DIR=${1:-""}

# If no directory provided, assume current dir is a project dir
if ! [ -d "$PROJECT_DIR" ]; then
    PROJECT_DIR="$(pwd -P)"
fi

# Fancy colors
blu="\e[1;95m" # Bold Magenta
bri="\e[97m"   # Bright
red="\e[1;91m" # Bold Red
clr="\e[0m"    # Reset
function printok() {
  printf "✅ ${blu}%s${clr}\n" "$*"
}
function printerr() {
  printf "❌ ${red}%s${clr}\n" "$*"
}
function printbright() {
  printf "${bri}%s${clr}\n" "$*"
}


# Guess project name from dir name
PROJECT_NAME=$(basename "$PROJECT_DIR")

# Guess project name from .idea/.name file
if [ -f "$PROJECT_DIR/.idea/.name" ]; then
    PROJECT_NAME=$(cat "$PROJECT_DIR/.idea/.name")
elif [ -d "$PROJECT_DIR/.idea" ]; then
    # Guess project name from *.iml file name
    IML_PATH=$(find "$PROJECT_DIR" "$PROJECT_DIR/.idea" -maxdepth 1 -type f -name "*.iml" | head -1)
    if [ -n "$IML_PATH" ]; then
        IML_FILENAME=$(basename "$IML_PATH")
        PROJECT_NAME="${IML_FILENAME%.iml}"
    fi
else 
    printerr "WARNNING: No .idea dir found! IDEA will treat $PROJECT_DIR as a directory outside the sources root"
fi


# Check dependencies
if [ -x "$IDEA_BIN" ]; then
    printok "IDEA binary found"
else
    printerr You need to install IDEA
    exit 1
fi

if hash yq 2>/dev/null; then
    printok "yq installed"
else
    printerr "You need to install yq: https://github.com/mikefarah/yq/#install"
    printf " - Homebrew: ${bri}%s${clr}\n" "brew install yq"
    printf " - Go: ${bri}%s${clr}\n" "go install github.com/mikefarah/yq/v4@latest"
    exit 2
fi

if hash "$CDN_LAYOUT_TOOL" 2>/dev/null; then
    printok "cdn-layout-tool found"
else
    echo "Downloading cdn-layout-tool..."
    # https://packages.jetbrains.team/maven/p/ij/intellij-shared-indexes-public/com/jetbrains/intellij/indexing/shared/cdn-layout-tool/
    CLT_VER=0.8.65
    curl -L -o "/tmp/cdn-layout-tool-$CLT_VER.zip" "https://packages.jetbrains.team/maven/p/ij/intellij-shared-indexes-public/com/jetbrains/intellij/indexing/shared/cdn-layout-tool/$CLT_VER/cdn-layout-tool-$CLT_VER.zip"

    unzip -u "/tmp/cdn-layout-tool-$CLT_VER.zip" -d "/tmp"
    rm -f "/tmp/cdn-layout-tool-$CLT_VER.zip"

    CDN_LAYOUT_TOOL="/tmp/cdn-layout-tool-$CLT_VER/bin/cdn-layout-tool"
    printf "Done: ${bri}%s${clr}\n" "$CDN_LAYOUT_TOOL"
fi


# Ok, let’s go
printf "Project dir: ${bri}%s${clr}\n" "$PROJECT_DIR"
printf "Project name: ${bri}%s${clr}\n" "$PROJECT_NAME"

# Create clean config for IDEA
export IDEA_PROPERTIES=/tmp/ide.properties

echo "idea.system.path=/tmp/ide-system" > "$IDEA_PROPERTIES"
echo "idea.config.path=/tmp/ide-config" >> "$IDEA_PROPERTIES"
echo "idea.log.path=/tmp/ide-log" >> "$IDEA_PROPERTIES"

# NB: Providing --git-dir="$PROJECT_DIR/.git" will not work when indexing .ijwb dir (which is a proper project dir for projects synced with Bazel plugin)
LAST_COMMIT=$(cd "$PROJECT_DIR" && git rev-parse HEAD 2>/dev/null || echo "")

# Generate indexes
echo Creating shared indexes at:
printbright "$TMP_OUTPUT_DIR"
printbright "You could later copy produced *.ijx files to the idea.system.path/shared-index"
"$IDEA_BIN" dump-shared-index project --output="$TMP_OUTPUT_DIR" --tmp="/tmp" --project-dir="$PROJECT_DIR" --project-id="$PROJECT_NAME" --commit="$LAST_COMMIT" --add-hash-to-output-names=false

PORT=9876
URL="http://0.0.0.0:$PORT/"
URL_PATH="$URL$PROJECT_NAME"

# Urlencode URL_PATH if jq is available
if hash jq 2>/dev/null; then
    URL_PATH=$(printf %s "$PROJECT_NAME" | jq -sRr @uri)
    URL_PATH="$URL$URL_PATH"
fi

# Generate CDN directory structure from indexes
echo Creating CDN layout from index chunks...
"$CDN_LAYOUT_TOOL" --indexes-dir="$TMP_OUTPUT_DIR" --url="$URL"

rm "$TMP_OUTPUT_DIR/project/list.json.xz"

# Match dotfiles, to be able to cp .../project/.ijwb ...
set -o dotglob
cp -Rp "$TMP_OUTPUT_DIR/project"/* "$CDN_ROOT"
rm -rf "$TMP_OUTPUT_DIR/project"
cp -p "$TMP_OUTPUT_DIR"/*.ijx.xz "$CDN_ROOT"
set +o dotglob

echo Updating project intellij.yaml
touch "$PROJECT_DIR/intellij.yaml"
yq e -i ".sharedIndex.project[0].url = \"$URL_PATH\"" "$PROJECT_DIR/intellij.yaml"
cat "$PROJECT_DIR/intellij.yaml"

echo Launching local web server with generated CDN layout at:
printbright "$CDN_ROOT"

if hash php 2>/dev/null; then
    cd "$CDN_ROOT"
    php -S "0.0.0.0:$PORT"
elif hash python3 2>/dev/null; then
    cd "$CDN_ROOT"
    python3 -m http.server "$PORT"
elif hash npx 2>/dev/null; then
    npx http-server "$CDN_ROOT" --port="$PORT"
else
    printerr "Could not find python3, php or npx, please launch local web server manually:"
    printbright "cd $CDN_ROOT"
    printbright "python3 -m http.server $PORT"
    exit 3
fi