Fix comment parsing for union variants

elm-tooling · Mar 6, 2020 · 8db2ef6 · 8db2ef6
1 parent a5bc8cf
commit 8db2ef6
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/util/hintHelper.ts b/src/util/hintHelper.ts
@@ -63,6 +63,9 @@ export class HintHelper {
       if (declaration.type === "type_declaration") {
         code = declaration.text;
       }
+      if (declaration.type === "union_variant") {
+        declaration = declaration.parent ? declaration.parent : declaration;
+      }
       if (declaration.previousNamedSibling) {
         if (declaration.previousNamedSibling.type === "type_annotation") {
           annotation = declaration.previousNamedSibling.text;