set -euo pipefail

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"

PROJECT_ROOT="$(cd "$SCRIPT_DIR/.." && pwd)"

RED='\033[0;31m'

GREEN='\033[0;32m'

YELLOW='\033[1;33m'

BLUE='\033[0;34m'

NC='\033[0m' # No Color

DRY_RUN=false

VERBOSE=false

log_info() {

echo -e "${BLUE}[INFO]${NC} $1"

}

log_success() {

echo -e "${GREEN}[✓]${NC} $1"

}

log_warn() {

echo -e "${YELLOW}[WARN]${NC} $1"

}

log_error() {

echo -e "${RED}[✗]${NC} $1"

}

log_step() {

echo -e "${BLUE}[STEP]${NC} $1"

}

usage() {

}

SKIP_TESTS=false

while [[ $# -gt 0 ]]; do

case $1 in

--dry-run)

DRY_RUN=true

shift

;;

--verbose)

VERBOSE=true

shift

;;

--skip-tests)

SKIP_TESTS=true

shift

;;

-h|--help)

usage

exit 0

;;

*)

log_error "Unknown option: $1"

usage

exit 1

;;

esac

cd "$PROJECT_ROOT"

FAILED=0

echo ""

echo "=============================================="

echo " DomainForge Pre-Release Checks"

echo "=============================================="

echo ""

if $DRY_RUN; then

log_warn "DRY RUN MODE - Showing what would be checked"

echo ""

log_step "Checking git status for uncommitted changes..."

if $DRY_RUN; then

log_info "Would run: git status --porcelain"

if [[ -n $(git status --porcelain) ]]; then

log_error "Uncommitted changes detected:"

git status --short

FAILED=1

else

log_success "No uncommitted changes"

fi

log_step "Validating current branch..."

if $DRY_RUN; then

log_info "Would check if branch matches: main or release/*"

BRANCH=$(git rev-parse --abbrev-ref HEAD)

if [[ "$BRANCH" =~ ^(main|release/.*)$ ]]; then

log_success "On valid branch: $BRANCH"

if [[ "$BRANCH" != "main" ]]; then

log_warn "Release preparation is allowed on $BRANCH, but scripts/release.sh must be run from main after promotion."

fi

else

log_warn "On branch '$BRANCH' - releases typically done from main or release/*"

# Not a hard failure, just a warning

fi

log_step "Checking for latest changes from remote..."

if $DRY_RUN; then

log_info "Would run: git fetch origin && git status -uno"

git fetch origin --quiet 2>/dev/null || log_warn "Could not fetch from origin"

LOCAL=$(git rev-parse HEAD)

REMOTE=$(git rev-parse "@{u}" 2>/dev/null || echo "")

if [[ -z "$REMOTE" ]]; then

log_warn "No upstream branch set"

elif [[ "$LOCAL" != "$REMOTE" ]]; then

BEHIND=$(git rev-list --count HEAD..@{u} 2>/dev/null || echo "0")

if [[ "$BEHIND" -gt 0 ]]; then

log_warn "Branch is $BEHIND commits behind remote. Consider pulling."

fi

else

log_success "Up to date with remote"

fi

log_step "Checking version consistency across files..."

if $DRY_RUN; then

log_info "Would compare versions in:"

log_info " - domainforge-core/Cargo.toml"

log_info " - pyproject.toml"

log_info " - package.json"

CARGO_VERSION=$(grep -m1 '^version = ' domainforge-core/Cargo.toml | sed 's/version = "\(.*\)"/\1/')

PYPROJECT_VERSION=$(grep -m1 '^version = ' pyproject.toml | sed 's/version = "\(.*\)"/\1/' | tr -d '\r')

PKG_VERSION=$(grep -m1 '"version"' package.json | sed 's/.*"\([0-9][^"]*\)".*/\1/')

log_info "Cargo.toml: $CARGO_VERSION"

log_info "pyproject.toml: $PYPROJECT_VERSION"

log_info "package.json: $PKG_VERSION"

if [[ "$CARGO_VERSION" == "$PYPROJECT_VERSION" && "$CARGO_VERSION" == "$PKG_VERSION" ]]; then

log_success "All versions match: $CARGO_VERSION"

else

log_error "Version mismatch across files!"

FAILED=1

fi

log_step "Checking CHANGELOG.md..."

if $DRY_RUN; then

log_info "Would verify CHANGELOG.md contains entry for current version"

if [[ -f "CHANGELOG.md" ]]; then

# Get the first version entry

FIRST_CHANGELOG_VERSION=$(grep -m1 '^\#\# \[' CHANGELOG.md | sed 's/.*\[\([0-9][^]]*\)\].*/\1/' || echo "")

if [[ -n "$FIRST_CHANGELOG_VERSION" ]]; then

if [[ "$FIRST_CHANGELOG_VERSION" == "$CARGO_VERSION" ]]; then

log_success "CHANGELOG.md has entry for current version ($CARGO_VERSION)"

else

log_warn "CHANGELOG.md top entry ($FIRST_CHANGELOG_VERSION) doesn't match current version ($CARGO_VERSION)"

fi

else

log_warn "Could not find version entry in CHANGELOG.md"

fi

else

log_error "CHANGELOG.md not found"

FAILED=1

fi

if $SKIP_TESTS; then

log_step "Skipping tests (--skip-tests flag)"

log_step "Running test suites..."

if $DRY_RUN; then

log_info "Would run: just all-tests (Rust + Python + TypeScript)"

else

if command -v just &> /dev/null; then

log_info "Running: just all-tests"

if just all-tests; then

log_success "All tests passed"

else

log_error "Tests failed"

FAILED=1

fi

else

log_warn "'just' not found, attempting individual test commands..."

# Try Rust tests

log_info "Running Rust tests..."

if cargo test -p domainforge-core --features cli; then

log_success "Rust tests passed"

else

log_error "Rust tests failed"

FAILED=1

fi

fi

fi

log_step "Verifying release build..."

if $DRY_RUN; then

log_info "Would run: cargo build -p domainforge-core --release --features cli"

if cargo build -p domainforge-core --release --features cli 2>/dev/null; then

log_success "Release build successful"

else

log_error "Release build failed"

FAILED=1

fi

echo ""

echo "=============================================="

if [[ $FAILED -eq 0 ]]; then

log_success "All pre-release checks passed!"

echo "=============================================="

exit 0

log_error "Some pre-release checks failed"

echo "=============================================="

exit 1