// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
// vespa status command
// author: bratseth

package cmd

import (
	"fmt"
	"log"
	"strconv"
	"time"

	"github.com/fatih/color"
	"github.com/spf13/cobra"
	"github.com/vespa-engine/vespa/client/go/vespa"
)

func newStatusCmd(cli *CLI) *cobra.Command {
	return &cobra.Command{
		Use:               "status",
		Short:             "Verify that a service is ready to use (query by default)",
		Example:           `$ vespa status query`,
		DisableAutoGenTag: true,
		SilenceUsage:      true,
		Args:              cobra.MaximumNArgs(1),
		RunE: func(cmd *cobra.Command, args []string) error {
			return printServiceStatus(cli, vespa.QueryService)
		},
	}
}

func newStatusQueryCmd(cli *CLI) *cobra.Command {
	return &cobra.Command{
		Use:               "query",
		Short:             "Verify that the query service is ready to use (default)",
		Example:           `$ vespa status query`,
		DisableAutoGenTag: true,
		SilenceUsage:      true,
		Args:              cobra.ExactArgs(0),
		RunE: func(cmd *cobra.Command, args []string) error {
			return printServiceStatus(cli, vespa.QueryService)
		},
	}
}

func newStatusDocumentCmd(cli *CLI) *cobra.Command {
	return &cobra.Command{
		Use:               "document",
		Short:             "Verify that the document service is ready to use",
		Example:           `$ vespa status document`,
		DisableAutoGenTag: true,
		SilenceUsage:      true,
		Args:              cobra.ExactArgs(0),
		RunE: func(cmd *cobra.Command, args []string) error {
			return printServiceStatus(cli, vespa.DocumentService)
		},
	}
}

func newStatusDeployCmd(cli *CLI) *cobra.Command {
	return &cobra.Command{
		Use:               "deploy",
		Short:             "Verify that the deploy service is ready to use",
		Example:           `$ vespa status deploy`,
		DisableAutoGenTag: true,
		SilenceUsage:      true,
		Args:              cobra.ExactArgs(0),
		RunE: func(cmd *cobra.Command, args []string) error {
			return printServiceStatus(cli, vespa.DeployService)
		},
	}
}

func printServiceStatus(cli *CLI, name string) error {
	t, err := cli.target(targetOptions{})
	if err != nil {
		return err
	}
	timeout := time.Duration(cli.flags.waitSecs) * time.Second
	if timeout > 0 {
		log.Printf("Waiting up to %s %s for service to become ready ...", color.CyanString(strconv.Itoa(cli.flags.waitSecs)), color.CyanString("seconds"))
	}
	s, err := t.Service(name, timeout, 0, "")
	if err != nil {
		return err
	}
	status, err := s.Wait(timeout)
	if status/100 == 2 {
		log.Print(s.Description(), " at ", color.CyanString(s.BaseURL), " is ", color.GreenString("ready"))
	} else {
		if err == nil {
			err = fmt.Errorf("status %d", status)
		}
		return fmt.Errorf("%s at %s is %s: %w", s.Description(), color.CyanString(s.BaseURL), color.RedString("not ready"), err)
	}
	return nil
}