mirror of https://github.com/qwc/backive.git
64 lines
1.3 KiB
Go
64 lines
1.3 KiB
Go
|
package backive
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"fmt"
|
||
|
|
"log"
|
||
|
|
"net"
|
||
|
|
"path"
|
||
|
|
)
|
||
|
|
|
||
|
|
type UIHandler struct {
|
||
|
|
ls net.Listener
|
||
|
|
client net.Conn
|
||
|
|
}
|
||
|
|
|
||
|
|
var mockUIAccept = func(uh *UIHandler) (net.Conn, error) {
|
||
|
|
log.Println("Calling eh.ls.Accept()")
|
||
|
|
return uh.ls.Accept()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (uh *UIHandler) Init(socketPath string) error {
|
||
|
|
log.Println("Initializing UIHandler")
|
||
|
|
var err error
|
||
|
|
dir, _ := path.Split(socketPath)
|
||
|
|
CreateDirectoryIfNotExists(dir)
|
||
|
|
uh.ls, err = net.Listen("unix", socketPath)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (uh *UIHandler) Listen() {
|
||
|
|
log.Println("Running UIHandler loop")
|
||
|
|
func() {
|
||
|
|
for {
|
||
|
|
var err error
|
||
|
|
uh.client, err = uh.ls.Accept()
|
||
|
|
if err != nil {
|
||
|
|
log.Printf("Accept failed %e\n", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (uh *UIHandler) DisplayMessage(header string, message string, level int) error {
|
||
|
|
if uh.client != nil {
|
||
|
|
var data map[string]interface{}
|
||
|
|
data["level"] = level
|
||
|
|
data["header"] = header
|
||
|
|
data["message"] = message
|
||
|
|
b, err := json.Marshal(data)
|
||
|
|
if err != nil {
|
||
|
|
log.Printf("Problem in sending message to UI: %e", err)
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
uh.client.Write(b)
|
||
|
|
return nil
|
||
|
|
} else {
|
||
|
|
log.Println("No UI client available, msg did not get delivered.")
|
||
|
|
return fmt.Errorf("No UI client available, msg not delivered.")
|
||
|
|
}
|
||
|
|
}
|