This repository has been archived on 2019-06-23. You can view files and clone it, but cannot push or open issues or pull requests.
mumbledj/bot/youtube_dl.go

82 lines
2.2 KiB
Go
Raw Permalink Normal View History

2016-06-21 02:16:05 +02:00
/*
* MumbleDJ
* By Matthieu Grieger
* bot/youtube_dl.go
* Copyright (c) 2016 Matthieu Grieger (MIT License)
*/
package bot
import (
"errors"
"os"
"os/exec"
"github.com/Sirupsen/logrus"
"github.com/matthieugrieger/mumbledj/interfaces"
"github.com/spf13/viper"
)
// YouTubeDL is a struct that gathers all methods related to the youtube-dl
// software.
// youtube-dl: https://rg3.github.io/youtube-dl/
type YouTubeDL struct{}
// Download downloads the audio associated with the incoming `track` object
// and stores it `track.Filename`.
func (yt *YouTubeDL) Download(t interfaces.Track) error {
player := "--prefer-ffmpeg"
if viper.GetString("defaults.player_command") == "avconv" {
player = "--prefer-avconv"
}
filepath := os.ExpandEnv(viper.GetString("cache.directory") + "/" + t.GetFilename())
// Determine which format to use.
format := "bestaudio"
for _, service := range DJ.AvailableServices {
if service.GetReadableName() == t.GetService() {
format = service.GetFormat()
}
}
// Check to see if track is already downloaded.
if _, err := os.Stat(filepath); os.IsNotExist(err) {
var cmd *exec.Cmd
if t.GetService() == "Mixcloud" {
cmd = exec.Command("youtube-dl", "--verbose", "--no-mtime", "--output", filepath, "--format", format, "--external-downloader", "aria2c", player, t.GetURL())
} else {
cmd = exec.Command("youtube-dl", "--verbose", "--no-mtime", "--output", filepath, "--format", format, player, t.GetURL())
}
output, err := cmd.CombinedOutput()
if err != nil {
args := ""
for s := range cmd.Args {
args += cmd.Args[s] + " "
}
logrus.Warnf("%s\n%s\nyoutube-dl: %s", args, string(output), err.Error())
return errors.New("Track download failed")
}
if viper.GetBool("cache.enabled") {
DJ.Cache.CheckDirectorySize()
}
}
return nil
}
// Delete deletes the audio file associated with the incoming `track` object.
func (yt *YouTubeDL) Delete(t interfaces.Track) error {
if !viper.GetBool("cache.enabled") {
filePath := os.ExpandEnv(viper.GetString("cache.directory") + "/" + t.GetFilename())
if _, err := os.Stat(filePath); err == nil {
if err := os.Remove(filePath); err == nil {
return nil
}
return errors.New("An error occurred while deleting the audio file")
}
}
return nil
}