Compare commits

..

7 Commits

9 changed files with 193 additions and 165 deletions

3
.gitignore vendored
View File

@ -16,3 +16,6 @@ ambiance
public/
node_modules/
src/node_modules
*.aac
*.opus
*.mp3

View File

@ -1,18 +1,61 @@
FROM golang:1.21-bookworm as builder
ENV FFMPEG_VERSION=5.1.2
ENV MPD_VERSION=0.23.14
RUN apt-get update && apt-get -y install \
libopus-dev libopusfile-dev \
meson g++ nasm \
libfmt-dev \
libpcre2-dev \
libmad0-dev libmpg123-dev libid3tag0-dev \
libflac-dev libvorbis-dev libopus-dev libogg-dev \
libadplug-dev libaudiofile-dev libsndfile1-dev libfaad-dev \
libsamplerate0-dev libsoxr-dev \
libcurl4-gnutls-dev \
libboost-dev \
zlib1g-dev
WORKDIR /tmp
RUN curl -LOs http://ffmpeg.org/releases/ffmpeg-${FFMPEG_VERSION}.tar.xz && tar xvf ffmpeg-${FFMPEG_VERSION}.tar.xz && \
cd ffmpeg-${FFMPEG_VERSION} && \
./configure \
--enable-version3 \
--enable-gpl \
--enable-nonfree \
--enable-small \
--enable-libvorbis \
--enable-libopus \
--enable-postproc \
--enable-openssl \
--disable-debug && \
make && \
make install
RUN curl -LOs https://www.musicpd.org/download/mpd/0.23/mpd-${MPD_VERSION}.tar.xz && tar xvf mpd-${MPD_VERSION}.tar.xz && \
cd mpd-${MPD_VERSION} && \
meson . output/release --buildtype=minsize -Db_ndebug=true && \
ninja -C output/release
WORKDIR /src
COPY . .
RUN apt-get update && apt-get -y install libopus-dev libopusfile-dev && \
go build
RUN go build
FROM debian:bookworm-slim
RUN apt-get update && apt-get -y install \
ca-certificates \
libopus-dev libopusfile-dev \
mpd ffmpeg curl && \
libfmt-dev \
libpcre2-dev \
libmad0-dev libmpg123-dev libid3tag0-dev \
libflac-dev libvorbis-dev libopus-dev libogg-dev \
libadplug-dev libaudiofile-dev libsndfile1-dev libfaad-dev \
libsamplerate0-dev libsoxr-dev \
libcurl4-gnutls-dev \
curl && \
curl -L https://github.com/badaix/snapcast/releases/download/v0.27.0/snapclient_0.27.0-1_without-pulse_amd64.deb -o /tmp/snapcast.deb && \
curl -L http://ftp.no.debian.org/debian/pool/main/f/flac/libflac8_1.3.3-2+deb11u2_amd64.deb -o /tmp/libflac8.deb && \
apt -y install /tmp/snapcast.deb /tmp/libflac8.deb
apt -y install /tmp/snapcast.deb /tmp/libflac8.deb && rm -rf /tmp/*.deb
COPY --from=builder /src/dndmusicbot /app/
COPY --from=builder /tmp/mpd-0.23.14/output/release/mpd /usr/local/bin
COPY --from=builder /usr/local/bin/ffmpeg /usr/local/bin
COPY mp3 /app/mp3
ADD tmpl /app/tmpl
WORKDIR /app
ENTRYPOINT [ "/app/dndmusicbot" ]

View File

@ -69,20 +69,17 @@ func GetAmbiances() (amb []Ambiance, err error) {
func AddAmbiance(uri, title string) (Ambiance, error) {
var amb Ambiance
msg := make(map[string]interface{})
msg["event"] = "ambiance_add_start"
data := make(map[string]string)
data["name"] = title
msg["payload"] = data
ws_msg <- msg
ev := Event{"ambiance_add_start", map[string]string{
"name": title,
}}
go ws.SendEvent(ev)
defer func() {
msg = make(map[string]interface{})
msg["event"] = "ambiance_add_finish"
data = make(map[string]string)
data["name"] = title
msg["payload"] = data
ws_msg <- msg
ev = Event{"ambiance_add_finish", map[string]string{
"name": title,
}}
go ws.SendEvent(ev)
}()
tmpfile, err := exec.Command("mktemp", "/tmp/dnd_XXXXXXXXXXXX.opus").Output()
@ -134,16 +131,13 @@ func AddAmbiance(uri, title string) (Ambiance, error) {
log.Printf("Start ffmpeg to extract audio to %s", string(tmpfile))
msg = make(map[string]interface{})
msg["event"] = "ambiance_encode_start"
data = make(map[string]string)
data["name"] = title
msg["payload"] = data
ws_msg <- msg
ev = Event{"ambiance_encode_start", map[string]string{
"name": title,
}}
msg = make(map[string]interface{})
msg["event"] = "ambiance_encode_progress"
data = make(map[string]string)
go ws.SendEvent(ev)
data := make(map[string]string)
data["name"] = title
scanner := bufio.NewScanner(ffprogress)
@ -162,8 +156,7 @@ func AddAmbiance(uri, title string) (Ambiance, error) {
data["percent"] = percent
}
msg["payload"] = data
ws_msg <- msg
go ws.SendEvent(Event{"ambiance_encode_progress", data})
})
}
@ -176,15 +169,14 @@ func AddAmbiance(uri, title string) (Ambiance, error) {
return amb, err
}
msg = make(map[string]interface{})
msg["event"] = "ambiance_encode_complete"
data = make(map[string]string)
data["name"] = title
msg["payload"] = data
ws_msg <- msg
ev = Event{"ambiance_encode_complete", map[string]string{
"name": title,
}}
go ws.SendEvent(ev)
id := uuid.New()
fn := filepath.Join(config.GetString("ambiance.path"), fmt.Sprintf("%s.opus", id.String()))
fn := filepath.Join(config.GetString("ambiance.path"), fmt.Sprintf("%s.aac", id.String()))
log.Printf("Moving to %s", fn)

121
events.go
View File

@ -148,74 +148,70 @@ func (app *App) volset(payload ...interface{}) {
amb_volume.Volume = vol
}
app.sendVolume()
ev := Event{"volume", map[string]float64{
"playlist": pl_volume.Volume,
"ambiance": amb_volume.Volume,
}}
go ws.SendEvent(ev)
}
func (app *App) sendVolume() {
msg := make(map[string]interface{})
out := make(map[string]float64)
msg["event"] = "volume"
out["playlist"] = pl_volume.Volume
out["ambiance"] = amb_volume.Volume
msg["payload"] = out
ws_msg <- msg
}
func (app *App) songInfoEvent(event string) (ev Event, err error) {
ev.Event = event
func (app *App) songInfoEvent(event string) map[string]interface{} {
msg := make(map[string]interface{})
msg["event"] = event
status, err := app.mpd.Status()
if err != nil {
log.Println(err)
return nil
return
}
cur, err := app.mpd.CurrentSong()
if err != nil {
log.Println(err)
return nil
return
}
info := new(SongInfo)
if status["state"] != "play" {
info.Pause = true
msg["payload"] = info
return msg
ev.Payload = info
return
}
duration, ok := status["duration"]
if ok && duration != "" {
slen, err := strconv.ParseFloat(duration, 64)
var slen float64
slen, err = strconv.ParseFloat(duration, 64)
if err != nil {
log.Println(err)
return nil
return
}
info.Length = time.Duration(slen * float64(time.Second)).Milliseconds()
}
elapsed, ok := status["elapsed"]
if ok && elapsed != "" {
spos, err := strconv.ParseFloat(elapsed, 64)
var spos float64
spos, err = strconv.ParseFloat(elapsed, 64)
if err != nil {
log.Println(err)
return nil
return
}
info.Position = time.Duration(spos * float64(time.Second)).Milliseconds()
}
album, ok := cur["Album"]
if ok {
plid, err := uuid.ParseBytes([]byte(album))
var plid uuid.UUID
plid, err = uuid.ParseBytes([]byte(album))
if err != nil {
log.Println(err)
return nil
return
}
pl, err := app.GetPlaylist(plid)
var pl *Playlist
pl, err = app.GetPlaylist(plid)
if err != nil {
log.Println(err)
return nil
return
}
info.Playlist = pl.Id
@ -237,9 +233,9 @@ func (app *App) songInfoEvent(event string) map[string]interface{} {
info.Song = location
}
msg["payload"] = *info
ev.Payload = *info
return msg
return ev, nil
}
func (app *App) ambiancePlay(payload ...interface{}) {
@ -292,15 +288,13 @@ func (app *App) ambiancePlay(payload ...interface{}) {
log.Fatal(err)
}
msg := make(map[string]interface{})
out := make(map[string]interface{})
app.curamb = amb
msg["event"] = "ambiance_play"
out["id"] = id
msg["payload"] = out
ws_msg <- msg
ev := Event{"ambiance_play", map[string]string{
"id": id,
}}
go ws.SendEvent(ev)
}
func (app *App) ambianceStop(payload ...interface{}) {
@ -319,10 +313,7 @@ func (app *App) ambianceStop(payload ...interface{}) {
log.Fatal(err)
}
msg := make(map[string]interface{})
msg["event"] = "ambiance_stop"
ws_msg <- msg
go ws.SendEvent(Event{"ambiance_stop", nil})
}
func (app *App) ambianceAdd(payload ...interface{}) {
@ -359,13 +350,12 @@ func (app *App) ambianceAdd(payload ...interface{}) {
return
}
msg := make(map[string]interface{})
out := make(map[string]interface{})
msg["event"] = "ambiance_add"
out["title"] = amb.Title
out["id"] = amb.Id
msg["payload"] = out
ws_msg <- msg
ev := Event{"ambiance_add", map[string]string{
"title": amb.Title,
"id": amb.Id,
}}
go ws.SendEvent(ev)
}
func (app *App) songPosition(payload ...interface{}) {
@ -380,18 +370,16 @@ func (app *App) songPosition(payload ...interface{}) {
}
l.Do(func() {
msg := make(map[string]interface{})
out := make(map[string]interface{})
slen, _ := strconv.ParseFloat(status["duration"], 64)
spos, _ := strconv.ParseFloat(status["elapsed"], 64)
msg["event"] = "song_position"
out["len"] = time.Duration(slen * float64(time.Second)).Milliseconds()
out["position"] = time.Duration(spos * float64(time.Second)).Milliseconds()
ev := Event{"song_position", map[string]int64{
"len": time.Duration(slen * float64(time.Second)).Milliseconds(),
"position": time.Duration(spos * float64(time.Second)).Milliseconds(),
}}
msg["payload"] = out
ws_msg <- msg
go ws.SendEvent(ev)
})
}
@ -399,10 +387,12 @@ func (app *App) songPosition(payload ...interface{}) {
func (app *App) songInfo(payload ...interface{}) {
log.Println("song_info event received")
msg := app.songInfoEvent("song_info")
if msg != nil {
ws_msg <- msg
ev, err := app.songInfoEvent("song_info")
if err != nil {
log.Println(err)
return
}
go ws.SendEvent(ev)
}
func (app *App) stop(payload ...interface{}) {
@ -415,10 +405,7 @@ func (app *App) stop(payload ...interface{}) {
app.mpd.Stop()
app.plmutex.Unlock()
msg := make(map[string]interface{})
msg["event"] = "stop"
ws_msg <- msg
go ws.SendEvent(Event{"stop", nil})
}
func (app *App) prevSong(payload ...interface{}) {
@ -492,12 +479,12 @@ func (app *App) addPlaylist(payload ...interface{}) {
log.Println("Error getting youtube playlist info,", plid)
}
msg := make(map[string]interface{})
ev := Event{"new_playlist", map[string]string{
"url": id.String(),
"title": pltitle,
}}
msg["event"] = "new_playlist"
msg["payload"] = map[string]string{"url": id.String(), "title": pltitle}
ws_msg <- msg
go ws.SendEvent(ev)
}
func (app *App) loadPlaylist(payload ...interface{}) {

1
mpd.go
View File

@ -41,7 +41,6 @@ func init() {
}
mpd_conf := config.GetStringMapString("mpd")
mpd_conf["proxy_port"] = strconv.Itoa(proxy_port)
err = t.Execute(f, mpd_conf)
if err != nil {

View File

@ -69,7 +69,7 @@ func init() {
return
}
err = handleWS(conn)
err = ws.join(conn)
if err != nil {
log.Printf("WS connection closed, %v\n", r.RemoteAddr)
}

View File

@ -1,6 +1,6 @@
{
"scripts": {
"build": "esbuild app/root.tsx --bundle --outfile=../public_test/script.js"
"build": "esbuild app/root.tsx --bundle --outfile=/public/script.js"
},
"dependencies": {
"@types/react": "^18.2.33",

66
ws.go
View File

@ -12,18 +12,21 @@ import (
"github.com/kataras/go-events"
)
type Websocket struct {
sync.Mutex
clients *bcast.Group
}
var ws *Websocket
func init() {
log.Println("ws.go loading..")
go ws_clients.Broadcast(0)
ws_msg = make(chan interface{})
ws = new(Websocket)
go func() {
var msg interface{}
for {
msg = <-ws_msg
ws_clients.Send(msg)
}
}()
ws.clients = bcast.NewGroup()
go ws.clients.Broadcast(0)
log.Println("ws.go done.")
}
@ -33,12 +36,19 @@ type WSmsg struct {
Payload json.RawMessage
}
var ws_clients = bcast.NewGroup()
var ws_msg chan interface{}
var WSMutex = &sync.Mutex{}
type Event struct {
Event string `json:"event"`
Payload any `json:"payload,omitempty"`
}
func handleWS(c *websocket.Conn) error {
memb := ws_clients.Join()
func (ws *Websocket) SendEvent(e Event) {
ws.Lock()
ws.clients.Send(e)
ws.Unlock()
}
func (ws *Websocket) join(c *websocket.Conn) error {
memb := ws.clients.Join()
defer memb.Close()
ctx, cancel := context.WithCancel(context.Background())
@ -64,29 +74,27 @@ func handleWS(c *websocket.Conn) error {
return nil
})
msg := app.songInfoEvent("song_info")
msg, err := app.songInfoEvent("song_info")
if err != nil {
return err
}
vol := make(map[string]interface{})
volout := make(map[string]float64)
vol["event"] = "volume"
volout["playlist"] = pl_volume.Volume
volout["ambiance"] = amb_volume.Volume
vol["payload"] = volout
vol := Event{"volume", map[string]float64{
"playlist": pl_volume.Volume,
"ambiance": amb_volume.Volume,
}}
c.SetWriteDeadline(time.Now().Add(10 * time.Second))
c.WriteJSON(msg)
c.WriteJSON(vol)
if app.ambiance.Len() > 0 {
msg := make(map[string]interface{})
out := make(map[string]interface{})
msg["event"] = "ambiance_play"
out["id"] = app.curamb.Id
msg["payload"] = out
msg := Event{"ambiance_play", map[string]string{
"id": app.curamb.Id,
}}
c.WriteJSON(msg)
} else {
msg := make(map[string]interface{})
msg["event"] = "ambiance_stop"
msg := Event{"ambiance_stop", nil}
c.WriteJSON(msg)
}
@ -97,6 +105,6 @@ func handleWS(c *websocket.Conn) error {
return err
}
app.events.Emit(events.EventName(msg.Event), msg.Payload)
app.events.Emit(events.EventName(msg.Event), msg.Payload, memb)
}
}

58
ytdl.go
View File

@ -1,23 +1,42 @@
package main
import (
"bufio"
"encoding/json"
"fmt"
"log"
"math"
"net/url"
"os"
"os/exec"
"strconv"
"time"
"golang.org/x/time/rate"
)
var prate = rate.Sometimes{Interval: 1 * time.Second}
var yturl = "https://youtu.be/%s"
// var yturl = "https://youtu.be/%s"
func YTUrl(uri string) (vid string, err error) {
u, err := url.Parse(uri)
if err != nil {
return "", err
}
switch u.Host {
case "youtu.be":
vid = u.Path[1:]
case "m.youtube.com":
fallthrough
case "youtube.com":
fallthrough
case "www.youtube.com":
vid = u.Query().Get("v")
}
if vid == "" {
return vid, fmt.Errorf("unable to parse vid")
}
return
}
/*
func NewYTdlUrl(vid string) ([]byte, error) {
ytdl := config.GetString("youtube.ytdl")
yt := exec.Command(
@ -132,29 +151,6 @@ func NewYTdl(vid string) ([]byte, error) {
return tmpfile, nil
}
func YTUrl(uri string) (vid string, err error) {
u, err := url.Parse(uri)
if err != nil {
return "", err
}
switch u.Host {
case "youtu.be":
vid = u.Path[1:]
case "m.youtube.com":
fallthrough
case "youtube.com":
fallthrough
case "www.youtube.com":
vid = u.Query().Get("v")
}
if vid == "" {
return vid, fmt.Errorf("unable to parse vid")
}
return
}
/*
func DownloadAmbiance(uri string, name string) error {