feat(compiler): Allow enums and unions in model indexers

Signed-off-by: Naoki Ikeguchi <[email protected]>
microsoft · Jan 12, 2025 · 7abf3ee · 7abf3ee
1 parent 3ac724c
commit 7abf3ee
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/packages/compiler/src/core/types.ts b/packages/compiler/src/core/types.ts
@@ -280,7 +280,7 @@ export type NeverIndexer = {
 };
 
 export type ModelIndexer = {
-  readonly key: Scalar;
+  readonly key: Enum | Scalar | Union;
   readonly value: Type;
 };
 

diff --git a/packages/compiler/src/lib/intrinsic/decorators.ts b/packages/compiler/src/lib/intrinsic/decorators.ts
@@ -1,12 +1,19 @@
 import { DocTarget, setDocData } from "../../core/intrinsic-type-state.js";
 import type { Program } from "../../core/program.js";
-import type { DecoratorContext, ModelIndexer, Scalar, Type } from "../../core/types.js";
+import type {
+  DecoratorContext,
+  Enum,
+  ModelIndexer,
+  Scalar,
+  Type,
+  Union,
+} from "../../core/types.js";
 
 const indexTypeKey = Symbol.for(`TypeSpec.index`);
 export const indexerDecorator = (
   context: DecoratorContext,
   target: Type,
-  key: Scalar,
+  key: Enum | Scalar | Union,
   value: Type,
 ) => {
   const indexer: ModelIndexer = { key, value };