release.sh

#!/bin/bash

set -e

# Check if version argument is provided
if [ -z "$1" ]; then
    echo "Please provide version argument (major|minor|patch) or --publish VERSION"
    exit 1
fi

# Handle publish mode
if [ "$1" = "--publish" ]; then
    if [ -z "$2" ]; then
        echo "Please provide version number for publish"
        exit 1
    fi
    NEW_VERSION=$2

    # Build the artifacts
    echo "Building artifacts..."
    make build_udfs
    make build_gateways

    # Create git tag
    echo "Creating git tag v$NEW_VERSION..."
    git tag -a "v$NEW_VERSION" -m "Release v$NEW_VERSION"
    git push origin "v$NEW_VERSION"
    # Create temporary directory for release assets
    TEMP_DIR=$(mktemp -d)

    # Copy and rename binaries for different architectures
    echo "Preparing release artifacts..."
    cp target/x86_64-unknown-linux-gnu/release/langdb_udf $TEMP_DIR/langdb_udf-x86_64
    cp target/aarch64-unknown-linux-gnu/release/langdb_udf $TEMP_DIR/langdb_udf-aarch64
    cp target/x86_64-unknown-linux-gnu/release/ai-gateway $TEMP_DIR/ai-gateway-x86_64
    cp target/aarch64-unknown-linux-gnu/release/ai-gateway $TEMP_DIR/ai-gateway-aarch64

    # Create GitHub release and upload assets
    echo "Creating GitHub release..."
    gh release create $NEW_VERSION \
        --title "Release $NEW_VERSION" \
        --notes-file CHANGELOG.md \
        $TEMP_DIR/langdb_udf-x86_64 \
        $TEMP_DIR/langdb_udf-aarch64 \
        $TEMP_DIR/ai-gateway-x86_64 \
        $TEMP_DIR/ai-gateway-aarch64

    # Cleanup
    rm -rf $TEMP_DIR
    echo "Release process completed successfully!"
    exit 0
fi

VERSION_TYPE=$1

# Get current version from core/Cargo.toml
CURRENT_VERSION=$(grep '^version = ' core/Cargo.toml | sed 's/version = "\(.*\)"/\1/')

if [ -z "$CURRENT_VERSION" ]; then
    echo "Error: Could not determine current version from core/Cargo.toml"
    exit 1
fi

# Calculate new version based on version type
case $VERSION_TYPE in
    major)
        NEW_VERSION=$(echo $CURRENT_VERSION | awk -F. '{$1 = $1 + 1; $2 = 0; $3 = 0} 1' OFS=.)
        ;;
    minor)
        NEW_VERSION=$(echo $CURRENT_VERSION | awk -F. '{$2 = $2 + 1; $3 = 0} 1' OFS=.)
        ;;
    patch)
        NEW_VERSION=$(echo $CURRENT_VERSION | awk -F. '{$3 = $3 + 1} 1' OFS=.)
        ;;
    *)
        echo "Invalid version type. Use major, minor, or patch"
        exit 1
        ;;
esac

# Update version in Cargo.toml files
echo "Updating version to $NEW_VERSION in Cargo.toml files..."
(cd core && cargo set-version $NEW_VERSION)
(cd udfs && cargo set-version $NEW_VERSION)
(cd gateway && cargo set-version $NEW_VERSION)
(cd guardrails && cargo set-version $NEW_VERSION)

# Install standard-version if not already installed
if ! command -v npx &> /dev/null; then
    echo "Installing npx..."
    npm install -g npx
fi

# Generate CHANGELOG
echo "Generating CHANGELOG..."
npx standard-version --release-as "$NEW_VERSION" --tag-prefix "" --skip.tag true

# Create and push PR for version bump and CHANGELOG
echo "Creating PR for version bump..."
BRANCH_NAME="release/v$NEW_VERSION"
git checkout -b $BRANCH_NAME
git add CHANGELOG.md core/Cargo.toml udfs/Cargo.toml gateway/Cargo.toml
git commit -m "chore: release v$NEW_VERSION"
git push origin $BRANCH_NAME

gh pr create \
    --title "Release v$NEW_VERSION" \
    --body "Automated PR for version v$NEW_VERSION release" \
    --base main \
    --head $BRANCH_NAME

echo "PR created. Please merge the PR before continuing with the release."
echo "After PR is merged, run: ./release.sh --publish $NEW_VERSION"