Add missing visitor for ConstantRefs and update prefix pass (#769)

Author: spinillos

Diff for: internal/ast/compiler/prefix_objects_names.go

@@ -25,6 +25,7 @@ func (pass *PrefixObjectNames) Process(schemas []*ast.Schema) ([]*ast.Schema, er
 		OnRef:         pass.processRef,
 		OnEnum:        pass.processEnum,
 		OnDisjunction: pass.processDisjunction,
+		OnConstantRef: pass.processConstantRef,
 	}
 
 	return visitor.VisitSchemas(schemas)
@@ -111,3 +112,11 @@ func (pass *PrefixObjectNames) processEnum(_ *Visitor, _ *ast.Schema, enum ast.T
 
 	return enum, nil
 }
+
+func (pass *PrefixObjectNames) processConstantRef(_ *Visitor, _ *ast.Schema, ref ast.Type) (ast.Type, error) {
+	originalName := ref.ConstantReference.ReferredType
+	ref.ConstantReference.ReferredType = pass.Prefix + originalName
+	ref.AddToPassesTrail(fmt.Sprintf("PrefixObjectNames[%s → %s]", originalName, ref.ConstantReference.ReferredType))
+
+	return ref, nil
+}

Diff for: internal/ast/compiler/visitor.go

@@ -25,6 +25,7 @@ type Visitor struct {
 	OnEnum         VisitTypeFunc
 	OnScalar       VisitTypeFunc
 	OnRef          VisitTypeFunc
+	OnConstantRef  VisitTypeFunc
 
 	newObjects *orderedmap.Map[string, ast.Object]
 }
@@ -149,6 +150,10 @@ func (visitor *Visitor) VisitType(schema *ast.Schema, def ast.Type) (ast.Type, e
 		return visitor.VisitRef(schema, def)
 	}
 
+	if def.IsConstantRef() {
+		return visitor.VisitConsantRef(schema, def)
+	}
+
 	return def, nil
 }
 
@@ -274,3 +279,11 @@ func (visitor *Visitor) VisitRef(schema *ast.Schema, def ast.Type) (ast.Type, er
 
 	return def, nil
 }
+
+func (visitor *Visitor) VisitConsantRef(schema *ast.Schema, def ast.Type) (ast.Type, error) {
+	if visitor.OnConstantRef != nil {
+		return visitor.OnConstantRef(visitor, schema, def)
+	}
+
+	return def, nil
+}