🎉 able to run on Heroku

2021-01-31 17:42:41 +08:00
parent 36055602b9
commit e808a3c1ee
4 changed files with 62 additions and 2 deletions
@@ -3,7 +3,25 @@ const { User } = require('../models');
 const tokenStore = new Map();

 async function initializeTokenStore() {
-  const users = await User.findAll();
+  let users = [];
+  if (process.env.MODE === '1') {
+    console.log('Current mode is Heroku mode.');
+    let user = {
+      wechatAppId: process.env.WECHAT_APP_ID,
+      wechatAppSecret: process.env.WECHAT_APP_SECRET,
+      wechatTemplateId: process.env.WECHAT_TEMPLATE_ID,
+      wechatOpenId: process.env.WECHAT_OPEN_ID,
+      wechatVerifyToken: process.env.WECHAT_VERIFY_TOKEN,
+      prefix: process.env.PREFIX,
+      email: process.env.EMAIL,
+      smtpServer: process.env.SMTP_SERVER,
+      smtpUser: process.env.SMTP_USER,
+      smtpPass: process.env.SMTP_PASS,
+    };
+    users.push(user);
+  } else {
+    users = await User.findAll();
+  }
  users.forEach((user) => {
    if (user.wechatAppId) {
      tokenStore.set(user.prefix, {