test: 💍

feat: 🎸 implement `onKey` in capacity estimators

Author: streamich

src/codegen/binary/AbstractBinaryCodegen.ts

@@ -8,8 +8,10 @@ import type {
   BinType,
   BoolType,
   ConType,
+  KeyType,
   MapType,
   NumType,
+  ObjType,
   OrType,
   RefType,
   StrType,

src/codegen/binary/cbor copy/CborCodegen.ts

@@ -1,7 +1,7 @@
 import {JsExpression} from '@jsonjoy.com/codegen/lib/util/JsExpression';
 import {normalizeAccessor} from '@jsonjoy.com/codegen/lib/util/normalizeAccessor';
 import {CborEncoder} from '@jsonjoy.com/json-pack/lib/cbor/CborEncoder';
-import {type MapType, KeyOptType, type ObjType, type Type} from '../../../type';
+import {type MapType, KeyOptType, type KeyType, type ObjType, type Type} from '../../../type';
 import type {CompiledBinaryEncoder, SchemaPath} from '../../types';
 import {lazyKeyedFactory} from '../../util';
 import {AbstractBinaryCodegen} from '../AbstractBinaryCodegen';
@@ -79,6 +79,10 @@ export class CborCodegen extends AbstractBinaryCodegen<CborEncoder> {
     }
   }
 
+  protected onKey(path: SchemaPath, r: JsExpression, type: KeyType<any, any>): void {
+    this.onNode([...path, type.key], r, type.val);
+  }
+
   protected genEncoder(type: Type): CompiledBinaryEncoder {
     return CborCodegen.get(type);
   }

src/codegen/binary/cbor/CborCodegen.ts

@@ -1,7 +1,7 @@
 import {JsExpression} from '@jsonjoy.com/codegen/lib/util/JsExpression';
 import {normalizeAccessor} from '@jsonjoy.com/codegen/lib/util/normalizeAccessor';
 import {CborEncoder} from '@jsonjoy.com/json-pack/lib/cbor/CborEncoder';
-import {KeyOptType, type ObjType, type Type} from '../../../type';
+import {KeyOptType, type KeyType, type ObjType, type Type} from '../../../type';
 import type {CompiledBinaryEncoder, SchemaPath} from '../../types';
 import {lazyKeyedFactory} from '../../util';
 import {AbstractBinaryCodegen} from '../AbstractBinaryCodegen';
@@ -79,6 +79,10 @@ export class CborCodegen extends AbstractBinaryCodegen<CborEncoder> {
     }
   }
 
+  protected onKey(path: SchemaPath, r: JsExpression, type: KeyType<any, any>): void {
+    this.onNode([...path, type.key], r, type.val);
+  }
+
   protected genEncoder(type: Type): CompiledBinaryEncoder {
     return CborCodegen.get(type);
   }

src/codegen/binary/json/JsonCodegen.ts

@@ -1,7 +1,7 @@
 import {JsExpression} from '@jsonjoy.com/codegen/lib/util/JsExpression';
 import {normalizeAccessor} from '@jsonjoy.com/codegen/lib/util/normalizeAccessor';
 import {JsonEncoder} from '@jsonjoy.com/json-pack/lib/json/JsonEncoder';
-import {type ArrType, type MapType, KeyOptType, type ObjType, type Type} from '../../../type';
+import {type ArrType, type MapType, KeyOptType, type KeyType, type ObjType, type Type} from '../../../type';
 import type {CompiledBinaryEncoder, SchemaPath} from '../../types';
 import {lazyKeyedFactory} from '../../util';
 import {AbstractBinaryCodegen} from '../AbstractBinaryCodegen';
@@ -218,6 +218,10 @@ export class JsonCodegen extends AbstractBinaryCodegen<JsonEncoder> {
     this.blob(objEndBlob);
   }
 
+  protected onKey(path: SchemaPath, r: JsExpression, type: KeyType<any, any>): void {
+    this.onNode([...path, type.key], r, type.val);
+  }
+
   protected genEncoder(type: Type): CompiledBinaryEncoder {
     return JsonCodegen.get(type);
   }

src/codegen/binary/msgpack/MsgPackCodegen.ts

@@ -1,7 +1,7 @@
 import {JsExpression} from '@jsonjoy.com/codegen/lib/util/JsExpression';
 import {normalizeAccessor} from '@jsonjoy.com/codegen/lib/util/normalizeAccessor';
 import {MsgPackEncoder} from '@jsonjoy.com/json-pack/lib/msgpack/MsgPackEncoder';
-import {KeyOptType, type ObjType, type Type} from '../../../type';
+import {KeyOptType, type KeyType, type ObjType, type Type} from '../../../type';
 import type {CompiledBinaryEncoder, SchemaPath} from '../../types';
 import {lazyKeyedFactory} from '../../util';
 import {AbstractBinaryCodegen} from '../AbstractBinaryCodegen';
@@ -88,6 +88,10 @@ export class MsgPackCodegen extends AbstractBinaryCodegen<MsgPackEncoder> {
     }
   }
 
+  protected onKey(path: SchemaPath, r: JsExpression, type: KeyType<any, any>): void {
+    this.onNode([...path, type.key], r, type.val);
+  }
+
   protected genEncoder(type: Type): CompiledBinaryEncoder {
     return MsgPackCodegen.get(type);
   }