Bring elemental-cli code into the agent (#13)

pkg/cloudinit/console.go

@@ -0,0 +1,83 @@
+/*
+Copyright © 2022 SUSE 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 cloudinit
+
+import (
+	"fmt"
+	"os/exec"
+
+	"github.com/hashicorp/go-multierror"
+	v1 "github.com/kairos-io/kairos/v2/pkg/types/v1"
+)
+
+// cloudInitConsole represents a yip's Console implementations using
+// the elemental v1.Runner interface.
+type cloudInitConsole struct {
+	runner v1.Runner
+	logger v1.Logger
+}
+
+// newCloudInitConsole returns an instance of the cloudInitConsole based on the
+// given v1.Runner and v1.Logger.
+func newCloudInitConsole(l v1.Logger, r v1.Runner) *cloudInitConsole {
+	return &cloudInitConsole{logger: l, runner: r}
+}
+
+// getRunner returns the internal runner used within this Console
+func (c cloudInitConsole) getRunner() v1.Runner {
+	return c.runner
+}
+
+// Run runs a command using the v1.Runner internal instance
+func (c cloudInitConsole) Run(command string, opts ...func(cmd *exec.Cmd)) (string, error) {
+	c.logger.Debugf("running command `%s`", command)
+	cmd := c.runner.InitCmd("sh", "-c", command)
+	for _, o := range opts {
+		o(cmd)
+	}
+	out, err := c.runner.RunCmd(cmd)
+	if err != nil {
+		return string(out), fmt.Errorf("failed to run %s: %v", command, err)
+	}
+
+	return string(out), err
+}
+
+// Start runs a non blocking command using the v1.Runner internal instance
+func (c cloudInitConsole) Start(cmd *exec.Cmd, opts ...func(cmd *exec.Cmd)) error {
+	c.logger.Debugf("running command `%s`", cmd)
+	for _, o := range opts {
+		o(cmd)
+	}
+	return cmd.Run()
+}
+
+// RunTemplate runs a sequence of non-blocking templated commands using the v1.Runner internal instance
+func (c cloudInitConsole) RunTemplate(st []string, template string) error {
+	var errs error
+
+	for _, svc := range st {
+		out, err := c.Run(fmt.Sprintf(template, svc))
+		if err != nil {
+			c.logger.Error(out)
+			c.logger.Error(err.Error())
+			errs = multierror.Append(errs, err)
+			continue
+		}
+	}
+	return errs
+}