init-cluster.sh

#!/bin/bash
# init-cluster.sh — Initialize a single-node Couchbase cluster for local development.
#
# Mounted into the container by docker-compose.yml and run manually after startup:
#   docker compose exec couchbase bash /init-cluster.sh
#
# What it does:
#   1. Waits for the Couchbase REST API to be ready
#   2. Initializes the cluster (services, memory quotas, credentials)
#   3. Creates a default bucket, scope, and collection
#   4. Creates a primary index
#   5. Optionally loads the travel-sample bucket
#
# Override defaults via environment variables:
#   CB_BUCKET, CB_SCOPE, CB_COLLECTION, KV_QUOTA, INDEX_QUOTA, FTS_QUOTA,
#   EVENTING_QUOTA, ANALYTICS_QUOTA, LOAD_TRAVEL_SAMPLE

set -euo pipefail

CB_HOST="${CB_HOST:-localhost}"
CB_USER="${COUCHBASE_ADMINISTRATOR_USERNAME:-Administrator}"
CB_PASS="${COUCHBASE_ADMINISTRATOR_PASSWORD:-password}"
CB_BUCKET="${CB_BUCKET:-myapp}"
CB_SCOPE="${CB_SCOPE:-_default}"
CB_COLLECTION="${CB_COLLECTION:-_default}"
LOAD_TRAVEL_SAMPLE="${LOAD_TRAVEL_SAMPLE:-false}"

KV_QUOTA="${KV_QUOTA:-512}"
INDEX_QUOTA="${INDEX_QUOTA:-256}"
FTS_QUOTA="${FTS_QUOTA:-256}"
EVENTING_QUOTA="${EVENTING_QUOTA:-256}"
ANALYTICS_QUOTA="${ANALYTICS_QUOTA:-1024}"

CLI="couchbase-cli"

wait_for_couchbase() {
    echo "Waiting for Couchbase REST API..."
    for i in $(seq 1 60); do
        if curl -sf "http://${CB_HOST}:8091/pools" > /dev/null 2>&1; then
            echo "Couchbase is ready."
            return 0
        fi
        sleep 2
    done
    echo "ERROR: Couchbase did not become ready in time." >&2
    exit 1
}

init_cluster() {
    echo "Initializing cluster..."

    $CLI cluster-init \
        --cluster "${CB_HOST}:8091" \
        --cluster-username "${CB_USER}" \
        --cluster-password "${CB_PASS}" \
        --cluster-name "dev-cluster" \
        --services "data,index,query,fts,eventing,analytics" \
        --cluster-ramsize "${KV_QUOTA}" \
        --cluster-index-ramsize "${INDEX_QUOTA}" \
        --cluster-fts-ramsize "${FTS_QUOTA}" \
        --cluster-eventing-ramsize "${EVENTING_QUOTA}" \
        --cluster-analytics-ramsize "${ANALYTICS_QUOTA}" \
        --index-storage-setting plasma

    echo "Cluster initialized."
}

create_bucket() {
    echo "Creating bucket '${CB_BUCKET}'..."

    $CLI bucket-create \
        --cluster "${CB_HOST}:8091" \
        --username "${CB_USER}" \
        --password "${CB_PASS}" \
        --bucket "${CB_BUCKET}" \
        --bucket-type couchbase \
        --bucket-ramsize "${KV_QUOTA}" \
        --bucket-replica 0 \
        --enable-flush 1 \
        --wait

    echo "Bucket '${CB_BUCKET}' created."
}

create_scope() {
    if [ "${CB_SCOPE}" = "_default" ]; then
        return
    fi
    echo "Creating scope '${CB_SCOPE}'..."
    $CLI collection-manage \
        --cluster "${CB_HOST}:8091" \
        --username "${CB_USER}" \
        --password "${CB_PASS}" \
        --bucket "${CB_BUCKET}" \
        --create-scope "${CB_SCOPE}"
    sleep 1
}

create_collection() {
    if [ "${CB_COLLECTION}" = "_default" ]; then
        return
    fi
    echo "Creating collection '${CB_COLLECTION}'..."
    $CLI collection-manage \
        --cluster "${CB_HOST}:8091" \
        --username "${CB_USER}" \
        --password "${CB_PASS}" \
        --bucket "${CB_BUCKET}" \
        --create-collection "${CB_SCOPE}.${CB_COLLECTION}"
    sleep 1
}

create_primary_index() {
    echo "Creating primary index on \`${CB_BUCKET}\`.\`${CB_SCOPE}\`.\`${CB_COLLECTION}\`..."
    # couchbase-cli does not manage indexes — use the Query REST endpoint directly
    curl -sf -X POST "http://${CB_HOST}:8093/query/service" \
        -u "${CB_USER}:${CB_PASS}" \
        --data-urlencode "statement=CREATE PRIMARY INDEX IF NOT EXISTS ON \`${CB_BUCKET}\`.\`${CB_SCOPE}\`.\`${CB_COLLECTION}\`" \
        > /dev/null
    echo "Primary index created."
}

load_travel_sample() {
    echo "Loading travel-sample..."
    $CLI sample-data-load \
        --cluster "${CB_HOST}:8091" \
        --username "${CB_USER}" \
        --password "${CB_PASS}" \
        --datasets travel-sample
    echo "travel-sample load initiated (runs in background)."
}

# --- Main ---
wait_for_couchbase
init_cluster
create_bucket
create_scope
create_collection
create_primary_index

if [ "${LOAD_TRAVEL_SAMPLE}" = "true" ]; then
    load_travel_sample
fi

echo ""
echo "Couchbase dev cluster ready."
echo "  Web Console : http://localhost:8091"
echo "  Credentials : ${CB_USER} / ${CB_PASS}"
echo "  Bucket      : ${CB_BUCKET}"
echo "  Scope       : ${CB_SCOPE}"
echo "  Collection  : ${CB_COLLECTION}"