package tasks

import (
	"fmt"
)

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

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

func (r TaskHandlerRegistry) Add(name TaskType, 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
// }

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