This commit is contained in:
Hex
2023-03-07 17:18:16 +08:00
parent a85e8f5ccd
commit 0ef1b8130f
4 changed files with 30 additions and 49 deletions

22
bot.js
View File

@@ -109,6 +109,25 @@ const updateCookie = (cookie, name, value) => {
return cookie.replace(re, `$1${name}=${encodeURIComponent(value)};`);
};
const getEnvCookies = () => {
let cookies = [];
// 判断环境变量里面是否有 cookie
if (process.env.SMZDM_COOKIE) {
if (process.env.SMZDM_COOKIE.indexOf('&') > -1) {
cookies = process.env.SMZDM_COOKIE.split('&');
}
else if (process.env.SMZDM_COOKIE.indexOf('\n') > -1) {
cookies = process.env.SMZDM_COOKIE.split('\n');
}
else {
cookies = [process.env.SMZDM_COOKIE];
}
}
return cookies[0] ? cookies : false;
};
// ------------------------------------
class SmzdmBot {
@@ -158,5 +177,6 @@ module.exports = {
SmzdmBot,
requestApi,
removeTags,
parseJSON
parseJSON,
getEnvCookies
};

View File

@@ -6,7 +6,7 @@ cron: 10 8 * * *
*/
const Env = require('./env');
const { SmzdmBot, requestApi, removeTags } = require('./bot');
const { SmzdmBot, requestApi, removeTags, getEnvCookies } = require('./bot');
const notify = require('./sendNotify');
// ------------------------------------
@@ -133,22 +133,9 @@ class SmzdmCheckinBot extends SmzdmBot {
}
!(async () => {
let cookies = [];
const cookies = getEnvCookies();
// 判断环境变量里面是否有 cookie
if (process.env.SMZDM_COOKIE) {
if (process.env.SMZDM_COOKIE.indexOf('&') > -1) {
cookies = process.env.SMZDM_COOKIE.split('&');
}
else if (process.env.SMZDM_COOKIE.indexOf('\n') > -1) {
cookies = process.env.SMZDM_COOKIE.split('\n');
}
else {
cookies = [process.env.SMZDM_COOKIE];
}
}
if (!cookies[0]) {
if (cookies === false) {
$.log('\n请先设置 SMZDM_COOKIE 环境变量');
return;

View File

@@ -6,7 +6,7 @@ cron: 20 8 * * *
*/
const Env = require('./env');
const { SmzdmBot, requestApi, parseJSON } = require('./bot');
const { SmzdmBot, requestApi, parseJSON, getEnvCookies } = require('./bot');
const notify = require('./sendNotify');
// ------------------------------------
@@ -99,22 +99,9 @@ class SmzdmLotteryBot extends SmzdmBot {
}
!(async () => {
let cookies = [];
const cookies = getEnvCookies();
// 判断环境变量里面是否有 cookie
if (process.env.SMZDM_COOKIE) {
if (process.env.SMZDM_COOKIE.indexOf('&') > -1) {
cookies = process.env.SMZDM_COOKIE.split('&');
}
else if (process.env.SMZDM_COOKIE.indexOf('\n') > -1) {
cookies = process.env.SMZDM_COOKIE.split('\n');
}
else {
cookies = [process.env.SMZDM_COOKIE];
}
}
if (!cookies[0]) {
if (cookies === false) {
$.log('\n请先设置 SMZDM_COOKIE 环境变量');
return;

View File

@@ -6,7 +6,7 @@ cron: 20 14 * * *
*/
const Env = require('./env');
const { SmzdmBot, requestApi, removeTags } = require('./bot');
const { SmzdmBot, requestApi, removeTags, getEnvCookies } = require('./bot');
const notify = require('./sendNotify');
// ------------------------------------
@@ -699,22 +699,9 @@ class SmzdmTaskBot extends SmzdmBot {
}
!(async () => {
let cookies = [];
const cookies = getEnvCookies();
// 判断环境变量里面是否有 cookie
if (process.env.SMZDM_COOKIE) {
if (process.env.SMZDM_COOKIE.indexOf('&') > -1) {
cookies = process.env.SMZDM_COOKIE.split('&');
}
else if (process.env.SMZDM_COOKIE.indexOf('\n') > -1) {
cookies = process.env.SMZDM_COOKIE.split('\n');
}
else {
cookies = [process.env.SMZDM_COOKIE];
}
}
if (!cookies[0]) {
if (cookies === false) {
$.log('\n请先设置 SMZDM_COOKIE 环境变量');
return;