mirror of
https://source.quilibrium.com/quilibrium/ceremonyclient.git
synced 2024-11-11 02:35:18 +00:00
140 lines
3.2 KiB
Go
140 lines
3.2 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
|
|
"github.com/libp2p/go-libp2p"
|
|
kaddht "github.com/libp2p/go-libp2p-kad-dht"
|
|
"github.com/libp2p/go-libp2p/core/host"
|
|
"github.com/libp2p/go-libp2p/core/network"
|
|
"github.com/libp2p/go-libp2p/core/peer"
|
|
"github.com/libp2p/go-libp2p/core/routing"
|
|
"github.com/libp2p/go-libp2p/p2p/discovery/mdns"
|
|
drouting "github.com/libp2p/go-libp2p/p2p/discovery/routing"
|
|
dutil "github.com/libp2p/go-libp2p/p2p/discovery/util"
|
|
"github.com/libp2p/go-libp2p/p2p/muxer/mplex"
|
|
"github.com/libp2p/go-libp2p/p2p/muxer/yamux"
|
|
tls "github.com/libp2p/go-libp2p/p2p/security/tls"
|
|
"github.com/libp2p/go-libp2p/p2p/transport/tcp"
|
|
"github.com/libp2p/go-libp2p/p2p/transport/websocket"
|
|
|
|
"github.com/multiformats/go-multiaddr"
|
|
)
|
|
|
|
type discoveryNotifee struct {
|
|
h host.Host
|
|
ctx context.Context
|
|
}
|
|
|
|
func (m *discoveryNotifee) HandlePeerFound(pi peer.AddrInfo) {
|
|
if m.h.Network().Connectedness(pi.ID) != network.Connected {
|
|
fmt.Printf("Found %s!\n", pi.ID.ShortString())
|
|
m.h.Connect(m.ctx, pi)
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
transports := libp2p.ChainOptions(
|
|
libp2p.Transport(tcp.NewTCPTransport),
|
|
libp2p.Transport(websocket.New),
|
|
)
|
|
|
|
muxers := libp2p.ChainOptions(
|
|
libp2p.Muxer("/yamux/1.0.0", yamux.DefaultTransport),
|
|
libp2p.Muxer("/mplex/6.7.0", mplex.DefaultTransport),
|
|
)
|
|
|
|
security := libp2p.Security(tls.ID, tls.New)
|
|
|
|
listenAddrs := libp2p.ListenAddrStrings(
|
|
"/ip4/0.0.0.0/tcp/0",
|
|
"/ip4/0.0.0.0/tcp/0/ws",
|
|
)
|
|
|
|
var dht *kaddht.IpfsDHT
|
|
newDHT := func(h host.Host) (routing.PeerRouting, error) {
|
|
var err error
|
|
dht, err = kaddht.New(ctx, h)
|
|
return dht, err
|
|
}
|
|
routing := libp2p.Routing(newDHT)
|
|
|
|
host, err := libp2p.New(
|
|
transports,
|
|
listenAddrs,
|
|
muxers,
|
|
security,
|
|
routing,
|
|
)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
// TODO: Replace our stream handler with a pubsub instance, and a handler
|
|
// to field incoming messages on our topic.
|
|
host.SetStreamHandler(chatProtocol, chatHandler)
|
|
|
|
for _, addr := range host.Addrs() {
|
|
fmt.Println("Listening on", addr)
|
|
}
|
|
|
|
targetAddr, err := multiaddr.NewMultiaddr("/ip4/127.0.0.1/tcp/63785/p2p/QmWjz6xb8v9K4KnYEwP5Yk75k5mMBCehzWFLCvvQpYxF3d")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
targetInfo, err := peer.AddrInfoFromP2pAddr(targetAddr)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
err = host.Connect(ctx, *targetInfo)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "connecting to bootstrap: %s", err)
|
|
} else {
|
|
fmt.Println("Connected to", targetInfo.ID)
|
|
}
|
|
|
|
notifee := &discoveryNotifee{h: host, ctx: ctx}
|
|
mdns := mdns.NewMdnsService(host, "", notifee)
|
|
if err := mdns.Start(); err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
err = dht.Bootstrap(ctx)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
routingDiscovery := drouting.NewRoutingDiscovery(dht)
|
|
dutil.Advertise(ctx, routingDiscovery, string(chatProtocol))
|
|
peers, err := dutil.FindPeers(ctx, routingDiscovery, string(chatProtocol))
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
for _, peer := range peers {
|
|
notifee.HandlePeerFound(peer)
|
|
}
|
|
|
|
donec := make(chan struct{}, 1)
|
|
go chatInputLoop(ctx, host, donec)
|
|
|
|
stop := make(chan os.Signal, 1)
|
|
signal.Notify(stop, syscall.SIGINT)
|
|
|
|
select {
|
|
case <-stop:
|
|
host.Close()
|
|
os.Exit(0)
|
|
case <-donec:
|
|
host.Close()
|
|
}
|
|
}
|