package context

import (
	gocontext "context"
	"net/http"

	"github.com/labstack/echo/v4"
	"github.com/medama-io/go-useragent"
	"github.com/onsonr/crypto/mpc"
	"github.com/onsonr/hway/pkg/common"
	hway "github.com/onsonr/hway/pkg/config"
	hwayorm "github.com/onsonr/hway/pkg/models"
)

type GatewayContext struct {
	echo.Context
	hwayorm.Querier
	id               string
	ipfsClient       common.IPFS
	agent            useragent.UserAgent
	tokenStore       common.IPFSTokenStore
	stagedEnclaves   map[string]mpc.Enclave
	grpcAddr         string
	turnstileSiteKey string
}

func GetGateway(c echo.Context) (*GatewayContext, error) {
	cc, ok := c.(*GatewayContext)
	if !ok {
		return nil, echo.NewHTTPError(http.StatusInternalServerError, "Gateway Context not found")
	}
	return cc, nil
}

func UseGateway(env hway.Hway, ipc common.IPFS, db *hwayorm.Queries) echo.MiddlewareFunc {
	return func(next echo.HandlerFunc) echo.HandlerFunc {
		return func(c echo.Context) error {
			ua := useragent.NewParser()
			ctx := &GatewayContext{
				Context:    c,
				Querier:    db,
				ipfsClient: ipc,
				agent:      ua.Parse(c.Request().UserAgent()),
				grpcAddr:   env.GetSonrGrpcUrl(),
				tokenStore: common.NewUCANStore(ipc),
				// turnstileSiteKey: env.GetTurnstileSiteKey(),
			}
			return next(ctx)
		}
	}
}

func BG() gocontext.Context {
	ctx := gocontext.Background()
	return ctx
}

func (cc *GatewayContext) ReadCookie(k common.CookieKey) string {
	return common.ReadCookieUnsafe(cc.Context, k)
}

func (cc *GatewayContext) WriteCookie(k common.CookieKey, v string) {
	common.WriteCookie(cc.Context, k, v)
}