104 lines
2.9 KiB
Go
104 lines
2.9 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 kubectl
|
|
|
|
import (
|
|
"bytes"
|
|
"os"
|
|
|
|
"k8s.io/cli-runtime/pkg/genericclioptions"
|
|
cmdutil "k8s.io/kubectl/pkg/cmd/util"
|
|
|
|
"opendev.org/airship/airshipctl/pkg/config"
|
|
"opendev.org/airship/airshipctl/pkg/document"
|
|
"opendev.org/airship/airshipctl/pkg/fs"
|
|
"opendev.org/airship/airshipctl/pkg/log"
|
|
utilyaml "opendev.org/airship/airshipctl/pkg/util/yaml"
|
|
)
|
|
|
|
// Kubectl container holds Factory, Streams and FileSystem to
|
|
// interact with upstream kubectl objects and serves as abstraction to kubectl project
|
|
type Kubectl struct {
|
|
cmdutil.Factory
|
|
genericclioptions.IOStreams
|
|
fs.FileSystem
|
|
// Directory to buffer documents before passing them to kubectl commands
|
|
// default is empty, this means that /tmp dir will be used
|
|
bufferDir string
|
|
}
|
|
|
|
// NewKubectl builds an instance
|
|
// of Kubectl struct from Path to kubeconfig file
|
|
func NewKubectl(f cmdutil.Factory) *Kubectl {
|
|
return &Kubectl{
|
|
Factory: f,
|
|
IOStreams: genericclioptions.IOStreams{
|
|
In: os.Stdin,
|
|
Out: os.Stdout,
|
|
ErrOut: os.Stderr,
|
|
},
|
|
FileSystem: fs.NewDocumentFs(),
|
|
}
|
|
}
|
|
|
|
// WithBufferDir sets buffer dir
|
|
func (kubectl *Kubectl) WithBufferDir(bd string) *Kubectl {
|
|
kubectl.bufferDir = bd
|
|
return kubectl
|
|
}
|
|
|
|
// ApplyDocs transform Document into YAML
|
|
// TODO: (anoskov) Add unit test for ApplyDocs, as of now it is calling ApplyYaml which meets the coverage test.
|
|
func (kubectl *Kubectl) ApplyDocs(docs []document.Document, ao *ApplyOptions) error {
|
|
buf := bytes.NewBuffer([]byte{})
|
|
for _, doc := range docs {
|
|
err := utilyaml.WriteOut(buf, doc)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return kubectl.ApplyYaml(buf.Bytes(), ao)
|
|
}
|
|
|
|
// ApplyYaml is abstraction to kubectl apply command
|
|
func (kubectl *Kubectl) ApplyYaml(yaml []byte, ao *ApplyOptions) error {
|
|
tf, err := kubectl.TempFile(kubectl.bufferDir, config.KubectlTempFilePrefix)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
defer func(f fs.File) {
|
|
fName := f.Name()
|
|
dErr := kubectl.RemoveAll(fName)
|
|
if dErr != nil {
|
|
log.Fatalf("Failed to cleanup temporary file %s during kubectl apply", fName)
|
|
}
|
|
}(tf)
|
|
|
|
defer tf.Close()
|
|
_, err = tf.Write(yaml)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
ao.SetSourceFiles([]string{tf.Name()})
|
|
return ao.Run()
|
|
}
|
|
|
|
// ApplyOptions is a wrapper over kubectl ApplyOptions, used to build
|
|
// new options from the factory and iostreams defined in Kubectl container
|
|
func (kubectl *Kubectl) ApplyOptions() (*ApplyOptions, error) {
|
|
return NewApplyOptions(kubectl.Factory, kubectl.IOStreams)
|
|
}
|