package controller

import (
	"github.com/go-openapi/runtime/middleware"
	"github.com/michaelquigley/df/dl"
	"github.com/openziti/zrok/v2/rest_model_zrok"
	"github.com/openziti/zrok/v2/rest_server_zrok/operations/metadata"
)

type environmentDetailHandler struct{}

func newEnvironmentDetailHandler() *environmentDetailHandler {
	return &environmentDetailHandler{}
}

func (h *environmentDetailHandler) Handle(params metadata.GetEnvironmentDetailParams, principal *rest_model_zrok.Principal) middleware.Responder {
	trx, err := str.Begin()
	if err != nil {
		dl.Errorf("error starting transaction for user '%v': %v", principal.Email, err)
		return metadata.NewGetEnvironmentDetailInternalServerError()
	}
	defer func() { _ = trx.Rollback() }()
	senv, err := str.FindEnvironmentForAccount(params.EnvZID, int(principal.ID), trx)
	if err != nil {
		dl.Errorf("environment '%v' not found for account '%v': %v", params.EnvZID, principal.Email, err)
		return metadata.NewGetEnvironmentDetailNotFound()
	}
	es := &rest_model_zrok.EnvironmentAndResources{
		Environment: &rest_model_zrok.Environment{
			Address:     senv.Address,
			CreatedAt:   senv.CreatedAt.UnixMilli(),
			Description: senv.Description,
			Host:        senv.Host,
			UpdatedAt:   senv.UpdatedAt.UnixMilli(),
			ZID:         senv.ZId,
		},
	}
	shrs, err := str.FindSharesForEnvironment(senv.Id, trx)
	if err != nil {
		dl.Errorf("error finding shares for environment '%v' for user '%v': %v", senv.ZId, principal.Email, err)
		return metadata.NewGetEnvironmentDetailInternalServerError()
	}
	sparkRx := make(map[string][]int64)
	sparkTx := make(map[string][]int64)
	if cfg.Metrics != nil && cfg.Metrics.Influx != nil {
		sparkRx, sparkTx, err = sparkDataForShares(shrs)
		if err != nil {
			dl.Errorf("error querying spark data for shares for user '%v': %v", principal.Email, err)
		}
	} else {
		dl.Debug("skipping spark data for shares; no influx configuration")
	}
	for _, shr := range shrs {
		frontendEndpoints := buildFrontendEndpointsForShare(shr.Id, shr.Token, shr.FrontendEndpoint, trx)
		target := ""
		if shr.BackendProxyEndpoint != nil {
			target = *shr.BackendProxyEndpoint
		}
		var sparkData []*rest_model_zrok.SparkDataSample
		for i := 0; i < len(sparkRx[shr.Token]) && i < len(sparkTx[shr.Token]); i++ {
			sparkData = append(sparkData, &rest_model_zrok.SparkDataSample{Rx: float64(sparkRx[shr.Token][i]), Tx: float64(sparkTx[shr.Token][i])})
		}
		es.Shares = append(es.Shares, &rest_model_zrok.Share{
			ShareToken:        shr.Token,
			ZID:               shr.ZId,
			EnvZID:            senv.ZId,
			ShareMode:         shr.ShareMode,
			BackendMode:       shr.BackendMode,
			FrontendEndpoints: frontendEndpoints,
			Target:            target,
			Activity:          sparkData,
			CreatedAt:         shr.CreatedAt.UnixMilli(),
			UpdatedAt:         shr.UpdatedAt.UnixMilli(),
		})
	}
	return metadata.NewGetEnvironmentDetailOK().WithPayload(es)
}