266 lines
8.4 KiB
Go
266 lines
8.4 KiB
Go
|
/*
|
||
|
|
Copyright 2014 The Kubernetes Authors.
|
||
|
|
|
||
|
|
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.
|
||
|
|
*/
|
||
|
|
|
||
|
|
// A set of common functions needed by cmd/kubectl and pkg/kubectl packages.
|
||
|
|
package kubectl
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"fmt"
|
||
|
|
"path"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"k8s.io/apimachinery/pkg/api/meta"
|
||
|
|
"k8s.io/apimachinery/pkg/runtime/schema"
|
||
|
|
"k8s.io/kubernetes/pkg/api"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
kubectlAnnotationPrefix = "kubectl.kubernetes.io/"
|
||
|
|
)
|
||
|
|
|
||
|
|
type NamespaceInfo struct {
|
||
|
|
Namespace string
|
||
|
|
}
|
||
|
|
|
||
|
|
func listOfImages(spec *api.PodSpec) []string {
|
||
|
|
images := make([]string, 0, len(spec.Containers))
|
||
|
|
for _, container := range spec.Containers {
|
||
|
|
images = append(images, container.Image)
|
||
|
|
}
|
||
|
|
return images
|
||
|
|
}
|
||
|
|
|
||
|
|
func makeImageList(spec *api.PodSpec) string {
|
||
|
|
return strings.Join(listOfImages(spec), ",")
|
||
|
|
}
|
||
|
|
|
||
|
|
// OutputVersionMapper is a RESTMapper that will prefer mappings that
|
||
|
|
// correspond to a preferred output version (if feasible)
|
||
|
|
type OutputVersionMapper struct {
|
||
|
|
meta.RESTMapper
|
||
|
|
|
||
|
|
// output versions takes a list of preferred GroupVersions. Only the first
|
||
|
|
// hit for a given group will have effect. This allows different output versions
|
||
|
|
// depending upon the group of the kind being requested
|
||
|
|
OutputVersions []schema.GroupVersion
|
||
|
|
}
|
||
|
|
|
||
|
|
// RESTMapping implements meta.RESTMapper by prepending the output version to the preferred version list.
|
||
|
|
func (m OutputVersionMapper) RESTMapping(gk schema.GroupKind, versions ...string) (*meta.RESTMapping, error) {
|
||
|
|
for _, preferredVersion := range m.OutputVersions {
|
||
|
|
if gk.Group == preferredVersion.Group {
|
||
|
|
mapping, err := m.RESTMapper.RESTMapping(gk, preferredVersion.Version)
|
||
|
|
if err == nil {
|
||
|
|
return mapping, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
break
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return m.RESTMapper.RESTMapping(gk, versions...)
|
||
|
|
}
|
||
|
|
|
||
|
|
// ResourceShortcuts represents a structure that holds the information how to
|
||
|
|
// transition from resource's shortcut to its full name.
|
||
|
|
type ResourceShortcuts struct {
|
||
|
|
ShortForm schema.GroupResource
|
||
|
|
LongForm schema.GroupResource
|
||
|
|
}
|
||
|
|
|
||
|
|
// ResourcesShortcutStatic is the list of short names to their expanded names.
|
||
|
|
// Note that the list is ordered by group.
|
||
|
|
var ResourcesShortcutStatic = []ResourceShortcuts{
|
||
|
|
// If you add an entry here, please also take a look at pkg/kubectl/cmd/cmd.go
|
||
|
|
// and add an entry to valid_resources when appropriate.
|
||
|
|
{
|
||
|
|
ShortForm: schema.GroupResource{Resource: "cm"},
|
||
|
|
LongForm: schema.GroupResource{Resource: "configmaps"},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
ShortForm: schema.GroupResource{Resource: "cs"},
|
||
|
|
LongForm: schema.GroupResource{Resource: "componentstatuses"},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
ShortForm: schema.GroupResource{Resource: "ep"},
|
||
|
|
LongForm: schema.GroupResource{Resource: "endpoints"},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
ShortForm: schema.GroupResource{Resource: "ev"},
|
||
|
|
LongForm: schema.GroupResource{Resource: "events"},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
ShortForm: schema.GroupResource{Resource: "limits"},
|
||
|
|
LongForm: schema.GroupResource{Resource: "limitranges"},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
ShortForm: schema.GroupResource{Resource: "no"},
|
||
|
|
LongForm: schema.GroupResource{Resource: "nodes"},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
ShortForm: schema.GroupResource{Resource: "ns"},
|
||
|
|
LongForm: schema.GroupResource{Resource: "namespaces"},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
ShortForm: schema.GroupResource{Resource: "po"},
|
||
|
|
LongForm: schema.GroupResource{Resource: "pods"},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
ShortForm: schema.GroupResource{Resource: "pvc"},
|
||
|
|
LongForm: schema.GroupResource{Resource: "persistentvolumeclaims"},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
ShortForm: schema.GroupResource{Resource: "pv"},
|
||
|
|
LongForm: schema.GroupResource{Resource: "persistentvolumes"},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
ShortForm: schema.GroupResource{Resource: "quota"},
|
||
|
|
LongForm: schema.GroupResource{Resource: "resourcequotas"},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
ShortForm: schema.GroupResource{Resource: "rc"},
|
||
|
|
LongForm: schema.GroupResource{Resource: "replicationcontrollers"},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
ShortForm: schema.GroupResource{Resource: "rs"},
|
||
|
|
LongForm: schema.GroupResource{Resource: "replicasets"},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
ShortForm: schema.GroupResource{Resource: "sa"},
|
||
|
|
LongForm: schema.GroupResource{Resource: "serviceaccounts"},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
ShortForm: schema.GroupResource{Resource: "svc"},
|
||
|
|
LongForm: schema.GroupResource{Resource: "services"},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
ShortForm: schema.GroupResource{Group: "autoscaling", Resource: "hpa"},
|
||
|
|
LongForm: schema.GroupResource{Group: "autoscaling", Resource: "horizontalpodautoscalers"},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
ShortForm: schema.GroupResource{Group: "certificates.k8s.io", Resource: "csr"},
|
||
|
|
LongForm: schema.GroupResource{Group: "certificates.k8s.io", Resource: "certificatesigningrequests"},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
ShortForm: schema.GroupResource{Group: "policy", Resource: "pdb"},
|
||
|
|
LongForm: schema.GroupResource{Group: "policy", Resource: "poddisruptionbudgets"},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
ShortForm: schema.GroupResource{Group: "extensions", Resource: "deploy"},
|
||
|
|
LongForm: schema.GroupResource{Group: "extensions", Resource: "deployments"},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
ShortForm: schema.GroupResource{Group: "extensions", Resource: "ds"},
|
||
|
|
LongForm: schema.GroupResource{Group: "extensions", Resource: "daemonsets"},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
ShortForm: schema.GroupResource{Group: "extensions", Resource: "hpa"},
|
||
|
|
LongForm: schema.GroupResource{Group: "extensions", Resource: "horizontalpodautoscalers"},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
ShortForm: schema.GroupResource{Group: "extensions", Resource: "ing"},
|
||
|
|
LongForm: schema.GroupResource{Group: "extensions", Resource: "ingresses"},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
ShortForm: schema.GroupResource{Group: "extensions", Resource: "psp"},
|
||
|
|
LongForm: schema.GroupResource{Group: "extensions", Resource: "podSecurityPolicies"},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
// ResourceShortFormFor looks up for a short form of resource names.
|
||
|
|
// TODO: Change the signature of this function so that it can
|
||
|
|
// make use of ResourceShortcuts.
|
||
|
|
func ResourceShortFormFor(resource string) (string, bool) {
|
||
|
|
var alias string
|
||
|
|
exists := false
|
||
|
|
for _, item := range ResourcesShortcutStatic {
|
||
|
|
if item.LongForm.Resource == resource {
|
||
|
|
alias = item.ShortForm.Resource
|
||
|
|
exists = true
|
||
|
|
break
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return alias, exists
|
||
|
|
}
|
||
|
|
|
||
|
|
// ResourceAliases returns the resource shortcuts and plural forms for the given resources.
|
||
|
|
func ResourceAliases(rs []string) []string {
|
||
|
|
as := make([]string, 0, len(rs))
|
||
|
|
plurals := make(map[string]struct{}, len(rs))
|
||
|
|
for _, r := range rs {
|
||
|
|
var plural string
|
||
|
|
switch {
|
||
|
|
case r == "endpoints":
|
||
|
|
plural = r // exception. "endpoint" does not exist. Why?
|
||
|
|
case strings.HasSuffix(r, "y"):
|
||
|
|
plural = r[0:len(r)-1] + "ies"
|
||
|
|
case strings.HasSuffix(r, "s"):
|
||
|
|
plural = r + "es"
|
||
|
|
default:
|
||
|
|
plural = r + "s"
|
||
|
|
}
|
||
|
|
as = append(as, plural)
|
||
|
|
|
||
|
|
plurals[plural] = struct{}{}
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, item := range ResourcesShortcutStatic {
|
||
|
|
if _, found := plurals[item.LongForm.Resource]; found {
|
||
|
|
as = append(as, item.ShortForm.Resource)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return as
|
||
|
|
}
|
||
|
|
|
||
|
|
// parseFileSource parses the source given. Acceptable formats include:
|
||
|
|
//
|
||
|
|
// 1. source-path: the basename will become the key name
|
||
|
|
// 2. source-name=source-path: the source-name will become the key name and source-path is the path to the key file
|
||
|
|
//
|
||
|
|
// Key names cannot include '='.
|
||
|
|
func parseFileSource(source string) (keyName, filePath string, err error) {
|
||
|
|
numSeparators := strings.Count(source, "=")
|
||
|
|
switch {
|
||
|
|
case numSeparators == 0:
|
||
|
|
return path.Base(source), source, nil
|
||
|
|
case numSeparators == 1 && strings.HasPrefix(source, "="):
|
||
|
|
return "", "", fmt.Errorf("key name for file path %v missing.", strings.TrimPrefix(source, "="))
|
||
|
|
case numSeparators == 1 && strings.HasSuffix(source, "="):
|
||
|
|
return "", "", fmt.Errorf("file path for key name %v missing.", strings.TrimSuffix(source, "="))
|
||
|
|
case numSeparators > 1:
|
||
|
|
return "", "", errors.New("Key names or file paths cannot contain '='.")
|
||
|
|
default:
|
||
|
|
components := strings.Split(source, "=")
|
||
|
|
return components[0], components[1], nil
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// parseLiteralSource parses the source key=val pair
|
||
|
|
func parseLiteralSource(source string) (keyName, value string, err error) {
|
||
|
|
// leading equal is invalid
|
||
|
|
if strings.Index(source, "=") == 0 {
|
||
|
|
return "", "", fmt.Errorf("invalid literal source %v, expected key=value", source)
|
||
|
|
}
|
||
|
|
// split after the first equal (so values can have the = character)
|
||
|
|
items := strings.SplitN(source, "=", 2)
|
||
|
|
if len(items) != 2 {
|
||
|
|
return "", "", fmt.Errorf("invalid literal source %v, expected key=value", source)
|
||
|
|
}
|
||
|
|
|
||
|
|
return items[0], items[1], nil
|
||
|
|
}
|