mirror of https://github.com/k3s-io/k3s
246 lines
8.2 KiB
Go
246 lines
8.2 KiB
Go
/*
|
|
Copyright 2014 Google Inc. All rights reserved.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package util
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/spf13/cobra"
|
|
|
|
"github.com/GoogleCloudPlatform/kubernetes/pkg/api/meta"
|
|
"github.com/GoogleCloudPlatform/kubernetes/pkg/api/validation"
|
|
"github.com/GoogleCloudPlatform/kubernetes/pkg/kubectl/resource"
|
|
"github.com/GoogleCloudPlatform/kubernetes/pkg/labels"
|
|
"github.com/GoogleCloudPlatform/kubernetes/pkg/runtime"
|
|
"github.com/GoogleCloudPlatform/kubernetes/pkg/util"
|
|
)
|
|
|
|
// ResourcesFromArgsOrFile computes a list of Resources by extracting info from filename or args. It will
|
|
// handle label selectors provided.
|
|
func ResourcesFromArgsOrFile(
|
|
cmd *cobra.Command,
|
|
args []string,
|
|
filename, selector string,
|
|
typer runtime.ObjectTyper,
|
|
mapper meta.RESTMapper,
|
|
clientBuilder func(cmd *cobra.Command, mapping *meta.RESTMapping) (resource.RESTClient, error),
|
|
schema validation.Schema,
|
|
requireNames bool,
|
|
cmdNamespace,
|
|
cmdVersion string,
|
|
) resource.Visitor {
|
|
|
|
// handling filename & resource id
|
|
if len(selector) == 0 {
|
|
if requireNames || len(filename) > 0 {
|
|
mapping, namespace, name := ResourceFromArgsOrFile(cmd, args, filename, typer, mapper, schema, cmdNamespace, cmdVersion)
|
|
client, err := clientBuilder(cmd, mapping)
|
|
checkErr(err)
|
|
return resource.NewInfo(client, mapping, namespace, name)
|
|
}
|
|
if len(args) == 2 {
|
|
mapping, namespace, name := ResourceOrTypeFromArgs(cmd, args, mapper, cmdNamespace, cmdVersion)
|
|
client, err := clientBuilder(cmd, mapping)
|
|
checkErr(err)
|
|
return resource.NewInfo(client, mapping, namespace, name)
|
|
}
|
|
}
|
|
|
|
labelSelector, err := labels.ParseSelector(selector)
|
|
checkErr(err)
|
|
|
|
namespace := cmdNamespace
|
|
visitors := resource.VisitorList{}
|
|
|
|
if len(args) < 1 {
|
|
usageError(cmd, "Must specify the type of resource")
|
|
}
|
|
if len(args) > 1 {
|
|
usageError(cmd, "Too many arguments")
|
|
}
|
|
types := SplitResourceArgument(args[0])
|
|
for _, arg := range types {
|
|
resourceName := arg
|
|
if len(resourceName) == 0 {
|
|
usageError(cmd, "Unknown resource %s", resourceName)
|
|
}
|
|
version, kind, err := mapper.VersionAndKindForResource(resourceName)
|
|
checkErr(err)
|
|
|
|
mapping, err := mapper.RESTMapping(kind, version)
|
|
checkErr(err)
|
|
|
|
client, err := clientBuilder(cmd, mapping)
|
|
checkErr(err)
|
|
|
|
visitors = append(visitors, resource.NewSelector(client, mapping, namespace, labelSelector))
|
|
}
|
|
return visitors
|
|
}
|
|
|
|
// ResourceFromArgsOrFile expects two arguments or a valid file with a given type, and extracts
|
|
// the fields necessary to uniquely locate a resource. Displays a usageError if that contract is
|
|
// not satisfied, or a generic error if any other problems occur.
|
|
func ResourceFromArgsOrFile(cmd *cobra.Command, args []string, filename string, typer runtime.ObjectTyper, mapper meta.RESTMapper, schema validation.Schema, cmdNamespace, cmdVersion string) (mapping *meta.RESTMapping, namespace, name string) {
|
|
// If command line args are passed in, use those preferentially.
|
|
if len(args) > 0 && len(args) != 2 {
|
|
usageError(cmd, "If passing in command line parameters, must be resource and name")
|
|
}
|
|
|
|
if len(args) == 2 {
|
|
resource := args[0]
|
|
namespace = cmdNamespace
|
|
name = args[1]
|
|
if len(name) == 0 || len(resource) == 0 {
|
|
usageError(cmd, "Must specify filename or command line params")
|
|
}
|
|
|
|
defaultVersion, kind, err := mapper.VersionAndKindForResource(resource)
|
|
if err != nil {
|
|
// The error returned by mapper is "no resource defined", which is a usage error
|
|
usageError(cmd, err.Error())
|
|
}
|
|
mapping, err = mapper.RESTMapping(kind, cmdVersion, defaultVersion)
|
|
checkErr(err)
|
|
return
|
|
}
|
|
|
|
if len(filename) == 0 {
|
|
usageError(cmd, "Must specify filename or command line params")
|
|
}
|
|
|
|
mapping, namespace, name, _ = ResourceFromFile(filename, typer, mapper, schema, cmdVersion)
|
|
if len(name) == 0 {
|
|
checkErr(fmt.Errorf("the resource in the provided file has no name (or ID) defined"))
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// ResourceFromArgs expects two arguments with a given type, and extracts the fields necessary
|
|
// to uniquely locate a resource. Displays a usageError if that contract is not satisfied, or
|
|
// a generic error if any other problems occur.
|
|
func ResourceFromArgs(cmd *cobra.Command, args []string, mapper meta.RESTMapper, cmdNamespace string) (mapping *meta.RESTMapping, namespace, name string) {
|
|
if len(args) != 2 {
|
|
usageError(cmd, "Must provide resource and name command line params")
|
|
}
|
|
|
|
resource := args[0]
|
|
namespace = cmdNamespace
|
|
name = args[1]
|
|
if len(name) == 0 || len(resource) == 0 {
|
|
usageError(cmd, "Must provide resource and name command line params")
|
|
}
|
|
|
|
version, kind, err := mapper.VersionAndKindForResource(resource)
|
|
checkErr(err)
|
|
|
|
mapping, err = mapper.RESTMapping(kind, version)
|
|
checkErr(err)
|
|
return
|
|
}
|
|
|
|
// ResourceFromArgs expects two arguments with a given type, and extracts the fields necessary
|
|
// to uniquely locate a resource. Displays a usageError if that contract is not satisfied, or
|
|
// a generic error if any other problems occur.
|
|
func ResourceOrTypeFromArgs(cmd *cobra.Command, args []string, mapper meta.RESTMapper, cmdNamespace, cmdVersion string) (mapping *meta.RESTMapping, namespace, name string) {
|
|
if len(args) == 0 || len(args) > 2 {
|
|
usageError(cmd, "Must provide resource or a resource and name as command line params")
|
|
}
|
|
|
|
resource := args[0]
|
|
if len(resource) == 0 {
|
|
usageError(cmd, "Must provide resource or a resource and name as command line params")
|
|
}
|
|
|
|
namespace = cmdNamespace
|
|
if len(args) == 2 {
|
|
name = args[1]
|
|
if len(name) == 0 {
|
|
usageError(cmd, "Must provide resource or a resource and name as command line params")
|
|
}
|
|
}
|
|
|
|
defaultVersion, kind, err := mapper.VersionAndKindForResource(resource)
|
|
checkErr(err)
|
|
|
|
mapping, err = mapper.RESTMapping(kind, cmdVersion, defaultVersion)
|
|
checkErr(err)
|
|
|
|
return
|
|
}
|
|
|
|
// ResourceFromFile retrieves the name and namespace from a valid file. If the file does not
|
|
// resolve to a known type an error is returned. The returned mapping can be used to determine
|
|
// the correct REST endpoint to modify this resource with.
|
|
func ResourceFromFile(filename string, typer runtime.ObjectTyper, mapper meta.RESTMapper, schema validation.Schema, cmdVersion string) (mapping *meta.RESTMapping, namespace, name string, data []byte) {
|
|
configData, err := ReadConfigData(filename)
|
|
checkErr(err)
|
|
data = configData
|
|
|
|
objVersion, kind, err := typer.DataVersionAndKind(data)
|
|
checkErr(err)
|
|
|
|
// TODO: allow unversioned objects?
|
|
if len(objVersion) == 0 {
|
|
checkErr(fmt.Errorf("the resource in the provided file has no apiVersion defined"))
|
|
}
|
|
|
|
err = schema.ValidateBytes(data)
|
|
checkErr(err)
|
|
|
|
// decode using the version stored with the object (allows codec to vary across versions)
|
|
mapping, err = mapper.RESTMapping(kind, objVersion)
|
|
checkErr(err)
|
|
|
|
obj, err := mapping.Codec.Decode(data)
|
|
checkErr(err)
|
|
|
|
meta := mapping.MetadataAccessor
|
|
namespace, err = meta.Namespace(obj)
|
|
checkErr(err)
|
|
name, err = meta.Name(obj)
|
|
checkErr(err)
|
|
|
|
// if the preferred API version differs, get a different mapper
|
|
if cmdVersion != objVersion {
|
|
mapping, err = mapper.RESTMapping(kind, cmdVersion)
|
|
checkErr(err)
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// CompareNamespace returns an error if the namespace the user has provided on the CLI
|
|
// or via the default namespace file does not match the namespace of an input file. This
|
|
// prevents a user from unintentionally updating the wrong namespace.
|
|
func CompareNamespace(defaultNamespace, namespace string) error {
|
|
if len(namespace) > 0 {
|
|
if defaultNamespace != namespace {
|
|
return fmt.Errorf("the namespace from the provided file %q does not match the namespace %q. You must pass '--namespace=%s' to perform this operation.", namespace, defaultNamespace, namespace)
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func SplitResourceArgument(arg string) []string {
|
|
set := util.NewStringSet()
|
|
set.Insert(strings.Split(arg, ",")...)
|
|
return set.List()
|
|
}
|