check.go

package check

import (
	"time"

	"github.com/ParetoSecurity/pareto-core/shared"
)

var AvailableChecks = 0

type Check interface {
	Name() string
	PassedMessage() string
	FailedMessage() string
	Run() error
	Passed() bool
	IsRunnable() bool
	UUID() string
	Status() string
	RequiresRoot() bool
}

func Register(c Check) Check {

	AvailableChecks = +1

	// If the check is already in the checks map, return it
	if found := shared.Config.Checks[c.UUID()]; found != (shared.CheckStatus{}) {
		return c
	}

	// If the checks map is nil, create it
	if shared.Config.Checks == nil {
		shared.Config.Checks = make(map[string]shared.CheckStatus)
	}

	// Add the check to the checks map
	shared.Config.Checks[c.UUID()] = shared.CheckStatus{
		UpdatedAt: time.Now(),
		Passed:    c.Passed(),
		Disabled:  !c.IsRunnable(),
	}
	return c
}

func Update(c Check) Check {
	shared.Config.Checks[c.UUID()] = shared.CheckStatus{
		UpdatedAt: time.Now(),
		Passed:    c.Passed(),
		Disabled:  !c.IsRunnable(),
	}
	return c
}