Add XMPP Jukebox demo

2016-03-14 11:31:31 +01:00 · 2016-03-14 11:31:31 +01:00 · c41ed1c32f
parent d26d066540
commit c41ed1c32f
1 changed files with 120 additions and 0 deletions
--- a/cmd/xmpp_jukebox/xmpp_jukebox.go
+++ b/cmd/xmpp_jukebox/xmpp_jukebox.go
@ -0,0 +1,120 @@
+// Can be launched with:
+//   ./xmpp_jukebox -jid=test@localhost/jukebox -password=test -address=localhost:5222
+package main
+
+import (
+	"flag"
+	"fmt"
+	"log"
+	"os"
+	"strings"
+
+	"github.com/processone/gox/xmpp"
+	"github.com/processone/gox/xmpp/iot"
+	"github.com/processone/gox/xmpp/pep"
+	"github.com/processone/mpg123"
+	"github.com/processone/soundcloud"
+)
+
+// Get the actual song Stream URL from SoundCloud website song URL and play it with mpg123 player.
+const scClientID = "dde6a0075614ac4f3bea423863076b22"
+
+func main() {
+	jid := flag.String("jid", "", "jukebok XMPP JID, resource is optional")
+	password := flag.String("password", "", "XMPP account password")
+	address := flag.String("address", "", "If needed, XMPP server DNSName or IP and optional port (ie myserver:5222)")
+	flag.Parse()
+
+	var client *xmpp.Client
+	var err error
+	if client, err = connectXmpp(*jid, *password, *address); err != nil {
+		log.Fatal("Could not connect to XMPP: ", err)
+	}
+
+	p, err := mpg123.NewPlayer()
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	// Iterator to receive packets coming from our XMPP connection
+	for packet := range client.Recv() {
+
+		switch packet := packet.(type) {
+		case *xmpp.ClientMessage:
+			processMessage(client, p, packet)
+		case *xmpp.ClientIQ:
+			processIq(client, p, packet)
+		case *xmpp.ClientPresence:
+			// Do nothing with received presence
+		default:
+			fmt.Fprintf(os.Stdout, "Ignoring packet: %T\n", packet)
+		}
+	}
+}
+
+func processMessage(client *xmpp.Client, p *mpg123.Player, packet *xmpp.ClientMessage) {
+	command := strings.Trim(packet.Body, " ")
+	if command == "stop" {
+		p.Stop()
+	} else {
+		playSCURL(p, command)
+		sendUserTune(client, "Radiohead", "Spectre")
+	}
+}
+
+func processIq(client *xmpp.Client, p *mpg123.Player, packet *xmpp.ClientIQ) {
+	switch payload := packet.Payload.(type) {
+	// We support IOT Control IQ
+	case *iot.ControlSet:
+		var url string
+		for _, element := range payload.Fields {
+			if element.XMLName.Local == "string" && element.Name == "url" {
+				url = strings.Trim(element.Value, " ")
+				break
+			}
+		}
+
+		playSCURL(p, url)
+		setResponse := new(iot.ControlSetResponse)
+		reply := xmpp.ClientIQ{Packet: xmpp.Packet{To: packet.From, Type: "result", Id: packet.Id}, Payload: setResponse}
+		client.Send(reply.XMPPFormat())
+		// TODO add Soundclound artist / title retrieval
+		sendUserTune(client, "Radiohead", "Spectre")
+	default:
+		fmt.Fprintf(os.Stdout, "Other IQ Payload: %T\n", packet.Payload)
+	}
+}
+
+func sendUserTune(client *xmpp.Client, artist string, title string) {
+	tune := pep.Tune{Artist: artist, Title: title}
+	client.Send(tune.XMPPFormat())
+}
+
+func playSCURL(p *mpg123.Player, rawURL string) {
+	songID, _ := soundcloud.GetSongID(rawURL)
+	// TODO: Maybe we need to check the track itself to get the stream URL from reply ?
+	url := soundcloud.FormatStreamURL(songID)
+
+	p.Play(url)
+}
+
+func connectXmpp(jid string, password string, address string) (client *xmpp.Client, err error) {
+	xmppOptions := xmpp.Options{Address: address,
+		Jid: jid, Password: password, PacketLogger: os.Stdout,
+		Retry: 10}
+
+	if client, err = xmpp.NewClient(xmppOptions); err != nil {
+		return
+	}
+
+	if _, err = client.Connect(); err != nil {
+		return
+	}
+	return
+}
+
+// TODO
+// - Have a player API to play, play next, or add to queue
+// - Have the ability to parse custom packet to play sound
+// - Use PEP to display tunes status
+// - Ability to "speak" messages