package screener

import (
	"context"
	"fmt"

	"github.com/synapsecns/sanguine/contrib/screener-api/chainalysis"
	"github.com/synapsecns/sanguine/contrib/screener-api/config"
	"github.com/synapsecns/sanguine/core/metrics"
)

type TestScreener interface {
	Screener
	SetClient(client chainalysis.Client)
}

func NewTestScreener(ctx context.Context, cfg config.Config, metricHandler metrics.Handler) (_ TestScreener, err error) {
	screener, err := NewScreener(ctx, cfg, metricHandler)
	if err != nil {
		return nil, fmt.Errorf("could not create screener: %w", err)
	}

	ts, ok := screener.(TestScreener)
	if !ok {
		return nil, fmt.Errorf("could not cast screener to test screener")
	}

	return ts, nil
}

func (s *screenerImpl) SetClient(client chainalysis.Client) {
	s.client = client
}