mirror of
https://github.com/foxcpp/maddy.git
synced 2025-04-03 05:07:38 +03:00
105 lines
2.7 KiB
Go
105 lines
2.7 KiB
Go
/*
|
|
Maddy Mail Server - Composable all-in-one email server.
|
|
Copyright © 2019-2020 Max Mazurov <fox.cpp@disroot.org>, Maddy Mail Server contributors
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
package module
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
|
|
"github.com/foxcpp/maddy/framework/config"
|
|
"github.com/foxcpp/maddy/framework/hooks"
|
|
"github.com/foxcpp/maddy/framework/log"
|
|
)
|
|
|
|
var (
|
|
instances = make(map[string]struct {
|
|
mod Module
|
|
cfg *config.Map
|
|
})
|
|
aliases = make(map[string]string)
|
|
|
|
Initialized = make(map[string]bool)
|
|
)
|
|
|
|
// RegisterInstance adds module instance to the global registry.
|
|
//
|
|
// Instance name must be unique. Second RegisterInstance with same instance
|
|
// name will replace previous.
|
|
func RegisterInstance(inst Module, cfg *config.Map) {
|
|
instances[inst.InstanceName()] = struct {
|
|
mod Module
|
|
cfg *config.Map
|
|
}{inst, cfg}
|
|
}
|
|
|
|
// RegisterAlias creates an association between a certain name and instance name.
|
|
//
|
|
// After RegisterAlias, module.GetInstance(aliasName) will return the same
|
|
// result as module.GetInstance(instName).
|
|
func RegisterAlias(aliasName, instName string) {
|
|
aliases[aliasName] = instName
|
|
}
|
|
|
|
func HasInstance(name string) bool {
|
|
aliasedName := aliases[name]
|
|
if aliasedName != "" {
|
|
name = aliasedName
|
|
}
|
|
|
|
_, ok := instances[name]
|
|
return ok
|
|
}
|
|
|
|
// GetInstance returns module instance from global registry, initializing it if
|
|
// necessary.
|
|
//
|
|
// Error is returned if module initialization fails or module instance does not
|
|
// exists.
|
|
func GetInstance(name string) (Module, error) {
|
|
aliasedName := aliases[name]
|
|
if aliasedName != "" {
|
|
name = aliasedName
|
|
}
|
|
|
|
mod, ok := instances[name]
|
|
if !ok {
|
|
return nil, fmt.Errorf("unknown config block: %s", name)
|
|
}
|
|
|
|
// Break circular dependencies.
|
|
if Initialized[name] {
|
|
return mod.mod, nil
|
|
}
|
|
|
|
Initialized[name] = true
|
|
if err := mod.mod.Init(mod.cfg); err != nil {
|
|
return mod.mod, err
|
|
}
|
|
|
|
if closer, ok := mod.mod.(io.Closer); ok {
|
|
hooks.AddHook(hooks.EventShutdown, func() {
|
|
log.Debugf("close %s (%s)", mod.mod.Name(), mod.mod.InstanceName())
|
|
if err := closer.Close(); err != nil {
|
|
log.Printf("module %s (%s) close failed: %v", mod.mod.Name(), mod.mod.InstanceName(), err)
|
|
}
|
|
})
|
|
}
|
|
|
|
return mod.mod, nil
|
|
}
|