fix writing via rootdocument for workspace.

Author: VisualBean

src/LEGO.AsyncAPI.Readers/V2/AsyncApiAvroSchemaDeserializer.cs

@@ -212,7 +212,7 @@ public class AsyncApiAvroSchemaDeserializer
             { s => s.StartsWith(string.Empty), (a, p, n) => a.Metadata[p] = n.CreateAny() },
         };
 
-        public static AvroSchema LoadSchema(ParseNode node)
+        public static AsyncApiAvroSchema LoadSchema(ParseNode node)
         {
             if (node is ValueNode valueNode)
             {
@@ -284,7 +284,7 @@ public static AvroSchema LoadSchema(ParseNode node)
             throw new AsyncApiReaderException("Invalid node type");
         }
 
-        private static AvroSchema LoadLogicalType(MapNode mapNode)
+        private static AsyncApiAvroSchema LoadLogicalType(MapNode mapNode)
         {
             var type = mapNode["logicalType"]?.Value.GetScalarValue();
             switch (type)

src/LEGO.AsyncAPI.Readers/V2/AsyncApiMessageDeserializer.cs

@@ -88,9 +88,9 @@ private static IAsyncApiMessagePayload LoadPayload(ParseNode n, string format)
                 case null:
                 case "":
                 case var _ when SupportedJsonSchemaFormats.Where(s => format.StartsWith(s)).Any():
-                    return new AsyncApiJsonSchemaPayload(AsyncApiSchemaDeserializer.LoadSchema(n));
+                    return AsyncApiSchemaDeserializer.LoadSchema(n);
                 case var _ when SupportedAvroSchemaFormats.Where(s => format.StartsWith(s)).Any():
-                    return new AsyncApiAvroSchemaPayload(AsyncApiAvroSchemaDeserializer.LoadSchema(n));
+                    return AsyncApiAvroSchemaDeserializer.LoadSchema(n);
                 default:
                     var supportedFormats = SupportedJsonSchemaFormats.Concat(SupportedAvroSchemaFormats);
                     throw new AsyncApiException($"'Could not deserialize Payload. Supported formats are {string.Join(", ", supportedFormats)}");

src/LEGO.AsyncAPI.Readers/V2/AsyncApiV2VersionService.cs

@@ -35,9 +35,9 @@ public AsyncApiV2VersionService(AsyncApiDiagnostic diagnostic)
             [typeof(AsyncApiOperation)] = AsyncApiV2Deserializer.LoadOperation,
             [typeof(AsyncApiParameter)] = AsyncApiV2Deserializer.LoadParameter,
             [typeof(AsyncApiJsonSchema)] = AsyncApiSchemaDeserializer.LoadSchema,
-            [typeof(AvroSchema)] = AsyncApiAvroSchemaDeserializer.LoadSchema,
-            [typeof(AsyncApiJsonSchemaPayload)] = AsyncApiV2Deserializer.LoadJsonSchemaPayload,
-            [typeof(AsyncApiAvroSchemaPayload)] = AsyncApiV2Deserializer.LoadAvroPayload,
+            [typeof(AsyncApiAvroSchema)] = AsyncApiAvroSchemaDeserializer.LoadSchema,
+            [typeof(AsyncApiJsonSchema)] = AsyncApiV2Deserializer.LoadJsonSchemaPayload,
+            [typeof(AsyncApiAvroSchema)] = AsyncApiV2Deserializer.LoadAvroPayload,
             [typeof(AsyncApiSecurityRequirement)] = AsyncApiV2Deserializer.LoadSecurityRequirement,
             [typeof(AsyncApiSecurityScheme)] = AsyncApiV2Deserializer.LoadSecurityScheme,
             [typeof(AsyncApiServer)] = AsyncApiV2Deserializer.LoadServer,

src/LEGO.AsyncAPI/AsyncApiWorkspace.cs

@@ -71,6 +71,7 @@ public void RegisterComponents(AsyncApiDocument document)
             {
                 location = baseUri + ReferenceType.SecurityScheme.GetDisplayName() + "/" + item.Key;
                 this.RegisterComponent(location, item.Value);
+                this.RegisterComponent(item.Key, item.Value);
             }
 
             // Register Parameters

src/LEGO.AsyncAPI/Models/AsyncApiAvroSchemaPayload.cs

@@ -12,7 +12,7 @@ namespace LEGO.AsyncAPI.Models
     /// </summary>
     public class AsyncApiDocument : IAsyncApiExtensible, IAsyncApiSerializable
     {
-        internal AsyncApiWorkspace Workspace { get; set; }
+        internal AsyncApiWorkspace Workspace { get; set; } = new AsyncApiWorkspace();
 
         /// <summary>
         /// REQUIRED. Specifies the AsyncAPI Specification version being used.
@@ -73,6 +73,9 @@ public void SerializeV2(IAsyncApiWriter writer)
                 throw new ArgumentNullException(nameof(writer));
             }
 
+            this.Workspace.RegisterComponents(this);
+
+            writer.RootDocument = this;
             writer.WriteStartObject();
 
             // asyncApi

src/LEGO.AsyncAPI/Models/AsyncApiDocument.cs

@@ -155,7 +155,6 @@ public void SerializeV2(IAsyncApiWriter writer)
 
         private string GetExternalReferenceV2()
         {
-
             return this.ExternalResource + (this.FragmentId != null ? "#" + this.FragmentId : string.Empty);
         }
 

src/LEGO.AsyncAPI/Models/AsyncApiJsonSchemaPayload.cs

@@ -7,7 +7,7 @@ namespace LEGO.AsyncAPI.Models
     using LEGO.AsyncAPI.Models.Interfaces;
     using LEGO.AsyncAPI.Writers;
 
-    public abstract class AvroSchema : IAsyncApiSerializable, IAsyncApiReferenceable
+    public abstract class AsyncApiAvroSchema : IAsyncApiSerializable, IAsyncApiReferenceable, IAsyncApiMessagePayload
     {
         public abstract string Type { get; }
 
@@ -20,7 +20,7 @@ public abstract class AvroSchema : IAsyncApiSerializable, IAsyncApiReferenceable
 
         public AsyncApiReference Reference { get; set; }
 
-        public static implicit operator AvroSchema(AvroPrimitiveType type)
+        public static implicit operator AsyncApiAvroSchema(AvroPrimitiveType type)
         {
             return new AvroPrimitive(type);
         }

src/LEGO.AsyncAPI/Models/AsyncApiReference.cs

@@ -6,14 +6,14 @@ namespace LEGO.AsyncAPI.Models
     using System.Linq;
     using LEGO.AsyncAPI.Writers;
 
-    public class AvroArray : AvroSchema
+    public class AvroArray : AsyncApiAvroSchema
     {
         public override string Type { get; } = "array";
 
         /// <summary>
         /// The schema of the array's items.
         /// </summary>
-        public AvroSchema Items { get; set; }
+        public AsyncApiAvroSchema Items { get; set; }
 
         /// <summary>
         /// A map of properties not in the schema, but added as additional metadata.