#!/usr/bin/env bash

set -e

# declare constants

COLOR_BLUE="\\e[34m"

COLOR_CYAN="\\e[96m"

COLOR_GREEN="\\e[32m"

COLOR_RED="\\e[31m"

COLOR_RESET="\\e[0m"

COLOR_YELLOW="\\e[93m"

HEADER_TEXT="${COLOR_YELLOW}=== Plexus v0.9.76 - https://github.com/wolveix/plexus ===$COLOR_RESET\\n${COLOR_RESET}"

OS=$(uname)

OS_DISTRO=""

function setup() {

# download config file if it doesn't already exist

if [ ! -f "$HOME"/.config/plexus/plexus.conf ]; then

mkdir -p "$HOME"/.config/plexus/

curl -O https://raw.githubusercontent.com/wolveix/plexus/main/plexus.conf 2>/dev/null

mv plexus.conf "$HOME"/.config/plexus/

printf "%bA config file could not be found. The default file has been downloaded.\\n%b" "$COLOR_GREEN" "$COLOR_RESET"

fi

# check config exists and contains necessary values

source "${HOME}/.config/plexus/plexus.conf"

required_values=("audio_codec" "convert_dir" "converted_dir" "ffmpeg_threads" "ffmpeg_preset" "force_overwrite" "list_file" "media_container" "media_dir" "video_codec" "video_library")

for value in "${required_values[@]}"; do

if [ -z "${!value}" ]; then

echo "$value"

printf "Your config file (%s/.config/plexus/plexus.conf) is missing required parameters.\\nMove or delete it and then run Plexus again.\\n" "$HOME"

exit 1

fi

done

# create the plexus temporary directory if it doesn't already exist

if [ ! -d "$HOME"/.plexus ]; then

mkdir -p "$HOME"/.plexus

fi

# determine OS distribution in-use

case "$OS" in

'FreeBSD' | 'Linux' | 'NetBSD' | 'OpenBSD')

if [ ! -f "/etc/os-release" ]; then

printf "\\nIt doesn't look like your distro is supported.\\nCreate an issue here: https://github.com/wolveix/plexus/issues/new\\n"

exit

else

OS_DISTRO=$(awk -F= '/^media_name/{print $2}' /etc/os-release)

fi

;;

'Darwin')

OS_DISTRO='macOS'

;;

*)

printf "\\nIt doesn't look like your distro is supported.\\nCreate an issue here: https://github.com/wolveix/plexus/issues/new\\n"

exit

;;

esac

# determine local package manager

case $OS_DISTRO in

'"Alpine Linux"')

packages="apk"

;;

'"Arch Linux"' | '"ArcoLinuxD"' | '"Manjaro Linux"')

packages="yes | LC_ALL=en_US.UTF-8 pacman"

;;

'"CentOS Linux"')

packages="yum -q -y"

;;

'"Debian GNU/Linux"' | '"Linux Mint"' | '"Raspbian GNU/Linux"' | '"Ubuntu"')

packages="apt-get -qq -y"

;;

'Fedora')

packages="dnf -q -y"

;;

'macOS')

packages="$(command -v brew)"

if [ ! "$(command -v brew)" ]; then

printf "\\nIt doesn't look like you have brew installed.\\nFind out more here: https://brew.sh/.\\n"

exit 1

fi

;;

*)

printf "\\nIt doesn't look like your distro is supported.\\nCreate an issue here: https://github.com/wolveix/plexus/issues/new\\n"

exit 1

;;

esac

# declare local function to reduce code repetition when configuring binary paths below

function set_binary_path() {

local binary=$1

local binary_path=$(command -v "$binary")

local config_field="${binary}_binary"

if [ -z "$binary_path" ] || [ ! -f "$binary_path" ]; then

eval $config_field=""

else

eval $config_field="$binary_path"

set_config "${config_field}=\"\"" "${config_field}=\"$binary_path\""

fi

}

# configure binary paths

set_binary_path "ffmpeg"

set_binary_path "ffprobe"

set_binary_path "fusermount"

set_binary_path "rclone"

set_binary_path "rsync"

}

function set_config() {

sed -ie "s/$(echo "$1" | sed 's_/_\\/_g')/$(echo "$2" | sed 's_/_\\/_g')/g" "$HOME"/.config/plexus/plexus.conf

}

function check_variable {

case $1 in

"audio")

case ${2,,} in

"" | "default" | "aac")

audio_codec="aac"

;;

"ac3" | "ac-3" | "atsc")

audio_codec="ac3"

;;

"dca" | "dts")

audio_codec="dts"

;;

"flac")

audio_codec="flac"

;;

"m4b")

audio_codec="m4b"

;;

"mp3" | "mpeg3")

audio_codec="mp3"

;;

"opus")

audio_codec="opus"

;;

*)

audio_codec="null"

;;

esac

;;

"container")

case ${2,,} in

"avi")

media_container="avi"

;;

"matroska" | "mkv")

media_container="mkv"

;;

"m4b")

media_container="m4b"

;;

"mp3" | "mpeg3")

media_container="mp3"

;;

"mp4" | "mpeg4")

media_container="mp4"

;;

"wmv")

media_container="wmv"

;;

*)

media_container="null"

;;

esac

;;

"ffmpeg_preset")

case ${2,,} in

"ultrafast" | "superfast" | "faster" | "fast" | "medium" | "slow" | "slower" | "veryslow" | "placebo")

ffmpeg_preset="$2"

;;

"" | "default" | "veryfast")

ffmpeg_preset="veryfast"

;;

*)

ffmpeg_preset="null"

;;

esac

;;

"hardware")

case ${2,,} in

"cuvid")

hardware_codec="cuvid"

;;

"" | "default" | "opencl")

hardware_codec="opencl"

;;

"qsv")

hardware_codec="qsv"

;;

"nvdec")

hardware_codec="nvdec"

;;

*)

hardware_codec="null"

;;

esac

;;

"video")

case ${2,,} in

"" | "default" | "x264" | "h.264" | "h264" | "avc" | "libx264")

video_codec="h264"

video_library="libx264"

;;

"x265" | "h.265" | "h265" | "hevc" | "libx265")

video_codec="h265"

video_library="libx265"

;;

"avi" | "divx" | "libxvid" | "mpeg4" | "mpeg-4" | "xvid")

video_codec="mpeg4"

video_library="libxvid"

;;

*)

video_codec="null"

video_library="null"

;;

esac

;;

*)

printf "%b\\nAn unexpected error has occurred.%b\\n" "$COLOR_RED" "$COLOR_RESET"

exit 1

;;

esac

}

function prompt_user {

answer=""

while [ -z "$answer" ]; do

read -r -p '' answer

if [ "$answer" == "exit" ] || [ "$answer" == "0" ]; then

exit 0

fi

case "$1" in

"directory")

if [ ! -d "$answer" ]; then

printf "\\nYou must enter a valid directory.\\n"

answer=""

else

if [ "${answer: -1}" == "/" ]; then

answer="${answer::-1}"

fi

fi

;;

"file")

if [ ! -f "$answer" ]; then

printf "\\nYou must enter a valid file.\\n"

answer=""

fi

;;

"audio"|"container"|"ffmpeg"|"hardware"|"video")

check_variable "$1" "$answer"

if [ "${1}_codec" == "null" ]; then

printf "\\nYou must enter a valid recognized value."

answer=""

fi

;;

*)

# Do nothing

;;

esac

done

}

function func_blacklist {

if grep -Fxq "$file" "$blacklist_file"; then

printf "%bFile already exists in the blacklist.\\n" "$COLOR_BLUE"

else

if [ -f "$file" ]; then

echo "$file" >> "$blacklist_file"

else

printf "%bThe file doesn't exist.\\n" "$COLOR_BLUE"

fi

fi

}

function func_config {

printf "\\nWhat config menu would you like to access?\\n"

select option in "Binaries (FFprobe, FFmpeg, Rclone)" "Codecs (Audio, Container, Video)" "Directories (Convert, Converted, Media, Mount)" "FFmpeg (CPU, Force Overwrite, Deinterlacing, Hardware Acceleration, Preset)" "Exit"; do

case "$option" in

"Binaries (FFprobe, FFmpeg, Rclone)")

printf "\\nWhich binary do you want to set?\\n"

select option in "FFprobe" "FFmpeg" "Rclone" "Exit"; do

case "$option" in

"FFprobe")

printf "\\n\\nWhat do you want to set the FFprobe binary to?\\n"

prompt_user "file"

set_config "ffprobe_binary=\"$ffprobe_binary\"" "ffprobe_binary=\"$answer\""

printf "\\nFFprobe binary successfully changed.\\n"

exit 0

;;

"FFmpeg")

printf "\\n\\nWhat do you want to set the FFmpeg binary to?\\n"

prompt_user "file"

set_config "ffmpeg_binary=\"$ffmpeg_binary\"" "ffmpeg_binary=\"$answer\""

printf "\\nFFmpeg binary successfully changed.\\n"

exit 0

;;

"Rclone")

printf "\\n\\nWhat do you want to set the Rclone binary to?\\n"

prompt_user "file"

set_config "rclone_binary=\"$rclone_binary\"" "rclone_binary=\"$answer\""

printf "\\nRclone binary successfully changed.\\n"

exit 0

;;

"Exit")

printf "\\nYou can find the config file here: %s/.config/plexus/plexus.conf\\n" "$HOME"

exit 0

;;

esac

done

;;

"Codecs (Audio, Container, Video)")

printf "\\nWhich codec do you want to set?\\n"

select option in "Audio" "Container" "Video" "Exit"; do

case "$option" in

"Audio")

printf "\\n\\nWhat do you want to set the audio codec to?\\n"

var=$audio_codec

prompt_user "audio"

set_config "audio_codec=\"$var\"" "audio_codec=\"$audio_codec\""

printf "\\nAudio codec successfully changed.\\n"

exit 0

;;

"Container")

printf "\\n\\nWhat do you want to set the container to?\\n"

var=$media_container

prompt_user "container"

set_config "media_container=\"$var\"" "media_container=\"$media_container\""

printf "\\nMedia container successfully changed.\\n"

exit 0

;;

"Video")

printf "\\n\\nWhat do you want to set the video codec to?\\n"

var=$video_codec

var2=$video_library

prompt_user "video"

set_config "video_codec=\"$var\"" "video_codec=\"$video_codec\""

set_config "video_library=\"$var2\"" "video_library=\"$video_library\""

printf "\\nVideo codec successfully changed.\\n"

exit 0

;;

"Exit")

printf "\\nYou can find the config file here: %s/.config/plexus/plexus.conf\\n" "$HOME"

exit 0

;;

esac

done

;;

"Directories (Convert, Converted, Media, Mount)")

printf "\\nWhich directory do you want to set?\\n"

select option in "Convert" "Converted" "Media" "Mount" "Exit"; do

case "$option" in

"Convert")

printf "\\n\\nWhat do you want to set the convert directory to?\\n"

prompt_user "directory"

set_config "convert_dir=\"$convert_dir\"" "convert_dir=\"$answer\""

printf "\\nConvert directory successfully changed.\\n"

exit 0

;;

"Converted")

printf "\\n\\nWhat do you want to set the converted directory to?\\n"

prompt_user "directory"

set_config "converted_dir=\"$converted_dir\"" "converted_dir=\"$answer\""

printf "\\nConverted directory successfully changed.\\n"

exit 0

;;

"Media")

printf "\\n\\nWhat do you want to set the media directory to?\\n"

prompt_user "directory"

set_config "media_dir=\"$media_dir\"" "media_dir=\"$answer\""

printf "\\nMedia directory successfully changed.\\n"

exit 0

;;

"Mount")

printf "\\n\\nWhat do you want to set the mount directory to?\\n"

prompt_user "directory"

set_config "mount_dir=\"$mount_dir\"" "mount_dir=\"$answer\""

printf "\\nMount directory successfully changed.\\n"

exit 0

;;

"Exit")

printf "\\nYou can find the config file here: %s/.config/plexus/plexus.conf\\n" "$HOME"

exit 0

;;

esac

done

;;

"FFmpeg (CPU, Force Overwrite, Deinterlacing, Hardware Acceleration, Preset)")

printf "\\nWhich option do you want to set?\\n"

select option in "CPU" "Force Overwrite" "Deinterlacing" "Hardware Acceleration" "Preset" "Exit"; do

case "$option" in

"CPU")

cpu_threads=$(grep -c processor /proc/cpuinfo)

printf "\\n\\nEnter a number between 1 - :\\n"

while [ "$answer" -eq 0 ] || [ "$answer" -gt "$cpu_threads" ]; do

prompt_user

done

set_config "ffmpeg_threads=\"$ffmpeg_threads\"" "ffmpeg_threads=\"$answer\""

printf "\\nFFmpeg thread allocation successfully set.\\n"

exit 0

;;

"Force Overwrite")

printf "\\n\\nEnter true or false: "

lock=0

while [ "$lock" -eq 0 ]; do

prompt_user "truth"

done

set_config "force_overwrite=\"$force_overwrite\"" "force_overwrite=\"$answer\""

printf "\\nForce overwrite successfully set.\\n"

exit 0

;;

"Deinterlacing")

printf "\\n\\nEnter true or false: "

lock=0

while [ "$lock" -eq 0 ]; do

prompt_user "truth"

done

set_config "deinterlacing=\"$deinterlacing\"" "deinterlacing=\"$answer\""

printf "\\nDeinterlacing successfully set.\\n"

exit 0

;;

"Hardware Acceleration")

printf "\\nWhich option do you want to set?\\n"

select option in "Codec" "Enabled" "Exit"; do

case "$option" in

"Codec")

printf "\\n\\nWhat do you want to set the hardware codec to?\\n"

var=$hardware_codec

prompt_user "hardware"

set_config "hardware_codec=\"$var\"" "hardware_codec=\"$hardware_codec\""

printf "\\nHardware codec successfully changed.\\n"

exit 0

;;

"Enabled")

printf "\\n\\nEnter true or false: "

lock=0

while [ "$lock" -eq 0 ]; do

prompt_user "truth"

done

set_config "hardware_acceleration=\"$hardware_acceleration\"" "hardware_acceleration=\"$answer\""

printf "\\nHardware Acceleration successfully set.\\n"

exit 0

;;

"Exit")

printf "\\nYou can find the config file here: %s/.config/plexus/plexus.conf\\n" "$HOME"

exit 0

;;

esac

done

;;

"Preset")

printf "\\n\\nEnter the desired FFmpeg preset: \\n"

var=$ffmpeg_preset

prompt_user "ffmpeg"

set_config "ffmpeg_preset=\"$var\"" "ffmpeg_preset=\"$ffmpeg_preset\""

printf "\\nFFmpeg preset successfully set.\\n"

exit 0

;;

"Exit")

printf "\\nYou can find the config file here: %s/.config/plexus/plexus.conf\\n" "$HOME"

exit 0

;;

esac

done

;;

"Exit")

printf "\\nYou can find the config file here: %s/.config/plexus/plexus.conf\\n" "$HOME"

exit 0

;;

esac

done

}

function func_encode {

ffmpeg_verbose=""

if [ "$verbose" != "true" ]; then

ffmpeg_verbose="-loglevel quiet"

fi

if [ -n "$rclone_bwlimit" ]; then

if ! [[ "$rclone_bwlimit" =~ ^[0-9]+$ ]]; then

printf "You have not entered a valid number for --bwlimit (do not include KB).\\n"

exit 1

fi

rclone_bwlimit="--bwlimit $rclone_bwlimit"

fi

if [ ! -e "$list_file" ]; then

printf "List file does not exist.\\nPlease manually specify a list file or run the list command to create one.\\n"

exit 1

fi

printf "\\n%bAudio codec: %s\\nList file: %s\\nMedia container: %s\\nMedia directory: %s\\nVideo codec: %s\\n\\n%b" "$COLOR_GREEN" "$audio_codec" "$list_file" "$media_container" "$media_dir" "$video_codec" "$COLOR_RESET"

while read -r line; do

if [ -n "$line" ]; then

DIRP=${line%/*}

EXT=${line##*.}

raw_file=${line##*/}

media_name=${raw_file%.*}

if [ -z "$media_container" ]; then

media_container=$EXT;

fi

if [ -d "$convert_dir" ] || [ -d "$converted_dir" ]; then

/bin/rm -rf "$convert_dir" "$converted_dir"

fi

/bin/mkdir -p "$convert_dir/$media_name/" "$converted_dir"

if [ "$remote_transfer" = true ] && [ -n "$rclone_remote" ]; then

if [ -z "$rclone_remote" ]; then

printf "%bPlease install Rclone to use an Rclone remote with this command%b\\n" "$COLOR_RED" "$COLOR_RESET"

exit 1

fi

printf "Downloading: %s\\n" "$raw_file"

if ! $rclone_binary copy "$rclone_remote$line" "$convert_dir/$media_name/" "$rclone_bwlimit" --stats-one-line -P; then

printf "\\n\e[31mError: File could not be downloaded.\\n\\n\e[0mMoving onto the next line.\\n"

/bin/sed -i 1d "$list_file"

echo "$line" >> "$list_file"

file_exists=0

else

printf "\\nFile downloaded.\\n"

file_exists=1

fi

else

file_exists=0

if [ -f "$line" ]; then

printf "%bCopying: %s%b\\n" "$COLOR_BLUE" "$raw_file" "$COLOR_RESET"

ln -sf "$line" "$convert_dir/$media_name/"

printf "\\n%bFile copied.%b\\n" "$COLOR_BLUE" "$COLOR_RESET"

file_exists=1

fi

fi

file_name="$media_name"

if [ "$force_overwrite" != true ]; then

file_name="$media_name [Plexus Encode]"

/bin/mv "$convert_dir/$media_name/$media_name.$EXT" "$convert_dir/$media_name/$file_name.$EXT"

fi

if [ $file_exists -eq 1 ]; then

if [ ! -d "$converted_dir/$media_name/" ]; then

/bin/mkdir "$converted_dir/$media_name/"

fi

if [ -f "$converted_dir/$media_name/$raw_file" ]; then

/bin/rm "$converted_dir/$media_name/$raw_file"

fi

exclude_stream=""

file_audio_codec=$("$ffprobe_binary" -v error -select_streams a:0 -show_entries stream=codec_name -of default=noprint_wrappers=1:nokey=1 "$convert_dir/$media_name/$file_name.$EXT")

file_video_codec=$("$ffprobe_binary" -v error -select_streams v:0 -show_entries stream=codec_name -of default=noprint_wrappers=1:nokey=1 "$convert_dir/$media_name/$file_name.$EXT")

stream_info=$("$ffprobe_binary" -select_streams s -show_entries stream=index,codec_name -of csv=p=0 "$convert_dir/$media_name/$file_name.$EXT" 2>&1)

stream_count=$(echo "$stream_info" | grep -cE 'Subtitle: dvd_subtitle|Subtitle: hdmv_pgs' || :)

if [ "$stream_count" -gt 0 ]; then

stream_id=$(echo "$stream_info" | grep -E 'Subtitle: dvd_subtitle|Subtitle: hdmv_pgs' || :)

counter=0

fail_counter=0

printf "%bExtracting subtitles...\\n%b" "$COLOR_BLUE" "$COLOR_RESET"

until [ "$counter" = "$stream_count" ]; do

counter=$((counter+1))

excluded_stream="$(echo "$stream_id" | grep -oP '0:[0-9]{1,3}' | sed -n "${counter}"p)"

subtitle_language="$(echo "$stream_id" | grep -oP '\([a-z]{3}\)' | sed -n "${counter}"p | sed -e 's/(//' -e 's/)//')"

if [ -n "$excluded_stream" ]; then

if [ "$exclude_stream" = "*$excluded_stream*" ]; then

counter="$stream_count"

else

exclude_stream="$exclude_stream -map -$excluded_stream"

if ! $ffmpeg_binary -y -i "$convert_dir/$media_name/$file_name.$EXT" -nostdin $ffmpeg_verbose -stats -map $excluded_stream -c copy "$converted_dir/$media_name/$file_name.$counter.$subtitle_language.srt"; then

if [ -f "$converted_dir/$media_name/$file_name.$counter.$subtitle_language.srt" ]; then

/bin/rm "$converted_dir/$media_name/$file_name.$counter.$subtitle_language.srt"

fi

if ! $ffmpeg_binary -y -i "$convert_dir/$media_name/$file_name.$EXT" -nostdin $ffmpeg_verbose -stats -map $excluded_stream -c copy "$converted_dir/$media_name/$file_name.$counter.$subtitle_language.sup"; then

if [ -f "$converted_dir/$media_name/$file_name.$counter.$subtitle_language.sup" ]; then

/bin/rm "$converted_dir/$media_name/$file_name.$counter.$subtitle_language.sup"

fi

fail_counter=$((fail_counter+1))

else

/bin/mv "$converted_dir/$media_name/$file_name.$counter.$subtitle_language.sup" "$converted_dir/$media_name/$file_name.$counter.$subtitle_language.srt"

fi

fi

fi

fi

excluded_stream=""

done

if [ "$fail_counter" -gt 0 ]; then

printf "%b%s subtitle stream(s) could not be extracted.\\n%b" "$COLOR_RED" "$fail_counter" "$COLOR_RESET"

else

printf "%bSubtitles extracted successfully.\\n%b" "$COLOR_BLUE" "$COLOR_RESET"

fi

fi

printf "\\n%bFile codecs:\\nAudio = %s -> %s\\nContainer = %s -> %s\\nVideo = %s -> %s%b\\n\\n" "$COLOR_YELLOW" "$file_audio_codec" "$audio_codec" "$EXT" "$media_container" "$file_video_codec" "$video_codec" "$COLOR_RESET"

if [ "$audio_codec" = "aac" ]; then

if [ -n "$($ffmpeg_binary -encoders | grep -i libfdk)" ]; then

audio_codec="libfdk_aac"

ffmpeg_error="\\nAn unknown error occurred with FFmpeg.\\n"

else

ffmpeg_error="\\nAn unknown error occurred with FFmpeg.\\nConsider installing the custom build of FFmpeg via the install command.\\n"

fi

fi

codec_copy=false

codecs=""

options="-map 0 $exclude_stream -preset $ffmpeg_preset -max_muxing_queue_size 1024 -movflags faststart -threads $ffmpeg_threads"

if [ -z "$audio_codec" ] || [ "$file_audio_codec" == "$audio_codec" ]; then

codec_copy=true

codecs="$codecs -c:a copy"

else

codecs="$codecs -c:a $audio_codec"

fi

hwaccel=""

if [ "$hardware_acceleration" = "true" ]; then

hwaccel="-hwaccel $hardware_codec"

fi

if [ -z "$video_codec" ] || [ "$file_video_codec" == "$video_codec" ]; then

codec_copy=true

codecs="$codecs -c:v copy"

else

codecs="$codecs -c:v $video_library"

fi

if [ "$video_codec" == "h264" ]; then

options="$options -crf 20 -level 4.1"

fi

if [ "$deinterlacing" != "false" ]; then

options="$options -vf yadif"

fi

encode_success=0

if ! $ffmpeg_binary $hwaccel -y -i "$convert_dir/$media_name/$file_name.$EXT" $options $codecs -nostdin $ffmpeg_verbose -stats "$converted_dir/$media_name/$file_name.$media_container"; then

if [ "$codec_copy" = true ]; then

printf "\\n\\nAn unknown error occurred with FFmpeg.\\nTrying with full conversion instead.\\n\\n"

if [ -n "$audio_codec" ]; then

encode_audio_codec="$audio_codec"

fi

if ! $ffmpeg_binary $hwaccel -y -i "$convert_dir/$media_name/$file_name.$EXT" $options -c:v $video_library -c:a $encode_audio_codec -nostdin $ffmpeg_verbose -stats "$converted_dir/$media_name/$file_name.$media_container"

then

printf "$ffmpeg_error"

encode_success=0

else

encode_success=1

fi

else

printf "$ffmpeg_error"

encode_success=0

fi

else

encode_success=1

fi

if [ $encode_success -eq 1 ]; then

if [ -f "$converted_dir/$media_name/$file_name.$media_container" ]; then

printf "\\n%bFile successfully converted.%b" "COLOR_GREEN" "$COLOR_RESET"

if [ -n "$rclone_remote" ]; then

if [ "$force_overwrite" = true ]; then

printf "\\n%bDeleting original file from Rclone remote.\\n%b" "$COLOR_BLUE" "$COLOR_RESET"

$rclone_binary delete "$rclone_remote$line" --stats-one-line -P

printf "%bOriginal file deleted.%b" "$COLOR_BLUE" "$COLOR_RESET"

fi

printf "\\n%bUploading converted file to Rclone remote.\\n%b" "$COLOR_BLUE" "$COLOR_RESET"

$rclone_binary move "$converted_dir/$media_name/" "$rclone_remote$DIRP/" "$rclone_bwlimit" --include "$media_name.*" --stats-one-line -P

printf "%bFile successfully uploaded.\\n%b" "$COLOR_BLUE" "$COLOR_RESET"

else

if [ "$force_overwrite" = true ]; then

printf "\\n%bDeleting original file.%b" "$COLOR_BLUE" "$COLOR_RESET"

/bin/rm "$line"

printf "\\n%bOriginal file deleted.%b" "$COLOR_BLUE" "$COLOR_RESET"

fi

printf "\\n%bMoving converted file to the correct location.%b\\n" "$COLOR_BLUE" "$COLOR_RESET"

$rsync_binary -ah --info=progress2 --remove-source-files "$converted_dir/$media_name/" "$DIRP"

printf "\\n%bFile successfully moved.%b" "$COLOR_BLUE" "$COLOR_RESET"

fi

printf "\\n%bDeleting cached file and delisting it." "$COLOR_BLUE"

/bin/rm -r "${convert_dir:?}/$media_name"

/bin/sed -i 1d "$list_file"

printf "\\n%bOperation successfully completed.%b\\n\\n" "$COLOR_GREEN" "$COLOR_RESET"

else

printf "\\nFile codecs are already correct. Delisting it.\\n"

/bin/sed -i 1d "$list_file"

printf "\\n%bOperation successfully completed.%b\\n\\n" "$COLOR_GREEN" "$COLOR_RESET"

fi

else

printf "\\n%bError: %s could not be encoded.\\n\\n%bMoving onto the next line.\\n\\n" "$COLOR_RED" "$line" "$COLOR_RESET"

if [ -d "$convert_dir" ] || [ -d "$converted_dir" ]; then

/bin/rm -rf "$convert_dir" "$converted_dir"

fi

/bin/sed -i 1d "$list_file"

echo "$line" >> "$list_file"

fi

else

printf "\\n%bError: %s does not exist.\\n\\n$bMoving onto the next line.\\n\\n" "$COLOR_RED" "$line" "$COLOR_RESET"

/bin/sed -i 1d "$list_file"

echo "$line" >> "$list_file"

fi

fi

done < "$list_file"

}

function func_help {

case "$(echo "$1" | tr '[:upper:]' '[:lower:]')" in

"about")

printf "This command tells you more about it's creation.\\n"

;;

"encode")

printf "\n%bAbout:\\n The encode command processes a list generated from the list\\n command and converts each listed file into two proposed codecs\\n (set to H264 and AAC by default)\\n\\nUsage:\\n plexus encode -l /binary_path/to/list.txt [flags]\\n plexus encode -l /binary_path/to/list.txt -r RcloneRemote: [flags]\\n\\nFlags:\\n -a Audio codec. Default = %s\\n This allows you to set a preferred audio codec\\n on runtime, rather than setting a default via\\n the config function\\n\\n -f Force overwrite. Default = %s\\n The newly encoded file will overwrite the\\n existing file\\n\\n -l List location. Default = %s\\n This is where your previously generated list file\\n is\\n\\n -p FFMpeg preset. Default = %s\\n This is the FFmpeg preset that will be used when\\n encoding your media\\n\\n -r Rclone remote. Use this for Rclone integration\\n Don't use this if you're converting local media.\\n This points Plexus to your pre-configured Rclone\\n remote\\n\\n -v Video codec. Default = %s\\n This allows you to set a preferred video codec\\n on runtime, rather than setting a default via\\n the config function\\n\\n --bwlimit Limit Rclone's active connection speed\\n To prevent using too much bandwidth, you can\\n set an active limit in KB/s\\n\\n --verbose Show detailed log\\n Shows you a more detailed log of what's happening.\\n Use this when experiencing issues\\n" "$COLOR_RESET" "$audio_codec" "$force_overwrite" "$list_file" "$ffmpeg_preset" "$video_codec"

;;

*)

printf "\\n%bUsage:\\n plexus [flags]\\n plexus [command]\\n\\nAvailable Commands:\\n about Learn more about the program\\n blacklist Add an entry to the blacklist\\n config Change the default variable values\\n encode Begin processing the encode queue\\n help Displays a list of available commands\\n install Reinstall Plexus or install any missing dependencies\\n list Build a .txt file containing media with incorrect codecs\\n mount Mount an Rclone remote or cache, set from the config function\\n unmount Unmount an Rclone remote or cache, set from the config function\\n\\nAdditional Information:\\n Use 'plexus help command' to find out more about a specific command\\n" "$COLOR_RESET"

;;

esac

}

function func_install {

printf "\\nWhat would you like to do?\\n"

select option in "Install missing dependencies" "Update Plexus" "Exit"

do

case "$option" in

"Install missing dependencies")

printf "\\nWhich dependency?\\n"

select option in "All" "FFmpeg" "FFmpeg (custom)" "Fuse" "Rclone" "Exit"

do

case "$option" in

"All")

printf "\\n%bInstalling all dependencies.%b\\n\\n" "$COLOR_CYAN" "$COLOR_RESET"

if [ "$OS_DISTRO" == '"Alpine Linux"' ]; then

$packages add curl ffmpeg fuse rsync unzip

else

$packages install curl ffmpeg fuse rsync unzip

fi

curl https://rclone.org/install.sh 2>/dev/null | sudo bash

;;

"FFmpeg")

printf "\\n%bInstalling FFmpeg.\\n\\n%b" "$COLOR_CYAN" "$COLOR_RESET"

if [ "$OS_DISTRO" == '"Alpine Linux"' ]; then

$packages add ffmpeg

elif [ "$OS_DISTRO" == '"CentOS Linux"' ]; then

if [ "$(rpm -qi ffmpeg)" = "package ffmpeg is not installed" ]; then

if [ "$(rpm --eval '%{centos_ver}')" -eq "8" ]; then

$packages install https://dl.fedoraproject.org/pub/epel/epel-release-latest-8.noarch.rpm

$packages install https://download1.rpmfusion.org/free/el/rpmfusion-free-release-8.noarch.rpm https://download1.rpmfusion.org/nonfree/el/rpmfusion-nonfree-release-8.noarch.rpm

$packages install http://mirror.leaseweb.com/centos/8/PowerTools/x86_64/os/Packages/SDL2-2.0.10-2.el8.x86_64.rpm

else

$packages install epel-release

$packages update && yum -q -y upgrade

$packages localinstall --nogpgcheck https://download1.rpmfusion.org/free/el/rpmfusion-free-release-7.noarch.rpm https://download1.rpmfusion.org/nonfree/el/rpmfusion-nonfree-release-7.noarch.rpm

fi

fi

$packages update

$packages install ffmpeg ffmpeg-devel

elif [ "$OS_DISTRO" == 'Fedora' ]; then

$packages install "https://download1.rpmfusion.org/free/fedora/rpmfusion-free-release-$(rpm -E %fedora).noarch.rpm"

$packages install "https://download1.rpmfusion.org/nonfree/fedora/rpmfusion-nonfree-release-$(rpm -E %fedora).noarch.rpm"

$packages install ffmpeg ffmpeg-devel

else

$packages install ffmpeg

fi

printf "%bFFmpeg has been installed!%b\\n" "$COLOR_GREEN" "$COLOR_RESET"

exit 0

;;

"FFmpeg (custom)")

printf "\\n%bInstalling FFmpeg (custom).\\nThis will take a few minutes as it's built from source with support for extended codecs\\n\\n%b" "$COLOR_CYAN" "$COLOR_RESET"

if [ "$OS_DISTRO" == '"CentOS Linux"' ] || [ "$OS_DISTRO" == 'Fedora' ]; then

$packages install build-essential bzip2 curl gcc-c++ make perl which

mkdir -p "$HOME/ffmpeg-build/packages/"

curl -O ftp://ftp.videolan.org/pub/x264/snapshots/last_x264.tar.bz2

tar xvjf last_x264.tar.bz2 -C "$HOME/ffmpeg-build/packages/"

elif [ "$OS_DISTRO" == '"Alpine Linux"' ]; then

printf "\\n\\nBuilding FFmpeg from source is currently unsupported on Alpine.%b\\n" "$COLOR_RESET"

exit 0

else

$packages install build-essential curl g++ make perl

fi

curl https://raw.githubusercontent.com/markus-perl/ffmpeg-build-script/master/web-install.sh?v1 2>/dev/null | sudo bash

printf "\\n\\nMoving FFmpeg into /usr/bin.\\n"

sudo /bin/mv -f "${HOME}/ffmpeg-build/workspace/bin/ffmpeg" "/usr/bin"

printf "\\nRemoving FFmpeg build files.\\n"

sudo /bin/rm -r "${HOME}/ffmpeg-build"

printf "\\n%bFFmpeg (custom) has been installed!%b\\n" "$COLOR_GREEN" "$COLOR_RESET"

exit 0

;;

"Fuse")

printf "\\n%bInstalling Fuse.\\n\\n%b" "$COLOR_CYAN" "$COLOR_RESET"

if [ "$OS_DISTRO" == '"Alpine Linux"' ]; then

$packages add fuse

else

$packages install fuse

fi

printf "%bFuse has been installed!%b\\n" "$COLOR_GREEN" "$COLOR_RESET"

exit 0

;;

"Rclone")

printf "\\n%bInstalling Rclone.\\n\\n%b" "$COLOR_CYAN" "$COLOR_RESET"

if [ "$OS_DISTRO" == '"Alpine Linux"' ]; then

$packages add curl unzip

else

$packages install curl unzip

fi

curl https://rclone.org/install.sh 2>/dev/null | sudo bash

exit 0

;;

"Exit")

printf "\\nYou can find the config file here: %s/.config/plexus/plexus.conf\\n" "$HOME"

exit 0

;;

esac

done

;;

"Update Plexus")

config_file="$HOME/.config/plexus/plexus.conf.bak"

if [ -f "$config_file" ]; then

i=1

while [ -f "$config_file.$i" ]; do

((i++))

done

config_file="$config_file.$i"

fi

/bin/mv "$HOME/.config/plexus/plexus.conf" "$config_file"

printf "\\n\\nYour current config file has been backed up (%s).\\n" "$config_file"

curl https://plexus.wolveix.com/install.sh 2>/dev/null | sudo bash

exit 0

;;

"Exit")

printf "\\nYou can find the config file here: %s/.config/plexus/plexus.conf\\n" "$HOME"

exit 0

;;

esac

done

}

function func_list {

printf "\\n%bAudio codec: %s\\nList file: %s\\nMedia container: %s\\nMedia directory: %s\\nMount Path: %s\\nVideo codec: %s\\n\\n%b" "$COLOR_GREEN" "$audio_codec" "$list_file" "$media_container" "$media_dir" "$mount_dir" "$video_codec" "$COLOR_RESET"

if [ ! -d "$media_dir" ]; then

printf "%bThe specified media directory (%s) does not exist.\\n%b" "$COLOR_RED" "$media_dir" "$COLOR_RESET"

exit 1

fi

if [ -f "$list_file" ]; then

printf "\\n%bThis will delete the current list file. Do you want to continue?\\n%b" "$COLOR_YELLOW" "$COLOR_RESET"

select option in "Yes" "No"; do

case "$option" in

"Yes")

rm "$list_file"

printf "\\n"

break

;;

"No")

printf "\\nYou can find the current list file here: %s\\n" "$list_file"

exit 0

;;

esac

done

fi

printf "%bScanning directory...\\n" "$COLOR_GREEN"

while IFS= read -r line; do

DIRP=${line%/*}

EXT=${line##*.}

raw_file=${line##*/}

media_name=${raw_file%.*}

if [[ "$media_name" != *"[Plexus Encode]"* ]]; then

if [ -f "$line" ]; then

printf "%bScanning: %s\\n" "$COLOR_BLUE" "$line"

if grep -Fxq "$line" "$blacklist_file"; then

printf "%bFound in blacklist. Skipping.\\n" "$COLOR_BLUE"

else

if [ -n "$media_container" ] && [ "$EXT" != "$media_container" ]; then

printf "%bAdding: %s\\n" "$COLOR_BLUE" "$line"

line=${line#"$mount_dir"}

echo "$line" >> "$list_file"

else

file_audio_codec=$($ffprobe_binary -v error -select_streams a:0 -show_entries stream=codec_name -of default=noprint_wrappers=1:nokey=1 "$line")

file_video_codec=$($ffprobe_binary -v error -select_streams v:0 -show_entries stream=codec_name -of default=noprint_wrappers=1:nokey=1 "$line")

if [ -n "$file_audio_codec" ] || [ -n "$file_video_codec" ]; then

if [ "$file_audio_codec" != "$audio_codec" ] || [ "$file_video_codec" != "$video_codec" ]; then

printf "%bAdding: %s\\n" "$COLOR_BLUE" "$line"

line=${line#"$mount_dir"}

echo "$line" >> "$list_file"

fi

fi

fi

EXT=""

file_audio_codec=""

file_video_codec=""

fi

fi

else

printf "%bSkipping (already encoded): %s\\n" "$COLOR_BLUE" "$line"

fi

done < <(find "$media_dir" -name '*.avi' -or -name '*.flv' -or -name '*.m4b' -or -name '*.mkv' -or -name '*.mov' -or -name '*.mp3' -or -name '*.mp4' -or -name '*.ts' -or -name '*.mpg' -or -name '*.wmv')

printf "%bScan complete! Run plexus encode to process the list.\\n%b" "$COLOR_GREEN" "$COLOR_RESET"

}

function func_mount {

printf "%bMounting Rclone remote to %s\\n%b" "$COLOR_CYAN" "$media_dir" "$COLOR_RESET"

if ! $rclone_binary mount "$rclone_remote" "$media_dir" --allow-other --buffer-size 256M --dir-cache-time 72h --drive-chunk-size 32M --log-level INFO --log-file "$HOME"/.plexus/logs/rclone.log --timeout 1h --umask 002 --vfs-read-chunk-size 128M --vfs-read-chunk-size-limit off &

then

printf "\\nAn uknown error occurred (does the remote exist?)\\n%s" "$COLOR_RESET"

else

printf "\\n%bThe Rclone remote has successfully been mounted!\\n%b" "$COLOR_GREEN" "$COLOR_RESET"

fi

}

function func_unmount {

if [ ! -d "$media_dir" ]; then

printf "%bUnmounting Rclone remote from %s\\n%b" "$COLOR_CYAN" "$media_dir" "$COLOR_RESET"

fusermount -uz "$media_dir"

printf "\\n%bThe Rclone remote has successfully been unmounted!\\n%b" "$COLOR_GREEN" "$COLOR_RESET"

else

printf "%bThe directory does not exist.\\n%b" "$COLOR_RED" "$COLOR_RESET"

fi

}

function main() {

printf "%b" "$HEADER_TEXT"

# run setup function to initialize the config

setup

# if no command is specified, print the default help text

if [ -z "$1" ]; then

func_help

exit 0

fi

# fill command variables to keep things readable

command=$1

subcommand=$2

shift

# handle different commands

case "$command" in

about)

printf "\\nPlexus is a simple tool to mass re-encode your media collection.\\nIt compiles a list of your media files which do not currently\\ncontain your preferred codecs. It can then re-encode all of that media\\nfor you, even if the media is stored on remote storage (via Rclone).\\n\\nI created Plexus because I needed a simple CLI tool for\\nre-encoding all of my media, and every other solution ended\\nup being more complex than I needed them to be.\\n\\n- Robert Thomas\\n Lead Software Engineer at Level Zero Technology\\n https://github.com/wolveix\\n"

;;

blacklist)

if [ $OPTIND -eq 1 ]; then

printf "\\n%bUsage:\\n plexus blacklist \"/binary_path/to/file.ext\"\\n" "$COLOR_RESET"

exit 0

else

file="${subcommand,,}"

func_blacklist

fi

;;

config)

func_config

;;

encode)

while getopts ":a:c:fl:p:r:v:-:" opt; do

case $opt in

a)

if [ -z "${OPTARG,,}" ]; then

if [ -z "$media_container" ] && [ -z "$video_codec" ]; then

printf "\\n%bPlease specify at least one of the following: audio code, media container, video codec.\\n" "$COLOR_RED"

exit 1

fi

printf "\\n%bYou've specified no audio codec, enabling video-only mode" "$COLOR_YELLOW"

encode_audio_codec="$audio_codec"

audio_codec=""

else

check_variable audio "${OPTARG,,}"

if [ "$audio_codec" = "null" ]; then

printf "You have not entered a supported audio codec.\\n"

exit 1

fi

fi

;;

c)

if [ -z "${OPTARG,,}" ]; then

if [ -z "$audio_codec" ] && [ -z "$video_codec" ]; then

printf "\\n%bYou've specified no audio or video codec, or a media container. Please specify at least one\\n" "$COLOR_RED"