From a432811b344d8cda8e82bb66e0ed8679efb304a1 Mon Sep 17 00:00:00 2001
From: fruchti <fruchti@gvfr.de>
Date: Mon, 21 Oct 2019 13:39:19 +0200
Subject: [PATCH] Refactor, allow MP3s as input files

Adds dependency on exiftool
---
 transcode.sh | 252 ++++++++++++++++++++++++++++++++-------------------
 1 file changed, 161 insertions(+), 91 deletions(-)

diff --git a/transcode.sh b/transcode.sh
index f39f186..3846014 100644
--- a/transcode.sh
+++ b/transcode.sh
@@ -1,106 +1,176 @@
-#!/bin/sh
+#!/bin/bash
 
-output_owner=fruchti:data
-lockfile=/tmp/music-transcode.lock
+output_owner=
+input_directory=/data/music
 
-if [ -f "$lockfile" ] ; then
+tmp_directory=/data/tmp
+lock_file=/tmp/music-transcode.lock
+
+function die()
+{
+    printf '%s\n' "$1" >&2
+    exit "${2-1}"
+}
+
+if [ -f "$lock_file" ] ; then
     echo "Another instance is already running. Exiting."
     echo "If you are certain this is the only instance, you can delete the lock file $lockfile"
     exit
 else
-    touch "$lockfile"
+    touch "$lock_file" || die "Could not create lock file."
 fi
 
-tmpdir=/data/tmp
-
-if [ ! -d "$tmpdir" ] ; then
-    mkdir -p "$tmpdir"
+if [ ! -d "$tmp_directory" ] ; then
+    mkdir -p "$tmp_directory" || die "Could not create temporary directory \"$tmp_directory\"."
 fi
 
+function mp3()
+{
+    output_directory="/data/mp3"
+
+    input_file="$1"
+    output_file="`echo "$input_file" | sed -E "s#^$input_directory(.+)\\.[a-z0-9]+\\$#$output_directory\\1.mp3#g"`"
+
+    directory="`dirname "$output_file"`"
+    if [ ! -d "$directory" ] ; then
+        echo "New Directory: \"$directory\""
+        mkdir -p "$directory" || die "Could not create directory \"$directory\"."
+    fi
+
+    transcode=false
+
+    if [ ! -f "$output_file" ] ; then
+        echo "New MP3: \"$output_file\""
+        transcode=true
+    elif [ "$input_file" -nt "$output_file" ] ; then
+        echo "Input file is newer: \"$output_file\""
+        transcode=true
+    fi
+
+    if [ "$transcode" = true ] ; then
+        input_type="`file -b --mime-type "$input_file"`"
+        case "$input_type" in
+            audio/flac)
+                tmp_file="$tmp_directory/`basename "$output_file"`"
+                echo "Transcoding MP3..."
+                ffmpeg -hide_banner -loglevel fatal -i "$input_file" -y -b:a 320k -qscale:a 2 -id3v2_version 3 -f mp3 "$tmp_file" || die "Failed to transcode to \"$tmp_file\"."
+                mv -f "$tmp_file" "$output_file" || die "Could not move file to \"$output_file\"."
+                ;;
+
+            audio/mpeg)
+                echo "Copying MP3..."
+                cp -f "$input_file" "$output_file"
+                ;;
+
+            *)
+                die "Unsupported input type $input_type (\"$input_file\")"
+                ;;
+        esac
+
+        if [ ! "$output_owner" = "" ] ; then
+            chown $output_owner "$output_file" || die "Could not change permissions on \"$output_file\"."
+        fi
+    fi
+}
+
+function ogg_add_cover()
+{
+    ogg_file="$1"
+    cover_file="$2"
+
+    tmp_metafile="$tmp_directory/metadata"
+    tmp_mdimg="$tmp_directory/image-with-header"
+
+    cover_mime_type="`file -b --mime-type "$cover_file"`"
+
+    description=""
+    vorbiscomment --list --raw "$ogg_file" > "$tmp_metafile" && \
+        sed -i -e '/^metadata_block_picture/d' "$tmp_metafile" && \
+        echo -n "" > "$tmp_mdimg" && \
+        printf "0: %.8x" 3 | xxd -r -g0 >> "$tmp_mdimg" && \
+        printf "0: %.8x" $(echo -n "$cover_mime_type" | wc -c) | xxd -r -g0 >> "$tmp_mdimg" && \
+        echo -n "$cover_mime_type" >> "$tmp_mdimg" && \
+        printf "0: %.8x" $(echo -n "$description" | wc -c) | xxd -r -g0 >> "$tmp_mdimg" && \
+        echo -n "$description" >> "$tmp_mdimg" && \
+        printf "0: %.8x" 0 | xxd -r -g0  >> "$tmp_mdimg" && \
+        printf "0: %.8x" 0 | xxd -r -g0  >> "$tmp_mdimg" && \
+        printf "0: %.8x" 0 | xxd -r -g0  >> "$tmp_mdimg" && \
+        printf "0: %.8x" 0 | xxd -r -g0  >> "$tmp_mdimg" && \
+        printf "0: %.8x" $(wc -c "$cover_file" | cut --delimiter=' ' --fields=1) | xxd -r -g0 >> "$tmp_mdimg" && \
+        cat "$tmp_cover" >> "$tmp_mdimg" && \
+        echo "metadata_block_picture=$(base64 --wrap=0 < "$tmp_mdimg")" >> "$tmp_metafile" && \
+        vorbiscomment --write --raw --commentfile "$tmp_metafile" "$ogg_file" && \
+        rm "$tmp_metafile" "$tmp_mdimg"
+}
+
+function ogg()
+{
+    output_directory="/data/ogg"
+
+    input_file="$1"
+    output_file="`echo "$input_file" | sed -E "s#^$input_directory(.+)\\.[a-z0-9]+\\$#$output_directory\\1.ogg#g"`"
+
+    directory="`dirname "$output_file"`"
+    if [ ! -d "$directory" ] ; then
+        echo "New Directory: \"$directory\""
+        mkdir -p "$directory" || die "Could not create directory \"$directory\"."
+    fi
+
+    transcode=false
+
+    if [ ! -f "$output_file" ] ; then
+        echo "New OGG: \"$output_file\""
+        transcode=true
+    elif [ "$input_file" -nt "$output_file" ] ; then
+        echo "Input file is newer: \"$output_file\""
+        transcode=true
+    fi
+
+    if [ "$transcode" = true ] ; then
+        input_type="`file -b --mime-type "$input_file"`"
+
+        tmp_ogg="$tmp_directory/`basename "$output_file"`"
+        tmp_cover="$tmp_directory/cover"
+
+        case "$input_type" in
+            audio/flac)
+                echo "Transcoding OGG..."
+                ffmpeg -hide_banner -loglevel fatal -i "$input_file" -y -map a -qscale:a 6 -id3v2_version 3 -f ogg "$tmp_ogg" || die "Could not transcode to \"$tmp_ogg\"."
+
+                echo "Adding OGG cover image..."
+                metaflac --export-picture-to="$tmp_cover" "$input_file" && \
+                    ogg_add_cover "$tmp_ogg" "$tmp_cover" || die "Could not add cover image to \"$tmp_ogg\"."
+
+                mv -f "$tmp_ogg" "$output_file" || die "Could not move file to \"$output_file\"."
+                ;;
+
+            audio/mpeg)
+                echo "Transcoding OGG..."
+                ffmpeg -hide_banner -loglevel fatal -i "$input_file" -y -map a -qscale:a 6 -id3v2_version 3 -f ogg "$tmp_ogg" || die "Could not transcode to \"$tmp_ogg\"."
+
+                echo "Adding OGG cover image..."
+                exiftool -Picture -b "$input_file" > "$tmp_cover" && \
+                    ogg_add_cover "$tmp_ogg" "$tmp_cover" || die "Could not add cover image to \"$tmp_ogg\"."
+
+                mv -f "$tmp_ogg" "$output_file" || die "Could not move file to \"$output_file\"."
+                ;;
+
+            *)
+                die "Unsupported input type $input_type (\"$input_file\")"
+                ;;
+        esac
+
+        if [ ! "$output_owner" = "" ] ; then
+            chown $output_owner "$output_file" || die "Could not change permissions on \"$output_file\"."
+        fi
+    fi
+}
+
 SAVEIFS=$IFS
 IFS=`echo -ne "\n\b"`
-for flac in `find /data/music -name '*.flac' | sort`; do
-    # MP3
-
-    mp3=`echo "$flac" | sed -e 's/^\/data\/music/\/data\/mp3/' -e 's/.flac$/.mp3/'`
-    tmpmp3="$tmpdir/`basename "$mp3"`"
-    mp3dir=`dirname "$mp3"`
-    transcodemp3=false
-
-    if [ ! -d "$mp3dir" ] ; then
-        echo "New Directory: \"$mp3dir\""
-        mkdir -p "$mp3dir"
-    fi
-
-    if [ ! -f "$mp3" ] ; then
-        echo "New MP3: \"$mp3\""
-        transcodemp3=true
-    elif [ "$flac" -nt "$mp3" ] ; then
-        echo "FLAC is newer: \"$mp3\""
-        transcodemp3=true
-    fi
-
-    if [ "$transcodemp3" = true ] ; then
-        echo "Transcoding MP3..."
-        # ffmpeg -i "$flac" -y -b:a 320k -qscale:a 2 -f mp3 "$mp3"
-        ffmpeg -hide_banner -loglevel fatal -i "$flac" -y -b:a 320k -qscale:a 2 -id3v2_version 3 -f mp3 "$tmpmp3" && \
-            mv -f "$tmpmp3" "$mp3" && \
-            chown $output_owner "$mp3"
-        echo -e "Done.\n"
-    fi
-
-    # OGG
-
-    ogg=`echo "$flac" | sed -e 's/^\/data\/music/\/data\/ogg/' -e 's/.flac$/.ogg/'`
-    tmpogg="$tmpdir/`basename "$ogg"`"
-    tmpcover="$tmpdir/cover"
-    tmpmetafile="$tmpdir/metadata"
-    tmpmdimg="$tmpdir/image-with-header"
-    oggdir=`dirname "$ogg"`
-    transcodeogg=false
-
-    if [ ! -d "$oggdir" ] ; then
-        echo "New Directory: \"$oggdir\""
-        mkdir -p "$oggdir"
-    fi
-
-    if [ ! -f "$ogg" ] ; then
-        echo "New OGG: \"$ogg\""
-        transcodeogg=true
-    elif [ "$flac" -nt "$ogg" ] ; then
-        echo "FLAC is newer: \"$ogg\""
-        transcodeogg=true
-    fi
-
-    if [ "$transcodeogg" = true ] ; then
-        echo "Transcoding OGG..."
-
-        ffmpeg -hide_banner -loglevel fatal -i "$flac" -y -map a -qscale:a 6 -id3v2_version 3 -f ogg "$tmpogg" && \
-            mv -f "$tmpogg" "$ogg" && \
-            metaflac --export-picture-to="$tmpcover" "$flac" && \
-            covermimetype="image/png" && \
-            covermimetype="`file -b --mime-type "$tmpcover"`" && \
-            description="" && \
-            vorbiscomment --list --raw "$ogg" > "$tmpmetafile" && \
-            sed -i -e '/^metadata_block_picture/d' "$tmpmetafile" && \
-            echo -n "" > "$tmpmdimg" && \
-            printf "0: %.8x" 3 | xxd -r -g0 >> "$tmpmdimg" && \
-            printf "0: %.8x" $(echo -n "$covermimetype" | wc -c) | xxd -r -g0 >> "$tmpmdimg" && \
-            echo -n "$covermimetype" >> "$tmpmdimg" && \
-            printf "0: %.8x" $(echo -n "$description" | wc -c) | xxd -r -g0 >> "$tmpmdimg" && \
-            echo -n "$description" >> "$tmpmdimg" && \
-            printf "0: %.8x" 0 | xxd -r -g0  >> "$tmpmdimg" && \
-            printf "0: %.8x" 0 | xxd -r -g0  >> "$tmpmdimg" && \
-            printf "0: %.8x" 0 | xxd -r -g0  >> "$tmpmdimg" && \
-            printf "0: %.8x" 0 | xxd -r -g0  >> "$tmpmdimg" && \
-            printf "0: %.8x" $(wc -c "$tmpcover" | cut --delimiter=' ' --fields=1) | xxd -r -g0 >> "$tmpmdimg" && \
-            cat "$tmpcover" >> "$tmpmdimg" && \
-            echo "metadata_block_picture=$(base64 --wrap=0 < "$tmpmdimg")" >> "$tmpmetafile" && \
-            vorbiscomment --write --raw --commentfile "$tmpmetafile" "$ogg" && \
-            chown $output_owner "$ogg"
-
-        echo -e "Done.\n"
-    fi
+for file in `find "$input_directory" -name '*.flac' -or -name '*.mp3' | sort`; do
+    mp3 "$file"
+    ogg "$file"
 done
 
 echo "Done Transcoding."