#!/bin/bash

######################################################################
# ks-mp3-album (ks-tools) - Convert folder album to MP3 Audio Format #
# Date: 30-03-2021                                                   #
# Author: q3aql                                                      #
# Contact: q3aql@protonmail.ch                                       #
######################################################################
VERSION="7.2"
M_DATE="300321"

# Variables
tempFile="/tmp/ks-mp4s-folder.txt"
tempFileTest="/tmp/ks-mp4s-folder-test.txt"
formatFiles="mp4 avi mpg mpeg mov wmv mkv ogv webm rm flv vob ogg oga mp3 mp2 aac ac3 flac opus wav"

# Global parameters
dirTemp="/tmp"
listTemp="ks-tools.list"
ksToolsTempFolder="/tmp/ks-tools"

# Basic parameters
acodec="libmp3lame"
b_acodec="128k"
v_ext="mp3"
track_init="1"
artist_init="Artist"
default_lang_audio="spa"

# Check cygwin alias (for Windows)
if [ -f "/usr/bin/cygwin-alias.sh" ] ; then
  shopt -s expand_aliases
  source "/usr/bin/cygwin-alias.sh"
fi

# Check if ffmpeg is installed
ffmpeg_test=$(ffmpeg --help 2>&1)
error_ffmpeg=$?
if [ ${error_ffmpeg} -ne 0 ] ; then
  echo ""
  echo "* ks-mp3-album (ks-tools) v${VERSION} (${M_DATE})"
  echo ""
  echo "+ The 'ffmpeg' tool is not installed!"
  echo ""
  exit
fi

# Funcion to show the name of file/folder from full path
# Syntax: extractFolderOrFile <full/path/file.txt>
function extractFolderOrFile() {
  pathToExtract="${1}/"
  findFolder=0
  count=1
  nameFolder=$(echo ${pathToExtract} | cut -d "/" -f ${count})
  count=$(expr $count + 1)
  while [ ${findFolder} -eq 0 ] ; do
    nameFolderTemp=$(echo ${pathToExtract} | cut -d "/" -f ${count})
    if [ -z "${nameFolderTemp}" ] ; then
      findFolder=1
    else
      nameFolder="${nameFolderTemp}"
      count=$(expr $count + 1)
    fi
  done
  echo "${nameFolder}"
}

# Function to show files with spaces.
# Syntax: showFileWithSpace <file number>
function showFileWithSpace() {
  echo "${1}" > ${dirTemp}/name.tmp
  sed -i 's/_/ /g' ${dirTemp}/name.tmp
  DisplayName=$(cat ${dirTemp}/name.tmp)
  rm -rf ${dirTemp}/name.tmp
  cutDisplayName=$(echo ${DisplayName} | cut -c1)
  if [ "${cutDisplayName}" == "-" ] ; then
    DisplayName=$(echo ${DisplayName} | cut -c2-100)
  else
    cutDisplayName=$(echo ${DisplayName} | cut -c2)
    if [ "${cutDisplayName}" == "-" ] ; then
      DisplayName=$(echo ${DisplayName} | cut -c3-100)
    else
      cutDisplayName=$(echo ${DisplayName} | cut -c3)
      if [ "${cutDisplayName}" == "-" ] ; then
        DisplayName=$(echo ${DisplayName} | cut -c4-100)
      fi
    fi
  fi
  echo ${DisplayName}
}

# Function to remove extension from file
# Syntax: removeExtension "<text>"
function removeExtension() {
  wordToConvert=${1}
  ksToolsSedFile="${ksToolsTempFolder}/ks-tools-${RANDOM}.txt"
  mkdir -p ${ksToolsTempFolder} && chmod 777 -R ${ksToolsTempFolder} 2> /dev/null
  echo "${wordToConvert}" > ${ksToolsSedFile}
  # Remove extensions
  sed -i 's/.avi//g' "${ksToolsSedFile}" &> /dev/null
  sed -i 's/.mp4//g' "${ksToolsSedFile}" &> /dev/null
  sed -i 's/.mkv//g' "${ksToolsSedFile}" &> /dev/null
  sed -i "s/.mov//g" "${ksToolsSedFile}" &> /dev/null
  sed -i 's/.vob//g' "${ksToolsSedFile}" &> /dev/null
  sed -i 's/.mpg//g' "${ksToolsSedFile}" &> /dev/null
  sed -i 's/.mpeg//g' "${ksToolsSedFile}" &> /dev/null
  sed -i 's/.wmv//g' "${ksToolsSedFile}" &> /dev/null
  sed -i 's/.ogv//g' "${ksToolsSedFile}" &> /dev/null
  sed -i 's/.webm//g' "${ksToolsSedFile}" &> /dev/null
  sed -i 's/.flv//g' "${ksToolsSedFile}" &> /dev/null
  sed -i 's/.mp3//g' "${ksToolsSedFile}" &> /dev/null
  sed -i 's/.mp2//g' "${ksToolsSedFile}" &> /dev/null
  sed -i 's/.oga//g' "${ksToolsSedFile}" &> /dev/null
  sed -i 's/.ogg//g' "${ksToolsSedFile}" &> /dev/null
  sed -i 's/.ac3//g' "${ksToolsSedFile}" &> /dev/null
  sed -i 's/.aac//g' "${ksToolsSedFile}" &> /dev/null
  sed -i 's/.flac//g' "${ksToolsSedFile}" &> /dev/null
  sed -i 's/.wav//g' "${ksToolsSedFile}" &> /dev/null
  sed -i 's/.opus//g' "${ksToolsSedFile}" &> /dev/null
  # Show file without extension
  wordToConvert=$(cat ${ksToolsSedFile})
  echo ${wordToConvert}
}

# Syntax: ks-mp3-album </absolute/path/album/
if [ -z "${1}" ] ; then
  echo ""
  echo "* ks-mp3-album (ks-tools) v${VERSION} (${M_DATE})"
  echo ""
  echo "- Convert folder album to MP3 Audio Format"
  echo ""
  echo "+ Config:"
  echo ""
  echo " - Audio codec: ${acodec}"
  echo " - Bitrate audio: ${b_acodec}"
  echo " - Container: ${v_ext}"
  echo ""
  echo "+ Syntax: "
  echo ""
  echo "  $ ks-mp3-album </absolute/path/album>"
  echo ""
  echo " + Example: ks-mp3-album /data/albums/matrix-soundtrack"
  echo ""
  exit
fi

# Check if folder exist
if [ -d "${1}" ] ; then
  echo null > /dev/null
else
  echo ""
  echo "* ks-mp3-album (ks-tools) v${VERSION} (${M_DATE})"
  echo ""
  echo "+ The folder '${1}' does not exist!"
  echo ""
  exit
fi

# Scan videos/audio files format
clear
echo ""
echo "* ks-mp3-album (ks-tools) v${VERSION} (${M_DATE})"
echo ""
echo "- Convert folder album to MP3 Audio Format"
echo ""
echo -n "* Scanning ${1} " && sleep 4
echo ""
rm -rf ${tempFile} && touch ${tempFile}
for format in ${formatFiles} ; do
  find "${1}"/*.${format} &> ${tempFileTest}
  if [ $? -ne 0 ] ; then
    echo "null" > /dev/null
  else
    echo "+ Video file(s) in .${format} found!"
    cd "${1}" && ls -1 *.${format} &>> ${tempFile}
    fail=0
  fi
done

# Set the global metadata album
current_date=$(date +%Y)
echo ""
echo "* Common METADATA configuration:"
echo ""
echo -n "* (Default: Album-Disc) Type name of album: " ; read album_file
if [ -z "${album_file}" ] ; then
  album_title="Album-Disc"
else
  album_title="${album_file}"
fi
echo -n "* (Default: ${current_date}) Type the year of album: " ; read year_file
if [ -z "${year_file}" ] ; then
  year_num="${current_date}"
else
  year_num="${year_file}"
fi
echo -n "* (Default: Pop) Type the genre of album: " ; read genre_file
if [ -z "${genre_file}" ] ; then
  genre_file="Pop"
else
  genre_file="${genre_file}"
fi

# Init the conversion files
convert_files=1
num_files=$(cat ${tempFile} | wc -l)
if [ ${num_files} -eq 0 ] ; then
  echo ""
  echo "+ No video/audio file(s) found in folder '${1}"
  echo ""
  exit
else
  echo ""
  echo "* List of files:"
  echo ""
  while [ ${convert_files} -le ${num_files} ] ; do
    show_file=$(cat ${tempFile} | head -${convert_files} | tail -1)
    echo "  + ${show_file}"
    convert_files=$(expr ${convert_files} + 1)
  done
  echo ""
  echo "* The output folder will be '${1}/to-mp3'"
  # Execute commands for conversion
  echo ""
  echo -n "* (Default: y) Do you want run the conversion? (y/n): " ; read run_commands_ffmpeg
  if [ "${run_commands_ffmpeg}" == "n" ] ; then
    exit
  else
    convert_files=1
    cd "${1}" && mkdir -p to-mp3 && rm -rf to-mp3/*
    while [ ${convert_files} -le ${num_files} ] ; do
      show_file=$(cat ${tempFile} | head -${convert_files} | tail -1)
      fname=$(echo "${show_file}" | cut -d "." -f 1)
      name_title_default=$(removeExtension "${show_file}")
      name_title_default=$(showFileWithSpace "${name_title_default}")
      echo ""
      echo "* METADATA for '${show_file}'"
      echo ""
      echo " - Album: ${album_title}"
      echo " - Year: ${year_num}"
      echo " - Genre: ${genre_file}"
      echo ""
      echo "* Information of '${show_file}':"
      echo ""
      echo "+ Audio Tracks:"
      echo ""
      ffmpeg -i "${show_file}" 2>&1 | grep Stream | tr -s " " | grep "Audio:" | cut -d "," -f 1
      echo ""
      # Check the audio track by default
      audio_default=$(ffmpeg -i "${show_file}" 2>&1 | grep Stream | tr -s " " | grep "Audio:" | grep "(${default_lang_audio})" | cut -d " " -f 3 | cut -c2-5 | cut -d "(" -f 1 | cut -d "[" -f 1 | head -1)
      if [ -z "${audio_default}" ] ; then
        audio_default=$(ffmpeg -i "${show_file}" 2>&1 | grep Stream | tr -s " " | grep "Audio:" | cut -d " " -f 3 | cut -c2-5 | cut -d "(" -f 1 | cut -d "[" -f 1 | head -1)
        if [ -z "${audio_default}" ] ; then
          audio_default="0:1"
        else
          audio_default_patch=$(echo ${audio_default} | cut -c4)
          if [ "${audio_default_patch}" == ":" ] ; then
            audio_default=$(echo ${audio_default} | cut -c1-3)
          else
            audio_default="${audio_default}"
          fi
        fi
      else
        audio_default_patch=$(echo ${audio_default} | cut -c4)
        if [ "${audio_default_patch}" == ":" ] ; then
          audio_default=$(echo ${audio_default} | cut -c1-3)
        else
          audio_default="${audio_default}"
        fi
      fi
      # Ask for audio
      echo -n "* (Default: ${audio_default}) Type the number of audio track: " ; read audio_track
      if [ -z "${audio_track}" ] ; then
        audio_track="${audio_default}"
      else
        audio_track="${audio_track}"
      fi
      echo -n "* (Default: ${name_title_default}) Type name of title: " ; read name_title
      if [ -z "${name_title}" ] ; then
        name_title="${name_title_default}"
      else
        name_title="${name_title}"
      fi
      echo -n "* (Default: ${track_init}) Type number of track: " ; read track_file
      if [ -z "${track_file}" ] ; then
        track_num="${track_init}"
      else
        track_num="${track_file}"
      fi
      echo -n "* (Default: ${artist_init}) Type name of artist: " ; read artist_file
      if [ -z "${artist_file}" ] ; then
        artist_title="${artist_init}"
      else
        artist_title="${artist_file}"
        artist_init="${artist_file}"
      fi
      # Check audio 5.1 or 7.1
      audio_5_7_1=$(${p_ffmpeg} "${show_file}" 2>&1 | grep Stream | tr -s " " | grep "Audio:" | grep "${audio_track}" | grep -i "5.1")
      audio_5_7_1_2=$(${p_ffmpeg} "${show_file}" 2>&1 | grep Stream | tr -s " " | grep "Audio:" | grep "${audio_track}" | grep -i "7.1")
      audio_5_7_1="${audio_5_7_1}${audio_5_7_1_2}"
      if [ -z "${audio_5_7_1}" ] ; then
        stereo_params="-ac 2"
      else
        stereo_params="-ac 2 -clev 3dB -slev -6dB"
      fi
      ffmpeg -i "${show_file}" -map ${audio_track} -c:a ${acodec} -b:a ${b_acodec} ${stereo_params} -metadata title="${name_title}" -metadata date="${year_num}" -metadata genre="${genre_file}" -metadata album="${album_title}" -metadata artist="${artist_title}" -metadata track="${track_num}" "to-mp3/${fname}.${v_ext}"
      # Increment variables
      convert_files=$(expr ${convert_files} + 1)
      track_init=$(expr ${track_num} + 1)
    done
  fi
fi
