hax/seek-tune
1
0
Fork 0
mirror of https://github.com/cgzirim/seek-tune.git synced 2025-12-17 08:54:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: implement audio snippet recording and fingerprint processing

Browse source
This commit is contained in:
Chigozirim Igweamaka 2025-04-01 17:53:17 +01:00
parent 6647fa1af7
commit 8a918c74cd
1 changed files with 39 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

42
socketHandlers.go
View file

@ -2,6 +2,7 @@ package main
import (
"context"
"encoding/base64"
"encoding/json"
"fmt"
"log/slog"
@ -10,7 +11,9 @@ import (
"song-recognition/shazam"
"song-recognition/spotify"
"song-recognition/utils"
"song-recognition/wav"
"strings"
"time"
socketio "github.com/googollee/go-socket.io"
"github.com/mdobak/go-xerrors"
@ -175,6 +178,7 @@ func handleSongDownload(socket socketio.Conn, spotifyURL string) {
}
}
// handleNewRecording saves new recorded audio snippet to a WAV file.
func handleNewRecording(socket socketio.Conn, recordData string) {
logger := utils.GetLogger()
ctx := context.Background()
@ -186,14 +190,46 @@ func handleNewRecording(socket socketio.Conn, recordData string) {
return
}
samples, err := utils.ProcessRecording(&recData, true)
err := utils.CreateFolder("recordings")
if err != nil {
err := xerrors.New(err)
logger.ErrorContext(ctx, "Failed to process recording.", slog.Any("error", err))
logger.ErrorContext(ctx, "Failed create folder.", slog.Any("error", err))
}
now := time.Now()
fileName := fmt.Sprintf("%04d_%02d_%02d_%02d_%02d_%02d.wav",
now.Second(), now.Minute(), now.Hour(),
now.Day(), now.Month(), now.Year(),
)
filePath := "recordings/" + fileName
decodedAudioData, err := base64.StdEncoding.DecodeString(recData.Audio)
if err != nil {
err := xerrors.New(err)
logger.ErrorContext(ctx, "Failed to decode base64", slog.Any("error", err))
}
err = wav.WriteWavFile(filePath, decodedAudioData, recData.SampleRate, recData.Channels, recData.SampleSize)
if err != nil {
err := xerrors.New(err)
logger.ErrorContext(ctx, "Failed write wav file.", slog.Any("error", err))
}
}
func handleNewFingerprint(socket socketio.Conn, fingerprintData string) {
logger := utils.GetLogger()
ctx := context.Background()
var data struct {
Fingerprint map[uint32]uint32 `json:"fingerprint"`
}
if err := json.Unmarshal([]byte(fingerprintData), &data); err != nil {
err := xerrors.New(err)
logger.ErrorContext(ctx, "Failed to unmarshal fingerprint data.", slog.Any("error", err))
return
}
matches, _, err := shazam.FindMatches(samples, recData.Duration, recData.SampleRate)
matches, _, err := shazam.FindMatchesFGP(data.Fingerprint)
if err != nil {
err := xerrors.New(err)
logger.ErrorContext(ctx, "failed to get matches.", slog.Any("error", err))