102 lines
3.0 KiB
Go
102 lines
3.0 KiB
Go
package client
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"forge.redroom.link/yves/meowlib"
|
|
"google.golang.org/protobuf/proto"
|
|
)
|
|
|
|
type InternalServer struct {
|
|
ServerData meowlib.Server `json:"server_data,omitempty"`
|
|
Presence bool `json:"presence,omitempty"`
|
|
LastCheck time.Time `json:"last_check,omitempty"`
|
|
Uptime time.Duration `json:"uptime,omitempty"`
|
|
Login string `json:"login,omitempty"`
|
|
Password string `json:"password,omitempty"`
|
|
Me meowlib.KeyPair `json:"me,omitempty"`
|
|
}
|
|
|
|
type InternalServerList struct {
|
|
Name string
|
|
Servers []InternalServer
|
|
}
|
|
|
|
func InternalServerFromUrl(url string) *InternalServer {
|
|
var is InternalServer
|
|
is.ServerData.Url = url
|
|
return &is
|
|
}
|
|
|
|
func InternalServerFromServer(server *meowlib.Server) *InternalServer {
|
|
var is InternalServer
|
|
is.ServerData = *server
|
|
is.Me = meowlib.NewKeyPair()
|
|
return &is
|
|
}
|
|
|
|
func (sl *InternalServerList) AddUrls(urls []string) {
|
|
for _, url := range urls {
|
|
sl.Servers = append(sl.Servers, *InternalServerFromUrl(url))
|
|
}
|
|
}
|
|
|
|
// AsymEncryptMessage prepares a message to send to a specific internal server
|
|
func (ints *InternalServer) AsymEncryptMessage(Message []byte) (EncryptedMessage []byte, Signature []byte, err error) {
|
|
EncryptedMessage, Signature, err = meowlib.AsymEncryptAndSign(ints.ServerData.PublicKey, ints.Me.Private, Message)
|
|
if err != nil {
|
|
fmt.Println(err.Error())
|
|
return nil, nil, err
|
|
}
|
|
return EncryptedMessage, Signature, err
|
|
}
|
|
|
|
// AsymDecryptMessage reads a message from a specific internal server
|
|
func (ints *InternalServer) AsymDecryptMessage(Message []byte, Signature []byte) (DecryptedMessage []byte, err error) {
|
|
DecryptedMessage, err = meowlib.AsymDecryptAndCheck(ints.Me.Private, ints.ServerData.PublicKey, Message, Signature)
|
|
if err != nil {
|
|
fmt.Println(err.Error())
|
|
return nil, err
|
|
}
|
|
return DecryptedMessage, err
|
|
}
|
|
|
|
// Creates a basic message to server from a single packed user message and returns it as protobuf serialized byte array
|
|
func (ints *InternalServer) BuildMessageSendingMessage(usermsg *meowlib.PackedUserMessage) ([]byte, error) {
|
|
var msg meowlib.ToServerMessage
|
|
msg.Type = "1"
|
|
msg.From = ints.Me.Public
|
|
msg.Messages = append(msg.Messages, usermsg)
|
|
out, err := proto.Marshal(&msg)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return out, nil
|
|
}
|
|
|
|
// Creates a basic message to server from a single packed user message and returns it as protobuf serialized byte array
|
|
func (ints *InternalServer) BuildMessageRequestMessage(lookupKeys []string) ([]byte, error) {
|
|
var msg meowlib.ToServerMessage
|
|
msg.Type = "1"
|
|
msg.From = ints.Me.Public
|
|
|
|
out, err := proto.Marshal(&msg)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return out, nil
|
|
}
|
|
|
|
func (ints *InternalServer) PackServerMessage(payload []byte, signature []byte) (protoPackedMessage []byte, err error) {
|
|
var msg meowlib.PackedServerMessage
|
|
msg.From = ints.Me.Public
|
|
msg.Payload = payload
|
|
msg.Signature = signature
|
|
out, err := proto.Marshal(&msg)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return out, nil
|
|
}
|