#!/bin/bash
set -e

# Create backup of original files
cp package.json package_original.json
cp package-lock.json package-lock_original.json

# Store the latest version after publishing
LATEST_VERSION=""

publish_package() {
    PACKAGE_NAME=$1
    BUILD_COMMAND=$2
    IS_REACT=$3

    echo "Preparing to publish ${PACKAGE_NAME}..."
    
    # Restore original package.json state before each publish
    cp package_original.json package.json
    cp package-lock_original.json package-lock.json

    # Update package name in package.json
    jq --arg name "$PACKAGE_NAME" '.name=$name' package.json > temp.json && mv temp.json package.json

    # Handle targets based on package type
    if [ "$IS_REACT" = "true" ]; then
        echo "Removing targets for React library build..."
        jq 'del(.targets)' package.json > temp.json && mv temp.json package.json
    fi

    # Clean dist directory
    if [ -d "dist" ]; then
        echo "Cleaning dist directory..."
        rm -rf dist
    fi

    # update version and store it
    LATEST_VERSION=$(npm version patch)
    echo "New version: ${LATEST_VERSION}"

    # Build package
    npm run "$BUILD_COMMAND"

    # Replace npm publish with npm pack for testing
    npm publish

    echo "Successfully packaged ${PACKAGE_NAME}"

    # Log the bundle size
    TARBALL="${PACKAGE_NAME}-${LATEST_VERSION#v}.tgz"
    if [ -f "$TARBALL" ]; then
        BUNDLE_SIZE=$(du -h "$TARBALL" | cut -f1)
        echo "Bundle size for ${PACKAGE_NAME}: ${BUNDLE_SIZE}"
    else
        echo "Error: ${TARBALL} not found."
        exit 1
    fi
}

# First publish docsgpt (HTML bundle)
publish_package "docsgpt" "build" "false"

# Then publish docsgpt-react (React library)
publish_package "docsgpt-react" "build:react" "true"

# Restore original state but keep the updated version
cp package_original.json package.json
cp package-lock_original.json package-lock.json

# Update the version in the final package.json
jq --arg version "${LATEST_VERSION#v}" '.version=$version' package.json > temp.json && mv temp.json package.json

# Run npm install to update package-lock.json with the new version
npm install --package-lock-only

# Cleanup backup files
rm -f package_original.json
rm -f package-lock_original.json
rm -f temp.json

echo "---Process completed---"
echo "Final version in package.json: $(jq -r '.version' package.json)"
echo "Final version in package-lock.json: $(jq -r '.version' package-lock.json)"
echo "Generated test packages:"
ls *.tgz