cluster.go

package main

import (
	"errors"
	"krak8s/app"

	"github.com/goadesign/goa"
)

// ClusterController implements the cluster resource.
type ClusterController struct {
	*goa.Controller
	ds      *DataStore
	backend *Runner
}

// NewClusterController creates a cluster controller.
func NewClusterController(service *goa.Service, store *DataStore, backend *Runner) *ClusterController {
	return &ClusterController{
		Controller: service.NewController("ClusterController"),
		ds:         store,
		backend:    backend,
	}
}

// MarshalResourcesObject to project media type
func MarshalResourcesObject(obj *ResourceObject) *app.Cluster {
	return &app.Cluster{
		ID:           obj.OID,
		Type:         obj.ObjType,
		NodePoolSize: obj.NodePoolSize,
		NamespaceID:  obj.NamespaceID,
		State:        obj.State,
		CreatedAt:    obj.CreatedAt,
		UpdatedAt:    obj.UpdatedAt,
	}
}

// Create runs the create action.
func (c *ClusterController) Create(ctx *app.CreateClusterContext) error {
	// ClusterController_Create: start_implement
	proj, ok := c.ds.Project(ctx.Projectid)
	if !ok {
		return ctx.NotFound()
	}
	ns, ok := c.ds.Namespace(ctx.Payload.NamespaceID)
	if !ok {
		return ctx.NotFound()
	} else if ns.Resources != nil {
		return ctx.Conflict()
	}

	found := false
	for _, val := range proj.Namespaces {
		if val.OID == ctx.Payload.NamespaceID {
			found = true
			break
		}
	}
	if !found {
		return ctx.BadRequest(errors.New("Inavlid Namespace Object ID specified in request"))
	}

	res := c.ds.NewResource(ctx.Payload.NamespaceID, ctx.Payload.NodePoolSize)
	if res == nil {
		return ctx.InternalServerError()
	}
	url := APIVersion + APIProjects + ctx.Projectid + APICluster + res.OID
	ns.Resources = &ObjectLink{OID: res.OID, URL: url}

	c.backend.ProjectRequest(AddProject, c.ds, proj, ns, res)

	return ctx.Accepted(MarshalResourcesObject(res))
	// ClusterController_Create: end_implement
}

// Delete runs the delete action.
func (c *ClusterController) Delete(ctx *app.DeleteClusterContext) error {
	// ClusterController_Delete: start_implement
	proj, ok := c.ds.Project(ctx.Projectid)
	if !ok {
		return ctx.NotFound()
	}
	res, ok := c.ds.Resource(ctx.ResourceID)
	if !ok {
		return ctx.NotFound()
	}
	ns, ok := c.ds.Namespace(res.NamespaceID)
	if !ok {
		return ctx.NotFound()
	}
	if ns.Resources.OID != ctx.ResourceID {
		return ctx.BadRequest(errors.New("Inavlid Cluster Resource Object ID specified in request"))
	}

	res.State = ResourceDeleteRequested
	c.backend.ProjectRequest(RemoveProject, c.ds, proj, ns, res)

	c.ds.DeleteResource(res)
	ns.Resources = nil

	return ctx.NoContent()
	// ClusterController_Delete: end_implement
}

// Get runs the get action.
func (c *ClusterController) Get(ctx *app.GetClusterContext) error {
	// ClusterController_Get: start_implement
	if _, ok := c.ds.Project(ctx.Projectid); !ok {
		return ctx.NotFound()
	}
	resource, ok := c.ds.Resource(ctx.ResourceID)
	if !ok {
		return ctx.NotFound()
	}
	res := MarshalResourcesObject(resource)
	return ctx.OK(res)
	// ClusterController_Get: end_implement
}