ricochet-go/backend/rpc.go

152 lines
4.0 KiB
Go

package main
import (
"errors"
ricochet "github.com/special/notricochet/core"
rpc "github.com/special/notricochet/rpc"
"golang.org/x/net/context"
"log"
"time"
)
var NotImplementedError error = errors.New("Not implemented")
type RicochetCore struct {
config *ricochet.Config
network *ricochet.Network
identity *ricochet.Identity
}
func (core *RicochetCore) Config() *ricochet.Config {
return core.config
}
func (core *RicochetCore) Network() *ricochet.Network {
return core.network
}
func (core *RicochetCore) Identity() *ricochet.Identity {
return core.identity
}
func (core *RicochetCore) GetServerStatus(ctx context.Context, req *rpc.ServerStatusRequest) (*rpc.ServerStatusReply, error) {
if req.RpcVersion != 1 {
return nil, errors.New("Unsupported RPC protocol version")
}
return &rpc.ServerStatusReply{
RpcVersion: 1,
ServerVersion: "0.0.0",
}, nil
}
func (core *RicochetCore) MonitorNetwork(req *rpc.MonitorNetworkRequest, stream rpc.RicochetCore_MonitorNetworkServer) error {
events := core.Network().EventMonitor().Subscribe(20)
defer core.Network().EventMonitor().Unsubscribe(events)
// Send initial status event
{
event := core.Network().GetStatus()
if err := stream.Send(&event); err != nil {
return err
}
}
for {
event, ok := (<-events).(rpc.NetworkStatus)
if !ok {
break
}
log.Printf("RPC monitor event: %v", event)
if err := stream.Send(&event); err != nil {
return err
}
}
return nil
}
func (core *RicochetCore) StartNetwork(ctx context.Context, req *rpc.StartNetworkRequest) (*rpc.NetworkStatus, error) {
// err represents the result of the first connection attempt, but as long
// as 'ok' is true, the network has started and this call was successful.
ok, err := core.Network().Start("tcp://127.0.0.1:9051", "")
if !ok {
return nil, err
}
status := core.Network().GetStatus()
return &status, nil
}
func (core *RicochetCore) StopNetwork(ctx context.Context, req *rpc.StopNetworkRequest) (*rpc.NetworkStatus, error) {
core.Network().Stop()
status := core.Network().GetStatus()
return &status, nil
}
func (core *RicochetCore) GetIdentity(ctx context.Context, req *rpc.IdentityRequest) (*rpc.Identity, error) {
reply := rpc.Identity{
Address: core.Identity().Address(),
}
return &reply, nil
}
func (core *RicochetCore) MonitorContacts(req *rpc.MonitorContactsRequest, stream rpc.RicochetCore_MonitorContactsServer) error {
// Populate
contacts := core.Identity().ContactList().Contacts()
for _, contact := range contacts {
data := &rpc.Contact{
Id: int32(contact.Id()),
Address: contact.Address(),
Nickname: contact.Nickname(),
WhenCreated: contact.WhenCreated().Format(time.RFC3339),
LastConnected: contact.LastConnected().Format(time.RFC3339),
}
event := &rpc.ContactEvent{
Type: rpc.ContactEvent_POPULATE,
Subject: &rpc.ContactEvent_Contact{
Contact: data,
},
}
if err := stream.Send(event); err != nil {
return err
}
}
// Terminate populate list with a null subject
{
event := &rpc.ContactEvent{
Type: rpc.ContactEvent_POPULATE,
}
if err := stream.Send(event); err != nil {
return err
}
}
return NotImplementedError
}
func (core *RicochetCore) AddContactRequest(ctx context.Context, req *rpc.ContactRequest) (*rpc.Contact, error) {
return nil, NotImplementedError
}
func (core *RicochetCore) UpdateContact(ctx context.Context, req *rpc.Contact) (*rpc.Contact, error) {
return nil, NotImplementedError
}
func (core *RicochetCore) DeleteContact(ctx context.Context, req *rpc.DeleteContactRequest) (*rpc.DeleteContactReply, error) {
return nil, NotImplementedError
}
func (core *RicochetCore) AcceptInboundRequest(ctx context.Context, req *rpc.ContactRequest) (*rpc.Contact, error) {
return nil, NotImplementedError
}
func (core *RicochetCore) RejectInboundRequest(ctx context.Context, req *rpc.ContactRequest) (*rpc.RejectInboundRequestReply, error) {
return nil, NotImplementedError
}
func (core *RicochetCore) StreamConversations(stream rpc.RicochetCore_StreamConversationsServer) error {
return NotImplementedError
}