package k8sapi

import (
	"context"
	"fmt"

	"github.com/coreos/go-systemd/v22/dbus"
)

func restartSystemdUnit(ctx context.Context, unit string) error {
	conn, err := dbus.NewSystemdConnectionContext(ctx)
	if err != nil {
		return fmt.Errorf("establishing systemd connection failed: %w", err)
	}

	restartChan := make(chan string)
	if _, err := conn.RestartUnitContext(ctx, unit, "replace", restartChan); err != nil {
		return fmt.Errorf("restarting systemd unit %q failed: %w", unit, err)
	}

	// Wait for the restart to finish and actually check if it was
	// successful or not.
	result := <-restartChan

	switch result {
	case "done":
		return nil

	default:
		return fmt.Errorf("restarting systemd unit %q failed: expected %v but received %v", unit, "done", result)
	}
}

func startSystemdUnit(ctx context.Context, unit string) error {
	conn, err := dbus.NewSystemdConnectionContext(ctx)
	if err != nil {
		return fmt.Errorf("establishing systemd connection failed: %w", err)
	}

	startChan := make(chan string)
	if _, err := conn.StartUnitContext(ctx, unit, "replace", startChan); err != nil {
		return fmt.Errorf("starting systemd unit %q failed: %w", unit, err)
	}

	// Wait for the enable to finish and actually check if it was
	// successful or not.
	result := <-startChan

	switch result {
	case "done":
		return nil

	default:
		return fmt.Errorf("starting systemd unit %q failed: expected %v but received %v", unit, "done", result)
	}
}

func enableSystemdUnit(ctx context.Context, unitPath string) error {
	conn, err := dbus.NewSystemdConnectionContext(ctx)
	if err != nil {
		return fmt.Errorf("establishing systemd connection failed: %w", err)
	}

	if _, _, err := conn.EnableUnitFilesContext(ctx, []string{unitPath}, true, true); err != nil {
		return fmt.Errorf("enabling systemd unit %q failed: %w", unitPath, err)
	}
	return nil
}