index.js

//'use strict';
exports.main_handler = async (event, context, callback) => {
    let params = {}
    let scripts = []
    if (event["TriggerName"] == 'remote') {
        console.log('remote触发:', event["Message"])
        const got = require('got')
        const links = ['https://raw.fastgit.org/zero205/JD_tencent_scf/main/','https://raw.githubusercontent.com/zero205/JD_tencent_scf/main/']
        for (let i = 0; i < links.length; i++) {
            try {
                const { body } = await got(`${links[i]}${event["Message"]}.js`, {
                    timeout: 5000,
                    retry: 2
                })
                eval(body)
                break
            } catch (error) {
                console.error(`got error:`, error)
            }
        }
        return
    } else if (event["TriggerName"] == 'config') {
        let now_hour = (new Date().getUTCHours() + 8) % 24
        console.log('hourly config触发,当前:', now_hour)
        if (event["Message"]) {
            const hour = Number(event["Message"])
            if (!isNaN(hour) && hour >= 0 && hour <= 23) {
                now_hour = hour
                console.log('hourly config触发,自定义触发小时:', now_hour)
            }
        }
        const { readFileSync, accessSync, constants } = require('fs')
        const config_file = process.cwd() + '/config.json'
        try {
            await accessSync(config_file, constants.F_OK)
            console.log(`${config_file} 存在`)
        } catch (err) {
            console.error(`${config_file} 不存在,结束`)
            return
        }
        let config
        try {
            config = JSON.parse(await readFileSync(config_file))
        } catch (e) {
            console.error(`read config error:${e}`)
            return
        }
        // console.debug(JSON.stringify(config))
        params = config['params']
        delete config['params']

        const config_diy_file = process.cwd() + '/config_diy.json'
        try {
            await accessSync(config_diy_file, constants.F_OK)
            console.log(`${config_diy_file} 存在`)
            const config_diy = JSON.parse(await readFileSync(config_diy_file))
            if (config_diy['params']) {
                params = { ...params, ...config_diy['params'] }
                delete config_diy['params']
            }
            config = { ...config, ...config_diy }
        } catch (err) {
            console.error(`${config_diy_file} 不存在或解析异常`)
        }
        console.log("params:", params)
        for (let script in config) {
            // console.debug(`script:${script}`)
            const cron = config[script]
            if (typeof cron == 'number') {
                // console.debug(`number param:${cron}`)
                if (now_hour % cron == 0) {
                    console.debug(`${script}:number cron triggered`)
                    scripts.push(script)
                }
            } else {
                // console.debug(`dict param:${cron}`)
                if (cron.includes(now_hour)) {
                    console.debug(`${script}:array cron triggered`)
                    scripts.push(script)
                }
            }
        }
    } else {
        if (!event["Message"]) {
            console.error('未接收到任何参数,请阅读@hshx123大佬教程的测试步骤,查看如何使用.')
            return
        }
        console.log('参数触发方式(不读取配置文件),触发参数:', event["Message"])
        scripts = event["Message"].split("&")
    }
    if (process.env.NOT_RUN) {
        const not_run = process.env.NOT_RUN.split("&")
        scripts = scripts.filter(script => {
            const flag = not_run.includes(script)
            if (flag) {
                console.log(`not run:${script}`)
            }
            return !flag
        })
    }
    if (!scripts.length) {
        console.log('No Script to Execute, Exit!')
        return
    }
    const is_sync = (params['global'] && params['global']['exec'] == 'sync')
    console.log('当前是', is_sync ? '同' : '异', '步执行')
    if (is_sync) {
        const { execFile } = require('child_process')
        const min = 1000 * 60
        const param_names = ['timeout']
        for (const script of scripts) {
            const name = './' + script + '.js'
            const param_run = {}
            const param = params[script]
            for (const param_name of param_names) {
                if (param) {
                    if (param[param_name]) {
                        console.debug(`${script} has specific ${param_name}:${param[param_name]}`)
                        param_run[param_name] = min * param[param_name]
                    }
                } else if (params['global'] && params['global'][param_name]) {
                    console.debug(`${script} use global ${param_name}`)
                    param_run[param_name] = min * params['global'][param_name]
                } else {
                    console.warn(`No global ${param_name}!`)
                }
            }
            console.log(`run script:${script}`)
            try {
                await (async () => {
                    return new Promise((resolve) => {
                        const child = execFile(process.execPath, [name], param_run)
                        child.stdout.on('data', function(data) {
                            console.log(data)
                        })
                        child.stderr.on('data', function(data) {
                            console.error(data)
                        })
                        child.on('close', function(code) {
                            console.log(`${script} finished`)
                            delete child
                            resolve()
                        })
                    })
                })()
            } catch (e) {
                console.error(`${script} ERROR:${e}`)
                console.error(`stdout:${e.stdout}`)
            }
        }
    } else {
        console.log('异步执行不支持params参数!');
        ['log', 'warn', 'error', 'debug', 'info'].forEach((methodName) => {
            const originalMethod = console[methodName]
            console[methodName] = (...args) => {
                try {
                    throw new Error()
                } catch (error) {
                    let stack = error
                        .stack // Grabs the stack trace
                        .split('\n')[2] // Grabs third line
                        .split("/").slice(-1)[0] // Grabs  file name and line number
                        .replace('.js', '')
                    stack = `${stack.substring(0, stack.lastIndexOf(':'))}:`
                    originalMethod.apply(
                        console,
                        [
                            stack,
                            ...args
                        ]
                    )
                }
            }
        })
        for (const script of scripts) {
            console.log(`run script:${script}`)
            const name = './' + script + '.js'
            try {
                require(name)
            } catch (e) {
                console.error(`异步${script}异常:`, e)
            }
        }
    }
}