im-kulikov/helium

package module

import (
    "go.uber.org/dig"
)

type (
    // Module type.
    Module []*Provider

    // Provider struct.
    Provider struct {
        Constructor interface{}
        Options     []dig.ProvideOption
    }
)

// New single module.
func New(fn interface{}, opts ...dig.ProvideOption) Module {
    return Module{
        {
            Constructor: fn,
            Options:     opts,
        },
    }
}

// Combine multiple modules into new one.
func Combine(mods ...Module) Module {
    var result Module
    for _, mod := range mods {
        result = append(result, mod...)
    }

    return result
}

// Append module to target module and return new module.
func (m Module) Append(mods ...Module) Module {
    result := m
    for _, mod := range mods {
        result = append(result, mod...)
    }

    return result
}

// AppendConstructor adds constructors into the module and return new Module.
func (m Module) AppendConstructor(constructors ...interface{}) Module {
    modules := make([]Module, 0, len(constructors))
    for _, c := range constructors {
        modules = append(modules, New(c))
    }

    return m.Append(modules...)
}

// Provide set providers functions to DI container.
func Provide(dic *dig.Container, providers Module) error {
    for _, p := range providers {
        if err := dic.Provide(p.Constructor, p.Options...); err != nil {
            return err
        }
    }

    return nil
}