init

Signed-off-by: ohotnikov.ivan <ohotnikov.ivan@e-queo.net>

Author: ohotnikov.ivan

internal/pkg/generator/generate.go

@@ -0,0 +1,102 @@
+package generator
+
+import (
+	"fmt"
+	"path/filepath"
+	"time"
+
+	"gopkg.in/yaml.v3"
+
+	"github.com/siderolabs/talos/cmd/talosctl/cmd/mgmt/gen"
+	"github.com/siderolabs/talos/pkg/machinery/config"
+	"github.com/siderolabs/talos/pkg/machinery/config/generate"
+	"github.com/siderolabs/talos/pkg/machinery/config/generate/secrets"
+)
+
+type Options struct {
+	RootDir      string
+	TalosVersion string
+	Preset       string
+	Force        bool
+
+	APIServerURL string
+}
+
+// Run performs full generation exactly like "talm init"
+func Run(opts Options) error {
+	var (
+		contract      *config.VersionContract
+		secretsBundle *secrets.Bundle
+		err           error
+	)
+
+	// Version contract
+	if opts.TalosVersion != "" {
+		contract, err = config.ParseContractFromVersion(opts.TalosVersion)
+		if err != nil {
+			return fmt.Errorf("invalid talos-version: %w", err)
+		}
+	}
+
+	// Secrets bundle
+	secretsBundle, err = secrets.NewBundle(secrets.NewFixedClock(time.Now()), contract)
+	if err != nil {
+		return fmt.Errorf("failed to create secrets bundle: %w", err)
+	}
+
+	// Write secrets.yaml
+	if err := writeSecretsBundle(opts, secretsBundle); err != nil {
+		return err
+	}
+
+	// Cluster name = name of directory
+	absolutePath, err := filepath.Abs(opts.RootDir)
+	if err != nil {
+		return err
+	}
+	clusterName := filepath.Base(absolutePath)
+
+	// Config generation
+	var genOptions []generate.Option
+	genOptions = append(genOptions, generate.WithSecretsBundle(secretsBundle))
+
+	if contract != nil {
+		genOptions = append(genOptions, generate.WithVersionContract(contract))
+	}
+
+	if opts.APIServerURL == "" {
+		opts.APIServerURL = "https://192.168.0.1:6443"
+	}
+
+	configBundle, err := gen.GenerateConfigBundle(
+		genOptions,
+		clusterName,
+		opts.APIServerURL,
+		"",
+		[]string{},
+		[]string{},
+		[]string{},
+	)
+	if err != nil {
+		return err
+	}
+
+	configBundle.TalosConfig().Contexts[clusterName].Endpoints = []string{"127.0.0.1"}
+
+	// Write talosconfig
+	content, err := yaml.Marshal(configBundle.TalosConfig())
+	if err != nil {
+		return err
+	}
+
+	if err := writeFile(opts, filepath.Join(opts.RootDir, "talosconfig"), content); err != nil {
+		return err
+	}
+
+	// Write preset files
+	if err := writePresets(opts, clusterName); err != nil {
+		return err
+	}
+
+	return nil
+}

internal/pkg/generator/write.go

@@ -0,0 +1,66 @@
+package generator
+
+import (
+	"fmt"
+	"os"
+	"path/filepath"
+	"strings"
+
+	"github.com/cozystack/talm/pkg/generated"
+	"gopkg.in/yaml.v3"
+
+	"github.com/siderolabs/talos/pkg/machinery/config/generate/secrets"
+)
+
+func writeSecretsBundle(opts Options, bundle *secrets.Bundle) error {
+	bytes, err := yaml.Marshal(bundle)
+	if err != nil {
+		return err
+	}
+
+	dest := filepath.Join(opts.RootDir, "secrets.yaml")
+	return writeFile(opts, dest, bytes)
+}
+
+func writeFile(opts Options, dest string, content []byte) error {
+	if !opts.Force {
+		if _, err := os.Stat(dest); err == nil {
+			return fmt.Errorf("%s already exists (use Force=true)", dest)
+		}
+	}
+
+	if err := os.MkdirAll(filepath.Dir(dest), os.ModePerm); err != nil {
+		return fmt.Errorf("failed to create dir: %w", err)
+	}
+
+	if err := os.WriteFile(dest, content, 0o644); err != nil {
+		return fmt.Errorf("failed to write file: %w", err)
+	}
+
+	fmt.Fprintf(os.Stderr, "Created %s\n", dest)
+	return nil
+}
+
+func writePresets(opts Options, clusterName string) error {
+	for path, content := range generated.PresetFiles {
+
+		parts := strings.SplitN(path, "/", 2)
+		chartName := parts[0]
+
+		if chartName != opts.Preset && chartName != "talm" {
+			continue
+		}
+
+		out := filepath.Join(opts.RootDir, parts[1])
+
+		// Template Chart.yaml
+		if strings.HasSuffix(path, "Chart.yaml") {
+			content = fmt.Sprintf(content, clusterName, "0.1.0")
+		}
+
+		if err := writeFile(opts, out, []byte(content)); err != nil {
+			return err
+		}
+	}
+	return nil
+}