package detect

import (
	"fmt"

	"github.com/andig/evcc/util"
)

type TaskHandler interface {
	Test(log *util.Logger, ip string) []interface{}
}

type TaskHandlerRegistry map[string]func(map[string]interface{}) (TaskHandler, error)

var registry TaskHandlerRegistry = make(map[string]func(map[string]interface{}) (TaskHandler, error))

func (r TaskHandlerRegistry) Add(name string, factory func(map[string]interface{}) (TaskHandler, error)) {
	if _, exists := r[name]; exists {
		panic(fmt.Sprintf("cannot register duplicate charger type: %s", name))
	}
	r[name] = factory
}

func (r TaskHandlerRegistry) Get(name string) (func(map[string]interface{}) (TaskHandler, error), error) {
	factory, exists := r[name]
	if !exists {
		return nil, fmt.Errorf("charger type not registered: %s", name)
	}
	return factory, nil
}