package main

import (
	"context"
	"fmt"

	"github.com/matrix-org/dendrite/common/basecomponent"

	"github.com/libp2p/go-libp2p"
	circuit "github.com/libp2p/go-libp2p-circuit"
	crypto "github.com/libp2p/go-libp2p-core/crypto"
	routing "github.com/libp2p/go-libp2p-core/routing"

	host "github.com/libp2p/go-libp2p-core/host"
	dht "github.com/libp2p/go-libp2p-kad-dht"
	pubsub "github.com/libp2p/go-libp2p-pubsub"
	"github.com/matrix-org/gomatrixserverlib"

	"github.com/matrix-org/dendrite/common/config"
)

// P2PDendrite is a Peer-to-Peer variant of BaseDendrite.
type P2PDendrite struct {
	Base basecomponent.BaseDendrite

	// Store our libp2p object so that we can make outgoing connections from it
	// later
	LibP2P        host.Host
	LibP2PContext context.Context
	LibP2PCancel  context.CancelFunc
	LibP2PDHT     *dht.IpfsDHT
	LibP2PPubsub  *pubsub.PubSub
}

// NewP2PDendrite creates a new instance to be used by a component.
// The componentName is used for logging purposes, and should be a friendly name
// of the component running, e.g. SyncAPI.
func NewP2PDendrite(cfg *config.Dendrite, componentName string) *P2PDendrite {
	// baseDendrite :=
	ctx, cancel := context.WithCancel(context.Background())

	privKey, err := crypto.UnmarshalEd25519PrivateKey(cfg.Matrix.PrivateKey[:])
	if err != nil {
		panic(err)
	}

	//defaultIP6ListenAddr, _ := multiaddr.NewMultiaddr("/ip6/::/tcp/0")
	var libp2pdht *dht.IpfsDHT
	libp2p, err := libp2p.New(ctx,
		libp2p.Identity(privKey),
		libp2p.DefaultListenAddrs,
		//libp2p.ListenAddrs(defaultIP6ListenAddr),
		libp2p.DefaultTransports,
		libp2p.Routing(func(h host.Host) (r routing.PeerRouting, err error) {
			libp2pdht, err = dht.New(ctx, h)
			if err != nil {
				return nil, err
			}
			libp2pdht.Validator = LibP2PValidator{}
			r = libp2pdht
			return
		}),
		libp2p.EnableAutoRelay(),
		libp2p.EnableRelay(circuit.OptHop),
	)
	if err != nil {
		panic(err)
	}

	libp2ppubsub, err := pubsub.NewFloodSub(context.Background(), libp2p, []pubsub.Option{
		pubsub.WithMessageSigning(true),
	}...)
	if err != nil {
		panic(err)
	}

	fmt.Println("Our public key:", privKey.GetPublic())
	fmt.Println("Our node ID:", libp2p.ID())
	fmt.Println("Our addresses:", libp2p.Addrs())

	cfg.Matrix.ServerName = gomatrixserverlib.ServerName(libp2p.ID().String())

	baseDendrite := basecomponent.NewBaseDendrite(cfg, componentName)

	return &P2PDendrite{
		Base:          *baseDendrite,
		LibP2P:        libp2p,
		LibP2PContext: ctx,
		LibP2PCancel:  cancel,
		LibP2PDHT:     libp2pdht,
		LibP2PPubsub:  libp2ppubsub,
	}
}