build-schema.sh

#!/bin/bash

# Build HyperFleet OpenAPI Schema
# Usage: ./build-schema.sh <provider> [--swagger|--openapi2]
#   provider: core, gcp, or any provider with aliases-{provider}.tsp file (required, must be first argument)
#   --swagger, --openapi2: Also generate OpenAPI 2.0 (Swagger) format

set -e

# Colors for output
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
NC='\033[0m' # No Color

# Parse arguments
GENERATE_SWAGGER=false

# Provider is required and must be the first argument
if [ $# -lt 1 ]; then
    echo -e "${RED}Error: Provider argument is required${NC}"
    echo "Usage: $0 <provider> [--swagger|--openapi2]"
    echo "  provider: core, gcp, or any provider with aliases-{provider}.tsp file"
    echo "  --swagger, --openapi2: Also generate OpenAPI 2.0 (Swagger) format"
    exit 1
fi

PROVIDER="$1"
shift

# Check if provider looks like an option (starts with -)
if [[ "$PROVIDER" == -* ]]; then
    echo -e "${RED}Error: Provider must be the first argument${NC}"
    echo "Usage: $0 <provider> [--swagger|--openapi2]"
    exit 1
fi

# Parse remaining options
for arg in "$@"; do
    case $arg in
        --swagger|--openapi2)
            GENERATE_SWAGGER=true
            ;;
        -*)
            echo -e "${RED}Error: Unknown option: $arg${NC}"
            echo "Usage: $0 <provider> [--swagger|--openapi2]"
            exit 1
            ;;
        *)
            echo -e "${RED}Error: Unexpected argument: $arg${NC}"
            echo "Usage: $0 <provider> [--swagger|--openapi2]"
            exit 1
            ;;
    esac
done

# Script directory (where the script is located)
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
cd "$SCRIPT_DIR"

# Define the aliases file for the provider
ALIASES_FILE="aliases-${PROVIDER}.tsp"

# Check if the aliases file exists
if [ ! -f "$ALIASES_FILE" ]; then
    echo -e "${RED}Error: Provider aliases file not found: ${ALIASES_FILE}${NC}"
    echo ""
    echo "Available providers:"
    for file in aliases-*.tsp; do
        if [ -f "$file" ]; then
            provider_name=$(echo "$file" | sed 's/aliases-\(.*\)\.tsp/\1/')
            echo "  - $provider_name"
        fi
    done
    exit 1
fi

# Check if main.tsp exists
if [ ! -f "main.tsp" ]; then
    echo -e "${RED}Error: main.tsp not found in current directory${NC}"
    exit 1
fi

# Check if tsp command is available
if ! command -v tsp &> /dev/null; then
    echo -e "${RED}Error: tsp command not found. Please install TypeSpec compiler.${NC}"
    echo "Install with: npm install -g @typespec/compiler"
    exit 1
fi

# Check if api-spec-converter is available when swagger output is requested
if [ "$GENERATE_SWAGGER" = true ]; then
    if ! npx api-spec-converter --version &> /dev/null; then
        echo -e "${RED}Error: api-spec-converter not found. Please install it.${NC}"
        echo "Install with: npm install --save-dev api-spec-converter"
        exit 1
    fi
fi

echo -e "${GREEN}Building HyperFleet API schema for provider: ${PROVIDER}${NC}"
if [ "$GENERATE_SWAGGER" = true ]; then
    echo -e "${GREEN}Output formats: OpenAPI 3.0 + OpenAPI 2.0 (Swagger)${NC}"
else
    echo -e "${GREEN}Output format: OpenAPI 3.0${NC}"
fi
echo ""

# Step 1: Re-link aliases.tsp to the provider-specific aliases file
echo -e "${YELLOW}Step 1: Linking aliases.tsp to ${ALIASES_FILE}${NC}"
if [ -L "aliases.tsp" ] || [ -f "aliases.tsp" ]; then
    rm -f aliases.tsp
fi
ln -sf "$ALIASES_FILE" aliases.tsp
echo -e "${GREEN}✓ Linked aliases.tsp → ${ALIASES_FILE}${NC}"
echo ""

# Step 2: Create output directory for the provider
OUTPUT_DIR="schemas/${PROVIDER}"
echo -e "${YELLOW}Step 2: Preparing output directory...${NC}"
mkdir -p "$OUTPUT_DIR"
echo -e "${GREEN}✓ Created output directory: ${OUTPUT_DIR}${NC}"
echo ""

# Step 3: Compile TypeSpec to generate OpenAPI schema
echo -e "${YELLOW}Step 3: Compiling TypeSpec...${NC}"
TEMP_OUTPUT_DIR="tsp-output-${PROVIDER}"

# Cleanup function to remove temporary directory on exit
cleanup() {
    if [ -d "$TEMP_OUTPUT_DIR" ]; then
        rm -rf "$TEMP_OUTPUT_DIR"
    fi
}
trap cleanup EXIT

if tsp compile main.tsp --output-dir "$TEMP_OUTPUT_DIR"; then
    # Move the generated schema to the provider-specific directory
    if [ -f "${TEMP_OUTPUT_DIR}/schema/openapi.yaml" ]; then
        mv "${TEMP_OUTPUT_DIR}/schema/openapi.yaml" "${OUTPUT_DIR}/openapi.yaml"
        echo ""
        echo -e "${GREEN}✓ Successfully generated OpenAPI 3.0 schema${NC}"
        echo -e "${GREEN}Output: ${OUTPUT_DIR}/openapi.yaml${NC}"
    else
        echo ""
        echo -e "${RED}✗ Generated schema file not found at expected location${NC}"
        echo "Expected: ${TEMP_OUTPUT_DIR}/schema/openapi.yaml"
        exit 1
    fi
else
    echo ""
    echo -e "${RED}✗ Failed to compile TypeSpec${NC}"
    exit 1
fi

# Step 4: Convert to OpenAPI 2.0 (Swagger) if requested
if [ "$GENERATE_SWAGGER" = true ]; then
    echo ""
    echo -e "${YELLOW}Step 4: Converting to OpenAPI 2.0 (Swagger)...${NC}"
    
    if npx api-spec-converter \
        --from=openapi_3 \
        --to=swagger_2 \
        --syntax=yaml \
        "${OUTPUT_DIR}/openapi.yaml" > "${OUTPUT_DIR}/swagger.yaml" 2>/dev/null; then
        echo -e "${GREEN}✓ Successfully generated OpenAPI 2.0 (Swagger) schema${NC}"
        echo -e "${GREEN}Output: ${OUTPUT_DIR}/swagger.yaml${NC}"
    else
        echo -e "${RED}✗ Failed to convert to OpenAPI 2.0 (Swagger)${NC}"
        echo "The OpenAPI 3.0 schema may contain features not supported in OpenAPI 2.0"
        rm -f "${OUTPUT_DIR}/swagger.yaml"
        exit 1
    fi
fi

echo ""
echo -e "${GREEN}Build complete!${NC}"