Working through the model builder

Shaping out the design as I go, what makes sence, what feels right, what do we need, how do we want to use it and how to we want to search it etc.

openapi/create_document.go

@@ -2,7 +2,9 @@ package openapi
 
 import (
     "github.com/pb33f/libopenapi/datamodel"
+    "github.com/pb33f/libopenapi/datamodel/low"
     v3 "github.com/pb33f/libopenapi/datamodel/low/3.0"
+    "github.com/pb33f/libopenapi/utils"
 )
 
 func CreateDocument(spec []byte) (*v3.Document, error) {
@@ -13,7 +15,64 @@ func CreateDocument(spec []byte) (*v3.Document, error) {
         return nil, err
     }
 
-    doc := &v3.Document{}
-    doc.Build(info.RootNode.Content[0])
-    return doc, nil
+    doc := v3.Document{Version: low.NodeReference[string]{Value: info.Version, ValueNode: info.RootNode}}
+
+    // build an index
+    //idx := index.NewSpecIndex(info.RootNode)
+    datamodel.BuildModel(info.RootNode.Content[0], &doc)
+
+    // extract info
+    extractErr := extractInfo(info, &doc)
+    if extractErr != nil {
+        return nil, extractErr
+    }
+
+    // extract servers
+    extractErr = extractServers(info, &doc)
+    if extractErr != nil {
+        return nil, extractErr
+    }
+
+    return &doc, nil
+}
+
+func extractInfo(info *datamodel.SpecInfo, doc *v3.Document) error {
+    _, ln, vn := utils.FindKeyNodeFull("info", info.RootNode.Content)
+    if vn != nil {
+        ir := v3.Info{}
+        err := datamodel.BuildModel(vn, &ir)
+        if err != nil {
+            return err
+        }
+        err = ir.Build(vn)
+        nr := low.NodeReference[*v3.Info]{Value: &ir, ValueNode: vn, KeyNode: ln}
+        doc.Info = nr
+    }
+    return nil
+}
+
+func extractServers(info *datamodel.SpecInfo, doc *v3.Document) error {
+    _, ln, vn := utils.FindKeyNodeFull("servers", info.RootNode.Content)
+    if vn != nil {
+        if utils.IsNodeArray(vn) {
+            var servers []low.NodeReference[*v3.Server]
+            for _, srvN := range vn.Content {
+                if utils.IsNodeMap(srvN) {
+                    srvr := v3.Server{}
+                    err := datamodel.BuildModel(srvN, &srvr)
+                    if err != nil {
+                        return err
+                    }
+                    srvr.Build(srvN)
+                    servers = append(servers, low.NodeReference[*v3.Server]{
+                        Value:     &srvr,
+                        ValueNode: srvN,
+                        KeyNode:   ln,
+                    })
+                }
+            }
+            doc.Servers = servers
+        }
+    }
+    return nil
 }