commandkit/integrations/ckconfig/sources.go

package ckconfig

import (
	"bytes"
	"fmt"
	"io"
	"strconv"

	commandkit "gitea.maximumdirect.net/eric/CommandKit"
	configkit "gitea.maximumdirect.net/eric/ConfigKit"
	"go.yaml.in/yaml/v4"
)

type sourceReport struct {
	Scope   string      `yaml:"scope"`
	Sources []sourceDTO `yaml:"sources"`
}

type sourceDTO struct {
	ID            string       `yaml:"id"`
	Label         string       `yaml:"label"`
	Path          string       `yaml:"path"`
	Layer         string       `yaml:"layer"`
	Kind          string       `yaml:"kind"`
	UnderlayIndex int          `yaml:"underlay_index"`
	ImportedFrom  *locationDTO `yaml:"imported_from"`
}

type locationDTO struct {
	SourceID string `yaml:"source_id"`
	Line     int    `yaml:"line"`
	Column   int    `yaml:"column"`
}

func runConfigSources[I, S any](invocation commandkit.Invocation, plan *configOperationPlan[I, S], state S) commandkit.Outcome {
	selected := plan.scopes[0]
	name := selected.scope.scope.Name
	if err := invocation.Context.Err(); err != nil {
		return configOperationFailure(err, nil, false)
	}
	if selected.scope.scope.Acquire == nil {
		return configOperationFailure(fmt.Errorf("configuration scope %q has no acquisition callback", name), nil, false)
	}
	acquired, acquireErr := selected.scope.scope.Acquire(invocation.Context, Request{
		Operation: Sources,
		Selection: plan.selection,
	}, selected.input, state)
	diagnostics := append([]commandkit.Diagnostic(nil), acquired.Diagnostics...)
	if acquireErr != nil {
		return configOperationFailure(acquireErr, diagnostics, false)
	}
	if err := invocation.Context.Err(); err != nil {
		return configOperationFailure(err, diagnostics, false)
	}
	if acquired.Value.Sources == nil {
		return configOperationFailure(fmt.Errorf("configuration scope %q did not provide a source reader", name), diagnostics, false)
	}

	sources := acquired.Value.Sources.Sources()
	document := sourceReport{Scope: name, Sources: make([]sourceDTO, 0, len(sources))}
	for _, source := range sources {
		item := sourceDTO{
			ID:            source.ID,
			Label:         source.Label,
			Path:          source.Path,
			Layer:         string(source.Layer),
			Kind:          string(source.Kind),
			UnderlayIndex: source.UnderlayIndex,
		}
		if source.ImportedFrom != nil {
			item.ImportedFrom = &locationDTO{SourceID: source.ImportedFrom.SourceID, Line: source.ImportedFrom.Line, Column: source.ImportedFrom.Column}
		}
		document.Sources = append(document.Sources, item)
	}
	var report bytes.Buffer
	if plan.format == "yaml" {
		if err := encodeYAML(&report, document); err != nil {
			return configOperationFailure(err, diagnostics, false)
		}
	} else {
		writeSourceReportText(&report, document)
	}
	if err := invocation.Context.Err(); err != nil {
		return configOperationFailure(err, diagnostics, false)
	}
	if err := writeConfigReport(invocation.Streams.Out, report.Bytes()); err != nil {
		return configDeliveryFailure(err, diagnostics, true)
	}
	outcome := commandkit.Success()
	outcome.Diagnostics = diagnostics
	return outcome
}

func writeSourceReportText(writer *bytes.Buffer, document sourceReport) {
	_, _ = fmt.Fprintf(writer, "Scope: %s\n", strconv.Quote(document.Scope))
	if len(document.Sources) == 0 {
		_, _ = writer.WriteString("Sources: []\n")
		return
	}
	_, _ = writer.WriteString("Sources:\n")
	for _, source := range document.Sources {
		_, _ = fmt.Fprintf(writer, "  - id: %s\n", strconv.Quote(source.ID))
		_, _ = fmt.Fprintf(writer, "    label: %s\n", strconv.Quote(source.Label))
		_, _ = fmt.Fprintf(writer, "    path: %s\n", strconv.Quote(source.Path))
		_, _ = fmt.Fprintf(writer, "    layer: %s\n", strconv.Quote(source.Layer))
		_, _ = fmt.Fprintf(writer, "    kind: %s\n", strconv.Quote(source.Kind))
		_, _ = fmt.Fprintf(writer, "    underlay_index: %d\n", source.UnderlayIndex)
		if source.ImportedFrom == nil {
			_, _ = writer.WriteString("    imported_from: null\n")
		} else {
			_, _ = fmt.Fprintf(writer, "    imported_from: {source_id: %s, line: %d, column: %d}\n",
				strconv.Quote(source.ImportedFrom.SourceID), source.ImportedFrom.Line, source.ImportedFrom.Column)
		}
	}
}

func encodeYAML(writer io.Writer, document any) error {
	encoder := yaml.NewEncoder(writer)
	encoder.SetIndent(2)
	return encoder.Encode(document)
}

func locationReports(locations []configkit.Location) []locationDTO {
	report := make([]locationDTO, 0, len(locations))
	for _, location := range locations {
		report = append(report, locationDTO{SourceID: location.SourceID, Line: location.Line, Column: location.Column})
	}
	return report
}