121 lines
2.9 KiB
Go
121 lines
2.9 KiB
Go
package app
|
|
|
|
import (
|
|
"encoding/gob"
|
|
"log"
|
|
"net"
|
|
"os"
|
|
"path"
|
|
"time"
|
|
)
|
|
|
|
func genClientStatus() ClientStatus {
|
|
cs := make(ClientStatus)
|
|
matchesLock.Lock()
|
|
|
|
// Painful data manipulation
|
|
for filter, filterMatches := range matches {
|
|
if cs[filter.stream.name] == nil {
|
|
cs[filter.stream.name] = make(map[string]MapPatternStatus)
|
|
}
|
|
if cs[filter.stream.name][filter.name] == nil {
|
|
cs[filter.stream.name][filter.name] = make(MapPatternStatus)
|
|
}
|
|
for pattern, patternMatches := range filterMatches {
|
|
var ps PatternStatus
|
|
cs[filter.stream.name][filter.name][pattern] = &ps
|
|
|
|
ps.Matches = len(patternMatches)
|
|
}
|
|
}
|
|
|
|
matchesLock.Unlock()
|
|
actionsLock.Lock()
|
|
|
|
// Painful data manipulation
|
|
for action, pendingActions := range actions {
|
|
if cs[action.filter.stream.name] == nil {
|
|
cs[action.filter.stream.name] = make(map[string]MapPatternStatus)
|
|
}
|
|
if cs[action.filter.stream.name][action.filter.name] == nil {
|
|
cs[action.filter.stream.name][action.filter.name] = make(MapPatternStatus)
|
|
}
|
|
for pattern, patternPendingActions := range pendingActions {
|
|
if cs[action.filter.stream.name][action.filter.name][pattern] == nil {
|
|
var ps PatternStatus
|
|
cs[action.filter.stream.name][action.filter.name][pattern] = &ps
|
|
}
|
|
var ps *PatternStatus
|
|
ps = cs[action.filter.stream.name][action.filter.name][pattern]
|
|
ps.Actions = make(map[string][]string)
|
|
|
|
for _, t := range patternPendingActions {
|
|
ps.Actions[action.name] = append(ps.Actions[action.name], t.Format(time.DateTime))
|
|
}
|
|
}
|
|
}
|
|
actionsLock.Unlock()
|
|
return cs
|
|
}
|
|
|
|
func createOpenSocket() net.Listener {
|
|
err := os.MkdirAll(path.Dir(*SocketPath), 0755)
|
|
if err != nil {
|
|
log.Fatalln("FATAL Failed to create socket directory")
|
|
}
|
|
_, err = os.Stat(*SocketPath)
|
|
if err == nil {
|
|
log.Println("WARN socket", SocketPath, "already exists: Is the daemon already running? Deleting.")
|
|
err = os.Remove(*SocketPath)
|
|
if err != nil {
|
|
log.Fatalln("FATAL Failed to remove socket:", err)
|
|
}
|
|
}
|
|
ln, err := net.Listen("unix", *SocketPath)
|
|
if err != nil {
|
|
log.Fatalln("FATAL Failed to create socket:", err)
|
|
}
|
|
return ln
|
|
}
|
|
|
|
// Handle connections
|
|
func SocketManager() {
|
|
ln := createOpenSocket()
|
|
defer ln.Close()
|
|
for {
|
|
conn, err := ln.Accept()
|
|
if err != nil {
|
|
log.Println("ERROR Failed to open connection from cli:", err)
|
|
continue
|
|
}
|
|
go func(conn net.Conn) {
|
|
defer conn.Close()
|
|
var request Request
|
|
var response Response
|
|
|
|
err := gob.NewDecoder(conn).Decode(&request)
|
|
if err != nil {
|
|
log.Println("ERROR Invalid Message from cli:", err)
|
|
return
|
|
}
|
|
|
|
switch request.Request {
|
|
case Show:
|
|
response.ClientStatus = genClientStatus()
|
|
case Flush:
|
|
// FIXME reimplement flush
|
|
response.Number = 0
|
|
default:
|
|
log.Println("ERROR Invalid Message from cli: unrecognised Request type")
|
|
return
|
|
}
|
|
|
|
err = gob.NewEncoder(conn).Encode(response)
|
|
if err != nil {
|
|
log.Println("ERROR Can't respond to cli:", err)
|
|
return
|
|
}
|
|
}(conn)
|
|
}
|
|
}
|