Add generated text state artifacts

internal/state/filesystem.go

@@ -27,12 +27,16 @@ type FilesystemStore struct {
 }
 
 type ArtifactPaths struct {
-	ModuleSnapshot string `json:"moduleSnapshot"`
-	Metadata       string `json:"metadata"`
-	DataPackage    string `json:"dataPackage"`
-	Preflight      string `json:"preflight"`
-	Notification   string `json:"notification,omitempty"`
-	RenderedReport string `json:"renderedReport,omitempty"`
+	ModuleSnapshot      string `json:"moduleSnapshot"`
+	Metadata            string `json:"metadata"`
+	DataPackage         string `json:"dataPackage"`
+	Preflight           string `json:"preflight"`
+	Notification        string `json:"notification,omitempty"`
+	RenderedReport      string `json:"renderedReport,omitempty"`
+	GeneratedTextRaw    string `json:"generatedTextRaw,omitempty"`
+	GeneratedTextResult string `json:"generatedTextResult,omitempty"`
+	GeneratedText       string `json:"generatedText,omitempty"`
+	RenderContext       string `json:"renderContext,omitempty"`
 }
 
 type ReportRecord struct {
@@ -89,12 +93,16 @@ func (s *FilesystemStore) Paths(resolved report.Resolved) (ArtifactPaths, error)
 	validDate := resolved.ValidPeriod.Start.Format("2006-01-02")
 	filenameBase := metadata.RunID
 	return ArtifactPaths{
-		ModuleSnapshot: s.join(s.snapshotsDir, group, validDate, filenameBase+".modules.json"),
-		Metadata:       s.join(s.snapshotsDir, group, validDate, filenameBase+".metadata.json"),
-		DataPackage:    s.join(s.dataPackagesDir, group, validDate, filenameBase+".data_package.yaml"),
-		Preflight:      s.join(s.preflightDir, group, validDate, filenameBase+".render.json"),
-		Notification:   s.join(s.notificationsDir, group, validDate, filenameBase+".distributor.json"),
-		RenderedReport: s.join(s.reportsDir, group, filenameBase+".md"),
+		ModuleSnapshot:      s.join(s.snapshotsDir, group, validDate, filenameBase+".modules.json"),
+		Metadata:            s.join(s.snapshotsDir, group, validDate, filenameBase+".metadata.json"),
+		DataPackage:         s.join(s.dataPackagesDir, group, validDate, filenameBase+".data_package.yaml"),
+		Preflight:           s.join(s.preflightDir, group, validDate, filenameBase+".render.json"),
+		Notification:        s.join(s.notificationsDir, group, validDate, filenameBase+".distributor.json"),
+		RenderedReport:      s.join(s.reportsDir, group, filenameBase+".md"),
+		GeneratedTextRaw:    s.join(s.snapshotsDir, group, validDate, filenameBase+".generated_text.raw.json"),
+		GeneratedTextResult: s.join(s.snapshotsDir, group, validDate, filenameBase+".generated_text.run.json"),
+		GeneratedText:       s.join(s.snapshotsDir, group, validDate, filenameBase+".generated_text.json"),
+		RenderContext:       s.join(s.snapshotsDir, group, validDate, filenameBase+".render_context.json"),
 	}, nil
 }
 
@@ -148,6 +156,50 @@ func (s *FilesystemStore) SaveDistributorNotification(_ context.Context, resolve
 	return paths.Notification, nil
 }
 
+func (s *FilesystemStore) SaveGeneratedTextRaw(_ context.Context, resolved report.Resolved, data []byte) (string, error) {
+	paths, err := s.Paths(resolved)
+	if err != nil {
+		return "", err
+	}
+	if err := fileutil.WriteFileAtomic(paths.GeneratedTextRaw, data); err != nil {
+		return "", err
+	}
+	return paths.GeneratedTextRaw, nil
+}
+
+func (s *FilesystemStore) SaveGeneratedTextResult(_ context.Context, resolved report.Resolved, value any) (string, error) {
+	paths, err := s.Paths(resolved)
+	if err != nil {
+		return "", err
+	}
+	if err := fileutil.WriteJSONAtomic(paths.GeneratedTextResult, value); err != nil {
+		return "", err
+	}
+	return paths.GeneratedTextResult, nil
+}
+
+func (s *FilesystemStore) SaveGeneratedText(_ context.Context, resolved report.Resolved, data []byte) (string, error) {
+	paths, err := s.Paths(resolved)
+	if err != nil {
+		return "", err
+	}
+	if err := fileutil.WriteFileAtomic(paths.GeneratedText, data); err != nil {
+		return "", err
+	}
+	return paths.GeneratedText, nil
+}
+
+func (s *FilesystemStore) SaveRenderContext(_ context.Context, resolved report.Resolved, value any) (string, error) {
+	paths, err := s.Paths(resolved)
+	if err != nil {
+		return "", err
+	}
+	if err := fileutil.WriteJSONAtomic(paths.RenderContext, value); err != nil {
+		return "", err
+	}
+	return paths.RenderContext, nil
+}
+
 func (s *FilesystemStore) PrepareRenderedReport(_ context.Context, resolved report.Resolved) (string, error) {
 	paths, err := s.Paths(resolved)
 	if err != nil {
@@ -325,6 +377,37 @@ func (s *FilesystemStore) LoadModuleSnapshot(_ context.Context, path string) (mo
 	return snapshot, nil
 }
 
+func (s *FilesystemStore) LoadGeneratedText(_ context.Context, path string) ([]byte, error) {
+	if path == "" {
+		return nil, fmt.Errorf("generated text path is required")
+	}
+	data, err := os.ReadFile(path)
+	if err != nil {
+		return nil, fmt.Errorf("read %q: %w", path, err)
+	}
+	return data, nil
+}
+
+func (s *FilesystemStore) LoadGeneratedTextResult(_ context.Context, path string, target any) error {
+	if path == "" {
+		return fmt.Errorf("generated text result path is required")
+	}
+	if target == nil {
+		return fmt.Errorf("generated text result target is required")
+	}
+	return readJSON(path, target)
+}
+
+func (s *FilesystemStore) LoadRenderContext(_ context.Context, path string, target any) error {
+	if path == "" {
+		return fmt.Errorf("render context path is required")
+	}
+	if target == nil {
+		return fmt.Errorf("render context target is required")
+	}
+	return readJSON(path, target)
+}
+
 func (s *FilesystemStore) reportRecord(path string) (ReportRecord, error) {
 	var metadata Metadata
 	if err := readJSON(path, &metadata); err != nil {