Adds block kind

flarebyte · Jan 7, 2025 · 330d457 · 330d457
1 parent 783e565
commit 330d457
Showing 1 changed file with 23 additions and 1 deletion.
diff --git a/src/incy-wincy-model.ts b/src/incy-wincy-model.ts
@@ -1,5 +1,5 @@
 import {describeEnum} from 'faora-kai';
-import {z} from 'zod';
+import { z } from 'zod';
 
 const programmingLanguageKey = ['js', 'ts', 'dart', 'python'] as const;
 
@@ -13,3 +13,25 @@ const programmingLanguageLabels = {
 const programmingLanguage = z
   .enum(programmingLanguageKey)
   .describe(describeEnum('Encryption strength:', programmingLanguageLabels));
+
+const blockKindKey = [
+  'class',
+  'interface',
+  'const',
+  'function',
+  'test',
+  'other',
+] as const;
+
+const blockKindlabels = {
+  class: 'Class',
+  interface: 'Interface',
+  const: 'Constant',
+  function: 'Function',
+  test: 'Test',
+  other: 'Other',
+};
+
+const blockKind = z
+  .enum(blockKindKey)
+  .describe(describeEnum('Encryption strength:', blockKindlabels));