add support for documentation of global variables

OpenZeppelin · Dec 2, 2021 · c3b7d8d · c3b7d8d
1 parent ef47ca5
commit c3b7d8d
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/src/solc.ts b/src/solc.ts
@@ -52,6 +52,7 @@ export namespace ast {
     nodeType: 'VariableDeclaration';
     visibility: 'internal' | 'public' | 'private';
     name: string;
+    documentation: string | null;
     constant: boolean;
     typeName: TypeName;
   }

diff --git a/src/source.ts b/src/source.ts
@@ -364,9 +364,13 @@ class SourceStateVariable implements Linkable {
     return `${this.type} ${this.name}`;
   }
 
-  get natspec(): {} {
-    warnStateVariableNatspec();
-    return {}
+  @memoize
+  get natspec(): NatSpec {
+    if (this.astNode.documentation === null || this.astNode.documentation === undefined) {
+      return {};
+    }
+
+    return parseNatSpec(this.astNode.documentation, this);
   }
 }
 
@@ -554,7 +558,7 @@ interface NatSpec {
   };
 }
 
-function parseNatSpec(doc: string, context: SourceFunctionLike | SourceContract): NatSpec {
+function parseNatSpec(doc: string, context: SourceFunctionLike | SourceContract | SourceStateVariable): NatSpec {
   const res: NatSpec = {};
 
   const tagMatches = execall(/^(?:@(\w+|custom:[a-z][a-z-]*) )?((?:(?!^@(?:\w+|custom:[a-z][a-z-]*) )[^])*)/m, doc);