Fix type alias highlighting in promptfiddle

BoundaryML · Jan 17, 2025 · 23e1015 · 23e1015
1 parent 243c834
commit 23e1015
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 5 deletions.
diff --git a/typescript/codemirror-lang-baml/src/index.ts b/typescript/codemirror-lang-baml/src/index.ts
@@ -51,6 +51,9 @@ export const BAMLLanguage = LRLanguage.define({
         ClassDecl: t.keyword,
         'ClassDecl/IdentifierDecl': t.typeName,
 
+        TypeAliasDecl: t.keyword,
+        'TypeAliasDecl/IdentifierDecl': t.typeName,
+
         ClientDecl: t.keyword,
         'ClientDecl/IdentifierDecl': t.typeName,
 
@@ -122,6 +125,7 @@ const exampleCompletion = BAMLLanguage.data.of({
     snippetCompletion('template_string MyString(${arg1}: string) #"\n  A jinja string\n"#', {
       label: 'template_string',
     }),
+    snippetCompletion('type ${AliasName} = ${TypeExpr};', { label: 'type alias' }),
   ],
 })
 

diff --git a/typescript/codemirror-lang-baml/src/syntax.grammar b/typescript/codemirror-lang-baml/src/syntax.grammar
@@ -1,12 +1,17 @@
 @top Baml { Decl* }
 
-Decl { ClassDecl | EnumDecl | FunctionDecl | ClientDecl | TemplateStringDecl | TestDecl }
+Decl { ClassDecl | EnumDecl | FunctionDecl | ClientDecl | TemplateStringDecl | TestDecl | TypeAliasDecl }
 
 ClassDecl { "class" IdentifierDecl "{" (BlockAttribute | ClassField)* "}" }
 
 ClassField {
   IdentifierDecl TypeExpr FieldAttribute*
 }
+
+TypeAliasDecl {
+  "type" IdentifierDecl "=" TypeExpr
+}
+
 TypeExpr { ComplexTypeExpr | UnionTypeExpr }
 ComplexTypeExpr {
   "(" UnionTypeExpr ")" |
@@ -51,7 +56,7 @@ ClientDecl {
 }
 
 TestDecl {
-  "test" IdentifierDecl "{" (BlockAttribute | TupleValue)* "}" 
+  "test" IdentifierDecl "{" (BlockAttribute | TupleValue)* "}"
 }
 
 LiteralDecl { NumericLiteral | QuotedString | UnquotedString }
@@ -87,15 +92,15 @@ TemplateStringDecl {
   whitespace { $[ \n\r\t] }
 
   UnquotedStringChar { ![#@{}()[\]<>,'\n\r\t/ ]}
-  
+
   TrailingComment { "//" ![\n]* }
   MultilineComment { "{//"![]*"//}" }
 
   "{" "}" "[" "]"
 }
 
 @skip {
-  whitespace | MultilineComment | TrailingComment 
+  whitespace | MultilineComment | TrailingComment
 }
 
 @skip {} {
@@ -108,4 +113,4 @@ TemplateStringDecl {
 
 }
 
-@detectDelim
+@detectDelim