#!/bin/bash

# Default values
TARGET_EXT="mp4"
TARGET_SIZE=10M

# Parse arguments
if [ -n "$1" ]; then
  INPUT_PATH="$1"
else
  INPUT_PATH="."
fi

if [ -n "$2" ]; then
  TARGET_EXT="$2"
fi

if [ -n "$3" ]; then
  TARGET_SIZE="$3"
fi

# Create output directory
OUTPUT_DIR="converted"
mkdir -p "$OUTPUT_DIR"

# Function to convert video
convert_video() {
  local input_file="$1"
  local output_file="$OUTPUT_DIR/${input_file%.*}.$TARGET_EXT"
  
  # Get video duration in seconds
  DURATION=$(ffprobe -v error -show_entries format=duration -of default=noprint_wrappers=1:nokey=1 "$input_file")
  echo "Duration: $DURATION seconds"
  
  # Convert target size to bytes
  TARGET_SIZE_BYTES=$(numfmt --from=iec "$TARGET_SIZE")
    
  # Calculate target bitrate in kilobits per second
  TARGET_BITRATE=$(echo "($TARGET_SIZE_BYTES * 8) / $DURATION / 2000" | bc)
  
  # Convert video
  ffmpeg -i "$input_file" -vcodec libx264 -b:v "${TARGET_BITRATE}k" -preset veryslow -acodec aac -c:a copy "$output_file"
  
  # Get original and converted video sizes
  ORIGINAL_SIZE=$(stat -c%s "$input_file")
  CONVERTED_SIZE=$(stat -c%s "$output_file")
  
  # Print out details
  echo "Original size: $(numfmt --to=iec $ORIGINAL_SIZE)"
  echo "Video length: $DURATION seconds"
  echo "Target size: $TARGET_SIZE"
  echo "Converted size: $(numfmt --to=iec $CONVERTED_SIZE)"
  echo "Target bitrate: ${TARGET_BITRATE}kbps"
}

# Function to move video if already below target size and in desired format
move_video() {
  local input_file="$1"
  local output_file="$OUTPUT_DIR/${input_file##*/}"
  
  # Get original video size
  ORIGINAL_SIZE=$(stat -c%s "$input_file")
  
  # Check if video is below target size and in desired format
  if [[ "$ORIGINAL_SIZE" -le "$TARGET_SIZE_BYTES" && "${input_file##*.}" == "$TARGET_EXT" ]]; then
    mv "$input_file" "$output_file"
    echo "Moved $input_file to $output_file"
  else
    convert_video "$input_file"
  fi
}

# Export functions for find command
export -f convert_video
export -f move_video
export TARGET_EXT
export TARGET_SIZE
export TARGET_SIZE_BYTES
export OUTPUT_DIR

# Find and process videos
if [ -d "$INPUT_PATH" ]; then
  find "$INPUT_PATH" \( -name "*.mkv" -o -name "*.mp4" -o -name "*.avi" -o -name "*.webm" \) -type f -exec bash -c 'move_video "$0"' {} \;
else
  move_video "$INPUT_PATH"
fi