Merge pull request #122 from solidity-parser/custom-storage-layout

Add support for custom storage layout

Author: Janther

antlr

@@ -54,6 +54,11 @@ export class ASTBuilder
 
     this._currentContract = name
 
+    const customLayoutStorageList = ctx.customStorageLayout_list()
+    if (customLayoutStorageList.length > 1) {
+      throw new Error('Only one custom storage layout is allowed per contract')
+    }
+
     const node: AST.ContractDefinition = {
       type: 'ContractDefinition',
       name,
@@ -64,6 +69,12 @@ export class ASTBuilder
       kind,
     }
 
+    if (customLayoutStorageList.length === 1) {
+      node.storageLayout = this.visitExpression(
+        customLayoutStorageList[0].expression()
+      )
+    }
+
     return this._addMeta(node, ctx)
   }