timeouts.go

package cmd

import (
	"fmt"
	"regexp"

	"github.com/drycc/pkg/prettyprint"

	"github.com/drycc/controller-sdk-go/api"
	"github.com/drycc/controller-sdk-go/config"
)

// TimeoutsList lists an app's timeouts.
func (d *DryccCmd) TimeoutsList(appID string) error {
	s, appID, err := load(d.ConfigFile, appID)

	if err != nil {
		return err
	}

	config, err := config.List(s.Client, appID)
	if d.checkAPICompatibility(s.Client, err) != nil {
		return err
	}

	d.Printf("=== %s Timeouts (sec)\n", appID)

	if len(config.Timeout) == 0 {
		d.Println("default (30 sec) or controlled by env KUBERNETES_POD_TERMINATION_GRACE_PERIOD_SECONDS")
	} else {
		timeoutsMap := make(map[string]string)

		for key, value := range config.Timeout {
			timeoutsMap[key] = fmt.Sprintf("%v", value)
		}

		d.Print(prettyprint.PrettyTabs(timeoutsMap, 5))
	}

	return nil
}

// TimeoutsSet sets an app's timeouts.
func (d *DryccCmd) TimeoutsSet(appID string, timeouts []string) error {
	s, appID, err := load(d.ConfigFile, appID)

	if err != nil {
		return err
	}

	timeoutsMap, err := parseTimeouts(timeouts)
	if err != nil {
		return err
	}

	d.Print("Applying timeouts... ")

	quit := progress(d.WOut)
	configObj := api.Config{}

	configObj.Timeout = timeoutsMap

	_, err = config.Set(s.Client, appID, configObj)
	quit <- true
	<-quit
	if d.checkAPICompatibility(s.Client, err) != nil {
		return err
	}

	d.Print("done\n\n")

	return d.TimeoutsList(appID)
}

// TimeoutsUnset removes an app's timeouts.
func (d *DryccCmd) TimeoutsUnset(appID string, timeouts []string) error {
	s, appID, err := load(d.ConfigFile, appID)

	if err != nil {
		return err
	}

	d.Print("Applying timeouts... ")

	quit := progress(d.WOut)

	configObj := api.Config{}

	valuesMap := make(map[string]interface{})

	for _, timeout := range timeouts {
		valuesMap[timeout] = nil
	}

	configObj.Timeout = valuesMap

	_, err = config.Set(s.Client, appID, configObj)
	quit <- true
	<-quit
	if d.checkAPICompatibility(s.Client, err) != nil {
		return err
	}

	d.Print("done\n\n")

	return d.TimeoutsList(appID)
}

func parseTimeouts(timeouts []string) (map[string]interface{}, error) {
	timeoutsMap := make(map[string]interface{})

	for _, timeout := range timeouts {
		key, value, err := parseTimeout(timeout)

		if err != nil {
			return nil, err
		}

		timeoutsMap[key] = value
	}

	return timeoutsMap, nil
}

func parseTimeout(timeout string) (string, string, error) {
	regex := regexp.MustCompile("^([a-z0-9]+(?:-[a-z0-9]+)*)=([0-9]+)$")

	if !regex.MatchString(timeout) {
		return "", "", fmt.Errorf(`%s doesn't fit format type=#
Examples: web=30 worker=300`, timeout)
	}

	capture := regex.FindStringSubmatch(timeout)

	return capture[1], capture[2], nil
}