0c4d8b80bf
It would be helpful to have a possibility to configure Timeout field, as well as others allowed. Change-Id: I1e0e895b7ca6a17e86a6ad36f405e2538775cbbb Closes: #553 Signed-off-by: Ruslan Aliev <raliev@mirantis.com>
118 lines
3.5 KiB
Go
118 lines
3.5 KiB
Go
/*
|
|
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
|
|
|
|
https://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 utils
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"os"
|
|
|
|
"k8s.io/apimachinery/pkg/api/meta"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
|
|
"k8s.io/cli-runtime/pkg/genericclioptions"
|
|
cmdutil "k8s.io/kubectl/pkg/cmd/util"
|
|
"sigs.k8s.io/cli-utils/pkg/manifestreader"
|
|
"sigs.k8s.io/cli-utils/pkg/util/factory"
|
|
|
|
"opendev.org/airship/airshipctl/pkg/document"
|
|
)
|
|
|
|
// ClientOption is a function that allows to modify ConfigFlags object which is used to create Client
|
|
type ClientOption func(*genericclioptions.ConfigFlags)
|
|
|
|
// SetTimeout sets Timeout option in ConfigFlags object
|
|
func SetTimeout(timeout string) ClientOption {
|
|
return func(co *genericclioptions.ConfigFlags) {
|
|
*co.Timeout = timeout
|
|
}
|
|
}
|
|
|
|
// FactoryFromKubeConfig returns a factory with the
|
|
// default Kubernetes resources for the given kube config path and context
|
|
func FactoryFromKubeConfig(path, context string, opts ...ClientOption) cmdutil.Factory {
|
|
kf := genericclioptions.NewConfigFlags(false)
|
|
kf.KubeConfig = &path
|
|
kf.Context = &context
|
|
for _, o := range opts {
|
|
o(kf)
|
|
}
|
|
|
|
return cmdutil.NewFactory(cmdutil.NewMatchVersionFlags(&factory.CachingRESTClientGetter{
|
|
Delegate: kf,
|
|
}))
|
|
}
|
|
|
|
// Streams returns default IO streams object, like stdout, stdin, stderr
|
|
func Streams() genericclioptions.IOStreams {
|
|
return genericclioptions.IOStreams{
|
|
In: os.Stdin,
|
|
Out: os.Stdout,
|
|
ErrOut: os.Stderr,
|
|
}
|
|
}
|
|
|
|
// ManifestReaderFactory factory function for manifestreader.ManifestReader
|
|
type ManifestReaderFactory func(
|
|
validate bool,
|
|
bundle document.Bundle,
|
|
mapper meta.RESTMapper) manifestreader.ManifestReader
|
|
|
|
// DefaultManifestReaderFactory default factory function for manifestreader.ManifestReader
|
|
var DefaultManifestReaderFactory ManifestReaderFactory = func(
|
|
validate bool,
|
|
bundle document.Bundle,
|
|
mapper meta.RESTMapper) manifestreader.ManifestReader {
|
|
return NewManifestBundleReader(validate, bundle, mapper)
|
|
}
|
|
|
|
// NewManifestBundleReader returns implementation of manifestreader interface
|
|
func NewManifestBundleReader(
|
|
validate bool,
|
|
bundle document.Bundle,
|
|
mapper meta.RESTMapper) *ManifestBundleReader {
|
|
opts := manifestreader.ReaderOptions{
|
|
Mapper: mapper,
|
|
Validate: validate,
|
|
Namespace: metav1.NamespaceDefault,
|
|
}
|
|
buffer := bytes.NewBuffer([]byte{})
|
|
return &ManifestBundleReader{
|
|
Bundle: bundle,
|
|
writer: buffer,
|
|
StreamReader: &manifestreader.StreamManifestReader{
|
|
ReaderName: "airship",
|
|
Reader: buffer,
|
|
ReaderOptions: opts,
|
|
},
|
|
}
|
|
}
|
|
|
|
// ManifestBundleReader implements manifestreader interface that to transform bundle to slice
|
|
// of *resource.Info objects using Read() method.
|
|
type ManifestBundleReader struct {
|
|
Bundle document.Bundle
|
|
StreamReader *manifestreader.StreamManifestReader
|
|
writer io.Writer
|
|
}
|
|
|
|
func (mbr *ManifestBundleReader) Read() ([]*unstructured.Unstructured, error) {
|
|
err := mbr.Bundle.Write(mbr.writer)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return mbr.StreamReader.Read()
|
|
}
|