package appmodule

import (
	"cosmossdk.io/core/internal"
	"cosmossdk.io/depinject"
)

// Option is a functional option for implementing modules.
type Option interface {
	apply(*internal.ModuleInitializer) error
}

type funcOption func(initializer *internal.ModuleInitializer) error

func (f funcOption) apply(initializer *internal.ModuleInitializer) error {
	return f(initializer)
}

// Provide registers providers with the dependency injection system that will be
// run within the module scope. See cosmossdk.io/depinject for
// documentation on the dependency injection system.
func Provide(providers ...interface{}) Option {
	return funcOption(func(initializer *internal.ModuleInitializer) error {
		for _, provider := range providers {
			desc, err := depinject.ExtractProviderDescriptor(provider)
			if err != nil {
				return err
			}

			initializer.Providers = append(initializer.Providers, desc)
		}
		return nil
	})
}

// Invoke registers invokers to run with depinject. Each invoker will be called
// at the end of dependency graph configuration in the order in which it was defined. Invokers may not define output
// parameters, although they may return an error, and all of their input parameters will be marked as optional so that
// invokers impose no additional constraints on the dependency graph. Invoker functions should nil-check all inputs.
func Invoke(invokers ...interface{}) Option {
	return funcOption(func(initializer *internal.ModuleInitializer) error {
		for _, invoker := range invokers {
			desc, err := depinject.ExtractInvokerDescriptor(invoker)
			if err != nil {
				return err
			}

			initializer.Invokers = append(initializer.Invokers, desc)
		}
		return nil
	})
}