node.go

package entities

import (
	"net"
	"time"
)

const (
	NodeStatusNew        = "NEW"
	NodeStatusDeploying  = "DEPLOYING"
	NodeStatusActive     = "ACTIVE"
	NodeStatusReady      = "READY"
	NodeStatusFailing    = "FAILING"
	NodeStatusRetrying   = "RETRYING"
	NodeStatusRestarting = "RESTARTING"
	NodeStatusStopped    = "STOPPED"
	NodeStatusError      = "ERROR"
	NodeStatusDeleted    = "DELETED"
)

type Node struct {
	ID              string       `json:"id"`
	ProviderID      string       `json:"providerId"`
	NodeToken       string       `json:"nodeToken"`
	IsMaster        bool         `json:"isMaster"`
	MasterIP        net.IP       `json:"masterIP"`
	Tags            []string     `json:"tags"`
	Name            string       `json:"name"`
	IPV4            net.IP       `json:"ipv4"`
	IPV6            net.IP       `json:"ipv6"`
	Status          string       `json:"status"`
	Retries         int          `json:"retries"`
	Error           string       `json:"error"`
	Created         time.Time    `json:"created"`
	Updated         time.Time    `json:"updated"`
	Deleted         *time.Time   `json:"deleted"`
	Features        NodeFeatures `json:"features"`
	Dependencies    []Dependency `json:"dependencies"`
	InstallPassword string       `json:"installPassword"`
	InstallUser     string       `json:"installUser"`
	Delete          *time.Time   `json:"delete"`
	//TODO: Add data as needed
}

type NodeFeatures map[string]bool

func (ma *Node) NeedsAdapting() bool {
	return ma.Status == NodeStatusNew || ma.Status == NodeStatusActive || ma.Status == NodeStatusRetrying
}

func (n *Node) Retry() *Node {
	if n.Retries > 10 {
		n.Status = NodeStatusFailing
		return n
	}

	n.Retries = n.Retries + 1
	n.Status = NodeStatusRetrying
	return n
}

func (ma *Node) Requires(dependency string) bool {
	for i := range ma.Dependencies {
		if ma.Dependencies[i].Name == dependency {
			if ma.Dependencies[i].Status == DependencyStatusNew ||
				ma.Dependencies[i].Status == DependencyStatusRetrying ||
				ma.Dependencies[i].Status == DependencyStatusWaiting {
				return true
			}
		}
	}
	return false
}

func (ma *Node) Fulfils(dependency string) bool {
	for i := range ma.Dependencies {
		if ma.Dependencies[i].Name == dependency {
			if ma.Dependencies[i].Status == DependencyStatusReady {
				return true
			}
		}
	}
	return false
}