twikoo-func 1.6.12 → 1.6.13

package/index.js

@@ -8,8 +8,7 @@ const { version: VERSION } = require('./package.json')
 const tcb = require('@cloudbase/node-sdk') // 云开发 SDK
 const {
   $,
-  JSDOM,
-  createDOMPurify,
+  getDomPurify,
   md5,
   xml2js
 } = require('./utils/lib')
@@ -35,6 +34,7 @@ const {
   commentImportValine,
   commentImportDisqus,
   commentImportArtalk,
+  commentImportArtalk2,
   commentImportTwikoo
 } = require('./utils/import')
 const { postCheckSpam } = require('./utils/spam')
@@ -46,10 +46,7 @@ const app = tcb.init({ env: tcb.SYMBOL_CURRENT_ENV })
 const auth = app.auth()
 const db = app.database()
 const _ = db.command
-
-// 初始化反 XSS
-const window = new JSDOM('').window
-const DOMPurify = createDOMPurify(window)
+const DOMPurify = getDomPurify()
 
 // 常量 / constants
 const { RES_CODE, MAX_REQUEST_TIMES } = require('./utils/constants')
@@ -131,6 +128,9 @@ exports.main = async (event, context) => {
       case 'UPLOAD_IMAGE': // >= 1.5.0
         res = await uploadImage(event, config)
         break
+      case 'COMMENT_EXPORT_FOR_ADMIN': // >= 1.6.13
+        res = await commentExportForAdmin(event)
+        break
       default:
         if (event.event) {
           res.code = RES_CODE.EVENT_NOT_EXIST
@@ -430,6 +430,11 @@ async function commentImportForAdmin (event) {
           comments = await commentImportArtalk(artalkDb, log)
           break
         }
+        case 'artalk2': {
+          const artalkDb = await readFile(event.fileId, 'json', log)
+          comments = await commentImportArtalk2(artalkDb, log)
+          break
+        }
         case 'twikoo': {
           const twikooDb = await readFile(event.fileId, 'json', log)
           comments = await commentImportTwikoo(twikooDb, log)
@@ -455,6 +460,23 @@ async function commentImportForAdmin (event) {
   return res
 }
 
+async function commentExportForAdmin (event) {
+  const res = {}
+  const isAdminUser = await isAdmin()
+  if (isAdminUser) {
+    const collection = event.collection || 'comment'
+    const data = await db
+      .collection(collection)
+      .get()
+    res.code = RES_CODE.SUCCESS
+    res.data = data.data
+  } else {
+    res.code = RES_CODE.NEED_LOGIN
+    res.message = '请先登录'
+  }
+  return res
+}
+
 // 读取云存储中的文件并转为 js object
 async function readFile (fileId, type, log) {
   try {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "twikoo-func",
-  "version": "1.6.12",
+  "version": "1.6.13",
   "description": "A simple comment system.",
   "author": "imaegoo <hello@imaegoo.com> (https://github.com/imaegoo)",
   "license": "MIT",

package/utils/import.js

@@ -1,5 +1,5 @@
 const { getRelativeUrl } = require('.')
-const { marked, DOMPurify, md5 } = require('./lib')
+const { marked, getDomPurify, md5 } = require('./lib')
 
 const fn = {
   // 兼容 Leancloud 两种 JSON 导出格式
@@ -135,13 +135,13 @@ const fn = {
       log('Artalk 评论文件格式有误')
       return
     }
+    const DOMPurify = getDomPurify()
     marked.setOptions({
       renderer: new marked.Renderer(),
       gfm: true,
       tables: true,
       breaks: true,
       pedantic: false,
-      sanitize: true,
       smartLists: true,
       smartypants: true
     })
@@ -160,7 +160,7 @@ const fn = {
           pid: comment.rid ? `artalk${comment.rid}` : '',
           rid: comment.rid ? `artalk${comment.rid}` : '',
           master: false,
-          comment: DOMPurify.sanitize(marked(comment.content)),
+          comment: DOMPurify.sanitize(marked.marked(comment.content)),
           url: getRelativeUrl(comment.page_key),
           href: comment.page_key,
           created: new Date(comment.date).getTime(),
@@ -175,6 +175,53 @@ const fn = {
     log(`解析成功 ${comments.length} 条评论`)
     return comments
   },
+  // Artalk v2 导入
+  async commentImportArtalk2 (artalkDb, log) {
+    const comments = []
+    if (!artalkDb || !artalkDb.length) {
+      log('Artalk v2 评论文件格式有误')
+      return
+    }
+    const DOMPurify = getDomPurify()
+    marked.setOptions({
+      renderer: new marked.Renderer(),
+      gfm: true,
+      tables: true,
+      breaks: true,
+      pedantic: false,
+      smartLists: true,
+      smartypants: true
+    })
+    log(`共 ${artalkDb.length} 条评论`)
+    for (const comment of artalkDb) {
+      try {
+        const parsed = {
+          _id: `artalk${comment.id}`,
+          nick: comment.nick,
+          ip: comment.ip,
+          mail: comment.email,
+          mailMd5: md5(comment.email),
+          isSpam: comment.is_pending,
+          ua: comment.ua || '',
+          link: comment.link,
+          pid: comment.rid ? `artalk${comment.rid}` : '',
+          rid: comment.rid ? `artalk${comment.rid}` : '',
+          master: false,
+          comment: DOMPurify.sanitize(marked.marked(comment.content)),
+          url: getRelativeUrl(comment.page_key),
+          href: comment.page_key,
+          created: new Date(comment.created_at).getTime(),
+          updated: Date.now()
+        }
+        comments.push(parsed)
+        log(`${comment.id} 解析成功`)
+      } catch (e) {
+        log(`${comment.id} 解析失败：${e.message}`)
+      }
+    }
+    log(`解析成功 ${comments.length} 条评论`)
+    return comments
+  },
   // Twikoo 导入
   async commentImportTwikoo (twikooDb, log) {
     let arr

package/utils/lib.js

@@ -14,15 +14,21 @@ const pushoo = require('pushoo').default // 即时消息通知
 const tencentcloud = require('tencentcloud-sdk-nodejs') // 腾讯云 API NODEJS SDK
 const xml2js = require('xml2js') // XML 解析
 
+function getDomPurify () {
+  // 初始化反 XSS
+  const window = new JSDOM('').window
+  const DOMPurify = createDOMPurify(window)
+  return DOMPurify
+}
+
 module.exports = {
   $,
   AkismetClient,
   CryptoJS,
   FormData,
-  JSDOM,
   axios,
   bowser,
-  createDOMPurify,
+  getDomPurify,
   ipToRegion,
   marked,
   md5,