generate.sh

#!/bin/bash

set -e  # Exit on any error

# return to the original directory on error or script exit
CWD=$(pwd)
trap '[ $? -ne 0 ] && echo "❌ Failed to update the schema"; cd "$CWD";' EXIT

# Pull latest registry
echo "Fetching latest registry..."
curl -s https://registry.thegraph.com/TheGraphNetworksRegistry.json > sample/TheGraphNetworksRegistry.json

# Extract and fetch schema URL
SCHEMA_URL=$(jq -r '."$schema"' sample/TheGraphNetworksRegistry.json)
echo "Fetching schema from $SCHEMA_URL..."
# weird name for the schema file but it's the only way to make quicktype generate acceptable type name
curl -s "$SCHEMA_URL" > "sample/Network.json"

# Extract schema version from filename (e.g., TheGraphNetworksRegistrySchema_v0_5.json)
SCHEMA_VERSION=$(echo "$SCHEMA_URL" | grep -o 'v[0-9]\+_[0-9]\+' | tr '_' '.')
MAJOR_MINOR_VERSION=${SCHEMA_VERSION#v}  # Remove the 'v' prefix
echo "Detected schema version: $MAJOR_MINOR_VERSION"

# Update version in TypeScript lib
echo "Updating TypeScript package version..."
jq ".version = \"$MAJOR_MINOR_VERSION.0\"" packages/typescript/package.json > temp.json
mv temp.json packages/typescript/package.json
echo "export const version = \"$MAJOR_MINOR_VERSION.0\";" > packages/typescript/src/version.ts

# Update Rust Cargo.toml
echo "Updating Rust package version..."
sed -i.bak "s/^version = \".*\"/version = \"$MAJOR_MINOR_VERSION.0\"/" packages/rust/Cargo.toml
rm packages/rust/Cargo.toml.bak

# Create/Update Go version file
echo "Updating Go version..."
cat > packages/golang/lib/version.go << EOF
package registry
const Version = "$MAJOR_MINOR_VERSION.0"
EOF

# Generate types for each language
echo "Generating TypeScript types..."
npx quicktype -s schema sample/Network.json --lang typescript --top-level NetworksRegistryElement --out packages/typescript/src/types.ts

echo "Generating Rust types..."
npx quicktype -s schema sample/Network.json --lang rust --top-level NetworksRegistry --density normal --visibility public --derive-debug --derive-clone --out packages/rust/src/types.rs

echo "Generating Go types..."
npx quicktype -s schema sample/Network.json --lang go --top-level NetworksRegistry --package registry --out packages/golang/lib/types.go

# Run tests for each package
echo "Running TypeScript tests..."
cd packages/typescript
bun install
bun run build
bun test
cd ../..

echo "Running Rust tests..."
cd packages/rust
cargo test
cd ../..

echo "Running Go tests..."
cd packages/golang/lib
go test
cd ../../..

echo "Cleaning up..."
rm sample/Network.json

# Run examples
cd packages/typescript/examples/local
bun install --no-save
bun run start
cd ../../../..

cd packages/rust/examples/local
cargo run
cd ../../../..

cd packages/golang/examples/local
go run main.go

echo "✅ Schema update completed successfully!"