chore: PoC - Model generation - support primitive types at root level

tools/codegen/codespec/api_spec.go

@@ -0,0 +1,104 @@
+package codespec
+
+import (
+	"context"
+	"fmt"
+	"strconv"
+
+	"github.com/pb33f/libopenapi/datamodel/high/base"
+	high "github.com/pb33f/libopenapi/datamodel/high/v3"
+	"github.com/pb33f/libopenapi/orderedmap"
+)
+
+var (
+	errSchemaNotFound = fmt.Errorf("schema not found")
+)
+
+// This function only builds the schema from a proxy and returns the basic type and format without handling oneOf, anyOf, allOf, or nullable types.
+func BuildSchema(proxy *base.SchemaProxy) (*APISpecSchema, error) {
+	resp := &APISpecSchema{}
+
+	schema, err := proxy.BuildSchema()
+	if err != nil {
+		return nil, fmt.Errorf("failed to build schema from proxy: %w", err)
+	}
+
+	if len(schema.Type) == 0 {
+		return nil, errInvalidSchema
+	}
+
+	resp.Type = schema.Type[0]
+	resp.Schema = schema
+	resp.Format = resp.Schema.Format
+
+	return resp, nil
+}
+
+func getSchemaFromMediaType(mediaTypes *orderedmap.Map[string, *high.MediaType]) (*APISpecSchema, error) {
+	if mediaTypes == nil {
+		return nil, errSchemaNotFound
+	}
+
+	// hard-coding API version for now, this should be dynamically handled to perhaps use the latest API version
+	jsonMediaType, ok := mediaTypes.Get("application/vnd.atlas.2023-01-01+json")
+	if ok && jsonMediaType.Schema != nil {
+		s, err := BuildSchema(jsonMediaType.Schema)
+		if err != nil {
+			return nil, err
+		}
+		return s, nil
+	}
+
+	sortedMediaTypes := orderedmap.SortAlpha(mediaTypes)
+	for pair := range orderedmap.Iterate(context.TODO(), sortedMediaTypes) {
+		mediaType := pair.Value()
+		if mediaType.Schema != nil {
+			s, err := BuildSchema(mediaType.Schema)
+			if err != nil {
+				return nil, err
+			}
+			return s, nil
+		}
+	}
+
+	return nil, errSchemaNotFound
+}
+
+func buildSchemaFromRequest(op *high.Operation) (*APISpecSchema, error) {
+	if op == nil || op.RequestBody == nil || op.RequestBody.Content == nil || op.RequestBody.Content.Len() == 0 {
+		return nil, errSchemaNotFound
+	}
+
+	return getSchemaFromMediaType(op.RequestBody.Content)
+}
+
+func buildSchemaFromResponse(op *high.Operation) (*APISpecSchema, error) {
+	if op == nil || op.Responses == nil || op.Responses.Codes == nil || op.Responses.Codes.Len() == 0 {
+		return nil, errSchemaNotFound
+	}
+
+	okResponse, ok := op.Responses.Codes.Get(OASResponseCodeOK)
+	if ok {
+		return getSchemaFromMediaType(okResponse.Content)
+	}
+
+	createdResponse, ok := op.Responses.Codes.Get(OASResponseCodeCreated)
+	if ok {
+		return getSchemaFromMediaType(createdResponse.Content)
+	}
+
+	sortedCodes := orderedmap.SortAlpha(op.Responses.Codes)
+	for pair := range orderedmap.Iterate(context.TODO(), sortedCodes) {
+		responseCode := pair.Value()
+		statusCode, err := strconv.Atoi(pair.Key())
+		if err != nil {
+			continue
+		}
+
+		if statusCode >= 200 && statusCode <= 299 {
+			return getSchemaFromMediaType(responseCode.Content)
+		}
+	}
+
+	return nil, errSchemaNotFound
+}

tools/codegen/codespec/api_spec_schema.go

@@ -0,0 +1,63 @@
+package codespec
+
+import (
+	"fmt"
+
+	"github.com/pb33f/libopenapi/datamodel/high/base"
+	high "github.com/pb33f/libopenapi/datamodel/high/v3"
+)
+
+var errInvalidSchema = fmt.Errorf("invalid schema")
+
+type APISpecSchema struct {
+	Schema *base.Schema
+	Type   string
+	Format string
+}
+
+type APISpecResource struct {
+	Description      *string
+	CreateOp         *high.Operation
+	ReadOp           *high.Operation
+	UpdateOp         *high.Operation
+	DeleteOp         *high.Operation
+	CommonParameters []*high.Parameter
+}
+
+func (s *APISpecSchema) GetComputability(name string) ComputedOptionalRequired {
+	for _, prop := range s.Schema.Required {
+		if name == prop {
+			return Required
+		}
+	}
+
+	return ComputedOptional
+}
+
+func (s *APISpecSchema) GetDeprecationMessage() *string {
+	if s.Schema.Deprecated == nil || !(*s.Schema.Deprecated) {
+		return nil
+	}
+
+	deprecationMessage := "This attribute has been deprecated"
+
+	return &deprecationMessage
+}
+
+func (s *APISpecSchema) GetDescription() *string {
+	if s.Schema.Description == "" {
+		return nil
+	}
+
+	return &s.Schema.Description
+}
+
+func (s *APISpecSchema) IsSensitive() *bool {
+	isSensitive := s.Format == OASFormatPassword
+
+	if !isSensitive {
+		return nil
+	}
+
+	return &isSensitive
+}