Refactor code and remove unnecessary scripts

background/index.js

@@ -38,16 +38,12 @@ importScripts('/background/icon.js')
 
   var mdWise = (tab) => {
     inject(tab.id);
-    chrome.scripting.executeScript({
-      target: { tabId: tab.id },
-      files: ["/content/mdwise.js", "/content/index.js"]
-    })
   }
 
   chrome.runtime.onInstalled.addListener(() => {
     chrome.contextMenus.create({
       id: "md-wise",
-      title: "Too Long Read It Short",
+      title: "Assist Me To Read",
       contexts: ["page"]
     });
   });

background/inject.js

@@ -18,9 +18,6 @@ md.inject = ({ storage: { state } }) => (id) => {
       pre = document.querySelector('pre');
       if (pre) {
         pre.style.visibility = 'hidden'
-        return
-      } else {
-        console.warn('No <pre> element found');
       }
     },
     injectImmediately: true
@@ -44,6 +41,7 @@ md.inject = ({ storage: { state } }) => (id) => {
       state.content.emoji && '/content/emoji.js',
       state.content.mermaid && ['/vendor/mermaid.min.js', '/vendor/panzoom.min.js', '/content/mermaid.js'],
       state.content.mathjax && ['/content/mathjax.js', '/vendor/mathjax/tex-mml-chtml.js'],
+      '/content/mdwise.js',
       '/content/index.js',
       '/content/scroll.js',
       state.content.autoreload && '/content/autoreload.js',

background/messages.js

@@ -53,12 +53,10 @@ md.messages = ({ storage: { defaults, state, set }, compilers, mathjax, xhr, web
         // files: ['/vendor/readability.min.js', '/vendor/turndown.min.js'],
         func: () => {
           try {
-            document.readyState = 'loading';
             // let content = new Readability(document).parse().content;
             let content = document.body;
             content = new TurndownService().turndown(content);
             document.body.innerHTML = `<pre>${content}</pre>`;
-            document.readyState = 'complete';
             return { content };
           } catch (err) {
             console.error('Readability error:', err);

content/index.js

@@ -169,6 +169,7 @@ function mount() {
     $('pre').style.display = 'none'
     var md = $('pre').innerText
   } catch (e) {
+    console.error('No <pre> element found')
     state.html = $('body').innerHTML
     var md = ''
   }
@@ -298,17 +299,15 @@ if (!document.querySelector('pre')) {
   // clean the HTML content
   cleanHtml();
   // convert the HTML content to markdown
-  chrome.runtime.sendMessage({ message: 'to-markdown' });
-}
-
-if (document.readyState === 'complete') {
-  mount()
-}
-else {
+  chrome.runtime.sendMessage(
+    { message: 'to-markdown' },
+    (_) => mount()
+  );
+} else {
   var timeout = setInterval(() => {
     if (document.readyState === 'complete') {
       clearInterval(timeout)
       mount()
     }
-  }, 0)
+  })
 }

content/mdwise.js

@@ -1,6 +1,12 @@
 function cleanHtml() {
-  window.document.querySelectorAll(
-    'link, style, script, meta, noscript, header, nav, span, footer, div[role="navigation"], figure, table'
+  if (articles = document.querySelectorAll('article')) {
+    document.body.innerHTML = '';
+    articles.forEach(e => document.body.appendChild(e));
+  } else if (main = document.querySelector('main')) {
+    document.body.innerHTML = main.innerHTML;
+  }
+  document.querySelectorAll(
+    'link, style, script, meta, noscript, header, nav, span, footer, figure, table'
   ).forEach(e => e.remove());
 }