mirror of
https://github.com/cgzirim/seek-tune.git
synced 2025-12-18 09:24:19 +00:00
refactor(youtube): remove deprecated kkdai/youtube code and unify audio download method
- Delete old downloadYTaudio1 function using github.com/kkdai/youtube - Rename downloadYTaudio2 → downloadYTaudio as the primary yt-dlp-based downloader
This commit is contained in:
parent
ae6e7a3805
commit
95218512a6
1 changed files with 2 additions and 58 deletions
|
|
@ -10,8 +10,6 @@ import (
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"song-recognition/utils"
|
"song-recognition/utils"
|
||||||
|
|
||||||
ytdlib "github.com/kkdai/youtube/v2"
|
|
||||||
|
|
||||||
"errors"
|
"errors"
|
||||||
"io"
|
"io"
|
||||||
"io/ioutil"
|
"io/ioutil"
|
||||||
|
|
@ -223,62 +221,8 @@ func ytSearch(searchTerm string, limit int) (results []*SearchResult, err error)
|
||||||
return results, nil
|
return results, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// downloadYTaudio1 downloads audio from a YouTube video using the github.com/kkdai/youtube library.
|
// downloadYTaudio downloads audio from a YouTube video using yt-dlp command line tool.
|
||||||
func downloadYTaudio1(id, outputFilePath string) (string, error) {
|
func downloadYTaudio(videoURL, outputFilePath string) (string, error) {
|
||||||
logger := utils.GetLogger()
|
|
||||||
|
|
||||||
dir := filepath.Dir(outputFilePath)
|
|
||||||
if stat, err := os.Stat(dir); err != nil || !stat.IsDir() {
|
|
||||||
logger.Error("Invalid directory for output file", slog.Any("error", err))
|
|
||||||
return "", errors.New("output directory does not exist or is not a directory")
|
|
||||||
}
|
|
||||||
|
|
||||||
client := ytdlib.Client{}
|
|
||||||
video, err := client.GetVideo(id)
|
|
||||||
if err != nil {
|
|
||||||
logger.Error("Error getting YouTube video", slog.Any("error", err))
|
|
||||||
return "", err
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
|
||||||
itag code: 140, container: m4a, content: audio, bitrate: 128k
|
|
||||||
change the FindByItag parameter to 139 if you want smaller files (but with a bitrate of 48k)
|
|
||||||
https://gist.github.com/sidneys/7095afe4da4ae58694d128b1034e01e2
|
|
||||||
*/
|
|
||||||
formats := video.Formats.Itag(140)
|
|
||||||
|
|
||||||
/* in some cases, when attempting to download the audio
|
|
||||||
using the library github.com/kkdai/youtube,
|
|
||||||
the download fails (and shows the file size as 0 bytes)
|
|
||||||
until the second or third attempt. */
|
|
||||||
var fileSize int64
|
|
||||||
file, err := os.Create(outputFilePath)
|
|
||||||
if err != nil {
|
|
||||||
logger.Error("Error creating file", slog.Any("error", err))
|
|
||||||
return "", err
|
|
||||||
}
|
|
||||||
|
|
||||||
for fileSize == 0 {
|
|
||||||
stream, _, err := client.GetStream(video, &formats[0])
|
|
||||||
if err != nil {
|
|
||||||
logger.Error("Error getting stream", slog.Any("error", err))
|
|
||||||
return "", err
|
|
||||||
}
|
|
||||||
|
|
||||||
if _, err = io.Copy(file, stream); err != nil {
|
|
||||||
logger.Error("Error copying stream to file", slog.Any("error", err))
|
|
||||||
return "", err
|
|
||||||
}
|
|
||||||
|
|
||||||
fileSize, _ = GetFileSize(outputFilePath)
|
|
||||||
}
|
|
||||||
defer file.Close()
|
|
||||||
|
|
||||||
return outputFilePath + "m4a", nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// downloadYTaudio2 downloads audio from a YouTube video using yt-dlp command line tool.
|
|
||||||
func downloadYTaudio2(videoURL, outputFilePath string) (string, error) {
|
|
||||||
logger := utils.GetLogger()
|
logger := utils.GetLogger()
|
||||||
|
|
||||||
dir := filepath.Dir(outputFilePath)
|
dir := filepath.Dir(outputFilePath)
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue