refactor(code_tags): make codeclimate happy

Reduce duplicated code and lines

Author: SukkaW

lib/plugins/tag/code.js

@@ -33,16 +33,25 @@ const rWrap = /\s*wrap:(\w+)/i;
  * @returns {String} Code snippet with code highlighting
  */
 
-function getHighlightOptions(config, arg) {
-  let lang = '';
+function parseArgs(arg) {
+  let _highlightEnable = true;
+  if (rHighlight.test(arg)) {
+    arg = arg.replace(rHighlight, (match, _enable) => {
+      _highlightEnable = _enable === 'true';
+      return '';
+    });
+  }
+
+  // Here we parse the args
+  let lang, line_number, wrap;
+
   if (rLang.test(arg)) {
     arg = arg.replace(rLang, (match, _lang) => {
       lang = _lang;
       return '';
     });
   }
 
-  let line_number = config.line_number;
   if (rLineNumber.test(arg)) {
     arg = arg.replace(rLineNumber, (match, _line_number) => {
       line_number = _line_number === 'true';
@@ -58,7 +67,6 @@ function getHighlightOptions(config, arg) {
     });
   }
 
-  let wrap = config.wrap;
   if (rWrap.test(arg)) {
     arg = arg.replace(rWrap, (match, _wrap) => {
       wrap = _wrap === 'true';
@@ -107,109 +115,58 @@ function getHighlightOptions(config, arg) {
   }
 
   return {
+    _highlightEnable,
     lang,
-    firstLine: first_line,
+    first_line,
     caption,
-    gutter: line_number,
-    hljs: config.hljs,
+    line_number,
     mark,
-    tab: config.tab_replace,
-    autoDetect: config.auto_detect,
     wrap
   };
 }
 
-function getPrismjsOptions(config, arg) {
-  let lang;
-  if (rLang.test(arg)) {
-    arg = arg.replace(rLang, (match, _lang) => {
-      lang = _lang;
-      return '';
-    });
-  }
+module.exports = ctx => function codeTag(args, content) {
+  const arg = args.join(' ');
+  const hljsCfg = ctx.config.highlight || {};
+  const prismjsCfg = ctx.config.prismjs || {};
 
-  let line_number = config.line_number;
-  if (rLineNumber.test(arg)) {
-    arg = arg.replace(rLineNumber, (match, _line_number) => {
-      line_number = _line_number === 'true';
-      return '';
-    });
-  }
+  const { _highlightEnable, lang, first_line, caption, line_number, mark, wrap } = parseArgs(arg);
 
-  let first_line = 1;
-  if (rFirstLine.test(arg)) {
-    arg = arg.replace(rFirstLine, (match, _first_line) => {
-      first_line = _first_line;
-      return '';
-    });
+  /*
+   * If highlight is not enabled inside code tag,
+   * or neither hljs nor prism is enabled,
+   * then return un-highlighted code directly.
+   */
+  if (!_highlightEnable || (!hljsCfg.enable && !prismjsCfg.enable)) {
+    return `<pre><code>${escapeHTML(content)}</code></pre>`;
   }
 
-  if (rWrap.test(arg)) arg = arg.replace(rWrap, '');
-
-  let mark = [];
-  if (rMark.test(arg)) {
-    arg = arg.replace(rMark, (match, _mark) => {
-      mark = _mark.split(',').reduce((prev, cur) => {
-        if (/-/.test(cur)) {
-          let a = Number(cur.substr(0, cur.indexOf('-')));
-          let b = Number(cur.substr(cur.indexOf('-') + 1));
-          if (b < a) { // switch a & b
-            const temp = a;
-            a = b;
-            b = temp;
-          }
-
-          for (; a <= b; a++) {
-            prev.push(a);
-          }
-
-          return prev;
-        }
-
-        prev.push(Number(cur));
-        return prev;
-      }, []);
-
-      return '';
-    });
-  }
+  const hljsOption = {
+    lang: typeof lang !== 'undefined' ? lang : '',
+    firstLine: first_line,
+    caption,
+    gutter: typeof line_number !== 'undefined' ? line_number : hljsCfg.line_number,
+    hljs: hljsCfg.hljs,
+    mark,
+    tab: hljsCfg.tab_replace,
+    autoDetect: hljsCfg.auto_detect,
+    wrap: typeof wrap === 'undefined' ? wrap : hljsCfg.wrap
+  };
 
-  return {
+  const prismjsOption = {
     lang,
     firstLine: first_line,
-    lineNumber: line_number,
+    lineNumber: typeof line_number !== 'undefined' ? line_number : prismjsCfg.line_number,
     mark,
-    tab: config.tab_replace,
-    isPreprocess: config.preprocess
+    tab: prismjsCfg.tab_replace,
+    isPreprocess: prismjsCfg.preprocess
   };
-}
-
-module.exports = ctx => function codeTag(args, content) {
-  let arg = args.join(' ');
-  const hljsCfg = ctx.config.highlight || {};
-  const prismjsCfg = ctx.config.prismjs || {};
-
-  let _hljsEnable = hljsCfg.enable;
-
-  if (rHighlight.test(arg)) {
-    arg = arg.replace(rHighlight, (match, _enable) => {
-      _hljsEnable = _enable === 'true';
-      return '';
-    });
-  }
-
-  if (!_hljsEnable && !prismjsCfg.enable) {
-    content = escapeHTML(content);
-    return `<pre><code>${content}</code></pre>`;
-  }
 
   if (prismjsCfg.enable) {
-    content = prismHighlight(content, getPrismjsOptions(prismjsCfg, arg));
+    content = prismHighlight(content, prismjsOption);
   } else {
-    content = highlight(content, getHighlightOptions(hljsCfg, arg));
+    content = highlight(content, hljsOption);
   }
 
-  content = content.replace(/{/g, '&#123;').replace(/}/g, '&#125;');
-
-  return content;
+  return content.replace(/{/g, '&#123;').replace(/}/g, '&#125;');
 };