34 lines
649 B
Go
34 lines
649 B
Go
|
package control
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io"
|
||
|
|
|
||
|
|
"github.com/pterodactyl/wings/api/websockets"
|
||
|
|
)
|
||
|
|
|
||
|
|
type ConsoleHandler struct {
|
||
|
|
Websockets *websockets.Collection
|
||
|
|
HandlerFunc *func(string)
|
||
|
|
}
|
||
|
|
|
||
|
|
var _ io.Writer = ConsoleHandler{}
|
||
|
|
|
||
|
|
func (c ConsoleHandler) Write(b []byte) (n int, e error) {
|
||
|
|
l := make([]byte, len(b))
|
||
|
|
copy(l, b)
|
||
|
|
line := string(l)
|
||
|
|
m := websockets.Message{
|
||
|
|
Type: websockets.MessageTypeConsole,
|
||
|
|
Payload: websockets.ConsolePayload{
|
||
|
|
Line: line,
|
||
|
|
Level: websockets.ConsoleLevelPlain,
|
||
|
|
Source: websockets.ConsoleSourceServer,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
c.Websockets.Broadcast <- m
|
||
|
|
if c.HandlerFunc != nil {
|
||
|
|
(*c.HandlerFunc)(line)
|
||
|
|
}
|
||
|
|
return len(b), nil
|
||
|
|
}
|