@@ -176,7 +176,7 @@ export function isDataModel(item: unknown): item is DataModel {
176176}
177177
178178export interface DataModelAttribute extends AstNode {
179- readonly $container : DataModel ;
179+ readonly $container : DataModel | Enum ;
180180 readonly $type : 'DataModelAttribute' ;
181181 args : Array < AttributeArg >
182182 decl : Reference < Attribute >
@@ -204,7 +204,7 @@ export function isDataModelField(item: unknown): item is DataModelField {
204204}
205205
206206export interface DataModelFieldAttribute extends AstNode {
207- readonly $container : DataModelField ;
207+ readonly $container : DataModelField | EnumField ;
208208 readonly $type : 'DataModelFieldAttribute' ;
209209 args : Array < AttributeArg >
210210 decl : Reference < Attribute >
@@ -260,6 +260,8 @@ export function isDataSourceField(item: unknown): item is DataSourceField {
260260export interface Enum extends AstNode {
261261 readonly $container : Model ;
262262 readonly $type : 'Enum' ;
263+ attributes : Array < DataModelAttribute >
264+ comments : Array < string >
263265 fields : Array < EnumField >
264266 name : string
265267}
@@ -273,6 +275,8 @@ export function isEnum(item: unknown): item is Enum {
273275export interface EnumField extends AstNode {
274276 readonly $container : DataModel | Enum | FunctionDecl ;
275277 readonly $type : 'EnumField' ;
278+ attributes : Array < DataModelFieldAttribute >
279+ comments : Array < string >
276280 name : string
277281}
278282
@@ -697,10 +701,21 @@ export class ZModelAstReflection extends AbstractAstReflection {
697701 return {
698702 name : 'Enum' ,
699703 mandatory : [
704+ { name : 'attributes' , type : 'array' } ,
705+ { name : 'comments' , type : 'array' } ,
700706 { name : 'fields' , type : 'array' }
701707 ]
702708 } ;
703709 }
710+ case 'EnumField' : {
711+ return {
712+ name : 'EnumField' ,
713+ mandatory : [
714+ { name : 'attributes' , type : 'array' } ,
715+ { name : 'comments' , type : 'array' }
716+ ]
717+ } ;
718+ }
704719 case 'FunctionDecl' : {
705720 return {
706721 name : 'FunctionDecl' ,
0 commit comments