Allow defaults modules to have defaults

Allow chaining defaults modules by making Defaults be a Defaultable.

Change-Id: I4ba86c07f7aad9c396ed33d55fe95d1fb78e487d
This commit is contained in:
Colin Cross
2016-07-27 10:15:06 -07:00
parent 627d83db71
commit e7b07137d4
2 changed files with 8 additions and 13 deletions

View File

@@ -26,7 +26,7 @@ import (
func init() {
RegisterBottomUpMutator("defaults_deps", defaultsDepsMutator)
RegisterTopDownMutator("defaults", defaultsMutator)
RegisterBottomUpMutator("defaults", defaultsMutator)
RegisterBottomUpMutator("arch", ArchMutator)
}

View File

@@ -45,7 +45,7 @@ func (d *DefaultableModule) setProperties(props []interface{}) {
type Defaultable interface {
defaults() *defaultsProperties
setProperties([]interface{})
applyDefaults(TopDownMutatorContext, Defaults)
applyDefaults(BottomUpMutatorContext, Defaults)
}
var _ Defaultable = (*DefaultableModule)(nil)
@@ -61,12 +61,13 @@ func InitDefaultableModule(module Module, d Defaultable,
}
type DefaultsModule struct {
DefaultableModule
defaultProperties []interface{}
}
type Defaults interface {
Defaultable
isDefaults() bool
setProperties([]interface{})
properties() []interface{}
}
@@ -75,22 +76,16 @@ func (d *DefaultsModule) isDefaults() bool {
}
func (d *DefaultsModule) properties() []interface{} {
return d.defaultProperties
}
func (d *DefaultsModule) setProperties(props []interface{}) {
d.defaultProperties = props
return d.defaultableProperties
}
func InitDefaultsModule(module Module, d Defaults, props ...interface{}) (blueprint.Module, []interface{}) {
d.setProperties(props)
return module, props
return InitDefaultableModule(module, d, props...)
}
var _ Defaults = (*DefaultsModule)(nil)
func (defaultable *DefaultableModule) applyDefaults(ctx TopDownMutatorContext,
func (defaultable *DefaultableModule) applyDefaults(ctx BottomUpMutatorContext,
defaults Defaults) {
for _, prop := range defaultable.defaultableProperties {
@@ -115,7 +110,7 @@ func defaultsDepsMutator(ctx BottomUpMutatorContext) {
}
}
func defaultsMutator(ctx TopDownMutatorContext) {
func defaultsMutator(ctx BottomUpMutatorContext) {
if defaultable, ok := ctx.Module().(Defaultable); ok {
for _, defaultsDep := range defaultable.defaults().Defaults {
ctx.VisitDirectDeps(func(m blueprint.Module) {