package plugins

import (
	"context"
	pkgHTTP "github.com/apache/apisix-go-plugin-runner/pkg/http"
	"github.com/go-saas/saas"
	"github.com/go-saas/sessions"
	"regexp"
)

type Resolver struct {
	r         pkgHTTP.Request
	key       string
	pathRegex string
}

func NewResolver(r pkgHTTP.Request, key, pathRegex string) *Resolver {
	return &Resolver{
		r:         r,
		key:       key,
		pathRegex: pathRegex,
	}
}

var _ saas.TenantResolver = (*Resolver)(nil)

func (r *Resolver) Resolve(ctx context.Context) (saas.TenantResolveResult, context.Context, error) {
	// default host side
	var t = ""
	if v := r.r.Header().Get(r.key); len(v) > 0 {
		t = v
	}
	if v := r.r.Args().Get(r.key); len(v) > 0 {
		t = v
	}
	if c, err := sessions.ReadCookie(r.r.Header().View(), r.key); err == nil {
		t = c.Value
	}
	if len(r.pathRegex) > 0 {
		host := r.r.Header().Get("Host")
		if len(host) > 0 {
			reg := regexp.MustCompile(r.pathRegex)
			f := reg.FindAllStringSubmatch(host, -1)
			if f != nil {
				t = f[0][1]
			}
		}
	}
	res := &saas.TenantResolveResult{TenantIdOrName: t}
	ctx = saas.NewTenantResolveRes(ctx, res)
	return *res, ctx, nil
}