move parse helpers to util

This commit is contained in:
Seth Pollack
2019-03-17 09:53:38 -04:00
parent 1bd7afe6e7
commit 29cbec37b8
3 changed files with 68 additions and 65 deletions

View File

@@ -18,7 +18,6 @@ package configmapandsecret
import ( import (
"fmt" "fmt"
"path"
"strings" "strings"
"github.com/pkg/errors" "github.com/pkg/errors"
@@ -82,7 +81,7 @@ func errIfInvalidKey(keyName string) error {
func keyValuesFromLiteralSources(sources []string) ([]kv.Pair, error) { func keyValuesFromLiteralSources(sources []string) ([]kv.Pair, error) {
var kvs []kv.Pair var kvs []kv.Pair
for _, s := range sources { for _, s := range sources {
k, v, err := parseLiteralSource(s) k, v, err := kv.ParseLiteralSource(s)
if err != nil { if err != nil {
return nil, err return nil, err
} }
@@ -110,7 +109,7 @@ func (bf baseFactory) keyValuesFromPlugins(sources []types.KVSource) ([]kv.Pair,
func (bf baseFactory) keyValuesFromFileSources(sources []string) ([]kv.Pair, error) { func (bf baseFactory) keyValuesFromFileSources(sources []string) ([]kv.Pair, error) {
var kvs []kv.Pair var kvs []kv.Pair
for _, s := range sources { for _, s := range sources {
k, fPath, err := parseFileSource(s) k, fPath, err := kv.ParseFileSource(s)
if err != nil { if err != nil {
return nil, err return nil, err
} }
@@ -133,44 +132,3 @@ func (bf baseFactory) keyValuesFromEnvFile(path string) ([]kv.Pair, error) {
} }
return kv.KeyValuesFromLines(content) return kv.KeyValuesFromLines(content)
} }
// 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 into its component pieces.
// This functionality is distinguished from strings.SplitN(source, "=", 2) since
// it returns an error in the case of empty keys, values, or a missing equals sign.
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], strings.Trim(items[1], "\"'"), nil
}

View File

@@ -17,23 +17,19 @@ limitations under the License.
package builtin package builtin
import ( import (
"fmt"
"strings"
"sigs.k8s.io/kustomize/k8sdeps/kv" "sigs.k8s.io/kustomize/k8sdeps/kv"
) )
// Literals takes a list of literals. // Literals takes a list of literals.
// Each literal source should be a key and literal value, // Each literal source should be a key and literal value,
// e.g. `somekey=somevalue` // e.g. `somekey=somevalue`
// It will be similar to kubectl create configmap|secret --from-literal
type Literals struct{} type Literals struct{}
// Get implements the interface for kv plugins. // Get implements the interface for kv plugins.
func (p Literals) Get(root string, args []string) ([]kv.Pair, error) { func (p Literals) Get(root string, args []string) ([]kv.Pair, error) {
var kvs []kv.Pair var kvs []kv.Pair
for _, s := range args { for _, s := range args {
k, v, err := parseLiteralSource(s) k, v, err := kv.ParseLiteralSource(s)
if err != nil { if err != nil {
return nil, err return nil, err
} }
@@ -41,19 +37,3 @@ func (p Literals) Get(root string, args []string) ([]kv.Pair, error) {
} }
return kvs, nil return kvs, nil
} }
// parseLiteralSource parses the source key=val pair into its component pieces.
// This functionality is distinguished from strings.SplitN(source, "=", 2) since
// it returns an error in the case of empty keys, values, or a missing equals sign.
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], strings.Trim(items[1], "\"'"), nil
}

65
k8sdeps/kv/util.go Normal file
View File

@@ -0,0 +1,65 @@
/*
Copyright 2019 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.
*/
package kv
import (
"errors"
"fmt"
"path"
"strings"
)
// 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 into its component pieces.
// This functionality is distinguished from strings.SplitN(source, "=", 2) since
// it returns an error in the case of empty keys, values, or a missing equals sign.
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], strings.Trim(items[1], "\"'"), nil
}