Allow dependency injection for nodemailer and DOMPurify (#699)

Co-authored-by: Tao Xin <alexander.xin@gmail.com>
twikoojs · May 20, 2024 · 5cb7be5 · 5cb7be5
1 parent 5ff1a33
commit 5cb7be5
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/src/server/function/twikoo/utils/lib.js b/src/server/function/twikoo/utils/lib.js
@@ -1,4 +1,9 @@
+let customLibs = {}
+
 module.exports = {
+  setCustomLibs (libs) {
+    customLibs = libs
+  },
   getCheerio () {
     const $ = require('cheerio') // jQuery 服务器版
     return $
@@ -24,6 +29,7 @@ module.exports = {
     return bowser
   },
   getDomPurify () {
+    if (customLibs.DOMPurify) return customLibs.DOMPurify
     // 初始化反 XSS
     const { JSDOM } = require('jsdom') // document.window 服务器版
     const createDOMPurify = require('dompurify') // 反 XSS
@@ -44,6 +50,7 @@ module.exports = {
     return md5
   },
   getNodemailer () {
+    if (customLibs.nodemailer) return customLibs.nodemailer
     const nodemailer = require('nodemailer') // 发送邮件
     return nodemailer
   },

diff --git a/src/server/function/twikoo/utils/notify.js b/src/server/function/twikoo/utils/notify.js
@@ -5,11 +5,16 @@ const {
   getPushoo
 } = require('./lib')
 const $ = getCheerio()
-const nodemailer = getNodemailer()
 const pushoo = getPushoo()
 const { RES_CODE } = require('./constants')
 const logger = require('./logger')
 
+let nodemailer
+
+function lazilyGetNodemailer () {
+  return nodemailer ?? (nodemailer = getNodemailer())
+}
+
 let transporter
 
 const fn = {
@@ -45,7 +50,7 @@ const fn = {
       } else {
         throw new Error('SMTP 服务器没有配置')
       }
-      transporter = nodemailer.createTransport(transportConfig)
+      transporter = lazilyGetNodemailer().createTransport(transportConfig)
       try {
         const success = await transporter.verify()
         if (success) logger.info('SMTP 邮箱配置正常')