closes #95

lib/scripts/mtw.js

@@ -1,6 +1,7 @@
 import { YandexTranslate } from './services/yandexTranslate';
 import { BingTranslate } from './services/bingTranslate';
 import { GoogleTranslate } from './services/googleTranslate';
+import './utils/mutationSummary';
 
 /** Class for content scriptcontroller */
 export class ContentScript {
@@ -22,8 +23,9 @@ export class ContentScript {
   /**
    * Loads data from storage and calls appropriate
    * functions as per the settings.
+   * @param {boolean} dynamicallyLoaded - flag to check if dynamic content
    */
-  translate() {
+  translate(dynamicallyLoaded) {
     chrome.storage.local.get(null, (res) => {
       if (res.activation === true) {
         this.ngramMin = res.ngramMin;
@@ -54,7 +56,7 @@ export class ContentScript {
           console.log('[MTW] No active pattern. Please select a pattern in the options page.');
         } else {
           this.injectCSS(res.translatedWordStyle);
-          var countedWords = this.getAllWords(this.ngramMin, this.ngramMax);
+          var countedWords = this.getAllWords(this.ngramMin, this.ngramMax, dynamicallyLoaded);
           var filteredWords;
           if (this.userDefinedOnly === true) {
             filteredWords = this.filterToUserDefined(countedWords,
@@ -136,21 +138,28 @@ export class ContentScript {
    * Retrieve all the words from current page
    * @param {number} ngramMin - minimum ngram for translation
    * @param {number} ngramMax - maximum ngram for translation
+   * @param {boolean} dynamicallyLoaded - flag to check if dynamic content
    * @returns {Object} countedWords - object with word counts
    */
-  getAllWords(ngramMin, ngramMax) {
-    var countedWords = {},
+  getAllWords(ngramMin, ngramMax, dynamicallyLoaded) {
+    var countedWords = {}, paragraphs;
+    if(dynamicallyLoaded){
+      paragraphs = dynamicallyLoadedPtags;
+    } else {
       paragraphs = document.getElementsByTagName('p');
-    console.log('Getting words from all ' + paragraphs.length + ' paragraphs');
-    for (var i = 0; i < paragraphs.length; i++) {
-      var words = paragraphs[i].innerText.split(/\s|,|[.()]|\d/g);
-      for (var j = 0; j < words.length; j++) {
-        for (var b = ngramMin; b <= ngramMax; b++) {
-          var word = words.slice(j, j + b).join(' ');
-          if (!(word in countedWords)) {
-            countedWords[word] = 0;
+    }
+    if (paragraphs.length > 0) {
+      console.log('Getting words from all ' + paragraphs.length + ' paragraphs');
+      for (var i = 0; i < paragraphs.length; i++) {
+        var words = paragraphs[i].innerText.split(/\s|,|[.()]|\d/g);
+        for (var j = 0; j < words.length; j++) {
+          for (var b = ngramMin; b <= ngramMax; b++) {
+            var word = words.slice(j, j + b).join(' ');
+            if (!(word in countedWords)) {
+              countedWords[word] = 0;
+            }
+            countedWords[word] += 1;
           }
-          countedWords[word] += 1;
         }
       }
     }
@@ -511,7 +520,23 @@ export class ContentScript {
 }
 
 var MTWTranslator = new ContentScript();
-MTWTranslator.translate();
+MTWTranslator.translate(false);
+
+var observer = new MutationSummary({
+  callback: handleDynamicChanges,
+  queries: [{ element: 'p' }]
+});
+
+var dynamicallyLoadedPtags = [];
+function handleDynamicChanges(summaries){
+  var dynamicChanges = summaries[0];
+  dynamicallyLoadedPtags = [];
+  var i=0;
+  dynamicChanges.added.forEach(function(newEl) {
+    dynamicallyLoadedPtags.push(newEl);
+  });
+  MTWTranslator.translate(true);
+}
 
 chrome.runtime.onMessage.addListener(function(request, sender, sendResponse) {
   if (request.type === 'toggleAllElements') {