46 lines
855 B
Go
46 lines
855 B
Go
package client
|
|
|
|
import (
|
|
"encoding/json"
|
|
|
|
"github.com/quexten/goldwarden/ipc/messages"
|
|
)
|
|
|
|
func NewVirtualClient(recv chan []byte, send chan []byte) VirtualClient {
|
|
return VirtualClient{
|
|
recv,
|
|
send,
|
|
}
|
|
}
|
|
|
|
type VirtualClient struct {
|
|
recv chan []byte
|
|
send chan []byte
|
|
}
|
|
|
|
func virtualReader(recv chan []byte) interface{} {
|
|
for {
|
|
var message messages.IPCMessage
|
|
err := json.Unmarshal(<-recv, &message)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return message
|
|
}
|
|
}
|
|
|
|
func (client VirtualClient) SendToAgent(request interface{}) (interface{}, error) {
|
|
message, err := messages.IPCMessageFromPayload(request)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
messageJson, err := json.Marshal(message)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
client.send <- messageJson
|
|
result := virtualReader(client.recv)
|
|
return messages.ParsePayload(result.(messages.IPCMessage)), nil
|
|
}
|