narratio/internal/app/plan.go

package app

import (
	"context"
	"flag"
	"fmt"
	"io"
	"log/slog"
	"os"

	"gitea.maximumdirect.net/eric/narratio/internal/artifacts"
	"gitea.maximumdirect.net/eric/narratio/internal/config"
	"gitea.maximumdirect.net/eric/narratio/internal/logging"
	"gitea.maximumdirect.net/eric/narratio/internal/manifest"
)

// Plan validates configuration, prepares the local workdir, and prints stage order.
func Plan(ctx context.Context, args []string, out io.Writer) error {
	fs := flag.NewFlagSet("plan", flag.ContinueOnError)
	fs.SetOutput(io.Discard)

	var flags commonConfigFlags
	var force bool
	addCommonConfigFlags(fs, &flags)
	fs.BoolVar(&force, "force", false, "show all stages as scheduled to rerun")

	if err := parseSessionAwareFlags("plan", fs, args, &flags.sessionID); err != nil {
		return err
	}
	if flags.sessionID == "" {
		return fmt.Errorf("plan: session_id is required")
	}
	loaded, err := loadCommandConfig(ctx, flags.pipelinePath, flags.campaignPath, flags.campaignFilePath, flags.sessionPath, flags.sessionOptions())
	if err != nil {
		return fmt.Errorf("plan: %w", err)
	}
	defer func() { _ = loaded.Close() }()
	cfg := loaded.Config
	if err := config.Validate(cfg); err != nil {
		return fmt.Errorf("plan: %w", err)
	}
	if _, err := loadSecretsFromConfig(cfg, logging.NewLogger(os.Stderr, slog.LevelInfo)); err != nil {
		return fmt.Errorf("plan: %w", err)
	}

	store := artifacts.NewLocalStore(cfg.Pipeline.Workspace.Root)
	paths, err := store.EnsureLayoutFor(cfg.Session.Campaign, cfg.Session.SessionID)
	if err != nil {
		return fmt.Errorf("plan: prepare workdir: %w", err)
	}

	stages := BuildFullPlan()
	var m *manifest.Manifest
	m, err = loadManifestIfPresent(ctx, cfg)
	if err != nil {
		return fmt.Errorf("plan: %w", err)
	}
	decisions := decideStageActions(stages, m, force)

	runCount := 0
	skipCount := 0
	if _, err := fmt.Fprintf(out, "narratio session plan: workdir prepared at %s\n", paths.Root); err != nil {
		return err
	}
	for _, d := range decisions {
		if d.Action == stageActionRun {
			runCount++
		} else {
			skipCount++
		}
		if _, err := fmt.Fprintf(out, "%s: %s\n", d.Stage.Name(), d.Action); err != nil {
			return err
		}
	}
	if _, err := fmt.Fprintf(out, "totals: run=%d skip=%d\n", runCount, skipCount); err != nil {
		return err
	}

	return nil
}