package main

import (
	"context"
	"errors"
	"strconv"
	"strings"

	"github.com/beckn-one/beckn-onix/pkg/plugin/definition"
	"github.com/beckn-one/beckn-onix/pkg/plugin/implementation/schemav2validator"
)

// schemav2ValidatorProvider provides instances of schemav2Validator.
type schemav2ValidatorProvider struct{}

// New initialises a new Schemav2Validator instance.
func (vp schemav2ValidatorProvider) New(ctx context.Context, config map[string]string) (definition.SchemaValidator, func() error, error) {
	if ctx == nil {
		return nil, nil, errors.New("context cannot be nil")
	}

	typeVal, hasType := config["type"]
	locVal, hasLoc := config["location"]

	if !hasType || typeVal == "" {
		return nil, nil, errors.New("type not configured")
	}
	if !hasLoc || locVal == "" {
		return nil, nil, errors.New("location not configured")
	}

	cfg := &schemav2validator.Config{
		Type:     typeVal,
		Location: locVal,
		CacheTTL: 3600,
	}

	if ttlStr, ok := config["cacheTTL"]; ok {
		if ttl, err := strconv.Atoi(ttlStr); err == nil && ttl > 0 {
			cfg.CacheTTL = ttl
		}
	}

	// NEW: Parse enableReferencedSchemas
	if enableStr, ok := config["enableReferencedSchemas"]; ok {
		cfg.EnableReferencedSchemas = enableStr == "true"
	}

	// NEW: Parse referencedSchemaConfig (if enabled)
	if cfg.EnableReferencedSchemas {
		if v, ok := config["referencedSchemaConfig.cacheTTL"]; ok {
			if ttl, err := strconv.Atoi(v); err == nil && ttl > 0 {
				cfg.ReferencedSchemaConfig.CacheTTL = ttl
			}
		}
		if v, ok := config["referencedSchemaConfig.maxCacheSize"]; ok {
			if size, err := strconv.Atoi(v); err == nil && size > 0 {
				cfg.ReferencedSchemaConfig.MaxCacheSize = size
			}
		}
		if v, ok := config["referencedSchemaConfig.downloadTimeout"]; ok {
			if timeout, err := strconv.Atoi(v); err == nil && timeout > 0 {
				cfg.ReferencedSchemaConfig.DownloadTimeout = timeout
			}
		}
		if v, ok := config["referencedSchemaConfig.allowedDomains"]; ok && v != "" {
			cfg.ReferencedSchemaConfig.AllowedDomains = strings.Split(v, ",")
		}
		if v, ok := config["referencedSchemaConfig.urlTransform"]; ok && v != "" {
			cfg.ReferencedSchemaConfig.URLTransform = v
		}
	}

	return schemav2validator.New(ctx, cfg)
}

// Provider is the exported plugin provider.
var Provider schemav2ValidatorProvider