mirror of
https://github.com/siderolabs/talos.git
synced 2025-12-27 12:21:14 +01:00
92eeaa4826
Update COSI, and stop using a fork of `gopkg.in/yaml.v3`, now we use new supported for of this library. Drop `MarshalYAMLBytes` for the machine config, as we actually marshal config as a string, and we don't need this at all. Make `talosctl` stop doing hacks on machine config for newer Talos, keep hacks for backwards compatibility. Signed-off-by: Andrey Smirnov <andrey.smirnov@siderolabs.com>
85 lines
2.1 KiB
Go
85 lines
2.1 KiB
Go
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
// Package executor implements overlay.Installer
|
|
package executor
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"os/exec"
|
|
|
|
"go.yaml.in/yaml/v4"
|
|
|
|
"github.com/siderolabs/talos/pkg/machinery/overlay"
|
|
)
|
|
|
|
var _ overlay.Installer[overlay.ExtraOptions] = (*Options)(nil)
|
|
|
|
// Options executor options.
|
|
type Options struct {
|
|
commandPath string
|
|
}
|
|
|
|
// New returns a new overlay installer executor.
|
|
func New(commandPath string) *Options {
|
|
return &Options{
|
|
commandPath: commandPath,
|
|
}
|
|
}
|
|
|
|
// GetOptions returns the options for the overlay installer.
|
|
func (o *Options) GetOptions(ctx context.Context, extra overlay.ExtraOptions) (overlay.Options, error) {
|
|
// parse extra as yaml
|
|
extraYAML, err := yaml.Marshal(extra)
|
|
if err != nil {
|
|
return overlay.Options{}, fmt.Errorf("failed to marshal extra: %w", err)
|
|
}
|
|
|
|
out, err := o.execute(ctx, bytes.NewReader(extraYAML), "get-options")
|
|
if err != nil {
|
|
return overlay.Options{}, err
|
|
}
|
|
|
|
var options overlay.Options
|
|
|
|
if err := yaml.Unmarshal(out, &options); err != nil {
|
|
return overlay.Options{}, fmt.Errorf("failed to unmarshal overlay options: %w", err)
|
|
}
|
|
|
|
return options, nil
|
|
}
|
|
|
|
// Install installs the overlay.
|
|
func (o *Options) Install(ctx context.Context, options overlay.InstallOptions[overlay.ExtraOptions]) error {
|
|
optionsBytes, err := yaml.Marshal(&options)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to marshal options: %w", err)
|
|
}
|
|
|
|
if _, err := o.execute(ctx, bytes.NewReader(optionsBytes), "install"); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (o *Options) execute(ctx context.Context, stdin io.Reader, args ...string) ([]byte, error) {
|
|
cmd := exec.CommandContext(ctx, o.commandPath, args...)
|
|
cmd.Stdin = stdin
|
|
|
|
var stdOut, stdErr bytes.Buffer
|
|
|
|
cmd.Stdout = &stdOut
|
|
cmd.Stderr = &stdErr
|
|
|
|
if err := cmd.Run(); err != nil {
|
|
return nil, fmt.Errorf("failed to run overlay installer: %w, stdErr: %s", err, stdErr.Bytes())
|
|
}
|
|
|
|
return stdOut.Bytes(), nil
|
|
}
|