You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
144 lines
3.3 KiB
Go
144 lines
3.3 KiB
Go
package data
|
|
|
|
import (
|
|
"encoding/hex"
|
|
"errors"
|
|
"math/rand"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/containrrr/watchtower/pkg/types"
|
|
)
|
|
|
|
type previewData struct {
|
|
rand *rand.Rand
|
|
lastTime time.Time
|
|
report *report
|
|
containerCount int
|
|
Entries []*logEntry
|
|
StaticData staticData
|
|
}
|
|
|
|
type staticData struct {
|
|
Title string
|
|
Host string
|
|
}
|
|
|
|
// New initializes a new preview data struct
|
|
func New() *previewData {
|
|
return &previewData{
|
|
rand: rand.New(rand.NewSource(1)),
|
|
lastTime: time.Now().Add(-30 * time.Minute),
|
|
report: nil,
|
|
containerCount: 0,
|
|
Entries: []*logEntry{},
|
|
StaticData: staticData{
|
|
Title: "Title",
|
|
Host: "Host",
|
|
},
|
|
}
|
|
}
|
|
|
|
// AddFromState adds a container status entry to the report with the given state
|
|
func (pb *previewData) AddFromState(state State) {
|
|
cid := types.ContainerID(pb.generateID())
|
|
old := types.ImageID(pb.generateID())
|
|
new := types.ImageID(pb.generateID())
|
|
name := pb.generateName()
|
|
image := pb.generateImageName(name)
|
|
var err error
|
|
if state == FailedState {
|
|
err = errors.New(pb.randomEntry(errorMessages))
|
|
} else if state == SkippedState {
|
|
err = errors.New(pb.randomEntry(skippedMessages))
|
|
}
|
|
pb.addContainer(containerStatus{
|
|
containerID: cid,
|
|
oldImage: old,
|
|
newImage: new,
|
|
containerName: name,
|
|
imageName: image,
|
|
error: err,
|
|
state: state,
|
|
})
|
|
}
|
|
|
|
func (pb *previewData) addContainer(c containerStatus) {
|
|
if pb.report == nil {
|
|
pb.report = &report{}
|
|
}
|
|
switch c.state {
|
|
case ScannedState:
|
|
pb.report.scanned = append(pb.report.scanned, &c)
|
|
case UpdatedState:
|
|
pb.report.updated = append(pb.report.updated, &c)
|
|
case FailedState:
|
|
pb.report.failed = append(pb.report.failed, &c)
|
|
case SkippedState:
|
|
pb.report.skipped = append(pb.report.skipped, &c)
|
|
case StaleState:
|
|
pb.report.stale = append(pb.report.stale, &c)
|
|
case FreshState:
|
|
pb.report.fresh = append(pb.report.fresh, &c)
|
|
default:
|
|
return
|
|
}
|
|
pb.containerCount += 1
|
|
}
|
|
|
|
// AddLogEntry adds a preview log entry of the given level
|
|
func (pd *previewData) AddLogEntry(level LogLevel) {
|
|
var msg string
|
|
switch level {
|
|
case FatalLevel:
|
|
fallthrough
|
|
case ErrorLevel:
|
|
fallthrough
|
|
case WarnLevel:
|
|
msg = pd.randomEntry(logErrors)
|
|
default:
|
|
msg = pd.randomEntry(logMessages)
|
|
}
|
|
pd.Entries = append(pd.Entries, &logEntry{
|
|
Message: msg,
|
|
Data: map[string]any{},
|
|
Time: pd.generateTime(),
|
|
Level: level,
|
|
})
|
|
}
|
|
|
|
// Report returns a preview report
|
|
func (pb *previewData) Report() types.Report {
|
|
return pb.report
|
|
}
|
|
|
|
func (pb *previewData) generateID() string {
|
|
buf := make([]byte, 32)
|
|
_, _ = pb.rand.Read(buf)
|
|
return hex.EncodeToString(buf)
|
|
}
|
|
|
|
func (pb *previewData) generateTime() time.Time {
|
|
pb.lastTime = pb.lastTime.Add(time.Duration(pb.rand.Intn(30)) * time.Second)
|
|
return pb.lastTime
|
|
}
|
|
|
|
func (pb *previewData) randomEntry(arr []string) string {
|
|
return arr[pb.rand.Intn(len(arr))]
|
|
}
|
|
|
|
func (pb *previewData) generateName() string {
|
|
index := pb.containerCount
|
|
if index <= len(containerNames) {
|
|
return "/" + containerNames[index]
|
|
}
|
|
suffix := index / len(containerNames)
|
|
index %= len(containerNames)
|
|
return "/" + containerNames[index] + strconv.FormatInt(int64(suffix), 10)
|
|
}
|
|
|
|
func (pb *previewData) generateImageName(name string) string {
|
|
index := pb.containerCount % len(organizationNames)
|
|
return organizationNames[index] + name + ":latest"
|
|
}
|