#!/bin/bash
set -e

BUILD_ROOT=$(realpath "${BASH_SOURCE[0]%/*}/..")
source "$BUILD_ROOT/helpers/message-helpers.sh"
source "$BUILD_ROOT/helpers/docker-helpers.sh"
source "$BUILD_ROOT/helpers/paths.sh"
source "$BUILD_ROOT/helpers/lock-helpers.sh"

print_header "Remove Package"

# Parse arguments
PACKAGE_NAME=""
while [[ $# -gt 0 ]]; do
  case $1 in
  --arch)
    ARCH="$2"
    update_arch_paths
    shift 2
    ;;
  --mirror)
    MIRROR="$2"
    require_valid_mirror "$MIRROR"
    update_arch_paths
    shift 2
    ;;
  -h | --help)
    echo "Usage: $0 <package-name> [OPTIONS]"
    echo ""
    echo "Options:"
    echo "  --arch <arch>      Target architecture (x86_64 or aarch64, default: x86_64)"
    echo "  --mirror <mirror>  Mirror to use (edge, rc, or stable, default: edge)"
    echo "  -h, --help         Show this help message"
    exit 0
    ;;
  *)
    if [[ -z "$PACKAGE_NAME" ]]; then
      PACKAGE_NAME="$1"
    else
      print_error "Unknown option: $1"
      exit 1
    fi
    shift
    ;;
  esac
done

if [[ -z "$PACKAGE_NAME" ]]; then
  print_error "Package name required"
  echo ""
  echo "Usage: $0 <package-name> [OPTIONS]"
  exit 1
fi

print_info "Target architecture: $ARCH"
print_info "Mirror: $MIRROR"
print_info "Repository: $REPO_DIR"

# Check if repository exists
if [[ ! -d "$REPO_DIR" ]]; then
  print_error "Repository directory not found: $REPO_DIR"
  exit 1
fi

# Check the selected container engine is available
check_engine

# Find package files to confirm before running Docker
cd "$REPO_DIR"
FILES=$(ls ${PACKAGE_NAME}-[0-9]*.pkg.tar.* 2>/dev/null || true)

if [[ -z "$FILES" ]]; then
  print_error "Package '$PACKAGE_NAME' not found"
  exit 1
fi

# Confirm removal
echo ""
echo "Package files to remove:"
for file in $FILES; do
  echo "  - $file"
done
echo ""
read -p "Remove package '$PACKAGE_NAME'? (y/N) " -n 1 -r
echo
if [[ ! $REPLY =~ ^[Yy]$ ]]; then
  print_info "Removal cancelled"
  exit 0
fi

# repo-remove is architecture-independent, so use a host-native edge image.
TOOL_ARCH=$(docker_native_arch) || {
  print_error "Unsupported host architecture: $(uname -m)"
  exit 1
}
build_docker_image "$BUILD_DIR" "$TOOL_ARCH" "edge"

acquire_release_lock || exit 1

print_info "Removing package..."

# Rootless Podman uses keep-id and leaves host ownership/modes intact.
if [[ "$CONTAINER_ENGINE" == "docker" ]]; then
  make_dir_writable "$REPO_DIR"
fi

# Run the removal script in the host-native image.
"$CONTAINER_ENGINE" run "${CONTAINER_RUN_ARGS[@]}" --rm "$(get_platform_arg "$TOOL_ARCH")" \
  -e ARCH="$ARCH" \
  -e MIRROR="$MIRROR" \
  -v "$REPO_ROOT:/pkgs.omarchy.org" \
  -v "$BUILD_DIR:/build:ro" \
  "omarchy-pkg-builder:latest-$TOOL_ARCH-edge" /build/remove-package.sh "$PACKAGE_NAME"

RESULT=$?

echo ""
if [[ $RESULT -eq 0 ]]; then
  print_success "Package removal completed successfully!"
else
  print_error "Package removal failed"
  exit $RESULT
fi