snapshot of https://github.com/ory/keto.git (master, 2026-09-16; blobs > 131072B omitted)

cmd/status/root.go

@@ -0,0 +1,94 @@
+// Copyright © 2023 Ory Corp
+// SPDX-License-Identifier: Apache-2.0
+
+package status
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"time"
+
+	"github.com/ory/x/cmdx"
+	"github.com/ory/x/stringsx"
+	"github.com/spf13/cobra"
+	"google.golang.org/grpc"
+	grpcHealthV1 "google.golang.org/grpc/health/grpc_health_v1"
+
+	cliclient "github.com/ory/keto/cmd/client"
+)
+
+const (
+	FlagEndpoint = "endpoint"
+)
+
+func newStatusCmd() *cobra.Command {
+	var endpoint string
+
+	cmd := &cobra.Command{
+		Use:   "status",
+		Short: "Get the status of the upstream Keto instance",
+		Long:  "Get a status report about the upstream Keto instance. Can also block until the service is healthy.",
+		Args:  cobra.ExactArgs(0),
+		RunE: func(cmd *cobra.Command, _ []string) error {
+			block, err := cmd.Flags().GetBool(cliclient.FlagBlock)
+			if err != nil {
+				return err
+			}
+
+			var connect func(*cobra.Command) (*grpc.ClientConn, error)
+
+			switch endpoints := stringsx.SwitchExact(endpoint); {
+			case endpoints.AddCase("read"):
+				connect = cliclient.GetReadConn
+			case endpoints.AddCase("write"):
+				connect = cliclient.GetWriteConn
+			default:
+				return endpoints.ToUnknownCaseErr()
+			}
+
+			loudPrinter := cmdx.NewLoudOutPrinter(cmd)
+
+			conn, err := connect(cmd)
+			for block && err != nil {
+				if !errors.Is(err, context.DeadlineExceeded) {
+					return err
+				}
+				_, _ = loudPrinter.Println("Context deadline exceeded, going to retry.")
+				conn, err = connect(cmd)
+			}
+
+			if err != nil {
+				_, _ = fmt.Fprint(cmd.ErrOrStderr(), err.Error())
+				return cmdx.FailSilently(cmd)
+			}
+
+			c := grpcHealthV1.NewHealthClient(conn)
+
+			for {
+				status, err := c.Check(cmd.Context(), &grpcHealthV1.HealthCheckRequest{})
+				if err != nil {
+					_, _ = fmt.Fprintf(cmd.ErrOrStderr(), "Unable to get a check response: %+v\n", err)
+					return cmdx.FailSilently(cmd)
+				}
+				if !block || status.GetStatus() == grpcHealthV1.HealthCheckResponse_SERVING {
+					_, _ = fmt.Fprintln(cmd.OutOrStdout(), status.GetStatus().String())
+					return nil
+				}
+				_, _ = loudPrinter.Printf("Got the status %s, waiting until %s.\n", status.GetStatus(), grpcHealthV1.HealthCheckResponse_SERVING)
+				time.Sleep(500 * time.Millisecond)
+			}
+		},
+	}
+
+	cliclient.RegisterRemoteURLFlags(cmd.Flags())
+	cmdx.RegisterNoiseFlags(cmd.Flags())
+
+	cmd.Flags().StringVar(&endpoint, FlagEndpoint, "read", "which endpoint to use; one of {read, write}")
+
+	return cmd
+}
+
+func RegisterCommandRecursive(parent *cobra.Command) {
+	parent.AddCommand(newStatusCmd())
+}