From 281bee231972c8127bcf5117c25298aff6f60db8 Mon Sep 17 00:00:00 2001
From: yann300 <yann.levreau@gmail.com>
Date: Sun, 24 Sep 2017 17:02:32 +0200
Subject: [PATCH] contextuallistener

---
 src/app/editor/contextualListener.js | 84 ++++++++++++++++++++++++++++
 1 file changed, 84 insertions(+)
 create mode 100644 src/app/editor/contextualListener.js

diff --git a/src/app/editor/contextualListener.js b/src/app/editor/contextualListener.js
new file mode 100644
index 0000000000..1f6bee23f9
--- /dev/null
+++ b/src/app/editor/contextualListener.js
@@ -0,0 +1,84 @@
+'use strict'
+var SourceMappingDecoder = require('ethereum-remix').util.SourceMappingDecoder
+var AstWalker = require('ethereum-remix').util.AstWalker
+
+class ContextualListener {
+  constructor (api, events) {
+    this._api = api
+    this._index = {
+      FunctionDefinition: {},
+      FunctionCalls: {},
+      FunctionCall: {}
+    }
+    this._events = []
+
+    events.compiler.register('compilationFinished', (success, data, source) => {
+      this._stopWarning()
+      if (success) {
+        this._buildIndex(data, source)
+      } else {
+        this._index = {
+          FunctionDefinition: {},
+          FunctionCalls: {},
+          FunctionCall: {}
+        }
+      }
+    })
+    this.sourceMappingDecoder = new SourceMappingDecoder()
+    this.astWalker = new AstWalker()
+    setInterval(() => {
+      this._context(api.getCursorPosition(), api.getCompilationResult())
+    }, 1000)
+  }
+
+  _context (cursorPosition, compilationResult) {
+    if (this.currentPosition === cursorPosition) return
+    this._stopWarning()
+    this.currentPosition = cursorPosition
+    if (compilationResult && compilationResult.data && compilationResult.source) {
+      var nodes = this.sourceMappingDecoder.nodesAtCursorPosition(null, cursorPosition, compilationResult.data.sources[compilationResult.source.target])
+      if (nodes && nodes['FunctionCall']) {
+        this._highlightFunctionCall(nodes['FunctionCall'], compilationResult)
+      }
+    }
+  }
+
+  _buildIndex (compilationResult, source) {
+    if (compilationResult && compilationResult.sources) {
+      var self = this
+      var callback = {}
+      callback['*'] = function (node) {
+        if (node && node.name && self._index[node.name]) {
+          self._index[node.name][node.id] = node
+          if (node.name === 'FunctionCall' && node.children[0] && node.children[0].attributes) {
+            var declaration = node.children[0].attributes.referencedDeclaration
+            if (!self._index['FunctionCalls'][declaration]) self._index['FunctionCalls'][declaration] = [] 
+            self._index['FunctionCalls'][declaration].push(node.id)
+          }
+        }
+        return true
+      }
+      this.astWalker.walk(compilationResult.sources[source.target].AST, callback)
+    }
+  }
+
+  _highlightFunctionCall (node, compilationResult) {
+    if (node.name === 'FunctionCall' && node.children[0] && node.children[0].attributes) {
+      var calls = this._index['FunctionCalls'][node.children[0].attributes.referencedDeclaration]
+      for (var call in calls) {
+        var position = this.sourceMappingDecoder.decode(this._index['FunctionCall'][calls[call]].src)
+        var eventId = this._api.warnFoundCall(position)
+        this._events.push({ eventId, position, fileTarget: compilationResult.source.target })
+      }
+    }
+  }
+
+  _stopWarning () {
+    for (var event in this._events) {
+      this._api.stopFoundCall(this._events[event])
+    }
+    this._events = []
+  }
+}
+
+module.exports = ContextualListener