package pushserver

import (
	"github.com/gorilla/mux"
	"github.com/matrix-org/dendrite/internal/pushgateway"
	"github.com/matrix-org/dendrite/pushserver/api"
	"github.com/matrix-org/dendrite/pushserver/consumers"
	"github.com/matrix-org/dendrite/pushserver/internal"
	"github.com/matrix-org/dendrite/pushserver/inthttp"
	"github.com/matrix-org/dendrite/pushserver/producers"
	"github.com/matrix-org/dendrite/pushserver/storage"
	roomserverAPI "github.com/matrix-org/dendrite/roomserver/api"
	"github.com/matrix-org/dendrite/setup/config"
	"github.com/matrix-org/dendrite/setup/kafka"
	"github.com/matrix-org/dendrite/setup/process"
	uapi "github.com/matrix-org/dendrite/userapi/api"
	"github.com/sirupsen/logrus"
)

// AddInternalRoutes registers HTTP handlers for the internal API. Invokes functions
// on the given input API.
func AddInternalRoutes(router *mux.Router, intAPI api.PushserverInternalAPI) {
	inthttp.AddRoutes(intAPI, router)
}

// NewInternalAPI returns a concerete implementation of the internal API. Callers
// can call functions directly on the returned API or via an HTTP interface using AddInternalRoutes.
func NewInternalAPI(
	cfg *config.PushServer,
	process *process.ProcessContext,
	pgClient pushgateway.Client,
	rsAPI roomserverAPI.RoomserverInternalAPI,
	userAPI uapi.UserInternalAPI,
) api.PushserverInternalAPI {
	consumer, _ := kafka.SetupConsumerProducer(&cfg.Matrix.Kafka)

	db, err := storage.Open(&cfg.Database)
	if err != nil {
		logrus.WithError(err).Panicf("failed to connect to push server db")
	}

	_, producer := kafka.SetupConsumerProducer(&cfg.Matrix.Kafka)
	syncProducer := producers.NewSyncAPI(
		db,
		producer,
		// TODO: user API should handle syncs for account data. Right now,
		// it's handled by clientapi, and hence uses its topic. When user
		// API handles it for all account data, we can remove it from
		// here.
		cfg.Matrix.Kafka.TopicFor(config.TopicOutputClientData),
		cfg.Matrix.Kafka.TopicFor(config.TopicOutputNotificationData),
	)

	psAPI := internal.NewPushserverAPI(
		cfg, db, userAPI, syncProducer,
	)

	caConsumer := consumers.NewOutputClientDataConsumer(
		process, cfg, consumer, db, pgClient, userAPI, syncProducer,
	)
	if err := caConsumer.Start(); err != nil {
		logrus.WithError(err).Panic("failed to start push server clientapi consumer")
	}

	eduConsumer := consumers.NewOutputReceiptEventConsumer(
		process, cfg, consumer, db, pgClient, syncProducer,
	)
	if err := eduConsumer.Start(); err != nil {
		logrus.WithError(err).Panic("failed to start push server EDU consumer")
	}

	rsConsumer := consumers.NewOutputRoomEventConsumer(
		process, cfg, consumer, db, pgClient, psAPI, rsAPI, syncProducer,
	)
	if err := rsConsumer.Start(); err != nil {
		logrus.WithError(err).Panic("failed to start push server room server consumer")
	}

	return psAPI
}