From ecd555976de8b3a78c985a3732eb2d30780e81c3 Mon Sep 17 00:00:00 2001 From: chenruilong Date: Thu, 20 Apr 2023 14:36:34 +0800 Subject: [PATCH] add cms-article schema.ext.js --- collection/uni-cms-articles/schema.ext.js | 243 ++++++++++++++++++++++ 1 file changed, 243 insertions(+) create mode 100644 collection/uni-cms-articles/schema.ext.js diff --git a/collection/uni-cms-articles/schema.ext.js b/collection/uni-cms-articles/schema.ext.js new file mode 100644 index 0000000..141ed56 --- /dev/null +++ b/collection/uni-cms-articles/schema.ext.js @@ -0,0 +1,243 @@ +// 获取配置 +const createConfig = safeRequire('uni-config-center') +const config = createConfig({ + pluginId: 'uni-cms' +}).config() + +// 获取数据库实例 +const db = uniCloud.database() + +// 文章数据库名称 +const articleDBName = 'uni-cms-articles' + +// 解锁内容数据库名称 +const unlockContentDBName = 'uni-cms-unlock-record' + +// 安全检测文本内容 +async function checkContentSec(content, requestId, errorMsg) { + // 安全引入内容安全检测模块 + const UniSecCheck = safeRequire('uni-sec-check') + // 创建内容安全检测实例 + const uniSecCheck = new UniSecCheck({ + provider: 'mp-weixin', + requestId + }) + // 调用文本安全检测接口 + const res = await uniSecCheck.textSecCheck({ + content, // 待检测的文本内容 + scene: 1, // 表示资料类场景 + version: 1 // 调用检测API的版本号 + }) + + // 如果存在敏感词，抛出异常 + if (res.errCode === uniSecCheck.ErrorCode.RISK_CONTENT) { + throw new Error(errorMsg || '存在敏感词，请修改后提交') + } else if (res.errCode !== 0) { + console.error(res) + throw new Error('内容安全检测异常：' + res.errCode) + } +} + +// 安全检测图片内容 +async function checkImageSec(image, requestId, errorMsg) { + // 安全引入内容安全检测模块 + const UniSecCheck = safeRequire('uni-sec-check') + // 创建内容安全检测实例 + const uniSecCheck = new UniSecCheck({ + provider: 'mp-weixin', + requestId + }) + + // 调用图片安全检测接口 + const res = await uniSecCheck.imgSecCheck({ + image, // 待检测的图片URL + scene: 1, // 表示资料类场景 + version: 1 // 调用检测API的版本号 + }) + + // 如果存在违规内容，抛出异常 + if (res.errCode === uniSecCheck.ErrorCode.RISK_CONTENT) { + throw new Error(errorMsg || '图片违规，请修改后提交') + } else if (res.errCode !== 0) { + console.error(res) + throw new Error('内容安全检测异常：' + res.errCode) + } +} + +// 检测内容安全开关 +function checkContentSecurityEnable(field) { + // 1. 从配置中心获取配置 + return config.contentSecurity && config.contentSecurity.allowCheckType && config.contentSecurity.allowCheckType.includes(field) +} + +// 安全require +function safeRequire(module) { + try { + return require(module) + } catch (e) { + if (e.code === 'MODULE_NOT_FOUND') { + throw new Error(`${module} 公共模块不存在，请在 uniCloud/database 目录右击"配置schema扩展公共模块"添加 ${module} 模块`) + } + } +} + +module.exports = { + trigger: { + // 创建文章前触发 + beforeCreate: async function ({ + clientInfo, + addDataList, + }) { + // addDataList 是一个数组，因为可以一次性创建多条数据 + if (addDataList.length <= 0) return + + // 遍历数组，对每一条数据进行安全检测 + for (const addData of addDataList) { + // 如果是草稿，不检测 + if (addData.article_status !== 1) continue + + // 并行检测 + const parallel = [] + // 检测标题 + if (addData.title && checkContentSecurityEnable('content')) { + parallel.push(checkContentSec(addData.title, clientInfo.requestId, '标题存在敏感字，请修改后提交')) + } + // 检测摘要 + if (addData.excerpt && checkContentSecurityEnable('content')) { + parallel.push(checkContentSec(addData.excerpt, clientInfo.requestId, '摘要存在敏感字，请修改后提交')) + } + // 检测内容 + if (addData.content && checkContentSecurityEnable('content')) { + parallel.push(checkContentSec(JSON.stringify(addData.content), clientInfo.requestId, '内容存在敏感字，请修改后提交')) + } + // 检测封面图 + if (addData.thumbnail && checkContentSecurityEnable('image')) { + parallel.push(checkImageSec(addData.thumbnail, clientInfo.requestId, '封面图存在违规，请修改后提交')) + } + // 等待所有并行检测完成 + await Promise.all(parallel) + } + }, + // 更新文章前触发 + beforeUpdate: async function ({ + clientInfo, + where, + updateData + }) { + const id = where && where._id + + if (!id) return + + // 如果是草稿，不检测 + if (updateData.article_status !== 1) return + + // 并行检测 + const parallel = [] + + // 检测标题 + if (updateData.title && checkContentSecurityEnable('content')) { + parallel.push(checkContentSec(updateData.title, clientInfo.requestId, '标题存在敏感字，请修改后提交')) + } + // 检测摘要 + if (updateData.excerpt && checkContentSecurityEnable('content')) { + parallel.push(checkContentSec(updateData.excerpt, clientInfo.requestId, '摘要存在敏感字，请修改后提交')) + } + // 检测内容 + if (updateData.content && checkContentSecurityEnable('content')) { + parallel.push(checkContentSec(JSON.stringify(updateData.content), clientInfo.requestId, '内容存在敏感字，请修改后提交')) + } + // 检测封面图 + if (updateData.thumbnail && checkContentSecurityEnable('image')) { + parallel.push(checkImageSec(updateData.thumbnail, clientInfo.requestId, '封面图存在违规，请修改后提交')) + } + + // 等待所有并行检测完成 + await Promise.all(parallel) + + }, + // 读取文章前触发 + afterRead: async function ({ userInfo, clientInfo, result, where, field }) { + // 检查是否配置了clientAppIds字段，如果没有则抛出错误 + if (!config.clientAppIds && clientInfo.uniPlatform !== 'web') { + throw new Error('请在 uni-cms 配置文件中配置 clientAppIds 字段后访问，详见：https://uniapp.dcloud.net.cn/uniCloud/uni-cms.html#uni-cms-config') + } + + // 如果clientAppIds字段未配置或当前appId不在clientAppIds中，则返回 + if (!config.clientAppIds || !config.clientAppIds.includes(clientInfo.appId)) return + + // 获取广告配置 + const adConfig = config.adConfig || {} + + // 获取文章id + const id = where && where._id + + // 如果id不存在或者field不包含content，则返回 + if (id && field.includes('content')) { + // 读取了content字段后view_count加1 + await db.collection(articleDBName).where(where).update({ + view_count: db.command.inc(1) + }) + } + + // 如果查询结果为空，则返回 + if (!result.data || result.data.length <= 0) return + + // 获取文章 + const article = result.data[0] + + // 如果文章内容不存在，则返回 + if (!article.content) return + + let needUnlock = false + let unlockContent = [] + + // 遍历文章内容，找到解锁内容 + for (const op of article.content.ops) { + unlockContent.push(op) + if (op.insert.unlockContent) { + needUnlock = true + break + } + } + + // 如果文章不需要解锁，则返回 + if (!needUnlock) return + + // 获取唯一标识符 + const uniqueId = adConfig.watchAdUniqueType === 'user' ? userInfo.uid : clientInfo.deviceId + + // 如果未登录或者文章未解锁，则返回解锁内容 + if (!uniqueId || !article._id) { + article.content = { + ops: unlockContent + } + return + } + + // 查询解锁记录 + const unlockRecord = await db.collection(unlockContentDBName).where({ + unique_id: uniqueId, + article_id: article._id + }).get() + + // 如果未解锁，则返回解锁内容 + if (unlockRecord.data && unlockRecord.data.length <= 0) { + article.content = { + ops: unlockContent + } + return + } + + // 将文章解锁替换为行结束符 \n + article.content = { + ops: article.content.ops.map(op => { + if (op.insert.unlockContent) { + op.insert = "\n" + } + return op + }) + } + } + + } +}