dokku/plugins/cron/report.go

package cron

import (
	"fmt"
	"strconv"
	"strings"

	"github.com/dokku/dokku/plugins/common"
)

// ReportSingleApp is an internal function that displays the cron report for one or more apps
func ReportSingleApp(appName string, format string, infoFlag string) error {
	if err := common.VerifyAppName(appName); err != nil {
		return err
	}

	flags := map[string]common.ReportFunc{
		"--cron-mailfrom":             reportMailfrom,
		"--cron-mailto":               reportMailto,
		"--cron-task-count":           reportTasks,
		"--cron-global-maintenance":   reportGlobalMaintenance,
		"--cron-computed-maintenance": reportComputedMaintenance,
		"--cron-maintenance":          reportMaintenance,
	}

	extraFlags := addCronMaintenanceFlags(appName, infoFlag)
	for flag, fn := range extraFlags {
		flags[flag] = fn
	}

	flagKeys := []string{}
	for flagKey := range flags {
		flagKeys = append(flagKeys, flagKey)
	}

	trimPrefix := false
	uppercaseFirstCharacter := true
	infoFlags := common.CollectReport(appName, infoFlag, flags)
	return common.ReportSingleApp("cron", appName, infoFlag, infoFlags, flagKeys, format, trimPrefix, uppercaseFirstCharacter)
}

func addCronMaintenanceFlags(appName string, infoFlag string) map[string]common.ReportFunc {
	flags := map[string]common.ReportFunc{}

	properties, err := common.PropertyGetAllByPrefix("cron", appName, MaintenancePropertyPrefix)
	if err != nil {
		return flags
	}

	for property, value := range properties {
		key := strings.Replace(property, MaintenancePropertyPrefix, "", 1)
		flags[fmt.Sprintf("--cron-maintenance-%s", key)] = func(appName string) string {
			return value
		}
	}

	return flags
}

func reportMailfrom(_ string) string {
	return common.PropertyGet("cron", "--global", "mailfrom")
}

func reportMailto(_ string) string {
	return common.PropertyGet("cron", "--global", "mailto")
}

func reportTasks(appName string) string {
	c, _ := FetchCronTasks(FetchCronTasksInput{AppName: appName})
	return strconv.Itoa(len(c))
}

func reportGlobalMaintenance(_ string) string {
	return common.PropertyGet("cron", "--global", "maintenance")
}

func reportComputedMaintenance(appName string) string {
	maintenance := common.PropertyGet("cron", appName, "maintenance")
	if maintenance == "true" {
		return "true"
	}

	return common.PropertyGetDefault("cron", "--global", "maintenance", DefaultProperties["maintenance"])
}

func reportMaintenance(appName string) string {
	return common.PropertyGet("cron", appName, "maintenance")
}