Merge pull request #3621 from natasha41575/OpenApiFetchCommand

added `kustomize openapi fetch` command to get schema from local cluster
This commit is contained in:
Kubernetes Prow Robot
2021-02-24 14:37:24 -08:00
committed by GitHub
3 changed files with 77 additions and 0 deletions

View File

@@ -0,0 +1,73 @@
package fetch
import (
"bytes"
"encoding/json"
"fmt"
"io"
"os/exec"
"time"
"github.com/spf13/cobra"
)
// NewCmdFetch makes a new fetch command.
func NewCmdFetch(w io.Writer) *cobra.Command {
infoCmd := 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)
},
Hidden: true,
}
return &infoCmd
}
func printSchema(w io.Writer) {
fmt.Fprintln(w, "Fetching schema from cluster")
errMsg := `
Error fetching schema from cluster.
Please make sure port 8081 is available, kubectl is installed, and its context is set correctly.
Installation and setup instructions: https://kubernetes.io/docs/tasks/tools/install-kubectl/`
command := exec.Command("kubectl", []string{"proxy", "--port=8081", "&"}...)
var stderr bytes.Buffer
command.Stderr = &stderr
err := command.Start()
defer killProcess(command)
// give the proxy a second to start up
time.Sleep(time.Second)
if err != nil || stderr.String() != "" {
fmt.Fprintln(w, err, stderr.String()+errMsg)
return
}
commandCurl := exec.Command("curl", []string{"http://localhost:8081/openapi/v2"}...)
var stdout bytes.Buffer
commandCurl.Stdout = &stdout
commandCurl.Stderr = &stderr
err = commandCurl.Run()
if err != nil || stdout.String() == "" {
fmt.Fprintln(w, err, stderr.String()+errMsg)
return
}
// format and output
var jsonSchema map[string]interface{}
output := stdout.Bytes()
json.Unmarshal(output, &jsonSchema)
output, _ = json.MarshalIndent(jsonSchema, "", " ")
fmt.Fprintln(w, string(output))
}
func killProcess(command *exec.Cmd) {
if command.Process != nil {
command.Process.Kill()
}
}

View File

@@ -8,6 +8,7 @@ import (
"github.com/spf13/cobra"
"sigs.k8s.io/kustomize/cmd/config/configcobra"
"sigs.k8s.io/kustomize/kustomize/v4/commands/openapi/fetch"
"sigs.k8s.io/kustomize/kustomize/v4/commands/openapi/info"
)
@@ -22,6 +23,7 @@ func NewCmdOpenAPI(w io.Writer) *cobra.Command {
}
openApiCmd.AddCommand(info.NewCmdInfo(w))
openApiCmd.AddCommand(fetch.NewCmdFetch(w))
configcobra.AddCommands(openApiCmd, "openapi")
return openApiCmd

View File

@@ -23,3 +23,5 @@ replace sigs.k8s.io/kustomize/api => ../api
replace sigs.k8s.io/kustomize/cmd/config => ../cmd/config
replace sigs.k8s.io/kustomize/kyaml => ../kyaml
replace sigs.k8s.io/kustomize/kustomize/v4/commands => ./commands