mirror of
https://github.com/kubernetes-sigs/kustomize.git
synced 2026-07-17 17:52:12 +00:00
support yaml formatted openapi schema (#4017)
* support yaml formatted openapi schema * suggested changes
This commit is contained in:
@@ -8,26 +8,38 @@ import (
|
||||
"os/exec"
|
||||
|
||||
"github.com/spf13/cobra"
|
||||
"sigs.k8s.io/yaml"
|
||||
)
|
||||
|
||||
var format string
|
||||
|
||||
// NewCmdFetch makes a new fetch command.
|
||||
func NewCmdFetch(w io.Writer) *cobra.Command {
|
||||
infoCmd := cobra.Command{
|
||||
fetchCmd := cobra.Command{
|
||||
Use: "fetch",
|
||||
Short: `Fetches the OpenAPI specification from the current kubernetes cluster specified
|
||||
in the user's kubeconfig`,
|
||||
Example: `kustomize openapi fetch`,
|
||||
Run: func(cmd *cobra.Command, args []string) {
|
||||
printSchema(w)
|
||||
RunE: func(cmd *cobra.Command, args []string) error {
|
||||
return printSchema(w)
|
||||
},
|
||||
}
|
||||
return &infoCmd
|
||||
fetchCmd.Flags().StringVar(
|
||||
&format,
|
||||
"format",
|
||||
"json",
|
||||
"Specify format for fetched schema ('json' or 'yaml')")
|
||||
return &fetchCmd
|
||||
}
|
||||
|
||||
func printSchema(w io.Writer) {
|
||||
func printSchema(w io.Writer) error {
|
||||
if format != "json" && format != "yaml" {
|
||||
return fmt.Errorf("format must be either 'json' or 'yaml'")
|
||||
}
|
||||
|
||||
errMsg := `
|
||||
Error fetching schema from cluster.
|
||||
Please make sure kubectl is installed and its context is set correctly.
|
||||
Please make sure kubectl is installed, its context is set correctly, and your cluster is up.
|
||||
Installation and setup instructions: https://kubernetes.io/docs/tasks/tools/install-kubectl/`
|
||||
|
||||
command := exec.Command("kubectl", []string{"get", "--raw", "/openapi/v2"}...)
|
||||
@@ -36,9 +48,10 @@ Installation and setup instructions: https://kubernetes.io/docs/tasks/tools/inst
|
||||
command.Stdout = &stdout
|
||||
command.Stderr = &stderr
|
||||
err := command.Run()
|
||||
if err != nil || stdout.String() == "" {
|
||||
fmt.Fprintln(w, err, stderr.String()+errMsg)
|
||||
return
|
||||
if err != nil {
|
||||
return fmt.Errorf("%w\n%s", err, stderr.String()+errMsg)
|
||||
} else if stdout.String() == "" {
|
||||
return fmt.Errorf(stderr.String() + errMsg)
|
||||
}
|
||||
|
||||
// format and output
|
||||
@@ -46,5 +59,14 @@ Installation and setup instructions: https://kubernetes.io/docs/tasks/tools/inst
|
||||
output := stdout.Bytes()
|
||||
json.Unmarshal(output, &jsonSchema)
|
||||
output, _ = json.MarshalIndent(jsonSchema, "", " ")
|
||||
|
||||
if format == "yaml" {
|
||||
output, err = yaml.JSONToYAML(output)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
fmt.Fprintln(w, string(output))
|
||||
return nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user