encoding-default

Author: AllyW

src/typespec-aaz/src/convertor.ts

@@ -6,11 +6,11 @@ import { AAZEmitterContext, AAZOperationEmitterContext, AAZSchemaEmitterContext
 import { resolveOperationId } from "./utils.js";
 import { TypeSpecPathItem } from "./model/path_item.js";
 import { CMDHttpOperation } from "./model/operation.js";
-import { DiagnosticTarget, Enum, EnumMember, Model, ModelProperty, Namespace, Program, Scalar, TwoLevelMap, Type, Union, Value, getDiscriminator, getDoc, getEncode, getFormat, getMaxItems, getMaxLength, getMaxValue, getMaxValueExclusive, getMinItems, getMinLength, getMinValue, getMinValueExclusive, getPattern, getProjectedName, getProperty, isArrayModelType, isNeverType, isNullType, isRecordModelType, isService, isTemplateDeclaration, isVoidType, resolveEncodedName, IntrinsicType } from "@typespec/compiler";
+import { DiagnosticTarget, Enum, EnumMember, Model, ModelProperty, Namespace, Program, Scalar, TwoLevelMap, Type, Union, Value, getDiscriminator, getDoc, getEncode, getFormat, getMaxItems, getMaxLength, getMaxValue, getMaxValueExclusive, getMinItems, getMinLength, getMinValue, getMinValueExclusive, getPattern, getProjectedName, getProperty, isArrayModelType, isNeverType, isNullType, isRecordModelType, isService, isTemplateDeclaration, isVoidType, resolveEncodedName, IntrinsicType, serializeValueAsJson } from "@typespec/compiler";
 import { LroMetadata, PagedResultMetadata, UnionEnum, getArmResourceIdentifierConfig, getLroMetadata, getPagedResult, getUnionAsEnum } from "@azure-tools/typespec-azure-core";
 import { XmsPageable } from "./model/x_ms_pageable.js";
 import { CMDHttpRequest, CMDHttpResponse } from "./model/http.js";
-import { CMDArraySchemaBase, CMDClsSchema, CMDClsSchemaBase, CMDObjectSchema, CMDObjectSchemaBase, CMDSchema, CMDSchemaBase, CMDStringSchema, CMDStringSchemaBase, CMDIntegerSchemaBase, Ref, ClsType, ArrayType, CMDObjectSchemaDiscriminator, CMDByteSchemaBase, CMDInteger32SchemaBase, CMDInteger64SchemaBase, CMDFloatSchemaBase, CMDFloat64SchemaBase, CMDFloat32SchemaBase, CMDUuidSchemaBase, CMDPasswordSchemaBase, CMDResourceIdSchemaBase, CMDDateSchemaBase, CMDDateTimeSchemaBase, CMDDurationSchemaBase, CMDResourceLocationSchema, CMDIdentityObjectSchemaBase, CMDBooleanSchemaBase, CMDAnyTypeSchemaBase } from "./model/schema.js";
+import { CMDArraySchemaBase, CMDClsSchema, CMDClsSchemaBase, CMDObjectSchema, CMDObjectSchemaBase, CMDSchema, CMDSchemaBase, CMDStringSchema, CMDStringSchemaBase, CMDIntegerSchemaBase, Ref, ClsType, ArrayType, CMDObjectSchemaDiscriminator, CMDByteSchemaBase, CMDInteger32SchemaBase, CMDInteger64SchemaBase, CMDFloatSchemaBase, CMDFloat64SchemaBase, CMDFloat32SchemaBase, CMDUuidSchemaBase, CMDPasswordSchemaBase, CMDResourceIdSchemaBase, CMDDateSchemaBase, CMDDateTimeSchemaBase, CMDDurationSchemaBase, CMDResourceLocationSchema, CMDIdentityObjectSchemaBase, CMDBooleanSchemaBase, CMDAnyTypeSchemaBase, CMDBinarySchema } from "./model/schema.js";
 import { reportDiagnostic } from "./lib.js";
 import {
   getExtensions,
@@ -267,9 +267,6 @@ function extractHttpRequest(context: AAZOperationEmitterContext, operation: Http
     if (body.bodyKind === "multipart") {
       throw new Error("NotImplementedError: Multipart form data payloads are not supported.");
     }
-    if (isBinaryPayload(body.type, consumes)) {
-      throw new Error("NotImplementedError: Binary payloads are not supported.");
-    }
     if (consumes.includes("multipart/form-data")) {
       throw new Error("NotImplementedError: Multipart form data payloads are not supported.");
     }
@@ -303,6 +300,12 @@ function extractHttpRequest(context: AAZOperationEmitterContext, operation: Http
           clientFlatten: true,
         } as CMDObjectSchema;
       }
+      if (isBinaryPayload(body.type, consumes)) {
+        schema = {
+          ...schema,
+          type: "binary"
+        } as CMDBinarySchema
+      }
       request.body = {
         json: {
           schema
@@ -554,7 +557,7 @@ function convert2CMDSchema(context: AAZSchemaEmitterContext, param: ModelPropert
 
     if (param.defaultValue) {
       schema.default = {
-        value: getDefaultValue(context, param.defaultValue),
+        value: getDefaultValue(context, param.defaultValue, param),
       }
     }
   }
@@ -563,6 +566,10 @@ function convert2CMDSchema(context: AAZSchemaEmitterContext, param: ModelPropert
       ...schema,
       ...applySchemaFormat(context, param, schema as CMDSchemaBase)
     }
+    schema = {
+      ...schema,
+      ...applyEncoding(context, param, schema)
+    }
     schema = {
       ...schema,
       ...applyExtensionsDecorators(context, param, schema)
@@ -614,6 +621,7 @@ function convert2CMDSchemaBase(context: AAZSchemaEmitterContext, type: Type): CM
   }
   if (schema) {
     schema = applySchemaFormat(context, type, schema);
+    schema = applyEncoding(context, type, schema);
     schema = applyExtensionsDecorators(context, type, schema);
   }
 
@@ -1686,6 +1694,24 @@ function emitArrayFormat(context: AAZSchemaEmitterContext, type: Model, targetFo
 
 // TODO: add emitResourceIdFormat
 
+function applyEncoding(
+  context: AAZSchemaEmitterContext,
+  type:Type,
+  schema: CMDSchemaBase
+): CMDSchemaBase {
+  if (type.kind !== "Scalar" && type.kind !== "ModelProperty") {
+    return schema;
+  }
+  const encodeData = getEncode(context.program, type);
+  if (encodeData !== undefined) {
+    schema = {
+      ...schema,
+      ...convertScalar2CMDSchemaBase(context, encodeData.type)
+    };
+  }
+  return schema;
+}
+
 // apply extension decorators
 function applyExtensionsDecorators(
   context: AAZSchemaEmitterContext,
@@ -1923,25 +1949,10 @@ function getClsDefinitionModel(schema: CMDClsSchemaBase): CMDObjectSchemaBase |
   return schema.type.pendingSchema.schema!
 }
 
-function getDefaultValue(content: AAZSchemaEmitterContext, defaultType: Value): unknown {
-  switch (defaultType.valueKind) {
-    case "StringValue":
-      return defaultType.value;
-    case "NumericValue":
-      return defaultType.value.asNumber() ?? undefined;
-    case "BooleanValue":
-      return defaultType.value;
-    case "ArrayValue":
-      return defaultType.values.map((x) => getDefaultValue(content, x));
-    case "NullValue":
-      return null;
-    case "EnumValue":
-      return defaultType.value.value ?? defaultType.value.name;
-    default:
-      reportDiagnostic(content.program, {
-        code: "invalid-default",
-        format: { type: defaultType.valueKind },
-        target: defaultType,
-      });
-  }
+function getDefaultValue(
+  context: AAZSchemaEmitterContext,
+  defaultType: Value,
+  modelProperty: ModelProperty,
+): any {
+  return serializeValueAsJson(context.program, defaultType, modelProperty);
 }