From e6ad51c68282bee30fe50998d48bf2baf85830ff Mon Sep 17 00:00:00 2001
From: James Westman <james@jwestman.net>
Date: Fri, 4 Mar 2022 15:05:02 -0600
Subject: [PATCH] Show docs for flag values

---
 blueprintcompiler/language/values.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/blueprintcompiler/language/values.py b/blueprintcompiler/language/values.py
index 7430991..ce5c406 100644
--- a/blueprintcompiler/language/values.py
+++ b/blueprintcompiler/language/values.py
@@ -109,6 +109,14 @@ class LiteralValue(Value):
 class Flag(AstNode):
     grammar = UseIdent("value")
 
+    @docs()
+    def docs(self):
+        type = self.parent.parent.value_type
+        if member := type.members.get(self.tokens["value"]):
+            return member.doc
+        else:
+            return type.doc
+
     @validate()
     def validate_for_type(self):
         type = self.parent.parent.value_type
@@ -175,7 +183,7 @@ class IdentValue(Value):
     @docs()
     def docs(self):
         type = self.parent.value_type
-        if isinstance(type, gir.Enumeration):
+        if isinstance(type, gir.Enumeration) or isinstance(type, gir.Bitfield):
             if member := type.members.get(self.tokens["value"]):
                 return member.doc
             else: