opfactory.go

// Copyright 2020 Canonical Ltd.
// Licensed under the AGPLv3, see LICENCE file for details.

package spaces

import (
	"github.com/juju/errors"

	"github.com/DavinZhang/juju/apiserver/common/networkingcommon"
	"github.com/DavinZhang/juju/state"
)

// OpFactory describes a source of model operations
// required by the spaces API.
type OpFactory interface {
	// NewRemoveSpaceOp returns an operation for removing a space.
	NewRemoveSpaceOp(fromName string) (state.ModelOperation, error)

	// NewRenameSpaceOp returns an operation for renaming a space.
	NewRenameSpaceOp(spaceName, toName string) (state.ModelOperation, error)

	// NewMoveSubnetsOp returns an operation for updating a space with new CIDRs.
	NewMoveSubnetsOp(spaceID string, subnets []MovingSubnet) (MoveSubnetsOp, error)
}

type opFactory struct {
	st *state.State
}

func newOpFactory(st *state.State) OpFactory {
	return &opFactory{st: st}
}

// NewRemoveSpaceOp (OpFactory) returns an operation
// for removing a space.
func (f *opFactory) NewRemoveSpaceOp(spaceName string) (state.ModelOperation, error) {
	space, err := f.st.SpaceByName(spaceName)
	if err != nil {
		return nil, errors.Trace(err)
	}
	subnets, err := space.Subnets()
	if err != nil {
		return nil, errors.Trace(err)
	}
	n := make([]Subnet, len(subnets))
	for i, subnet := range subnets {
		n[i] = subnet
	}
	return NewRemoveSpaceOp(space, n), nil
}

// NewRenameSpaceOp (OpFactory) returns an operation
// for renaming a space.
func (f *opFactory) NewRenameSpaceOp(fromName, toName string) (state.ModelOperation, error) {
	space, err := f.st.SpaceByName(fromName)
	if err != nil {
		return nil, errors.Trace(err)
	}
	return NewRenameSpaceOp(
		f.st.IsController(), f.st.NewControllerSettings(), &renameSpaceState{f.st}, space, toName), nil
}

// NewMoveSubnetsOp (OpFactory) returns an operation
// to move a list of subnets to a space.
func (f *opFactory) NewMoveSubnetsOp(spaceName string, subnets []MovingSubnet) (MoveSubnetsOp, error) {
	space, err := f.st.SpaceByName(spaceName)
	if err != nil {
		return nil, errors.Trace(err)
	}
	return NewMoveSubnetsOp(networkingcommon.NewSpaceShim(space), subnets), nil
}