#--------------------------------
# filename:     /zsh/ffmpeg
#--------------------------------

# Original script: https://yohanes.gultom.me/2016/05/21/bash-script-to-batch-convert-mkv-to-mp4-linux/
# Rewritten with ChatGPT 🙈

function ff-mp4 () {
  local findpath="${1:-.}"
  local file_list=()
  local temp_file=$(mktemp)

  # Find all video files and write them to a temporary file
  find "$findpath" \( -iname '*.mkv' -o -iname '*.flv' -o -iname '*.mov' -o -iname '*.mp4' \) -print0 > "$temp_file"

  # Read the file contents into the file_list array
  while IFS= read -r -d $'\0' file; do
    file_list+=("$file")
  done < "$temp_file"
  rm "$temp_file" # Delete the temporary file

  # Process each file in the file_list array
  local num_files=${#file_list[@]}
  local counter=1
  for file in "${file_list[@]}"; do
    local dir=$(dirname "$file")
    local file_name=$(basename "$file")
    local name="${file_name%.*}"
    local ext="${file_name##*.}"

	echo "Processing file [$counter/$num_files]: $file_name"
	counter=$((counter+1))

    # Re-container the video file to mp4 using ffmpeg
    ffmpeg -hide_banner -loglevel error -i "$file" -map 0 -c copy -pix_fmt yuv420p -movflags faststart "$dir/temp_$name.mp4"

    # Rename the original file and move the new mp4 file to its place
    mv -n "$file" "$dir/_$file_name"
    mv -n "$dir/temp_$name.mp4" "$dir/$file_name"
  done
}

# ffmpeg-convert
# source: https://yohanes.gultom.me/2016/05/21/bash-script-to-batch-convert-mkv-to-mp4-linux/
function ff-ac3-to-stereo() {
        findpath=$1
        : "${findpath:="."}"
        find "$findpath" \( -iname '*.mkv' -o -iname '*.flv' -o -iname '*.mov' -o -iname '*.mp4' \) | while read -r f ; do
                dir=$(dirname "$f");
                file=$(basename "$f");
                name="${file%.*}";
                ext="${file##*.}";

                ffmpeg -hide_banner -loglevel error -i "$f" -map 0 -c:v copy -c:a libfdk_aac -b:a 192k -ac 2 -af volume=2 -movflags faststart "$dir/temp_$name.mp4";

                mv "$f" "$dir/_$file";
                mv "$dir/temp_$name.mp4" "$dir/$file";
        done
}

# ffmpeg audio-to-ac3
function ff-input-to-ac3() {
	ffmpeg -i "$1" -movflags faststart -map 0 -c:v copy -c:s copy -c:a ac3 -b:a 448k "temp_$1"
	mv "$1" "_$1" && mv "temp_$1" "$1"	
}

# ffmpeg fix 16:9 AR
function ff-ar16-9() {	
	ffmpeg -i "$1" -c copy -map 0 -bsf:v "h264_metadata=sample_aspect_ratio=4/3" -aspect 16:9 -pix_fmt yuv420p -movflags faststart "temp_$1"
        mv "$1" "_$1" && mv "temp_$1" "$1"
}

# ffmpeg fix 5:4 AR
function ff-ar5-4() {
        ffmpeg -i "$1" -c copy -map 0 -bsf:v "h264_metadata=sample_aspect_ratio=1/1" -aspect 5:4 -pix_fmt yuv420p -movflags faststart "temp_$1"
        mv "$1" "_$1" && mv "temp_$1" "$1"
}

# ffmpeg fix 4:3 AR
function ff-ar4-3() {
        ffmpeg -i "$1" -c copy -map 0 -bsf:v "h264_metadata=sample_aspect_ratio=1/1" -aspect 4:3 -pix_fmt yuv420p -movflags faststart "temp_$1"
        mv "$1" "_$1" && mv "temp_$1" "$1"
}

# ffmpeg fix AR ($1 = file, $2 1/1, $4:3) 
function ff-ar-custom() {
	if [ $# -lt 3 ]; then
	    >&2 echo "Not enough arguments (\$1 = 'file', \$2 '1/1', \$3 '4:3')"
	    return 0
	else
	        ffmpeg -i "$1" -c copy -map 0 -bsf:v "h264_metadata=sample_aspect_ratio=$2" -aspect $3 -pix_fmt yuv420p -movflags faststart "_$1"
	fi
}

# ffmpeg fix Non-monotonous DTS
function ff-dts() {
	ffmpeg -fflags +igndts -i "$1" -c copy -pix_fmt yuv420p -movflags faststart "temp_$1"
        mv "$1" "_$1" && mv "temp_$1" "$1"
}

## ffmpeg wma-to-m4a
function ff-wma-m4a() {
        findpath=$1
        : "${findpath:="."}"
        find "$findpath" -name '*.wma' | while read -r f ; do
                dir=$(dirname "$f");
                file=$(basename "$f");
                name="${file%.*}";
                ext="${file##*.}";

                ffmpeg -hide_banner -loglevel error -i "$f" -map 0 -vn -b:a 192k -c:a libfdk_aac "$dir/temp_$name.mp4";

                mv "$f" "$dir/_$file";
                mv "$dir/temp_$name.mp4" "$dir/$file";
        done
}