ceremonyclient/go-libp2p/examples/ipfs-camp-2019/07-Messaging/main.go
2023-08-20 23:07:43 -05:00

139 lines
2.8 KiB
Go

package main
import (
"context"
"fmt"
"os"
"os/signal"
"syscall"
"github.com/libp2p/go-libp2p"
"github.com/libp2p/go-libp2p/p2p/discovery/mdns"
"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"
kaddht "github.com/libp2p/go-libp2p-kad-dht"
pubsub "github.com/libp2p/go-libp2p-pubsub"
"github.com/libp2p/go-libp2p/core/host"
"github.com/libp2p/go-libp2p/core/peer"
"github.com/libp2p/go-libp2p/core/routing"
"github.com/multiformats/go-multiaddr"
)
type mdnsNotifee struct {
h host.Host
ctx context.Context
}
func (m *mdnsNotifee) HandlePeerFound(pi peer.AddrInfo) {
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)
}
ps, err := pubsub.NewGossipSub(ctx, host)
if err != nil {
panic(err)
}
topic, err := ps.Join(pubsubTopic)
if err != nil {
panic(err)
}
defer topic.Close()
sub, err := topic.Subscribe()
if err != nil {
panic(err)
}
// TODO: Modify this handler to use the protobufs defined in this folder
go pubsubHandler(ctx, sub)
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 {
panic(err)
}
fmt.Println("Connected to", targetInfo.ID)
mdns := mdns.NewMdnsService(host, "", &mdnsNotifee{h: host, ctx: ctx})
if err := mdns.Start(); err != nil {
panic(err)
}
err = dht.Bootstrap(ctx)
if err != nil {
panic(err)
}
donec := make(chan struct{}, 1)
// TODO: modify this chat input loop to use the protobufs defined in this
// folder.
go chatInputLoop(ctx, topic, donec)
stop := make(chan os.Signal, 1)
signal.Notify(stop, syscall.SIGINT)
select {
case <-stop:
host.Close()
os.Exit(0)
case <-donec:
host.Close()
}
}