mirror of
https://github.com/siderolabs/talos.git
synced 2025-10-27 14:31:11 +01:00
0359c8537c
Drop BurntSushi one, and use /v2 of pelletier package. There is indirect use of v1 which should hopefully go away once we move away from sonobouy. Signed-off-by: Andrey Smirnov <andrey.smirnov@siderolabs.com>
61 lines
1.4 KiB
Go
61 lines
1.4 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 toml
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/pelletier/go-toml/v2"
|
|
|
|
"github.com/siderolabs/talos/pkg/machinery/config/merge"
|
|
)
|
|
|
|
func tomlDecodeFile(path string, dest any) error {
|
|
f, err := os.Open(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
defer f.Close() //nolint:errcheck
|
|
|
|
return toml.NewDecoder(f).Decode(dest)
|
|
}
|
|
|
|
// Merge several TOML documents in files into one.
|
|
//
|
|
// Merge process relies on generic map[string]any merge which might not always be correct.
|
|
func Merge(parts []string) ([]byte, error) {
|
|
merged := map[string]any{}
|
|
|
|
var header []byte
|
|
|
|
for _, part := range parts {
|
|
partial := map[string]any{}
|
|
|
|
if err := tomlDecodeFile(part, &partial); err != nil {
|
|
return nil, fmt.Errorf("error decoding %q: %w", part, err)
|
|
}
|
|
|
|
if err := merge.Merge(merged, partial); err != nil {
|
|
return nil, fmt.Errorf("error merging %q: %w", part, err)
|
|
}
|
|
|
|
header = append(header, []byte(fmt.Sprintf("## %s\n", part))...)
|
|
}
|
|
|
|
var out bytes.Buffer
|
|
|
|
_, _ = out.Write(header)
|
|
_ = out.WriteByte('\n')
|
|
|
|
if err := toml.NewEncoder(&out).SetIndentTables(true).Encode(merged); err != nil {
|
|
return nil, fmt.Errorf("error encoding merged config: %w", err)
|
|
}
|
|
|
|
return out.Bytes(), nil
|
|
}
|