86 lines
2 KiB
Go
86 lines
2 KiB
Go
package info
|
|
|
|
import (
|
|
"github.com/FAU-CDI/wisski-distillery/internal/component/exporter"
|
|
"github.com/FAU-CDI/wisski-distillery/internal/wisski"
|
|
"github.com/FAU-CDI/wisski-distillery/pkg/httpx"
|
|
"github.com/tkw1536/goprogram/status"
|
|
"github.com/tkw1536/goprogram/stream"
|
|
)
|
|
|
|
type instanceActionFunc = func(info *Info, instance *wisski.WissKI, str stream.IOStream) error
|
|
|
|
var socketInstanceActions = map[string]instanceActionFunc{
|
|
"snapshot": func(info *Info, instance *wisski.WissKI, str stream.IOStream) error {
|
|
return info.Exporter.MakeExport(
|
|
str,
|
|
exporter.ExportTask{
|
|
Dest: "",
|
|
Instance: instance,
|
|
|
|
StagingOnly: false,
|
|
},
|
|
)
|
|
},
|
|
"rebuild": func(_ *Info, instance *wisski.WissKI, str stream.IOStream) error {
|
|
return instance.Build(str, true)
|
|
},
|
|
"update": func(_ *Info, instance *wisski.WissKI, str stream.IOStream) error {
|
|
return instance.BlindUpdate(str)
|
|
},
|
|
"cron": func(_ *Info, instance *wisski.WissKI, str stream.IOStream) error {
|
|
return instance.Cron(str)
|
|
},
|
|
}
|
|
|
|
func (info *Info) serveSocket(conn httpx.WebSocketConnection) {
|
|
// read the next message to act on
|
|
message, ok := <-conn.Read()
|
|
if !ok {
|
|
return
|
|
}
|
|
|
|
// perform an action if it exists!
|
|
if action, ok := socketInstanceActions[string(message.Bytes)]; ok {
|
|
info.handleInstanceAction(conn, action)
|
|
return
|
|
}
|
|
}
|
|
|
|
func (info *Info) handleInstanceAction(conn httpx.WebSocketConnection, action instanceActionFunc) {
|
|
|
|
// read the slug
|
|
slug, ok := <-conn.Read()
|
|
if !ok {
|
|
conn.WriteText("Error reading slug")
|
|
return
|
|
}
|
|
|
|
// resolve the instance
|
|
instance, err := info.Instances.WissKI(string(slug.Bytes))
|
|
if err != nil {
|
|
conn.WriteText("Instance not found")
|
|
return
|
|
}
|
|
|
|
// build a stream
|
|
writer := &status.LineBuffer{
|
|
Line: func(line string) {
|
|
<-conn.WriteText(line)
|
|
},
|
|
FlushLineOnClose: true,
|
|
}
|
|
defer writer.Close()
|
|
|
|
str := stream.NewIOStream(writer, writer, nil, 0)
|
|
|
|
// and perform the action
|
|
{
|
|
err := action(info, instance, str)
|
|
if err != nil {
|
|
str.EPrintln(err)
|
|
return
|
|
}
|
|
str.Println("done")
|
|
}
|
|
}
|