instance_table.go

package cmd

import (
	"fmt"
	"strings"

	boshdir "github.com/cloudfoundry/bosh-cli/director"
	boshtbl "github.com/cloudfoundry/bosh-cli/ui/table"
)

type InstanceTableValues struct {
	Name    boshtbl.Value
	Process boshtbl.Value

	ProcessState boshtbl.Value
	State        boshtbl.Value
	AZ           boshtbl.Value
	VMType       boshtbl.Value
	IPs          boshtbl.Value

	// Details
	VMCID        boshtbl.Value
	DiskCIDs     boshtbl.Value
	AgentID      boshtbl.Value
	Resurrection boshtbl.Value
	Ignore       boshtbl.Value

	// DNS
	DNS boshtbl.Value

	// Vitals
	Uptime boshtbl.Value // only for Process
	Load   boshtbl.Value

	CPUTotal boshtbl.Value // only for Process
	CPUUser  boshtbl.Value
	CPUSys   boshtbl.Value
	CPUWait  boshtbl.Value

	Memory boshtbl.Value
	Swap   boshtbl.Value

	SystemDisk     boshtbl.Value
	EphemeralDisk  boshtbl.Value
	PersistentDisk boshtbl.Value
}

var InstanceTableHeader = InstanceTableValues{
	Name:    boshtbl.NewValueString("Instance"),
	Process: boshtbl.NewValueString("Process"),

	ProcessState: boshtbl.NewValueString("Process State"),
	AZ:           boshtbl.NewValueString("AZ"),
	VMType:       boshtbl.NewValueString("VM Type"),
	IPs:          boshtbl.NewValueString("IPs"),

	// Details
	State:        boshtbl.NewValueString("State"),
	VMCID:        boshtbl.NewValueString("VM CID"),
	DiskCIDs:     boshtbl.NewValueString("Disk CIDs"),
	AgentID:      boshtbl.NewValueString("Agent ID"),
	Resurrection: boshtbl.NewValueString("Resurrection\nPaused"),
	Ignore:       boshtbl.NewValueString("Ignore"),

	// DNS
	DNS: boshtbl.NewValueString("DNS A Records"),

	// Vitals
	Uptime: boshtbl.NewValueString("Uptime"), // only for Process
	Load:   boshtbl.NewValueString("Load\n(1m, 5m, 15m)"),

	CPUTotal: boshtbl.NewValueString("CPU\nTotal"), // only for Process
	CPUUser:  boshtbl.NewValueString("CPU\nUser"),
	CPUSys:   boshtbl.NewValueString("CPU\nSys"),
	CPUWait:  boshtbl.NewValueString("CPU\nWait"),

	Memory: boshtbl.NewValueString("Memory\nUsage"),
	Swap:   boshtbl.NewValueString("Swap\nUsage"),

	SystemDisk:     boshtbl.NewValueString("System\nDisk Usage"),
	EphemeralDisk:  boshtbl.NewValueString("Ephemeral\nDisk Usage"),
	PersistentDisk: boshtbl.NewValueString("Persistent\nDisk Usage"),
}

type InstanceTable struct {
	Processes, VMDetails, Details, DNS, Vitals bool
}

func (t InstanceTable) Header() InstanceTableValues {
	return InstanceTableHeader
}

func (t InstanceTable) ForVMInfo(i boshdir.VMInfo) InstanceTableValues {
	vals := InstanceTableValues{
		Name:    t.buildName(i),
		Process: boshtbl.ValueString{},

		ProcessState: boshtbl.ValueFmt{
			V:     boshtbl.NewValueString(i.ProcessState),
			Error: !i.IsRunning(),
		},

		AZ:     boshtbl.NewValueString(i.AZ),
		VMType: boshtbl.NewValueString(i.VMType),
		IPs:    boshtbl.NewValueStrings(i.IPs),

		// Details
		State:        boshtbl.NewValueString(i.State),
		VMCID:        boshtbl.NewValueString(i.VMID),
		DiskCIDs:     boshtbl.NewValueStrings(i.DiskIDs),
		AgentID:      boshtbl.NewValueString(i.AgentID),
		Resurrection: boshtbl.NewValueBool(i.ResurrectionPaused),
		Ignore:       boshtbl.NewValueBool(i.Ignore),

		// DNS
		DNS: boshtbl.NewValueStrings(i.DNS),

		// Vitals
		Uptime: ValueUptime{i.Vitals.Uptime.Seconds},
		Load:   boshtbl.NewValueString(strings.Join(i.Vitals.Load, ", ")),

		CPUTotal: ValueCPUTotal{i.Vitals.CPU.Total},
		CPUUser:  NewValueStringPercent(i.Vitals.CPU.User),
		CPUSys:   NewValueStringPercent(i.Vitals.CPU.Sys),
		CPUWait:  NewValueStringPercent(i.Vitals.CPU.Wait),

		Memory: ValueMemSize{i.Vitals.Mem},
		Swap:   ValueMemSize{i.Vitals.Swap},

		SystemDisk:     ValueDiskSize{i.Vitals.SystemDisk()},
		EphemeralDisk:  ValueDiskSize{i.Vitals.EphemeralDisk()},
		PersistentDisk: ValueDiskSize{i.Vitals.PersistentDisk()},
	}

	if len(i.VMType) == 0 {
		vals.VMType = boshtbl.NewValueString(i.ResourcePool)
	}

	return vals
}

func (t InstanceTable) buildName(i boshdir.VMInfo) boshtbl.ValueString {
	name := "?"

	if len(i.JobName) > 0 {
		name = i.JobName
	}

	if len(i.ID) > 0 {
		name += "/" + i.ID

		if i.Bootstrap {
			name += "*"
		}

		if i.Index != nil {
			name += fmt.Sprintf(" (%d)", *i.Index)
		}
	} else {
		if i.Index == nil {
			name += "/?"
		} else {
			name += fmt.Sprintf("/%d", *i.Index)
		}

		if i.Bootstrap {
			name += "*"
		}
	}

	return boshtbl.NewValueString(name)
}

func (t InstanceTable) ForProcess(p boshdir.VMInfoProcess) InstanceTableValues {
	return InstanceTableValues{
		Name:    boshtbl.ValueString{},
		Process: boshtbl.NewValueString(p.Name),

		ProcessState: boshtbl.ValueFmt{
			V:     boshtbl.NewValueString(p.State),
			Error: !p.IsRunning(),
		},

		// Vitals
		Uptime:   ValueUptime{p.Uptime.Seconds},
		Memory:   ValueMemIntSize{p.Mem},
		CPUTotal: ValueCPUTotal{p.CPU.Total},
	}
}

// AsValues is public instead of being private to aid ease of accessing vals in tests
func (t InstanceTable) AsValues(v InstanceTableValues) []boshtbl.Value {
	result := []boshtbl.Value{v.Name}

	if t.Processes {
		result = append(result, v.Process)
	}

	result = append(result, []boshtbl.Value{v.ProcessState, v.AZ, v.IPs}...)

	if t.Details {
		result = append(result, []boshtbl.Value{v.State, v.VMCID, v.VMType, v.DiskCIDs, v.AgentID, v.Resurrection, v.Ignore}...)
	} else if t.VMDetails {
		result = append(result, []boshtbl.Value{v.VMCID, v.VMType}...)
	}

	if t.DNS {
		result = append(result, v.DNS)
	}

	if t.Vitals {
		result = append(result, []boshtbl.Value{v.Uptime, v.Load}...)
		result = append(result, []boshtbl.Value{v.CPUTotal, v.CPUUser, v.CPUSys, v.CPUWait}...)
		result = append(result, []boshtbl.Value{v.Memory, v.Swap}...)
		result = append(result, []boshtbl.Value{v.SystemDisk, v.EphemeralDisk, v.PersistentDisk}...)
	}

	return result
}