refactor(musicRecognition): replace parec/ffmpeg pipeline with songrec listen fix:#3057 (#3058)

This commit is contained in:
Minh
2026-04-05 21:45:18 +02:00
committed by GitHub
@@ -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
AUDIO_DEVICE=$(pactl get-default-sink).monitor
elif [ "$SOURCE_TYPE" = "input" ]; then
MONITOR_SOURCE=$(pactl info | grep "Default Source:" | awk '{print $3}' || true)
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))
( sleep "$TOTAL_DURATION" && kill "$SONGREC_PID" 2>/dev/null ) &
if (( ELAPSED >= TOTAL_DURATION )); then
while IFS= read -r line; do
if echo "$line" | grep -q '"matches": \['; then
echo "$line"
exit 0
fi
done < "$FIFO"
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)
if echo "$RESULT" | grep -q '"matches": \[' && [ ${#RESULT} -gt $MIN_VALID_RESULT_LENGTH ]; then
echo "$RESULT"
exit 0
fi
done
exit 0