package extends

import (
	"fmt"

	"gitee.com/meng_mengs_boys/eosc"
)

type IExtenderRegister interface {
	eosc.IExtenderDriverManager
	eosc.IExtenderDrivers
}

type ExtenderRegister struct {
	group   string
	project string
	data    map[string]eosc.IExtenderDriverFactory
}

func NewExtenderRegister(group string, project string) *ExtenderRegister {
	return &ExtenderRegister{group: group, project: project, data: make(map[string]eosc.IExtenderDriverFactory)}
}

func (r *ExtenderRegister) RegisterExtenderDriver(name string, factory eosc.IExtenderDriverFactory) error {
	_, has := r.data[name]
	if has {
		return fmt.Errorf("%s:%w", name, ErrorExtenderNameDuplicate)
	}
	r.data[name] = factory
	return nil
}

func (r *ExtenderRegister) RegisterTo(register eosc.IExtenderDriverRegister) {
	for n, f := range r.data {
		id := FormatDriverId(r.group, r.project, n)
		register.RegisterExtenderDriver(id, f)
	}
}

func (r *ExtenderRegister) All() map[string]eosc.IExtenderDriverFactory {
	return r.data
}