diff --git a/pkg/osbuild/container_deploy_stage.go b/pkg/osbuild/container_deploy_stage.go index 9a0a926dc1..38b3eacf26 100644 --- a/pkg/osbuild/container_deploy_stage.go +++ b/pkg/osbuild/container_deploy_stage.go @@ -12,7 +12,7 @@ func (inputs ContainerDeployInputs) validate() error { if inputs.Images.References == nil { return fmt.Errorf("stage requires exactly 1 input container (got nil References)") } - if ncontainers := inputs.Images.References.Len(); ncontainers != 1 { + if ncontainers := len(inputs.Images.References); ncontainers != 1 { return fmt.Errorf("stage requires exactly 1 input container (got %d)", ncontainers) } return nil diff --git a/pkg/osbuild/containers_input.go b/pkg/osbuild/containers_input.go index 153e72bd96..a0dc9306b3 100644 --- a/pkg/osbuild/containers_input.go +++ b/pkg/osbuild/containers_input.go @@ -4,32 +4,17 @@ import ( "github.com/osbuild/images/pkg/container" ) -type ContainersInputReferences interface { - isContainersInputReferences() - Len() int -} - type ContainersInputSourceRef struct { Name string `json:"name"` } -type ContainersInputSourceMap map[string]ContainersInputSourceRef - -func (ContainersInputSourceMap) isContainersInputReferences() {} - -func (cism ContainersInputSourceMap) Len() int { - return len(cism) -} - type ContainersInput struct { inputCommon - References ContainersInputReferences `json:"references"` + References map[string]ContainersInputSourceRef `json:"references"` } -const InputTypeContainers string = "org.osbuild.containers" - func NewContainersInputForSources(containers []container.Spec) ContainersInput { - refs := make(ContainersInputSourceMap, len(containers)) + refs := make(map[string]ContainersInputSourceRef, len(containers)) for _, c := range containers { ref := ContainersInputSourceRef{ Name: c.LocalName, @@ -40,7 +25,7 @@ func NewContainersInputForSources(containers []container.Spec) ContainersInput { return ContainersInput{ References: refs, inputCommon: inputCommon{ - Type: InputTypeContainers, + Type: "org.osbuild.containers", Origin: InputOriginSource, }, } diff --git a/pkg/osbuild/ostree_deploy_container_stage.go b/pkg/osbuild/ostree_deploy_container_stage.go index 43e0b76e95..c814b27a12 100644 --- a/pkg/osbuild/ostree_deploy_container_stage.go +++ b/pkg/osbuild/ostree_deploy_container_stage.go @@ -51,7 +51,7 @@ func (inputs OSTreeDeployContainerInputs) validate() error { if inputs.Images.References == nil { return fmt.Errorf("stage requires exactly 1 input container (got nil References)") } - if ncontainers := inputs.Images.References.Len(); ncontainers != 1 { + if ncontainers := len(inputs.Images.References); ncontainers != 1 { return fmt.Errorf("stage requires exactly 1 input container (got %d)", ncontainers) } return nil