#!/bin/bash

# Abort if anything fails
set -e

# Source common functions
BUILD_ROOT=$(realpath "${BASH_SOURCE[0]%/*}/..")
source "$BUILD_ROOT/helpers/message-helpers.sh"
source "$BUILD_ROOT/helpers/docker-helpers.sh"

ARCH=${ARCH:-x86_64}

BUILD_OUTPUT_DIR="$BUILD_ROOT/build-output/$ARCH"
FINAL_OUTPUT_DIR="$BUILD_ROOT/pkgs.omarchy.org/$ARCH"
SRC_DIR="$BUILD_ROOT/src"
BUILD_DIR="$BUILD_ROOT/build"

# Create directories if they don't exist
mkdir -p "$BUILD_OUTPUT_DIR" "$FINAL_OUTPUT_DIR" "$SRC_DIR"

print_header "Omarchy AUR Package Builder"

# Check Docker is available
check_docker

# Parse command line arguments
while [[ $# -gt 0 ]]; do
  case $1 in
  --arch)
    ARCH="$2"
    shift 2
    ;;
  --package)
    SINGLE_PACKAGE="$2"
    shift 2
    ;;
  -h | --help)
    echo "Usage: $0 [OPTIONS]"
    echo ""
    echo "Options:"
    echo "  --arch <arch>        Target architecture (x86_64 or aarch64, default: x86_64)"
    echo "  --package <name>     Build only the specified package (with optional options)"
    echo "  -h, --help           Show this help message"
    echo ""
    echo "This script builds AUR packages from:"
    echo "  build/packages/omarchy-aur.packages"
    echo ""
    echo "Or build a single package with --package:"
    echo "  Package can be: AUR package, GitHub repo (owner/repo), or local PKGBUILD"
    echo "  Options can be appended after package name (skip-pgp, install, always-build, etc.)"
    echo ""
    echo "Examples:"
    echo "  $0 --arch aarch64"
    echo "  $0 --package yay"
    echo "  $0 --package \"elephant install\""
    echo "  $0 --package \"owner/repo skip-pgp\""
    echo ""
    exit 0
    ;;
  *)
    print_error "Unknown option: $1"
    exit 1
    ;;
  esac
done

# Show target architecture after parsing args
print_info "Target architecture: $ARCH"
print_info "Build workspace: $BUILD_OUTPUT_DIR"
print_info "Final output: $FINAL_OUTPUT_DIR"

# Clean build-output directory to start fresh
print_info "Cleaning build workspace..."
rm -rf "$BUILD_OUTPUT_DIR"/*
mkdir -p "$BUILD_OUTPUT_DIR"

# Check if AUR package list exists (only required if not building single package)
if [[ -z "$SINGLE_PACKAGE" ]]; then
  if [[ ! -f "$BUILD_DIR/packages/omarchy-aur.packages" ]]; then
    print_error "omarchy-aur.packages not found at $BUILD_DIR/packages/"
    exit 1
  fi
else
  print_info "Building single package: $SINGLE_PACKAGE"
fi

# Build/update the Docker image
build_docker_image "$BUILD_DIR"

print_info "Running AUR package build..."

# Ensure output directories are writable by container user
make_dir_writable "$BUILD_OUTPUT_DIR"
make_dir_writable "$FINAL_OUTPUT_DIR"

# Build Docker arguments
DOCKER_ARGS=(
  --rm
  -e ARCH="$ARCH"
  -e SINGLE_PACKAGE="$SINGLE_PACKAGE"
  -v "$BUILD_ROOT/build-output:/build-output"
  -v "$BUILD_ROOT/pkgs.omarchy.org:/pkgs.omarchy.org"
  -v "$BUILD_DIR:/build:ro"
  -v "$BUILD_ROOT/pkgbuilds:/pkgbuilds:ro"
)

# Use local pacman cache if you already have one on host to speed up repeat runs
if [ -d "/var/cache/pacman/pkg" ]; then
  DOCKER_ARGS+=(-v "/var/cache/pacman/pkg:/var/cache/pacman/pkg")
fi

# Run the builder with assembled args
docker run "${DOCKER_ARGS[@]}" omarchy-aur-builder:latest /build/build.sh

BUILD_RESULT=$?

# Summary
echo ""
if [[ $BUILD_RESULT -eq 0 ]]; then
  print_success "AUR build completed successfully!"
else
  print_warning "Some AUR packages failed (see details above)"
  exit $BUILD_RESULT
fi