From a20f30b404a5406293fb472393aa44132fa927a5 Mon Sep 17 00:00:00 2001
From: Matej Genci <mg918@ic.ac.uk>
Date: Fri, 13 Mar 2020 16:47:38 +0000
Subject: [PATCH] Function to generate markers to highlight node

---
 src/Comm.js | 22 +++++++++++++---------
 1 file changed, 13 insertions(+), 9 deletions(-)

diff --git a/src/Comm.js b/src/Comm.js
index 4e2f0de..216fdbd 100644
--- a/src/Comm.js
+++ b/src/Comm.js
@@ -9,19 +9,27 @@ export function astMetaToGraphData(astMeta) {
     return graphData
 }
 
-function generateMarkerObject(start, end, off = 0) {
+function generateMarkerObject(start, end, off = 0, style="ast-node-highlight") {
     return {
       startRow: start.lineNum - 1,
       startCol: start.charNum + off,
       endRow: end.lineNum - 1,
       endCol: end.charNum,
-      className: "ast-node-highlight",
+      className: style,
       type: "text"
     }
 }
 
-function generateArmMarkers(lineNums) {
-    return lineNums.map(ln => ({startRow: ln - 1, className: "ast-node-highlight", type: "background"}))
+function generateArmMarkers(lineNums, style = "ast-node-highlight") {
+    return lineNums.map(ln => ({startRow: ln - 1, className: style, type: "background"}))
+}
+
+function highlightNode(node, style = "ast-node-highlight") {
+    return {
+        wacc: [generateMarkerObject(node.waccStart, node.waccEnd, style)],
+        js:   typeof node.jsStart === "undefined" ? [{}] : [generateMarkerObject(node.jsStart, node.jsEnd, -1, style)],
+        arm:  generateArmMarkers(node.armLineNums, style)
+    }
 }
 
 function createGraphNode(node, astMeta){
@@ -41,11 +49,7 @@ function createGraphNode(node, astMeta){
     return {
         name: name,
         attributes: {value: value},
-        highlighting: {
-            wacc:   [generateMarkerObject(node.waccStart, node.waccEnd)],
-            js:     typeof node.jsStart === "undefined" ? [{}] : [generateMarkerObject(node.jsStart, node.jsEnd, -1)],
-            arm:    generateArmMarkers(node.armLineNums)
-        },
+        highlighting: highlightNode(node),
         children: children
     }
 
-- 
GitLab