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/main.go

129 lines
3.7 KiB
Go
Raw Normal View History

2014-12-09 06:45:22 +01:00
/*
* MumbleDJ
* By Matthieu Grieger
* main.go
* Copyright (c) 2014 Matthieu Grieger (MIT License)
*/
package main
import (
2014-12-10 00:41:50 +01:00
"flag"
"fmt"
2014-12-09 06:45:22 +01:00
"github.com/layeh/gumble/gumble"
"github.com/layeh/gumble/gumble_ffmpeg"
2014-12-09 06:45:22 +01:00
"github.com/layeh/gumble/gumbleutil"
"os/user"
2014-12-09 06:45:22 +01:00
)
// MumbleDJ type declaration
type mumbledj struct {
2014-12-16 01:40:31 +01:00
config gumble.Config
client *gumble.Client
keepAlive chan bool
defaultChannel string
2014-12-16 01:40:31 +01:00
conf DjConfig
queue *SongQueue
audioStream *gumble_ffmpeg.Stream
homeDir string
}
2014-12-27 19:05:13 +01:00
// OnConnect event. First moves MumbleDJ into the default channel specified
// via commandline args, and moves to root channel if the channel does not exist. The current
// user's homedir path is stored, configuration is loaded, and the audio stream is set up.
func (dj *mumbledj) OnConnect(e *gumble.ConnectEvent) {
if dj.client.Channels().Find(dj.defaultChannel) != nil {
dj.client.Self().Move(dj.client.Channels().Find(dj.defaultChannel))
} else {
2014-12-27 19:05:13 +01:00
fmt.Println("Channel doesn't exist or one was not provided, staying in root channel...")
}
2014-12-16 01:40:31 +01:00
if currentUser, err := user.Current(); err == nil {
dj.homeDir = currentUser.HomeDir
}
if err := loadConfiguration(); err == nil {
2014-12-13 05:52:07 +01:00
fmt.Println("Configuration successfully loaded!")
2014-12-13 18:22:30 +01:00
} else {
panic(err)
2014-12-13 05:52:07 +01:00
}
if audioStream, err := gumble_ffmpeg.New(dj.client); err == nil {
dj.audioStream = audioStream
dj.audioStream.Done = dj.queue.OnItemFinished
2014-12-27 18:41:27 +01:00
dj.audioStream.SetVolume(dj.conf.Volume.DefaultVolume)
} else {
panic(err)
}
}
2014-12-27 19:05:13 +01:00
// OnDisconnect event. Terminates MumbleDJ thread.
func (dj *mumbledj) OnDisconnect(e *gumble.DisconnectEvent) {
dj.keepAlive <- true
}
2014-12-27 19:05:13 +01:00
// OnTextMessage event. Checks for command prefix, and calls parseCommand if it exists. Ignores
// the incoming message otherwise.
func (dj *mumbledj) OnTextMessage(e *gumble.TextMessageEvent) {
2014-12-27 19:40:32 +01:00
if e.Message[0] == dj.conf.General.CommandPrefix[0] {
2014-12-16 01:40:31 +01:00
parseCommand(e.Sender, e.Sender.Name(), e.Message[1:])
}
}
2014-12-27 19:05:13 +01:00
// Checks if username has the permissions to execute a command. Permissions are specified in
// mumbledj.gcfg.
2014-12-16 01:40:31 +01:00
func (dj *mumbledj) HasPermission(username string, command bool) bool {
if dj.conf.Permissions.AdminsEnabled && command {
for _, adminName := range dj.conf.Permissions.Admins {
if username == adminName {
return true
}
}
return false
} else {
return true
2014-12-13 05:52:07 +01:00
}
}
2014-12-27 19:05:13 +01:00
// dj variable declaration. This is done outside of main() to allow global use.
2014-12-16 01:40:31 +01:00
var dj = mumbledj{
2014-12-13 05:52:07 +01:00
keepAlive: make(chan bool),
queue: NewSongQueue(),
2014-12-09 06:45:22 +01:00
}
2014-12-10 00:41:50 +01:00
2014-12-27 19:05:13 +01:00
// Main function, but only really performs startup tasks. Grabs and parses commandline
// args, sets up the gumble client and its listeners, and then connects to the server.
2014-12-10 00:41:50 +01:00
func main() {
2014-12-20 06:57:48 +01:00
var address, port, username, password, channel string
2014-12-10 00:41:50 +01:00
flag.StringVar(&address, "server", "localhost", "address for Mumble server")
flag.StringVar(&port, "port", "64738", "port for Mumble server")
flag.StringVar(&username, "username", "MumbleDJ", "username of MumbleDJ on server")
flag.StringVar(&password, "password", "", "password for Mumble server (if needed)")
flag.StringVar(&channel, "channel", "root", "default channel for MumbleDJ")
2014-12-10 00:41:50 +01:00
flag.Parse()
2014-12-16 01:40:31 +01:00
dj.client = gumble.NewClient(&dj.config)
dj.config = gumble.Config{
2014-12-10 00:41:50 +01:00
Username: username,
Password: password,
2014-12-16 01:40:31 +01:00
Address: address + ":" + port,
2014-12-10 00:41:50 +01:00
}
dj.defaultChannel = channel
2014-12-16 01:40:31 +01:00
dj.client.Attach(gumbleutil.Listener{
2014-12-16 01:40:31 +01:00
Connect: dj.OnConnect,
Disconnect: dj.OnDisconnect,
TextMessage: dj.OnTextMessage,
})
2014-12-16 01:40:31 +01:00
// IMPORTANT NOTE: This will be changed later once released. Not really safe at the
// moment.
dj.config.TLSConfig.InsecureSkipVerify = true
if err := dj.client.Connect(); err != nil {
2014-12-10 00:41:50 +01:00
panic(err)
}
2014-12-16 01:40:31 +01:00
<-dj.keepAlive
2014-12-10 00:41:50 +01:00
}