forked from cwtch.im/cwtch
98 lines
2.7 KiB
Go
98 lines
2.7 KiB
Go
package connections
|
|
|
|
import (
|
|
"git.mascherari.press/cwtch/peer/fetch"
|
|
"git.mascherari.press/cwtch/peer/listen"
|
|
"git.mascherari.press/cwtch/peer/send"
|
|
"git.mascherari.press/cwtch/protocol"
|
|
"github.com/s-rah/go-ricochet"
|
|
"github.com/s-rah/go-ricochet/channels"
|
|
"github.com/s-rah/go-ricochet/connection"
|
|
"github.com/s-rah/go-ricochet/identity"
|
|
"github.com/s-rah/go-ricochet/utils"
|
|
"time"
|
|
)
|
|
|
|
type PeerServerConnection struct {
|
|
connection.AutoConnectionHandler
|
|
Server string
|
|
state ConnectionState
|
|
connection connection.Connection
|
|
|
|
GroupMessageHandler func(string, *protocol.GroupMessage)
|
|
}
|
|
|
|
func NewPeerServerConnection(serverhostname string) *PeerServerConnection {
|
|
psc := new(PeerServerConnection)
|
|
psc.Server = serverhostname
|
|
psc.Init()
|
|
return psc
|
|
}
|
|
|
|
// GetState returns the current connection state
|
|
func (psc *PeerServerConnection) GetState() ConnectionState {
|
|
return psc.state
|
|
}
|
|
|
|
// Run manages the setup and teardown of a peer server connection
|
|
func (psc *PeerServerConnection) Run() error {
|
|
rc, err := goricochet.Open(psc.Server)
|
|
if err == nil {
|
|
rc.TraceLog(true)
|
|
psc.connection = *rc
|
|
psc.state = CONNECTED
|
|
pk, err := utils.GeneratePrivateKey()
|
|
if err == nil {
|
|
_, err := connection.HandleOutboundConnection(&psc.connection).ProcessAuthAsClient(identity.Initialize("cwtchpeer", pk))
|
|
if err == nil {
|
|
psc.state = AUTHENTICATED
|
|
|
|
go func() {
|
|
psc.connection.Do(func() error {
|
|
psc.connection.RequestOpenChannel("im.cwtch.server.fetch", &fetch.CwtchPeerFetchChannel{Handler: psc})
|
|
return nil
|
|
})
|
|
|
|
psc.connection.Do(func() error {
|
|
psc.connection.RequestOpenChannel("im.cwtch.server.listen", &listen.CwtchPeerListenChannel{Handler: psc})
|
|
return nil
|
|
})
|
|
}()
|
|
|
|
psc.connection.Process(psc)
|
|
}
|
|
}
|
|
}
|
|
psc.state = FAILED
|
|
return err
|
|
}
|
|
|
|
// Break makes Run() return and prevents processing, but doesn't close the connection.
|
|
func (psc *PeerServerConnection) Break() error {
|
|
return psc.connection.Break()
|
|
}
|
|
|
|
func (psc *PeerServerConnection) SendGroupMessage(gm *protocol.GroupMessage) {
|
|
psc.connection.Do(func() error {
|
|
psc.connection.RequestOpenChannel("im.cwtch.server.send", &send.CwtchPeerSendChannel{})
|
|
return nil
|
|
})
|
|
// TODO We have to wait to receive the channel result before we can continue
|
|
// We should have a better mechanism for this kindof interaction
|
|
time.Sleep(time.Second * 1)
|
|
psc.connection.Do(func() error {
|
|
channel := psc.connection.Channel("im.cwtch.server.send", channels.Outbound)
|
|
if channel != nil {
|
|
sendchannel, ok := channel.Handler.(*send.CwtchPeerSendChannel)
|
|
if ok {
|
|
sendchannel.SendGroupMessage(gm)
|
|
}
|
|
}
|
|
return nil
|
|
})
|
|
}
|
|
|
|
func (psc *PeerServerConnection) HandleGroupMessage(gm *protocol.GroupMessage) {
|
|
psc.GroupMessageHandler(psc.Server, gm)
|
|
}
|