Fix and refactor extracting content using metaMarked directly might lead in invalid object

1 files changed, 14 insertions, 45 deletions

diff --git a/lib/response.js b/lib/response.js
index 54e2a337..a0dc8b1f 100755
--- a/lib/response.js
+++ b/lib/response.js
@@ -6,7 +6,6 @@ var markdownpdf = require("markdown-pdf");
 var LZString = require('lz-string');
 var S = require('string');
 var shortId = require('shortid');
-var metaMarked = require('meta-marked');
 var querystring = require('querystring');
 var request = require('request');
 var moment = require('moment');
@@ -76,13 +75,8 @@ function showIndex(req, res, next) {
 
 function responseHackMD(res, note) {
     var body = note.content;
-    var meta = null;
-    try {
-        meta = models.Note.parseMeta(metaMarked(body).meta);
-    } catch(err) {
-        //na
-    }
-    if (!meta) meta = {};
+    var extracted = models.Note.extractMeta(body);
+    var meta = models.Note.parseMeta(extracted.meta);
     var title = models.Note.decodeTitle(note.title);
     title = models.Note.generateWebTitle(meta.title || title);
     res.set({
@@ -192,16 +186,9 @@ function showPublishNote(req, res, next) {
                 return response.errorNotFound(res);
             }
             var body = note.content;
-            var meta = null;
-            var markdown = null;
-            try {
-                var obj = metaMarked(body);
-                markdown = obj.markdown;
-                meta = models.Note.parseMeta(obj.meta);
-            } catch(err) {
-                //na
-            }
-            if (!meta) meta = {};
+            var extracted = models.Note.extractMeta(body);
+            markdown = extracted.markdown;
+            meta = models.Note.parseMeta(extracted.meta);
             var createtime = note.createdAt;
             var updatetime = note.lastchangeAt;
             var title = models.Note.decodeTitle(note.title);
@@ -266,16 +253,9 @@ function actionDownload(req, res, note) {
 
 function actionInfo(req, res, note) {
     var body = note.content;
-    var meta = null;
-    var markdown = null;
-    try {
-        var obj = metaMarked(body);
-        markdown = obj.markdown;
-        meta = models.Note.parseMeta(obj.meta);
-    } catch(err) {
-        //na
-    }
-    if (!meta) meta = {};
+    var extracted = models.Note.extractMeta(body);
+    var markdown = extracted.markdown;
+    var meta = models.Note.parseMeta(extracted.meta);
     var createtime = note.createdAt;
     var updatetime = note.lastchangeAt;
     var title = models.Note.decodeTitle(note.title);
@@ -298,18 +278,14 @@ function actionInfo(req, res, note) {
 
 function actionPDF(req, res, note) {
     var body = note.content;
-    try {
-        body = metaMarked(body).markdown;
-    } catch(err) {
-        //na
-    }
+    var extracted = models.Note.extractMeta(body);
     var title = models.Note.decodeTitle(note.title);
 
     if (!fs.existsSync(config.tmppath)) {
         fs.mkdirSync(config.tmppath);
     }
     var path = config.tmppath + '/' + Date.now() + '.pdf';
-    markdownpdf().from.string(body).to(path, function () {
+    markdownpdf().from.string(extracted.markdown).to(path, function () {
         var stream = fs.createReadStream(path);
         var filename = title;
         // Be careful of special characters
@@ -580,16 +556,9 @@ function showPublishSlide(req, res, next) {
                 return response.errorNotFound(res);
             }
             var body = note.content;
-            var meta = null;
-            var markdown = null;
-            try {
-                var obj = metaMarked(body);
-                markdown = obj.markdown;
-                meta = models.Note.parseMeta(obj.meta);
-            } catch(err) {
-                //na
-            }
-            if (!meta) meta = {};
+            var extracted = models.Note.extractMeta(body);
+            markdown = extracted.markdown;
+            meta = models.Note.parseMeta(extracted.meta);
             var createtime = note.createdAt;
             var updatetime = note.lastchangeAt;
             var title = models.Note.decodeTitle(note.title);
@@ -603,7 +572,7 @@ function showPublishSlide(req, res, next) {
                 updatetime: updatetime,
                 url: origin,
                 body: markdown,
-                meta: JSON.stringify(obj.meta || {}),
+                meta: JSON.stringify(extracted.meta),
                 useCDN: config.usecdn,
                 owner: note.owner ? note.owner.id : null,
                 ownerprofile: note.owner ? models.User.getProfile(note.owner) : null,