130 lines
3.7 KiB
Go
130 lines
3.7 KiB
Go
/*
|
|
* MumbleDJ
|
|
* By Matthieu Grieger
|
|
* song.go
|
|
* Copyright (c) 2014 Matthieu Grieger (MIT License)
|
|
*/
|
|
|
|
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"github.com/jmoiron/jsonq"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"os"
|
|
"os/exec"
|
|
"strings"
|
|
)
|
|
|
|
// Song type declaration.
|
|
type Song struct {
|
|
submitter string
|
|
title string
|
|
youtubeId string
|
|
duration string
|
|
thumbnailUrl string
|
|
skippers []string
|
|
}
|
|
|
|
// Returns a new Song type. Before returning the new type, the song's metadata is collected
|
|
// via the YouTube Gdata API.
|
|
func NewSong(user, id string) *Song {
|
|
jsonUrl := fmt.Sprintf("http://gdata.youtube.com/feeds/api/videos/%s?v=2&alt=jsonc", id)
|
|
jsonString := ""
|
|
|
|
if response, err := http.Get(jsonUrl); err == nil {
|
|
defer response.Body.Close()
|
|
if body, err := ioutil.ReadAll(response.Body); err == nil {
|
|
jsonString = string(body)
|
|
}
|
|
}
|
|
|
|
jsonData := map[string]interface{}{}
|
|
decoder := json.NewDecoder(strings.NewReader(jsonString))
|
|
decoder.Decode(&jsonData)
|
|
jq := jsonq.NewQuery(jsonData)
|
|
|
|
videoTitle, _ := jq.String("data", "title")
|
|
videoThumbnail, _ := jq.String("data", "thumbnail", "hqDefault")
|
|
duration, _ := jq.Int("data", "duration")
|
|
videoDuration := fmt.Sprintf("%d:%02d", duration/60, duration%60)
|
|
|
|
song := &Song{
|
|
submitter: user,
|
|
title: videoTitle,
|
|
youtubeId: id,
|
|
duration: videoDuration,
|
|
thumbnailUrl: videoThumbnail,
|
|
}
|
|
return song
|
|
}
|
|
|
|
// Downloads the song via youtube-dl. All downloaded songs are stored in ~/.mumbledj/songs and should be automatically cleaned.
|
|
func (s *Song) Download() error {
|
|
cmd := exec.Command("youtube-dl", "--output", fmt.Sprintf(`~/.mumbledj/songs/%s.m4a`, s.youtubeId), "--format", "m4a", s.youtubeId)
|
|
if err := cmd.Run(); err == nil {
|
|
return nil
|
|
} else {
|
|
return errors.New("Song download failed.")
|
|
}
|
|
}
|
|
|
|
// Plays the song. Once the song is playing, a notification is displayed in a text message that features the video thumbnail, URL, title,
|
|
// duration, and submitter.
|
|
func (s *Song) Play() {
|
|
dj.audioStream.Play(fmt.Sprintf("%s/.mumbledj/songs/%s.m4a", dj.homeDir, s.youtubeId))
|
|
dj.client.Self().Channel().Send(fmt.Sprintf(NOW_PLAYING_HTML, s.thumbnailUrl, s.youtubeId, s.title, s.duration, s.submitter), false)
|
|
}
|
|
|
|
// Deletes the song from ~/.mumbledj/songs.
|
|
func (s *Song) Delete() error {
|
|
filePath := fmt.Sprintf("%s/.mumbledj/songs/%s.m4a", dj.homeDir, s.youtubeId)
|
|
if _, err := os.Stat(filePath); err == nil {
|
|
if err := os.Remove(filePath); err == nil {
|
|
return nil
|
|
} else {
|
|
return errors.New("Error occurred while deleting audio file.")
|
|
}
|
|
} else {
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// Adds a skip to the skippers slice. If the user is already in the slice AddSkip() returns
|
|
// an error and does not add a duplicate skip.
|
|
func (s *Song) AddSkip(username string) error {
|
|
for _, user := range s.skippers {
|
|
if username == user {
|
|
return errors.New("This user has already skipped the current song.")
|
|
}
|
|
}
|
|
s.skippers = append(s.skippers, username)
|
|
return nil
|
|
}
|
|
|
|
// Removes a skip from the skippers slice. If username is not in the slice, an error is
|
|
// returned.
|
|
func (s *Song) RemoveSkip(username string) error {
|
|
for i, user := range s.skippers {
|
|
if username == user {
|
|
s.skippers = append(s.skippers[:i], s.skippers[i+1:]...)
|
|
return nil
|
|
}
|
|
}
|
|
return errors.New("This user has not skipped the song.")
|
|
}
|
|
|
|
// Calculates current skip ratio based on number of users within MumbleDJ's channel and the
|
|
// amount of values in the skippers slice. If the value is greater than or equal to the skip ratio
|
|
// defined in mumbledj.gcfg, the function returns true. Returns false otherwise.
|
|
func (s *Song) SkipReached(channelUsers int) bool {
|
|
if float32(len(s.skippers))/float32(channelUsers) >= dj.conf.General.SkipRatio {
|
|
return true
|
|
} else {
|
|
return false
|
|
}
|
|
}
|