Introduce secret factory.

This commit is contained in:
Jeffrey Regan
2018-07-20 10:01:58 -07:00
parent 82f2cf9124
commit 7f22f187f8
6 changed files with 88 additions and 65 deletions

View File

@@ -17,73 +17,31 @@ limitations under the License.
package resmap
import (
"context"
"os/exec"
"path/filepath"
"time"
"os"
"github.com/kubernetes-sigs/kustomize/pkg/configmapandsecret"
"github.com/kubernetes-sigs/kustomize/pkg/fs"
"github.com/kubernetes-sigs/kustomize/pkg/resource"
"github.com/kubernetes-sigs/kustomize/pkg/types"
"github.com/pkg/errors"
corev1 "k8s.io/api/core/v1"
)
func newResourceFromSecretGenerator(p string, sArgs types.SecretArgs) (*resource.Resource, error) {
s, err := makeSecret(p, sArgs)
if err != nil {
return nil, errors.Wrap(err, "makeSecret")
}
if sArgs.Behavior == "" {
sArgs.Behavior = "create"
}
return resource.NewResourceWithBehavior(
s, resource.NewGenerationBehavior(sArgs.Behavior))
}
func makeSecret(p string, sArgs types.SecretArgs) (*corev1.Secret, error) {
s := &corev1.Secret{}
s.APIVersion = "v1"
s.Kind = "Secret"
s.Name = sArgs.Name
s.Type = corev1.SecretType(sArgs.Type)
if s.Type == "" {
s.Type = corev1.SecretTypeOpaque
}
s.Data = map[string][]byte{}
for k, v := range sArgs.Commands {
out, err := createSecretKey(p, v)
if err != nil {
return nil, errors.Wrap(err, "createSecretKey")
}
s.Data[k] = out
}
return s, nil
}
// Run a command, return its output as the secret.
func createSecretKey(wd string, command string) ([]byte, error) {
fi, err := os.Stat(wd)
if err != nil || !fi.IsDir() {
wd = filepath.Dir(wd)
}
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()
cmd := exec.CommandContext(ctx, "sh", "-c", command)
cmd.Dir = wd
return cmd.Output()
}
// NewResMapFromSecretArgs takes a SecretArgs slice and executes its command in directory p
// then writes the output to a Resource slice and return it.
func NewResMapFromSecretArgs(p string, secretList []types.SecretArgs) (ResMap, error) {
// NewResMapFromSecretArgs takes a SecretArgs slice and executes its command in directory
// wd then writes the output to a Resource slice and return it.
func NewResMapFromSecretArgs(
wd string, fSys fs.FileSystem,
secretList []types.SecretArgs) (ResMap, error) {
var allResources []*resource.Resource
for _, secret := range secretList {
res, err := newResourceFromSecretGenerator(p, secret)
for _, args := range secretList {
s, err := configmapandsecret.NewSecretFactory(args, fSys).MakeSecret(wd)
if err != nil {
return nil, errors.Wrap(err, "newResourceFromSecretGenerator")
return nil, errors.Wrap(err, "makeSecret")
}
if args.Behavior == "" {
args.Behavior = "create"
}
res, err := resource.NewResourceWithBehavior(
s, resource.NewGenerationBehavior(args.Behavior))
if err != nil {
return nil, errors.Wrap(err, "NewResourceWithBehavior")
}
allResources = append(allResources, res)
}