Skip to content

Commit

Permalink
Fixes issue 1392, kpt can't handle kind: List. (#3654)
Browse files Browse the repository at this point in the history
Co-authored-by: Pieter Emmelot <p.emmelot@gmail.com>
  • Loading branch information
mmlt and Pieter Emmelot committed Nov 20, 2022
1 parent c67b11f commit 23de8ff
Show file tree
Hide file tree
Showing 3 changed files with 126 additions and 0 deletions.
6 changes: 6 additions & 0 deletions commands/live/apply/cmdapply.go
Original file line number Diff line number Diff line change
Expand Up @@ -175,6 +175,12 @@ func (r *Runner) runE(c *cobra.Command, args []string) error {
return err
}

// objs may contain kind List
objs, err = live.Flatten(objs)
if err != nil {
return err
}

invInfo, err := live.ToInventoryInfo(inv)
if err != nil {
return err
Expand Down
42 changes: 42 additions & 0 deletions pkg/live/flatten.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,42 @@
// Copyright 2022 Google LLC
//
// 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 live

import (
"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
"k8s.io/apimachinery/pkg/runtime"
)

// Flatten returns a list containing 'in' objects with objects of kind List
// replaced by their members.
func Flatten(in []*unstructured.Unstructured) ([]*unstructured.Unstructured, error) {
var out []*unstructured.Unstructured

for _, o := range in {
if o.IsList() {
err := o.EachListItem(func(item runtime.Object) error {
item2 := item.(*unstructured.Unstructured)
out = append(out, item2)
return nil
})
if err != nil {
return nil, err
}
} else {
out = append(out, o)
}
}
return out, nil
}
78 changes: 78 additions & 0 deletions pkg/live/flatten_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,78 @@
// Copyright 2022 Google LLC
//
// 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 live

import (
"testing"

"github.com/stretchr/testify/assert"
"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
)

func Test_Flatten(t *testing.T) {
tests := []struct {
name string
in []*unstructured.Unstructured
want []*unstructured.Unstructured
}{
{
name: "simple list",
in: simpleList,
want: simpleList,
},
{
name: "list with list",
in: []*unstructured.Unstructured{
{
Object: map[string]interface{}{
"kind": "List",
"items": anySlice(simpleList),
},
},
},
want: simpleList,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
got, err := Flatten(tt.in)
if assert.NoError(t, err) {
assert.Equal(t, tt.want, got)
}
})
}
}

func anySlice(in []*unstructured.Unstructured) (out []interface{}) {
for _, o := range in {
out = append(out, o.Object)
}
return
}

var simpleList = []*unstructured.Unstructured{
{
Object: map[string]interface{}{
"kind": "ConfigMap",
"version": "v1",
},
},
{
Object: map[string]interface{}{
"kind": "Pod",
"version": "v1",
},
},
}

0 comments on commit 23de8ff

Please sign in to comment.