niveum/packages/mp3player-write.nix

# Convert and transfer audio files to an MP3 player
{
  lib,
  writers,
  ffmpeg,
  coreutils,
  gnugrep,
  gnused,
}:
writers.writeBashBin "mp3player-write" ''
  set -e

  SPEED=1.0

  while getopts ":s:" opt; do
    case $opt in
      s) SPEED=$OPTARG ;;
      \?) echo "Invalid option: -$OPTARG" >&2; exit 1 ;;
      :) echo "Option -$OPTARG requires a value." >&2; exit 1 ;;
    esac
  done
  shift $((OPTIND -1))

  if [ "$#" -lt 2 ]; then
    echo "Usage: mp3player-write [-s speed] MOUNT_POINT FILE1 [FILE2 ...]"
    exit 1
  fi

  MOUNT_POINT=$1
  shift
  FILES=("$@")

  if [ ! -d "$MOUNT_POINT" ]; then
    echo "Error: Mount point '$MOUNT_POINT' does not exist."
    exit 1
  fi

  sanitize_filename() {
    local name
    name=$(${lib.getExe' coreutils "basename"} "$1")
    name=''${name%.*}
    name=$(echo "$name" | ${lib.getExe' coreutils "tr"} ' ' '_' | ${lib.getExe' coreutils "tr"} -cd '[:alnum:]_-')
    echo "''${name:0:50}"
  }

  for f in "''${FILES[@]}"; do
    [ -f "$f" ] || continue

    existing_prefixes=$(${lib.getExe' coreutils "ls"} "$MOUNT_POINT" | ${lib.getExe gnugrep} -E '^[0-9].*\.mp3$' | ${lib.getExe gnused} -E 's/^([0-9]).*/\1/' | ${lib.getExe' coreutils "sort"} -n | ${lib.getExe' coreutils "uniq"})
    for i in {0..9}; do
      if ! echo "$existing_prefixes" | ${lib.getExe gnugrep} -q "^$i$"; then
        PREFIX=$i
        break
      fi
    done

    BASENAME=$(sanitize_filename "$f")
    OUT_PATTERN="$MOUNT_POINT/''${PREFIX}_%03d_''${BASENAME}.mp3"

    echo "Converting '$f' to '$OUT_PATTERN' at speed $SPEED..."

    ${lib.getExe ffmpeg} -nostdin -i "$f" \
      -filter:a "atempo=$SPEED" \
      -ar 22050 -ac 1 -c:a libmp3lame -b:a 32k \
      -f segment -segment_time 300 \
      "$OUT_PATTERN"
  done

  echo "All files processed successfully."
''