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

refactor: simplify WAV conversion and fingerprinting process

Browse source
This commit is contained in:
Chigozirim Igweamaka 2025-07-15 15:36:35 +01:00
parent 6635647061
commit c2fc834a0b
1 changed files with 9 additions and 33 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

42
server/spotify/downloader.go
View file

@ -12,7 +12,6 @@ import (
"song-recognition/shazam"
"song-recognition/utils"
"song-recognition/wav"
"strconv"
"strings"
"sync"
"time"
@ -228,49 +227,26 @@ func ProcessAndSaveSong(songFilePath, songTitle, songArtist, ytID string) error
}
defer dbclient.Close()
channelsStr := utils.GetEnv("SHZ_CHANNELS", "1")
channels, err := strconv.Atoi(channelsStr)
if err != nil {
logger.Error("Failed to convert channels to int", slog.Any("error", err))
return err
}
wavFilePath, err := wav.ConvertToWAV(songFilePath, channels)
wavFilePath, err := wav.ConvertToWAV(songFilePath)
if err != nil {
logger.Error("Failed to convert to WAV", slog.Any("error", err))
return err
}
wavInfo, err := wav.ReadWavInfo(wavFilePath)
if err != nil {
logger.Error("Failed to read WAV info", slog.Any("error", err))
return err
}
// TODO: handle the case where there're 2 channels.
samples, err := wav.WavBytesToSamples(wavInfo.Data)
if err != nil {
logger.Error("Error converting WAV bytes to samples", slog.Any("error", err))
return fmt.Errorf("error converting wav bytes to float64: %v", err)
}
spectro, err := shazam.Spectrogram(samples, wavInfo.SampleRate)
if err != nil {
logger.Error("Error creating spectrogram", slog.Any("error", err))
return fmt.Errorf("error creating spectrogram: %v", err)
}
songID, err := dbclient.RegisterSong(songTitle, songArtist, ytID)
if err != nil {
logger.Error("Failed to register song", slog.Any("error", err))
return err
return fmt.Errorf("error registering song '%s' by '%s': %v", songTitle, songArtist, err)
}
peaks := shazam.ExtractPeaks(spectro, wavInfo.Duration)
fingerprints := shazam.Fingerprint(peaks, songID)
fingerprint, err := shazam.FingerprintAudio(wavFilePath, songID)
if err != nil {
dbclient.DeleteSongByID(songID)
logger.Error("Failed to create fingerprint", slog.String("wavFilePath", wavFilePath))
return fmt.Errorf("error generating fingerprint for %s by %s", songTitle, songArtist)
}
err = dbclient.StoreFingerprints(fingerprints)
err = dbclient.StoreFingerprints(fingerprint)
if err != nil {
dbclient.DeleteSongByID(songID)
logger.Error("Failed to store fingerprints", slog.Any("error", err))