Emit variants as enumMember.

Author: cristianoc

analysis/src/SemanticTokens.ml

@@ -160,6 +160,12 @@ let emitRecordLabel ~(label : Longident.t Location.loc) ~debug emitter =
        ~pos:(Utils.tupleOfLexing label.loc.loc_start)
        ~lid:label.txt ~debug
 
+let emitVariant ~(name : Longident.t Location.loc) ~debug emitter =
+  emitter
+  |> emitLongident ~upperCaseToken:Token.EnumMember
+       ~pos:(Utils.tupleOfLexing name.loc.loc_start)
+       ~lid:name.txt ~debug
+
 let parser ~debug ~emitter ~path =
   let processTypeArg (coreType : Parsetree.core_type) =
     if debug then Printf.printf "TypeArg: %s\n" (locToString coreType.ptyp_loc)
@@ -189,6 +195,9 @@ let parser ~debug ~emitter ~path =
       cases
       |> List.iter (fun (label, _) -> emitter |> emitRecordLabel ~label ~debug);
       Ast_mapper.default_mapper.pat mapper p
+    | Ppat_construct (name, _) ->
+      emitter |> emitVariant ~name ~debug;
+      Ast_mapper.default_mapper.pat mapper p
     | _ -> Ast_mapper.default_mapper.pat mapper p
   in
   let expr (mapper : Ast_mapper.mapper) (e : Parsetree.expression) =
@@ -241,6 +250,9 @@ let parser ~debug ~emitter ~path =
     | Pexp_field (_, label) | Pexp_setfield (_, label, _) ->
       emitter |> emitRecordLabel ~label ~debug;
       Ast_mapper.default_mapper.expr mapper e
+    | Pexp_construct (name, _) ->
+      emitter |> emitVariant ~name ~debug;
+      Ast_mapper.default_mapper.expr mapper e
     | _ -> Ast_mapper.default_mapper.expr mapper e
   in
   let module_expr (mapper : Ast_mapper.mapper) (me : Parsetree.module_expr) =
@@ -301,10 +313,19 @@ let parser ~debug ~emitter ~path =
          ~debug;
     Ast_mapper.default_mapper.label_declaration mapper ld
   in
+  let constructor_declaration (mapper : Ast_mapper.mapper)
+      (cd : Parsetree.constructor_declaration) =
+    emitter
+    |> emitVariant
+         ~name:{loc = cd.pcd_name.loc; txt = Longident.Lident cd.pcd_name.txt}
+         ~debug;
+    Ast_mapper.default_mapper.constructor_declaration mapper cd
+  in
 
   let mapper =
     {
       Ast_mapper.default_mapper with
+      constructor_declaration;
       expr;
       label_declaration;
       module_declaration;

analysis/tests/src/expected/Parser.res.txt

@@ -60,6 +60,9 @@ Lident: someOtherField (60,4) Property
 Type: string (60,20)->(60,26)
 Lident: theParam (61,4) Property
 Type: someEnum (64,7)->(64,15)
+Lident: A (64,18) EnumMember
+Lident: B (64,22) EnumMember
+Lident: C (64,26) EnumMember
 Ldot: T (67,17) Namespace
 Lident: someField (67,19) Property
 Lident: x (67,15) Variable