Make Notarius validation transactional

This commit is contained in:
2026-09-05 00:26:54 +00:00
parent 3136c26d4e
commit fddf78dc80
3 changed files with 187 additions and 67 deletions

View File

@@ -235,7 +235,7 @@ the internal structure of Notarius or Scriptorium validation.
## Stage 3 — Decompose Notarius Validation And Normalization ## Stage 3 — Decompose Notarius Validation And Normalization
**Status:** Pending **Status:** Completed
### Goal ### Goal

View File

@@ -4,6 +4,7 @@ import (
"fmt" "fmt"
"os" "os"
"path/filepath" "path/filepath"
"reflect"
"strings" "strings"
"testing" "testing"
) )
@@ -252,6 +253,99 @@ func TestNotariusNilAndEmptyReferencesAreValid(t *testing.T) {
} }
} }
func TestNotariusValidationFailureDoesNotMutateConfig(t *testing.T) {
tests := []struct {
name string
mutate func(*NotariusConfig)
want string
}{
{
name: "reference",
mutate: func(cfg *NotariusConfig) {
cfg.Binary = " notarius "
cfg.References = map[string]string{" party ": "unsupported"}
},
want: "pipeline.notarius.references.party",
},
{
name: "output",
mutate: func(cfg *NotariusConfig) {
cfg.Binary = " notarius "
cfg.References = map[string]string{" party ": " narratio.input.party "}
cfg.Outputs["npc_registry"] = NotariusOutputConfig{LaneID: " missing-schema "}
},
want: "pipeline.notarius.outputs.npc_registry.media_type",
},
}
for _, test := range tests {
t.Run(test.name, func(t *testing.T) {
cfg := validNotariusConfig()
test.mutate(cfg)
before := cloneNotariusConfigForTest(cfg)
err := validateNotarius(cfg, nil)
if err == nil || !strings.Contains(err.Error(), test.want) {
t.Fatalf("validateNotarius() error = %v, want field containing %q", err, test.want)
}
if !reflect.DeepEqual(cfg, before) {
t.Fatalf("config mutated after validation failure:\n got: %#v\nwant: %#v", cfg, before)
}
})
}
}
func TestNotariusNormalizationIsIndependentOfMapInsertionOrder(t *testing.T) {
first := validNotariusConfig()
first.References = make(map[string]string)
first.References[" party "] = " narratio.input.party "
first.References["chunk . players"] = " narratio.input.players "
first.Outputs = make(map[string]NotariusOutputConfig)
first.Outputs[" npc_registry "] = NotariusOutputConfig{
LaneID: " npc-registry ", MediaType: " application/json ",
SchemaID: " notarius.dnd.npc_registry ", SchemaVersion: " v1 ",
}
first.Outputs["spells"] = NotariusOutputConfig{
LaneID: " spells ", MediaType: " application/json ",
SchemaID: " notarius.dnd.spells ", SchemaVersion: " v1 ",
}
second := validNotariusConfig()
second.References = make(map[string]string)
second.References["chunk . players"] = " narratio.input.players "
second.References[" party "] = " narratio.input.party "
second.Outputs = make(map[string]NotariusOutputConfig)
second.Outputs["spells"] = first.Outputs["spells"]
second.Outputs[" npc_registry "] = first.Outputs[" npc_registry "]
if err := validateNotarius(first, nil); err != nil {
t.Fatalf("validateNotarius(first) error = %v", err)
}
if err := validateNotarius(second, nil); err != nil {
t.Fatalf("validateNotarius(second) error = %v", err)
}
if !reflect.DeepEqual(first, second) {
t.Fatalf("normalized configs differ:\nfirst: %#v\nsecond: %#v", first, second)
}
}
func cloneNotariusConfigForTest(cfg *NotariusConfig) *NotariusConfig {
clone := *cfg
if cfg.References != nil {
clone.References = make(map[string]string, len(cfg.References))
for key, value := range cfg.References {
clone.References[key] = value
}
}
if cfg.Outputs != nil {
clone.Outputs = make(map[string]NotariusOutputConfig, len(cfg.Outputs))
for key, value := range cfg.Outputs {
clone.Outputs[key] = value
}
}
return &clone
}
func TestNotariusSpellCatalogReferenceRequiresEffectiveInput(t *testing.T) { func TestNotariusSpellCatalogReferenceRequiresEffectiveInput(t *testing.T) {
cfg := loadedValidConfig(t) cfg := loadedValidConfig(t)
cfg.Pipeline.Notarius = validNotariusConfig() cfg.Pipeline.Notarius = validNotariusConfig()

View File

@@ -5,7 +5,6 @@ import (
"path/filepath" "path/filepath"
"sort" "sort"
"strings" "strings"
"time"
"gitea.maximumdirect.net/eric/narratio/internal/artifactmodel" "gitea.maximumdirect.net/eric/narratio/internal/artifactmodel"
"gitea.maximumdirect.net/eric/narratio/internal/artifactpolicy" "gitea.maximumdirect.net/eric/narratio/internal/artifactpolicy"
@@ -16,95 +15,113 @@ func validateNotarius(cfg *NotariusConfig, scriptorium *ScriptoriumConfig) error
if cfg == nil || !cfg.Enabled { if cfg == nil || !cfg.Enabled {
return nil return nil
} }
if strings.TrimSpace(cfg.Binary) == "" {
binary := strings.TrimSpace(cfg.Binary)
configPath := filepath.Clean(cfg.ConfigPath)
pipelineID := strings.TrimSpace(cfg.PipelineID)
timeout := strings.TrimSpace(cfg.Timeout)
workingDirectory := filepath.Clean(cfg.WorkingDirectory)
if binary == "" {
return fmt.Errorf("pipeline.notarius.binary is required when pipeline.notarius.enabled is true") return fmt.Errorf("pipeline.notarius.binary is required when pipeline.notarius.enabled is true")
} }
if strings.TrimSpace(cfg.ConfigPath) == "" { if strings.TrimSpace(cfg.ConfigPath) == "" {
return fmt.Errorf("pipeline.notarius.config_path is required when pipeline.notarius.enabled is true") return fmt.Errorf("pipeline.notarius.config_path is required when pipeline.notarius.enabled is true")
} }
if strings.TrimSpace(cfg.PipelineID) == "" { if pipelineID == "" {
return fmt.Errorf("pipeline.notarius.pipeline_id is required when pipeline.notarius.enabled is true") return fmt.Errorf("pipeline.notarius.pipeline_id is required when pipeline.notarius.enabled is true")
} }
if len(cfg.Outputs) == 0 { if timeout == "" {
return fmt.Errorf("pipeline.notarius.outputs must contain at least one output when pipeline.notarius.enabled is true") return fmt.Errorf("pipeline.notarius.timeout must be a valid duration")
} }
duration, err := time.ParseDuration(strings.TrimSpace(cfg.Timeout)) if err := validateDuration("pipeline.notarius.timeout", timeout); err != nil {
if err != nil { return err
return fmt.Errorf("pipeline.notarius.timeout must be a valid duration: %w", err)
}
if duration <= 0 {
return fmt.Errorf("pipeline.notarius.timeout must be positive")
} }
if strings.TrimSpace(cfg.WorkingDirectory) == "" { if strings.TrimSpace(cfg.WorkingDirectory) == "" {
return fmt.Errorf("pipeline.notarius.working_directory is required when pipeline.notarius.enabled is true") return fmt.Errorf("pipeline.notarius.working_directory is required when pipeline.notarius.enabled is true")
} }
if len(cfg.Outputs) == 0 {
return fmt.Errorf("pipeline.notarius.outputs must contain at least one output when pipeline.notarius.enabled is true")
}
if len(cfg.References) > MaxNotariusReferenceBindings { if len(cfg.References) > MaxNotariusReferenceBindings {
return fmt.Errorf("pipeline.notarius.references must contain at most %d bindings", MaxNotariusReferenceBindings) return fmt.Errorf("pipeline.notarius.references must contain at most %d bindings", MaxNotariusReferenceBindings)
} }
referenceKeys := make([]string, 0, len(cfg.References)) references, err := normalizeNotariusReferences(cfg.References)
for selector := range cfg.References { if err != nil {
referenceKeys = append(referenceKeys, selector) return err
} }
sort.Strings(referenceKeys) outputs, err := normalizeNotariusOutputs(cfg.Outputs, scriptorium)
var normalizedReferences map[string]string if err != nil {
if cfg.References != nil { return err
normalizedReferences = make(map[string]string, len(cfg.References))
} }
referenceOwners := make(map[string]string, len(cfg.References))
for _, rawSelector := range referenceKeys { cfg.Binary = binary
cfg.ConfigPath = configPath
cfg.PipelineID = pipelineID
cfg.Timeout = timeout
cfg.WorkingDirectory = workingDirectory
cfg.References = references
cfg.Outputs = outputs
return nil
}
func normalizeNotariusReferences(references map[string]string) (map[string]string, error) {
selectors := make([]string, 0, len(references))
for selector := range references {
selectors = append(selectors, selector)
}
sort.Strings(selectors)
var normalized map[string]string
if references != nil {
normalized = make(map[string]string, len(references))
}
owners := make(map[string]string, len(references))
for _, rawSelector := range selectors {
selector, err := notariusref.NormalizeSelector(rawSelector) selector, err := notariusref.NormalizeSelector(rawSelector)
if err != nil { if err != nil {
return fmt.Errorf("pipeline.notarius.references selector %q is invalid: %w", rawSelector, err) return nil, fmt.Errorf("pipeline.notarius.references selector %q is invalid: %w", rawSelector, err)
} }
if previous, ok := referenceOwners[selector]; ok { if previous, ok := owners[selector]; ok {
return fmt.Errorf("pipeline.notarius.references selectors %q and %q normalize to %q", previous, rawSelector, selector) return nil, fmt.Errorf("pipeline.notarius.references selectors %q and %q normalize to %q", previous, rawSelector, selector)
} }
referenceOwners[selector] = rawSelector owners[selector] = rawSelector
source := strings.TrimSpace(cfg.References[rawSelector]) source := strings.TrimSpace(references[rawSelector])
if source == "" { if source == "" {
return fmt.Errorf("pipeline.notarius.references.%s source is required", selector) return nil, fmt.Errorf("pipeline.notarius.references.%s source is required", selector)
} }
if _, ok := artifactpolicy.DescribePreparedInputSource(source); !ok { if _, ok := artifactpolicy.DescribePreparedInputSource(source); !ok {
return fmt.Errorf("pipeline.notarius.references.%s source %q is not a supported prepared input source", selector, source) return nil, fmt.Errorf("pipeline.notarius.references.%s source %q is not a supported prepared input source", selector, source)
} }
normalizedReferences[selector] = source normalized[selector] = source
} }
return normalized, nil
}
reservedSources := map[string]string{} func normalizeNotariusOutputs(outputs map[string]NotariusOutputConfig, scriptorium *ScriptoriumConfig) (map[string]NotariusOutputConfig, error) {
for _, spec := range artifactmodel.RuntimeTranscriptArtifacts() { reservedSources := notariusReservedSources(scriptorium)
reservedSources[spec.SourceID] = "built-in source" keys := make([]string, 0, len(outputs))
} for key := range outputs {
reservedSources[artifactpolicy.SourceBoundsSession] = "built-in source" keys = append(keys, key)
if scriptorium != nil {
for key := range scriptorium.Artifacts {
normalizedKey := strings.TrimSpace(key)
reservedSources[artifactpolicy.ConfiguredSourceID(normalizedKey)] = "configured Scriptorium source"
reservedSources[artifactpolicy.PreviousSessionSourceID(normalizedKey)] = "previous-session source"
}
} }
sort.Strings(keys)
rawKeys := make([]string, 0, len(cfg.Outputs)) normalized := make(map[string]NotariusOutputConfig, len(outputs))
for key := range cfg.Outputs { sourceOwners := make(map[string]string, len(outputs))
rawKeys = append(rawKeys, key) laneOwners := make(map[string]string, len(outputs))
} for _, rawKey := range keys {
sort.Strings(rawKeys) output := outputs[rawKey]
normalizedOutputs := make(map[string]NotariusOutputConfig, len(cfg.Outputs))
sourceOwners := map[string]string{}
laneOwners := map[string]string{}
for _, rawKey := range rawKeys {
output := cfg.Outputs[rawKey]
key := strings.TrimSpace(rawKey) key := strings.TrimSpace(rawKey)
if !artifactpolicy.IsConfiguredKey(key) { if !artifactpolicy.IsConfiguredKey(key) {
return fmt.Errorf("pipeline.notarius.outputs keys must match ^[a-z][a-z0-9_]*$") return nil, fmt.Errorf("pipeline.notarius.outputs keys must match ^[a-z][a-z0-9_]*$")
} }
sourceID := artifactpolicy.ExtractionSourceID(key) sourceID := artifactpolicy.ExtractionSourceID(key)
if previousKey, ok := sourceOwners[sourceID]; ok { if previousKey, ok := sourceOwners[sourceID]; ok {
return fmt.Errorf("pipeline.notarius.outputs keys %q and %q normalize to duplicate source %q", previousKey, rawKey, sourceID) return nil, fmt.Errorf("pipeline.notarius.outputs keys %q and %q normalize to duplicate source %q", previousKey, rawKey, sourceID)
} }
if owner, ok := reservedSources[sourceID]; ok { if owner, ok := reservedSources[sourceID]; ok {
return fmt.Errorf("pipeline.notarius.outputs.%s source %q collides with %s", key, sourceID, owner) return nil, fmt.Errorf("pipeline.notarius.outputs.%s source %q collides with %s", key, sourceID, owner)
} }
sourceOwners[sourceID] = rawKey sourceOwners[sourceID] = rawKey
@@ -115,31 +132,40 @@ func validateNotarius(cfg *NotariusConfig, scriptorium *ScriptoriumConfig) error
output.ModuleKey = strings.TrimSpace(output.ModuleKey) output.ModuleKey = strings.TrimSpace(output.ModuleKey)
prefix := "pipeline.notarius.outputs." + key prefix := "pipeline.notarius.outputs." + key
if output.LaneID == "" { if output.LaneID == "" {
return fmt.Errorf("%s.lane_id is required", prefix) return nil, fmt.Errorf("%s.lane_id is required", prefix)
} }
if previousKey, ok := laneOwners[output.LaneID]; ok { if previousKey, ok := laneOwners[output.LaneID]; ok {
return fmt.Errorf("%s.lane_id %q duplicates pipeline.notarius.outputs.%s.lane_id", prefix, output.LaneID, previousKey) return nil, fmt.Errorf("%s.lane_id %q duplicates pipeline.notarius.outputs.%s.lane_id", prefix, output.LaneID, previousKey)
} }
laneOwners[output.LaneID] = key laneOwners[output.LaneID] = key
if output.MediaType == "" { if output.MediaType == "" {
return fmt.Errorf("%s.media_type is required", prefix) return nil, fmt.Errorf("%s.media_type is required", prefix)
} }
if output.SchemaID == "" { if output.SchemaID == "" {
return fmt.Errorf("%s.schema_id is required", prefix) return nil, fmt.Errorf("%s.schema_id is required", prefix)
} }
if output.SchemaVersion == "" { if output.SchemaVersion == "" {
return fmt.Errorf("%s.schema_version is required", prefix) return nil, fmt.Errorf("%s.schema_version is required", prefix)
} }
normalizedOutputs[key] = output normalized[key] = output
} }
cfg.Binary = strings.TrimSpace(cfg.Binary) return normalized, nil
cfg.ConfigPath = filepath.Clean(cfg.ConfigPath) }
cfg.PipelineID = strings.TrimSpace(cfg.PipelineID)
cfg.Timeout = strings.TrimSpace(cfg.Timeout) func notariusReservedSources(scriptorium *ScriptoriumConfig) map[string]string {
cfg.WorkingDirectory = filepath.Clean(cfg.WorkingDirectory) reserved := make(map[string]string)
cfg.References = normalizedReferences for _, spec := range artifactmodel.RuntimeTranscriptArtifacts() {
cfg.Outputs = normalizedOutputs reserved[spec.SourceID] = "built-in source"
return nil }
reserved[artifactpolicy.SourceBoundsSession] = "built-in source"
if scriptorium != nil {
for key := range scriptorium.Artifacts {
normalizedKey := strings.TrimSpace(key)
reserved[artifactpolicy.ConfiguredSourceID(normalizedKey)] = "configured Scriptorium source"
reserved[artifactpolicy.PreviousSessionSourceID(normalizedKey)] = "previous-session source"
}
}
return reserved
} }
func notariusOutputKeySet(notarius *NotariusConfig) map[string]struct{} { func notariusOutputKeySet(notarius *NotariusConfig) map[string]struct{} {