package main

type PluginRegistry struct {
	CorePlugin    map[string]CoreInterface
	HandlerPlugin map[string]HandlerInterface
}

func NewPluginRegistry() *PluginRegistry {
	r := new(PluginRegistry)
	r.CorePlugin = make(map[string]CoreInterface)
	r.HandlerPlugin = make(map[string]HandlerInterface)
	return r
}

func (r *PluginRegistry) RegisterCorePlugin(plugin CoreInterface) error {
	name, e := plugin.Register()
	if e != nil {
		return e
	}
	r.CorePlugin[name] = plugin
	return nil
}

func (r *PluginRegistry) RegisterHandlerPlugin(plugin HandlerInterface) error {
	name, e := plugin.Register()
	if e != nil {
		return e
	}
	r.HandlerPlugin[name] = plugin
	return nil
}

func (r *PluginRegistry) GetCorePlugin(name string) CoreInterface {
	return r.CorePlugin[name]
}

func (r *PluginRegistry) GetHandlerPlugin(name string) HandlerInterface {
	return r.HandlerPlugin[name]
}

func (r *PluginRegistry) RemoveCorePlugin(name string) {
	delete(r.CorePlugin, name)
}

func (r *PluginRegistry) RemoveHandlerPlugin(name string) {
	delete(r.HandlerPlugin, name)
}