wtf/modules/opsgenie/settings.go

package opsgenie

import (
	"os"

	"github.com/olebedev/config"
	"github.com/wtfutil/wtf/cfg"
)

const defaultTitle = "OpsGenie"

type Settings struct {
	common *cfg.Common

	apiKey                 string   `help:"Your OpsGenie API token."`
	region                 string   `help:"Defines region to use. Possible options: us (by default), eu." optional:"true"`
	displayEmpty           bool     `help:"Whether schedules with no assigned person on-call should be displayed." optional:"true"`
	schedule               []string `help:"A list of names of the schedule(s) to retrieve."`
	scheduleIdentifierType string   `help:"Type of the schedule identifier." values:"id or name" optional:"true"`
}

func NewSettingsFromYAML(name string, ymlConfig *config.Config, globalConfig *config.Config) *Settings {

	settings := Settings{
		common: cfg.NewCommonSettingsFromModule(name, defaultTitle, ymlConfig, globalConfig),

		apiKey:                 ymlConfig.UString("apiKey", os.Getenv("WTF_OPS_GENIE_API_KEY")),
		region:                 ymlConfig.UString("region", "us"),
		displayEmpty:           ymlConfig.UBool("displayEmpty", true),
		scheduleIdentifierType: ymlConfig.UString("scheduleIdentifierType", "id"),
	}

	settings.schedule = settings.arrayifySchedules(ymlConfig, globalConfig)

	return &settings
}

// arrayifySchedules figures out if we're dealing with a single project or an array of projects
func (settings *Settings) arrayifySchedules(ymlConfig *config.Config, globalConfig *config.Config) []string {
	schedules := []string{}

	// Single schedule
	schedule, err := ymlConfig.String("schedule")
	if err == nil {
		schedules = append(schedules, schedule)
		return schedules
	}

	// Array of schedules
	scheduleList := ymlConfig.UList("schedule")
	for _, scheduleName := range scheduleList {
		if schedule, ok := scheduleName.(string); ok {
			schedules = append(schedules, schedule)
		}
	}

	return schedules
}