#!/bin/bash
trap "kill 0" SIGINT # Kill all spawned subprocesses on ctrl^c
LOCATION=$(dirname "$0")
source "$LOCATION"/../common_tools

screenshot(){
  ios_device_info "$1"
  FILENAME="$MANUFACTURER-$MODEL-iOS$VERSION-$(date +%Y-%m-%d-%H-%M-%S).png"
  echo "📸 Saving screenshot into $FILENAME..."
  go-ios screenshot --udid="$1" --output="$HOME/Desktop/$FILENAME" &> /dev/null
  EXIT_CODE=$?
  if [ $EXIT_CODE -ne 0 ]; then
    echo "🤷 Screenshot failed, error code: $EXIT_CODE"
  fi
}

screenshot_all(){
  check_for_update
  ios_get_devices
  for ID in "${IOS_USB_DEVICES[@]}"
   do
    screenshot "$ID" &
  done
  wait
}

if [ "$1" == "-a" ];
then
  screenshot_all
else
  ios_choose_device
  screenshot "$SELECTED_DEVICE"
fi
