package cmd

import (
  "fmt"
  "regexp"

  "github.com/teamhephy/pkg/prettyprint"

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

// TimeoutsList lists an app's timeouts.
func (d *DeisCmd) 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 *DeisCmd) 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 *DeisCmd) 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
}