Integrate source version checks for main inputs

internal/engine/load_input.go

@@ -8,12 +8,14 @@ import (
 
 // assembledInput is the source/document handoff across schema and version checks.
 type assembledInput struct {
-	parsed      parsedSource
-	document    *document.Node
-	composition *composedDocument
-	inheritance *inheritanceScope
-	sources     []SourceInfo
-	contexts    map[string]ResourceContext
+	parsed             parsedSource
+	document           *document.Node
+	composition        *composedDocument
+	inheritance        *inheritanceScope
+	sources            []SourceInfo
+	contexts           map[string]ResourceContext
+	versionEvidence    *sourceVersionEvidence
+	versionDiagnostics []Diagnostic
 }
 
 func assembleBaseInput[T any](source Source, options Options[T]) (assembledInput, []Diagnostic) {
@@ -21,6 +23,10 @@ func assembleBaseInput[T any](source Source, options Options[T]) (assembledInput
 }
 
 func assembleBaseInputTracked[T any](source Source, options Options[T], destination Path, accounting *importAccounting, visited map[resourceIdentity]importRoute, hostToken *namespaceToken) (assembledInput, []Diagnostic) {
+	return assembleBaseInputTrackedWithVersion(source, options, destination, accounting, visited, hostToken, nil, sourceVersionModeDefault)
+}
+
+func assembleBaseInputTrackedWithVersion[T any](source Source, options Options[T], destination Path, accounting *importAccounting, visited map[resourceIdentity]importRoute, hostToken *namespaceToken, localVersion *VersionRule, mode sourceVersionMode) (assembledInput, []Diagnostic) {
 	limits := defaultImportLimits()
 	parseLimits := defaultParseLimits()
 	parseLimits.source.remaining = limits.bytes - accounting.bytes
@@ -42,6 +48,14 @@ func assembleBaseInputTracked[T any](source Source, options Options[T], destinat
 		}
 		return input, diagnostics
 	}
+	if mode != sourceVersionModeDefault {
+		processed := processSourceVersion(parsed.document, localVersion, sourceVersionMain, destination, mode)
+		input.versionEvidence = processed.evidence
+		input.versionDiagnostics = processed.diagnostics
+		if hasErrors(processed.diagnostics) {
+			return input, processed.diagnostics
+		}
+	}
 	if options.Imports {
 		input.composition, input.sources, diagnostics = loadImportedDocumentsTracked(source, options.Filesystem, parsed, destination, options.Definition.Version, limits, accounting, visited, hostToken)
 		if len(diagnostics) != 0 {