1 files changed, 106 insertions, 6 deletions

diff --git a/juick-www/src/main/js/killy/index.js b/juick-www/src/main/js/killy/index.js
index c2ce20fb..634dfc42 100644
--- a/juick-www/src/main/js/killy/index.js
+++ b/juick-www/src/main/js/killy/index.js
@@ -3,12 +3,39 @@ function insertAfter(newNode, referenceNode) {
   referenceNode.parentNode.insertBefore(newNode, referenceNode.nextSibling);
 }
 
-function moveAll(fromNode, toNode) {
-  for (let c; c = fromNode.firstChild; ) { toNode.appendChild(c); }
+function htmlEscape(html) {
+  let textarea = document.createElement('textarea');
+  textarea.textContent = html;
+  return textarea.innerHTML;
 }
 
-function removeAllFrom(fromNode) {
-  for (let c; c = fromNode.lastChild; ) { fromNode.removeChild(c); }
+// rules :: [{pr: number, re: RegExp, with: string}]
+// rules :: [{pr: number, re: RegExp, with: Function}]
+// rules :: [{pr: number, re: RegExp, brackets: true, with: [string, string]}]
+// rules :: [{pr: number, re: RegExp, brackets: true, with: [string, string, Function]}]
+function formatText(txt, rules) {
+  let idCounter = 0;
+  function nextId() { return idCounter++; }
+  function ft(txt, rules) {
+    let matches = rules.map(r => { r.re.lastIndex = 0; return [r, r.re.exec(txt)]; })
+                       .filter(([,m]) => m !== null)
+                       .sort(([r1,m1],[r2,m2]) => (r1.pr - r2.pr) || (m1.index - m2.index));
+    if (matches && matches.length > 0) {
+      let [rule, match] = matches[0];
+      let idStr = `<>(${nextId()})<>`;
+      let outerStr = txt.substring(0, match.index) + idStr + txt.substring(rule.re.lastIndex);
+      let innerStr = (rule.brackets)
+        ? (() => { let [l ,r ,f] = rule.with; return l + ft((f ? f(match[1]) : match[1]), rules) + r; })()
+        : match[0].replace(rule.re, rule.with);
+      return ft(outerStr, rules).replace(idStr, innerStr);
+    }
+    return txt;
+  }
+  return ft(htmlEscape(txt), rules); // idStr above relies on the fact the text is escaped
+}
+
+function fixWwwLink(url) {
+  return url.replace(/^(?!([a-z]+:)?\/\/)/i, '//');
 }
 
 function makeIframe(src, w, h, scrolling='no') {
@@ -45,6 +72,62 @@ function makeResizable(element, calcHeight) {
   setHeight(element);
 }
 
+function extractDomain(url) {
+  const domainRe = /^(?:https?:\/\/)?(?:[^@\/\n]+@)?(?:www\.)?([^:\/\n]+)/i;
+  return domainRe.exec(url)[1];
+}
+
+function urlReplace(match, p1, p2, p3) {
+  let isBrackets = (p1 !== undefined);
+  return (isBrackets)
+    ? `<a href="${fixWwwLink(p2 || p3)}">${p1}</a>`
+    : `<a href="${fixWwwLink(match)}">${extractDomain(match)}</a>`;
+}
+
+function urlReplaceInCode(match, p1, p2, p3) {
+  let isBrackets = (p1 !== undefined);
+  return (isBrackets)
+    ? `<a href="${fixWwwLink(p2 || p3)}">${match}</a>`
+    : `<a href="${fixWwwLink(match)}">${match}</a>`;
+}
+
+function messageReplyReplace(messageId) {
+  return function(match, mid, rid) {
+    let replyPart = (rid && rid != '0') ? '#' + rid : '';
+    return `<a href="//${window.location.hostname}/${mid || messageId}${replyPart}">${match}</a>`;
+  };
+}
+
+/**
+ * Given "txt" message in unescaped plaintext with Juick markup, this function
+ * returns escaped formatted HTML string.
+ *
+ * @param {string} txt
+ * @param {string} messageId - current message id
+ * @param {boolean} isCode
+ * @returns {string}
+ */
+function juickFormat(txt, messageId, isCode) {
+  const urlRe = /(?:\[([^\]\[]+)\](?:\[([^\]]+)\]|\(((?:[a-z]+:\/\/|www\.|ftp\.)(?:\([-\w+*&@#/%=~|$?!:;,.]*\)|[-\w+*&@#/%=~|$?!:;,.])*(?:\([-\w+*&@#/%=~|$?!:;,.]*\)|[\w+*&@#/%=~|$]))\))|\b(?:[a-z]+:\/\/|www\.|ftp\.)(?:\([-\w+*&@#/%=~|$?!:;,.]*\)|[-\w+*&@#/%=~|$?!:;,.])*(?:\([-\w+*&@#/%=~|$?!:;,.]*\)|[\w+*&@#/%=~|$]))/gi;
+  const bqReplace = m => m.replace(/^(?:>|&gt;)\s?/gmi, '');
+  return (isCode)
+    ? formatText(txt, [
+      { pr: 1, re: urlRe, with: urlReplaceInCode },
+      { pr: 1, re: /\B(?:#(\d+))?(?:\/(\d+))?\b/g, with: messageReplyReplace(messageId) },
+      { pr: 1, re: /\B@([\w-]+)\b/gi, with: '<a href="//' + window.location.hostname + '/$1">@$1</a>' },
+    ])
+    : formatText(txt, [
+      { pr: 0, re: /((?:^(?:>|&gt;)\s?[\s\S]+?$\n?)+)/gmi, brackets: true, with: ['<q>', '</q>', bqReplace] },
+      { pr: 1, re: urlRe, with: urlReplace },
+      { pr: 1, re: /\B(?:#(\d+))?(?:\/(\d+))?\b/g, with: messageReplyReplace(messageId) },
+      { pr: 1, re: /\B@([\w-]+)\b/gi, with: '<a href="//' + window.location.hostname + '/$1">@$1</a>' },
+      { pr: 2, re: /\B\*([^\*\n<>]+)\*((?=\s)|(?=$)|(?=[!\"#$%&'*+,\-./:;<=>?@[\]^_`{|}~()]+))/g, brackets: true, with: ['<b>', '</b>'] },
+      { pr: 2, re: /\B\/([^\/\n<>]+)\/((?=\s)|(?=$)|(?=[!\"#$%&'*+,\-./:;<=>?@[\]^_`{|}~()]+))/g, brackets: true, with: ['<i>', '</i>'] },
+      { pr: 2, re: /\b\_([^\_\n<>]+)\_((?=\s)|(?=$)|(?=[!\"#$%&'*+,\-./:;<=>?@[\]^_`{|}~()]+))/g, brackets: true, with: ['<span class="u">', '</span>'] },
+      { pr: 3, re: /\n/g, with: '<br/>' },
+    ]);
+}
+
 function getEmbeddableLinkTypes() {
   return [
     {
@@ -184,6 +267,16 @@ function embedLinks(aNodes, container) {
   return anyEmbed;
 }
 
+/**
+ * Embed all the links inside element "x" that match to "allLinksSelector".
+ * All the embedded media is placed inside "div.embedContainer".
+ * "div.embedContainer" is inserted before an element matched by "beforeNodeSelector"
+ * if not present. Existing container is used otherwise.
+ *
+ * @param {Element} x
+ * @param {string} beforeNodeSelector
+ * @param {string} allLinksSelector
+ */
 function embedLinksToX(x, beforeNodeSelector, allLinksSelector) {
   let isCtsPost = false;
   let allLinks = x.querySelectorAll(allLinksSelector);
@@ -219,10 +312,17 @@ function embedLinksToPost() {
   });
 }
 
-exports.embed = function() {
+/**
+ * Embed all the links in all messages/replies on the page.
+ */
+function embedAll() {
   if (document.querySelectorAll('#content > article').length) {
     embedLinksToArticles();
   } else {
     embedLinksToPost();
   }
-};
+}
+
+exports.embedAll = embedAll;
+exports.embedLinksToX = embedLinksToX;
+exports.format = juickFormat;