package isready

import (
	"fmt"
	"github.com/spf13/cobra"
	"github.com/teamhanko/hanko/backend/config"
	"log"
	"net"
	"net/http"
)

func NewIsReadyCommand() *cobra.Command {
	var (
		configFile string
	)

	cmd := &cobra.Command{
		Use:       "isready",
		Args:      cobra.OnlyValidArgs,
		ValidArgs: []string{"admin", "public"},
		Short:     "Health check a service",
		Long: `Checks if the specified service is healthy. Possible values are "admin" and "public".
Uses the "/health/ready" endpoint to check if the service is ready to serve requests.
		`,
		Run: func(cmd *cobra.Command, args []string) {
			if len(args) != 1 {
				log.Fatalf("Please specify a service to check")
			}
			service := args[0]

			cfg, err := config.Load(&configFile)
			if err != nil {
				log.Fatal(err)
			}

			address := ""
			if service == "admin" {
				address = cfg.Server.Admin.Address
			}
			if service == "public" {
				address = cfg.Server.Public.Address
			}
			host, port, err := net.SplitHostPort(address)
			if err != nil {
				log.Fatalf("Could not parse address %s", address)
			}
			if host == "" {
				host = "localhost"
			}
			requestUrl := fmt.Sprintf("http://%s:%s/health/ready", host, port)
			res, err := http.Get(requestUrl)
			if err != nil {
				log.Fatalf("Service %s is not ready", service)
			} else {
				if res.StatusCode != 200 {
					log.Fatalf("Service %s is not ready", service)
				} else {
					log.Println(fmt.Sprintf("Service %s is ready", service))
				}
			}
		},
	}

	cmd.Flags().StringVar(&configFile, "config", config.DefaultConfigFilePath, "config file")

	return cmd
}

func RegisterCommands(parent *cobra.Command) {
	cmd := NewIsReadyCommand()
	parent.AddCommand(cmd)
}