119 lines
3.6 KiB
Go
119 lines
3.6 KiB
Go
package download
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"path/filepath"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/mathismqn/godeez/internal/deezer"
|
|
"github.com/mathismqn/godeez/internal/fsutil"
|
|
"github.com/mathismqn/godeez/internal/store"
|
|
"github.com/mathismqn/godeez/internal/tag"
|
|
)
|
|
|
|
func (d *Downloader) downloadTrack(ctx context.Context, resource deezer.Resource, track *deezer.Track, opts Options, outputDir string) downloadResult {
|
|
media, err := d.deezerClient.FetchMedia(ctx, track, opts.Quality)
|
|
if err != nil {
|
|
return downloadResult{err: fmt.Errorf("failed to fetch media: %w", err)}
|
|
}
|
|
|
|
mediaFormat := media.Format()
|
|
if opts.Strict && strings.ToLower(mediaFormat) != opts.Quality {
|
|
return downloadResult{err: fmt.Errorf("requested quality '%s' not available", opts.Quality)}
|
|
}
|
|
|
|
if skipPath, skip := d.shouldSkipDownload(ctx, track.ID, mediaFormat); skip {
|
|
return downloadResult{skipped: true, path: skipPath}
|
|
}
|
|
|
|
metadataChan := make(chan metadataResult, 1)
|
|
go func() {
|
|
metadataChan <- fetchMetadata(ctx, d.deezerClient.Session.HTTPClient, track, opts)
|
|
}()
|
|
|
|
dlCtx, cancel := context.WithTimeout(ctx, opts.Timeout)
|
|
defer cancel()
|
|
|
|
stream, err := d.deezerClient.MediaStream(dlCtx, media)
|
|
if err != nil {
|
|
return downloadResult{err: fmt.Errorf("failed to get media stream: %w", err)}
|
|
}
|
|
|
|
fileName := track.Filename(d.kind, mediaFormat)
|
|
outputPath := d.uniqueOutputPath(track.ID, filepath.Join(outputDir, fileName))
|
|
|
|
key := deezer.BlowfishKey(track.ID)
|
|
if err := d.streamToFile(dlCtx, stream, outputPath, key); err != nil {
|
|
return downloadResult{err: fmt.Errorf("failed to stream to file: %w", err)}
|
|
}
|
|
|
|
var warnings []string
|
|
|
|
if opts.Quality != strings.ToLower(mediaFormat) {
|
|
warnings = append(warnings, fmt.Sprintf("requested quality '%s' not available, using '%s' instead", opts.Quality, strings.ToLower(mediaFormat)))
|
|
}
|
|
|
|
cover, err := d.deezerClient.FetchCoverImage(ctx, track)
|
|
if err != nil && !errors.Is(err, context.Canceled) {
|
|
warnings = append(warnings, fmt.Sprintf("failed to fetch cover image: %v", err))
|
|
}
|
|
|
|
metadata := <-metadataChan
|
|
|
|
if err := ctx.Err(); err != nil {
|
|
fsutil.Remove(outputPath)
|
|
return downloadResult{err: err}
|
|
}
|
|
|
|
warnings = append(warnings, metadata.warnings...)
|
|
warnings = append(warnings, d.finalizeDownload(resource, track, outputPath, mediaFormat, metadata.genre, cover, metadata.bpmKey)...)
|
|
|
|
return downloadResult{warnings: warnings}
|
|
}
|
|
|
|
func (d *Downloader) uniqueOutputPath(trackID, path string) string {
|
|
owned := ""
|
|
if info, err := d.store.DownloadInfo(trackID); err == nil {
|
|
owned = info.Path
|
|
}
|
|
|
|
ext := filepath.Ext(path)
|
|
stem := strings.TrimSuffix(path, ext)
|
|
candidate := path
|
|
for i := 2; candidate != owned && fsutil.Exists(candidate); i++ {
|
|
candidate = fmt.Sprintf("%s (%d)%s", stem, i, ext)
|
|
}
|
|
|
|
return candidate
|
|
}
|
|
|
|
func (d *Downloader) finalizeDownload(resource deezer.Resource, track *deezer.Track, outputPath, mediaFormat, genre string, cover []byte, bpmKey bpmKey) []string {
|
|
var warnings []string
|
|
|
|
if err := tag.Write(outputPath, buildTagMetadata(resource, track, cover, bpmKey, genre)); err != nil {
|
|
warnings = append(warnings, fmt.Sprintf("failed to add tags: %v", err))
|
|
}
|
|
|
|
hash, err := hashFile(outputPath)
|
|
if err != nil {
|
|
warnings = append(warnings, fmt.Sprintf("failed to get file hash: %v", err))
|
|
}
|
|
|
|
info := &store.DownloadInfo{
|
|
TrackID: track.ID,
|
|
Quality: mediaFormat,
|
|
Path: outputPath,
|
|
Hash: hash,
|
|
Downloaded: time.Now(),
|
|
}
|
|
|
|
if err := d.store.PutDownloadInfo(info); err != nil {
|
|
warnings = append(warnings, fmt.Sprintf("failed to save download info: %v", err))
|
|
}
|
|
|
|
return warnings
|
|
}
|