/* * MumbleDJ * By Matthieu Grieger * commands/add.go * Copyright (c) 2016 Matthieu Grieger (MIT License) */ package commands import ( "errors" "fmt" "github.com/layeh/gumble/gumble" "github.com/matthieugrieger/mumbledj/interfaces" "github.com/spf13/viper" ) // AddCommand is a command that adds an audio track associated with a supported // URL to the queue. type AddCommand struct{} // Aliases returns the current aliases for the command. func (c *AddCommand) Aliases() []string { return viper.GetStringSlice("commands.add.aliases") } // Description returns the description for the command. func (c *AddCommand) Description() string { return viper.GetString("commands.add.description") } // IsAdminCommand returns true if the command is only for admin use, and // returns false otherwise. func (c *AddCommand) IsAdminCommand() bool { return viper.GetBool("commands.add.is_admin") } // Execute executes the command with the given user and arguments. // Return value descriptions: // string: A message to be returned to the user upon successful execution. // bool: Whether the message should be private or not. true = private, // false = public (sent to whole channel). // error: An error message to be returned upon unsuccessful execution. // If no error has occurred, pass nil instead. // Example return statement: // return "This is a private message!", true, nil func (c *AddCommand) Execute(user *gumble.User, args ...string) (string, bool, error) { var ( allTracks []interfaces.Track tracks []interfaces.Track service interfaces.Service err error lastTrackAdded interfaces.Track ) if len(args) == 0 { return "", true, errors.New("A URL must be supplied with the add command") } for _, arg := range args { if service, err = DJ.GetService(arg); err == nil { tracks, err = service.GetTracks(arg, user) if err == nil { allTracks = append(allTracks, tracks...) } } } if len(allTracks) == 0 { return "", true, errors.New("No valid tracks were found with the provided URL(s)") } numTooLong := 0 numAdded := 0 for _, track := range allTracks { if err = DJ.Queue.AppendTrack(track); err != nil { numTooLong++ } else { numAdded++ lastTrackAdded = track } } if numAdded == 0 { return "", true, errors.New("Your track(s) were either too long or an error occurred while processing them. No track(s) have been added.") } else if numAdded == 1 { return fmt.Sprintf("%s added 1 track to the queue:
\"%s\" from %s", user.Name, lastTrackAdded.GetTitle(), lastTrackAdded.GetService()), false, nil } retString := fmt.Sprintf("%s added %d tracks to the queue.", user.Name, numAdded) if numTooLong != 0 { retString += fmt.Sprintf("
%d tracks could not be added due to error or because they are too long.", numTooLong) } return retString, false, nil }