merge recognize-music.sh

dots/.config/quickshell/ii/scripts/musicRecognition/recognize-music.sh

@@ -2,11 +2,8 @@
 
 INTERVAL=2
 TOTAL_DURATION=30
-MIN_VALID_RESULT_LENGTH=300
 SOURCE_TYPE="monitor"  # monitor | input
-TMP_PATH="/tmp/quickshell/media/songrec"
-TMP_RAW="$TMP_PATH/recording.raw"
-TMP_MP3="$TMP_PATH/recording.mp3"
+FIFO=$(mktemp -u /tmp/songrec_out_XXXXXX)
 
 while getopts "i:t:s:" opt; do
     case $opt in
@@ -16,47 +13,43 @@ while getopts "i:t:s:" opt; do
         *) exit 1 ;;
     esac
 done
+
+if ! command -v songrec >/dev/null 2>&1; then
+    exit 1
+fi
+
 if [ "$SOURCE_TYPE" = "monitor" ]; then
-    MONITOR_SOURCE=$(pactl get-default-sink).monitor
-    elif [ "$SOURCE_TYPE" = "input" ]; then
-    MONITOR_SOURCE=$(pactl info | grep "Default Source:" | awk '{print $3}' || true)
+    AUDIO_DEVICE=$(pactl get-default-sink).monitor
+elif [ "$SOURCE_TYPE" = "input" ]; then
+    AUDIO_DEVICE=$(pactl info | grep "Default Source:" | awk '{print $3}' || true)
 else
     echo "Invalid source type"
     exit 1
 fi
 
-if ! command -v songrec >/dev/null 2>&1 || ! command -v parec >/dev/null 2>&1 || ! command -v ffmpeg >/dev/null 2>&1; then
+if [ -z "$AUDIO_DEVICE" ] || ! pactl list short sources | grep -q "$AUDIO_DEVICE"; then
     exit 1
 fi
 
-if [ -z "$MONITOR_SOURCE" ] || ! pactl list short sources | grep -q "$MONITOR_SOURCE"; then
-    exit 1
-fi
+mkfifo "$FIFO"
 
 cleanup() {
-    rm -f "$TMP_RAW" "$TMP_MP3"
-    pkill -P $$ parec >/dev/null 2>&1 || true
+    kill "$SONGREC_PID" 2>/dev/null || true
+    wait "$SONGREC_PID" 2>/dev/null
+    rm -f "$FIFO"
 }
 trap cleanup EXIT
 
-mkdir -p "$TMP_PATH"
-parec --device="$MONITOR_SOURCE" --format=s16le --rate=44100 --channels=2 > "$TMP_RAW" &
-START_TIME=$(date +%s)
+songrec listen --audio-device "$AUDIO_DEVICE" --request-interval "$INTERVAL" --json --disable-mpris > "$FIFO" &
+SONGREC_PID=$!
 
-while true; do
-    sleep "$INTERVAL"
-    CURRENT_TIME=$(date +%s)
-    ELAPSED=$((CURRENT_TIME - START_TIME))
-    
-    if (( ELAPSED >= TOTAL_DURATION )); then
+( sleep "$TOTAL_DURATION" && kill "$SONGREC_PID" 2>/dev/null ) &
+
+while IFS= read -r line; do
+    if echo "$line" | grep -q '"matches": \['; then
+        echo "$line"
         exit 0
     fi
-    
-    ffmpeg -f s16le -ar 44100 -ac 2 -i "$TMP_RAW" -acodec libmp3lame -y -hide_banner -loglevel error "$TMP_MP3" 2>/dev/null
-    RESULT=$(songrec recognize --json "$TMP_MP3" 2>/dev/null || true)
+done < "$FIFO"
 
-    if echo "$RESULT" | grep -q '"matches": \[' && [ ${#RESULT} -gt $MIN_VALID_RESULT_LENGTH ]; then
-        echo "$RESULT"
-        exit 0
-    fi
-done
+exit 0