model/status.go

Rama main · réplica de lectura del repositorio original.

package model

import (
	"encoding/json"
)

const (
	UNKNOWN Status = iota
	OK
	INFO
	WARNING
	CRITICAL
)

type Status int

func (s Status) String() string {
	switch s {
	case OK:
		return "ok"
	case INFO:
		return "info"
	case WARNING:
		return "warning"
	case CRITICAL:
		return "critical"
	default:
		return "unknown"
	}
}

func (s Status) MarshalJSON() ([]byte, error) {
	return json.Marshal(s.String())
}

func (s *Status) UnmarshalJSON(data []byte) error {
	var str string
	if err := json.Unmarshal(data, &str); err != nil {
		return err
	}
	switch str {
	case "ok":
		*s = OK
	case "info":
		*s = INFO
	case "warning":
		*s = WARNING
	case "critical":
		*s = CRITICAL
	default:
		*s = UNKNOWN
	}
	return nil
}

func (s Status) Color() string {
	switch s {
	case OK:
		return "#23d160"
	case WARNING:
		return "#ffdd57"
	case CRITICAL:
		return "#f44034"
	}
	return "#d3d3d3"
}

type Indicator struct {
	Status  Status `json:"status"`
	Message string `json:"message"`
}

func CalcIndicators(app *Application) []Indicator {
	if app == nil {
		return nil
	}
	var res []Indicator
	for _, r := range app.Reports {
		if r.Status == UNKNOWN {
			continue
		}
		res = append(res, Indicator{
			Status:  r.Status,
			Message: string(r.Name),
		})
	}
	return res
}