5f33292496
This change just moves the assertion/failure functions from the pkg/util package to pkg/util/contract, so things read a bit nicer (i.e., `contract.Assert(x)` versus `util.Assert(x)`).
50 lines
1.4 KiB
Go
50 lines
1.4 KiB
Go
// Copyright 2016 Marapongo, Inc. All rights reserved.
|
|
|
|
package compiler
|
|
|
|
import (
|
|
"github.com/marapongo/mu/pkg/ast"
|
|
"github.com/marapongo/mu/pkg/compiler/backends"
|
|
"github.com/marapongo/mu/pkg/util/contract"
|
|
)
|
|
|
|
// Context holds all state available to any templates or code evaluated at compile-time.
|
|
type Context struct {
|
|
Options *Options // compiler options supplied.
|
|
Cluster *ast.Cluster // the cluster that we will deploy to.
|
|
Arch backends.Arch // the target cloud architecture.
|
|
Properties ast.PropertyBag // properties supplied at stack construction time.
|
|
}
|
|
|
|
// NewContext returns a new, empty context.
|
|
func NewContext(opts *Options) *Context {
|
|
return &Context{
|
|
Options: opts,
|
|
Properties: make(ast.PropertyBag),
|
|
}
|
|
}
|
|
|
|
// WithClusterArch returns a clone of this Context with the given cluster and architecture attached to it.
|
|
func (c *Context) WithClusterArch(cl *ast.Cluster, a backends.Arch) *Context {
|
|
contract.Assert(cl != nil)
|
|
return &Context{
|
|
Cluster: cl,
|
|
Arch: a,
|
|
Options: c.Options,
|
|
Properties: c.Properties,
|
|
}
|
|
}
|
|
|
|
// WithProps returns a clone of this Context with the given properties attached to it.
|
|
func (c *Context) WithProps(props ast.PropertyBag) *Context {
|
|
if props == nil {
|
|
props = make(ast.PropertyBag)
|
|
}
|
|
return &Context{
|
|
Cluster: c.Cluster,
|
|
Arch: c.Arch,
|
|
Options: c.Options,
|
|
Properties: props,
|
|
}
|
|
}
|