package messaging

import (
	"context"

	"github.com/go-chi/chi"
	_ "github.com/joho/godotenv/autoload"
	"github.com/spf13/cobra"
	"github.com/titpetric/factory"
	"go.uber.org/zap"

	"github.com/cortezaproject/corteza-server/messaging/commands"
	migrate "github.com/cortezaproject/corteza-server/messaging/db"
	"github.com/cortezaproject/corteza-server/messaging/rest"
	"github.com/cortezaproject/corteza-server/messaging/service"
	"github.com/cortezaproject/corteza-server/messaging/service/event"
	"github.com/cortezaproject/corteza-server/messaging/websocket"
	"github.com/cortezaproject/corteza-server/pkg/app"
	"github.com/cortezaproject/corteza-server/pkg/auth"
	"github.com/cortezaproject/corteza-server/pkg/corredor"
	"github.com/cortezaproject/corteza-server/pkg/scheduler"
)

type (
	App struct {
		Opts *app.Options
		Log  *zap.Logger

		ws *websocket.Websocket
	}
)

const SERVICE = "messaging"

func (app *App) Setup(log *zap.Logger, opts *app.Options) (err error) {
	app.Log = log.Named(SERVICE)
	app.Opts = opts

	scheduler.Service().OnTick(
		event.MessagingOnInterval(),
		event.MessagingOnTimestamp(),
	)

	app.ws = websocket.New(&websocket.Config{
		Timeout:     opts.Websocket.Timeout,
		PingTimeout: opts.Websocket.PingTimeout,
		PingPeriod:  opts.Websocket.PingPeriod,
	})

	// @todo Wire in cross-service JWT maker for Corredor
	corredor.Service().SetUserFinder(nil)
	corredor.Service().SetRoleFinder(nil)

	return
}

func (app *App) Upgrade(ctx context.Context) (err error) {
	db := factory.Database.MustGet(SERVICE, "default").With(ctx).Quiet()
	err = migrate.Migrate(db, app.Log)
	if err != nil {
		return
	}

	return
}

func (app *App) Initialize(ctx context.Context) (err error) {
	// Connects to all services it needs to
	err = service.Initialize(ctx, app.Log, service.Config{
		Storage: app.Opts.Storage,
	})

	if err != nil {
		return
	}

	return
}

func (app *App) Activate(ctx context.Context) (err error) {
	if err = service.Activate(ctx); err != nil {
		return
	}

	service.Watchers(ctx)
	websocket.Watch(ctx)

	return
}

func (app *App) Provision(ctx context.Context) (err error) {
	ctx = auth.SetSuperUserContext(ctx)

	if err = provisionConfig(ctx, app.Log); err != nil {
		return
	}

	return
}

func (app *App) MountApiRoutes(r chi.Router) {
	rest.MountRoutes(r)
	app.ws.ApiServerRoutes(r)
}

func (app *App) RegisterCliCommands(p *cobra.Command) {
	p.AddCommand(
		commands.Importer(),
		commands.Exporter(),
	)
}