-
Notifications
You must be signed in to change notification settings - Fork 326
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Backport of [NET-11256] Add
gateway list
command to consul-k8s CLI …
…into release/1.6.x (#4439) * backport of commit ffbecb3 * backport of commit d5a0b69 --------- Co-authored-by: Nathan Coleman <nathan.coleman@hashicorp.com>
- Loading branch information
1 parent
7314e75
commit 78ba2c3
Showing
4 changed files
with
481 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,3 @@ | ||
```release-note:enhancement | ||
cli: Introduce `gateway list` for collecting multiple components of all gateways' configuration by running a single command. | ||
``` |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,312 @@ | ||
package read | ||
|
||
import ( | ||
"archive/zip" | ||
"context" | ||
"encoding/json" | ||
"fmt" | ||
"os" | ||
"strings" | ||
"sync" | ||
|
||
helmcli "helm.sh/helm/v3/pkg/cli" | ||
"k8s.io/client-go/rest" | ||
"sigs.k8s.io/controller-runtime/pkg/client" | ||
gwv1alpha2 "sigs.k8s.io/gateway-api/apis/v1alpha2" | ||
gwv1beta1 "sigs.k8s.io/gateway-api/apis/v1beta1" | ||
"sigs.k8s.io/yaml" | ||
|
||
"github.com/hashicorp/consul-k8s/cli/common" | ||
"github.com/hashicorp/consul-k8s/cli/common/flag" | ||
"github.com/hashicorp/consul-k8s/cli/common/terminal" | ||
) | ||
|
||
type Command struct { | ||
*common.BaseCommand | ||
|
||
kubernetes client.Client | ||
restConfig *rest.Config | ||
|
||
set *flag.Sets | ||
|
||
flagAllNamespaces bool | ||
flagGatewayKind string | ||
flagGatewayNamespace string | ||
flagKubeConfig string | ||
flagKubeContext string | ||
flagOutput string | ||
|
||
initOnce sync.Once | ||
help string | ||
} | ||
|
||
func (c *Command) Help() string { | ||
c.initOnce.Do(c.init) | ||
return fmt.Sprintf("%s\n\nUsage: consul-k8s gateway list [flags]\n\n%s", c.Synopsis(), c.help) | ||
} | ||
|
||
func (c *Command) Synopsis() string { | ||
return "Inspect the configuration for all Gateways in the Kubernetes cluster or a given Kubernetes namespace." | ||
} | ||
|
||
// init establishes the flags for Command | ||
func (c *Command) init() { | ||
c.set = flag.NewSets() | ||
|
||
f := c.set.NewSet("Command Options") | ||
f.StringVar(&flag.StringVar{ | ||
Name: "namespace", | ||
Target: &c.flagGatewayNamespace, | ||
Usage: "The Kubernetes namespace to list Gateways in.", | ||
Aliases: []string{"n"}, | ||
}) | ||
f.BoolVar(&flag.BoolVar{ | ||
Name: "all-namespaces", | ||
Target: &c.flagAllNamespaces, | ||
Default: false, | ||
Usage: "List Gateways in all Kubernetes namespaces.", | ||
Aliases: []string{"A"}, | ||
}) | ||
f.StringVar(&flag.StringVar{ | ||
Name: "output", | ||
Target: &c.flagOutput, | ||
Usage: "Output the Gateway configuration as 'json' in the terminal or 'archive' as a zip archive named 'gateways.zip' in the current directory.", | ||
Default: "archive", | ||
Aliases: []string{"o"}, | ||
}) | ||
|
||
f = c.set.NewSet("Global Options") | ||
f.StringVar(&flag.StringVar{ | ||
Name: "kubeconfig", | ||
Aliases: []string{"c"}, | ||
Target: &c.flagKubeConfig, | ||
Usage: "Set the path to a kubeconfig file.", | ||
}) | ||
f.StringVar(&flag.StringVar{ | ||
Name: "context", | ||
Target: &c.flagKubeContext, | ||
Usage: "Set the Kubernetes context to use.", | ||
}) | ||
|
||
c.help = c.set.Help() | ||
} | ||
|
||
// Run runs the command | ||
func (c *Command) Run(args []string) int { | ||
c.initOnce.Do(c.init) | ||
c.Log.ResetNamed("read") | ||
defer common.CloseWithError(c.BaseCommand) | ||
|
||
if err := c.set.Parse(args); err != nil { | ||
c.UI.Output(err.Error(), terminal.WithErrorStyle()) | ||
return 1 | ||
} | ||
|
||
if err := c.initKubernetes(); err != nil { | ||
c.UI.Output(err.Error(), terminal.WithErrorStyle()) | ||
return 1 | ||
} | ||
|
||
if err := c.fetchCRDs(); err != nil { | ||
c.UI.Output(err.Error(), terminal.WithErrorStyle()) | ||
return 1 | ||
} | ||
|
||
return 0 | ||
} | ||
|
||
type gatewayWithRoutes struct { | ||
Gateway gwv1beta1.Gateway `json:"gateway"` | ||
GatewayClass gwv1beta1.GatewayClass `json:"gatewayClass"` | ||
HTTPRoutes []gwv1beta1.HTTPRoute `json:"httpRoutes"` | ||
TCPRoutes []gwv1alpha2.TCPRoute `json:"tcpRoutes"` | ||
} | ||
|
||
func (c *Command) fetchCRDs() error { | ||
// Fetch Gateways | ||
var gateways gwv1beta1.GatewayList | ||
if err := c.kubernetes.List(context.Background(), &gateways, client.InNamespace(c.flagGatewayNamespace)); err != nil { | ||
return fmt.Errorf("error fetching Gateway CRD: %w", err) | ||
} | ||
|
||
// Fetch all HTTPRoutes in all namespaces | ||
var httpRoutes gwv1beta1.HTTPRouteList | ||
if err := c.kubernetes.List(context.Background(), &httpRoutes); err != nil { | ||
return fmt.Errorf("error fetching HTTPRoute CRDs: %w", err) | ||
} | ||
|
||
// Fetch all TCPRoutes in all namespaces | ||
var tcpRoutes gwv1alpha2.TCPRouteList | ||
if err := c.kubernetes.List(context.Background(), &tcpRoutes); err != nil { | ||
return fmt.Errorf("error fetching TCPRoute CRDs: %w", err) | ||
} | ||
|
||
var gws []gatewayWithRoutes | ||
|
||
for _, gateway := range gateways.Items { | ||
// Fetch GatewayClass referenced by Gateway | ||
var gatewayClass gwv1beta1.GatewayClass | ||
if err := c.kubernetes.Get(context.Background(), client.ObjectKey{Namespace: "", Name: string(gateway.Spec.GatewayClassName)}, &gatewayClass); err != nil { | ||
return fmt.Errorf("error fetching GatewayClass CRD: %w", err) | ||
} | ||
|
||
// FUTURE Fetch GatewayClassConfig referenced by GatewayClass | ||
// This import requires resolving hairy dependency discrepancies between modules in this repo | ||
//var gatewayClassConfig v1alpha1.GatewayClassConfig | ||
//if err := c.kubernetes.Get(context.Background(), client.ObjectKey{Namespace: "", Name: gatewayClass.Spec.ParametersRef.Name}, &gatewayClassConfig); err != nil { | ||
// return fmt.Errorf("error fetching GatewayClassConfig CRD: %w", err) | ||
//} | ||
|
||
// FUTURE Fetch MeshServices referenced by HTTPRoutes or TCPRoutes | ||
// This import requires resolving hairy dependency discrepancies between modules in this repo | ||
// var meshServices v1alpha1.MeshServiceList | ||
// if err := c.kubernetes.List(context.Background(), &meshServices); err != nil { | ||
// return fmt.Errorf("error fetching MeshService CRDs: %w", err) | ||
// } | ||
|
||
gw := gatewayWithRoutes{ | ||
Gateway: gateway, | ||
GatewayClass: gatewayClass, | ||
HTTPRoutes: make([]gwv1beta1.HTTPRoute, 0, len(httpRoutes.Items)), | ||
TCPRoutes: make([]gwv1alpha2.TCPRoute, 0, len(tcpRoutes.Items)), | ||
} | ||
|
||
for _, route := range httpRoutes.Items { | ||
for _, ref := range route.Spec.ParentRefs { | ||
switch { | ||
case string(ref.Name) != gateway.Name: | ||
// Route parent references gateway with different name | ||
continue | ||
case ref.Namespace != nil && string(*ref.Namespace) == gateway.Namespace: | ||
// Route parent explicitly references gateway with same name and namespace | ||
gw.HTTPRoutes = append(gw.HTTPRoutes, route) | ||
case ref.Namespace == nil && route.Namespace == gateway.Namespace: | ||
// Route parent implicitly references gateway with same name in local namespace | ||
gw.HTTPRoutes = append(gw.HTTPRoutes, route) | ||
} | ||
} | ||
} | ||
|
||
for _, route := range tcpRoutes.Items { | ||
for _, ref := range route.Spec.ParentRefs { | ||
switch { | ||
case string(ref.Name) != gateway.Name: | ||
// Route parent references gateway with different name | ||
continue | ||
case ref.Namespace != nil && string(*ref.Namespace) == gateway.Namespace: | ||
// Route parent explicitly references gateway with same name and namespace | ||
gw.TCPRoutes = append(gw.TCPRoutes, route) | ||
case ref.Namespace == nil && route.Namespace == gateway.Namespace: | ||
// Route parent implicitly references gateway with same name in local namespace | ||
gw.TCPRoutes = append(gw.TCPRoutes, route) | ||
} | ||
} | ||
} | ||
|
||
gws = append(gws, gw) | ||
} | ||
|
||
switch strings.ToLower(c.flagOutput) { | ||
case "json": | ||
if err := c.writeJSONOutput(gws); err != nil { | ||
return fmt.Errorf("error writing CRDs as JSON: %w", err) | ||
} | ||
default: | ||
file, err := os.Create("./gateways.zip") | ||
if err != nil { | ||
return fmt.Errorf("error creating output file: %w", err) | ||
} | ||
|
||
zipw := zip.NewWriter(file) | ||
defer zipw.Close() | ||
|
||
if err := c.writeArchive(zipw, gws); err != nil { | ||
return fmt.Errorf("error writing CRDs to zip archive: %w", err) | ||
} | ||
c.UI.Output("Wrote to zip archive " + file.Name()) | ||
return zipw.Close() | ||
} | ||
|
||
return nil | ||
} | ||
|
||
func (c *Command) writeJSONOutput(obj interface{}) error { | ||
output, err := json.MarshalIndent(obj, "", "\t") | ||
if err != nil { | ||
return err | ||
} | ||
|
||
c.UI.Output(string(output)) | ||
return nil | ||
} | ||
|
||
// writeArchive writes one file to the zip archive for each Gateway in the list. | ||
// The files have name `<namespace>-<name>.yaml`. | ||
func (c *Command) writeArchive(zipw *zip.Writer, gws []gatewayWithRoutes) error { | ||
for _, gw := range gws { | ||
name := fmt.Sprintf("%s-%s.yaml", gw.Gateway.Namespace, gw.Gateway.Name) | ||
|
||
w, err := zipw.Create(name) | ||
if err != nil { | ||
return fmt.Errorf("error creating zip entry for %s: %w", name, err) | ||
} | ||
|
||
objYaml, err := yaml.Marshal(gw) | ||
if err != nil { | ||
return fmt.Errorf("error marshalling %s: %w", name, err) | ||
} | ||
|
||
_, err = w.Write(objYaml) | ||
if err != nil { | ||
return fmt.Errorf("error writing %s to zip archive: %w", name, err) | ||
} | ||
} | ||
|
||
return nil | ||
} | ||
|
||
// initKubernetes initializes the REST config and uses it to initialize the k8s client. | ||
func (c *Command) initKubernetes() (err error) { | ||
settings := helmcli.New() | ||
|
||
// If a kubeconfig was specified, use it | ||
if c.flagKubeConfig != "" { | ||
settings.KubeConfig = c.flagKubeConfig | ||
} | ||
|
||
// If a kube context was specified, use it | ||
if c.flagKubeContext != "" { | ||
settings.KubeContext = c.flagKubeContext | ||
} | ||
|
||
// Create a REST config from the settings for our Kubernetes client | ||
if c.restConfig == nil { | ||
if c.restConfig, err = settings.RESTClientGetter().ToRESTConfig(); err != nil { | ||
return fmt.Errorf("error creating Kubernetes REST config: %w", err) | ||
} | ||
} | ||
|
||
// Create a controller-runtime client from c.restConfig | ||
if c.kubernetes == nil { | ||
if c.kubernetes, err = client.New(c.restConfig, client.Options{}); err != nil { | ||
return fmt.Errorf("error creating controller-runtime client: %w", err) | ||
} | ||
// FUTURE Fix dependency discrepancies between modules in this repo so that this scheme can be added (see above) | ||
//_ = v1alpha1.AddToScheme(c.kubernetes.Scheme()) | ||
_ = gwv1alpha2.AddToScheme(c.kubernetes.Scheme()) | ||
_ = gwv1beta1.AddToScheme(c.kubernetes.Scheme()) | ||
} | ||
|
||
// If all namespaces specified, use empty namespace; otherwise, if | ||
// no namespace was specified, use the one from the kube context. | ||
if c.flagAllNamespaces { | ||
c.flagGatewayNamespace = "" | ||
} else if c.flagGatewayNamespace == "" { | ||
if c.flagOutput != "json" { | ||
c.UI.Output("No namespace specified, using current kube context namespace: %s", settings.Namespace()) | ||
} | ||
c.flagGatewayNamespace = settings.Namespace() | ||
} | ||
|
||
return nil | ||
} |
Oops, something went wrong.