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

cmd/jsonnet/format.go

@@ -0,0 +1,61 @@
+// Copyright © 2023 Ory Corp
+// SPDX-License-Identifier: Apache-2.0
+
+package jsonnet
+
+import (
+	"fmt"
+	"os"
+	"path/filepath"
+
+	"github.com/google/go-jsonnet/formatter"
+	"github.com/spf13/cobra"
+
+	"github.com/ory/x/cmdx"
+	"github.com/ory/x/flagx"
+)
+
+func NewFormatCmd() *cobra.Command {
+	c := &cobra.Command{
+		Use:   "format",
+		Short: "Helpers for formatting code",
+	}
+	c.AddCommand(NewJsonnetFormatCmd())
+	return c
+}
+
+func NewJsonnetFormatCmd() *cobra.Command {
+	c := &cobra.Command{
+		Use: "jsonnet path/to/files/*.jsonnet [more/files.jsonnet] [supports/**/{foo,bar}.jsonnet]",
+		Long: `Formats JSONNet files using the official JSONNet formatter.
+
+Use -w or --write to write output back to files instead of stdout.
+
+` + GlobHelp,
+		Args: cobra.MinimumNArgs(1),
+		Run: func(cmd *cobra.Command, args []string) {
+			for _, pattern := range args {
+				files, err := filepath.Glob(pattern)
+				cmdx.Must(err, `Glob path "%s" is not valid: %s`, pattern, err)
+
+				shouldWrite := flagx.MustGetBool(cmd, "write")
+				for _, file := range files {
+					content, err := os.ReadFile(file) // #nosec G304 -- the file is provided by the user
+					cmdx.Must(err, `Unable to read file "%s" because: %s`, file, err)
+
+					output, err := formatter.Format(file, string(content), formatter.DefaultOptions())
+					cmdx.Must(err, `JSONNet file "%s" could not be formatted: %s`, file, err)
+
+					if shouldWrite {
+						err := os.WriteFile(file, []byte(output), 0o644) //#nosec
+						cmdx.Must(err, `Could not write to file "%s" because: %s`, file, err)
+					} else {
+						fmt.Println(output)
+					}
+				}
+			}
+		},
+	}
+	c.Flags().BoolP("write", "w", false, "Write formatted output back to file.")
+	return c
+}