danmaku/routes/airportsub.js

const express = require('express');
const router = express.Router();
const oss = require('../utils/oss');
const yaml = require('js-yaml');
const cookie = require('cookie');
const {filesize} = require('filesize');
const moment = require('moment');
const axios = require('axios');

function getscheme(req) {
    return req.headers['x-forwarded-proto'] || req.protocol;
}

function getuserinfo(headers) {
    if (!headers)
        return undefined
    const str = headers["subscription-userinfo"]
    if (str === undefined) {
        return undefined
    }
    var dic = cookie.parse(str);
    dic['total_use'] = 1 * dic.upload + 1 * dic.download
    dic['use_percent'] = (100.0 * dic.total_use / dic.total).toFixed(2)
    const now_time = Math.floor(Date.now() / 1000)
    const end_time = dic.expire
    dic['date_percent'] = (100.0 * (now_time - end_time + 3600 * 24 * 365.0) / (3600 * 24 * 365.0)).toFixed(2)

    dic['total_use'] = filesize(dic['total_use'], {base: 2, standard: "jedec"})
    dic.total = filesize(dic.total, {base: 2, standard: "jedec"})
    dic.expire = moment(dic.expire * 1000).format('YYYY-MM-DD')
    return dic
}

async function updateDatabase() {
    const database = await oss.get('SUB/database.yaml');
    try {
        const doc = yaml.load(database);
        return doc
    } catch (e) {
        console.log(e);
    }
}

/* GET users listing. */
router.get('/', async function (req, res, next) {
    const database = await updateDatabase();
    if (req.query.user) {
        const userinfo = database.user[req.query.user]
        if (userinfo) {
            const expireDate = new Date(userinfo.expire);
            const now = new Date();
            if (now < expireDate) {
                if (req.query.ctype) {
                    const subinfo = database.suburl[req.query.ctype]
                    //返回指定订阅信息
                    if (subinfo) {
                        const ua = req.headers['user-agent'];
                        if (ua.indexOf('Mozilla') === -1) {
                            const url = await oss.signurl('SUB/' + req.query.ctype, true)
                            res.redirect(url)
                        } else {
                            const ret = await oss.get('SUB/' + req.query.ctype)
                            res.type('text/plain').end(ret);
                        }
                    } else {
                        res.status(404).send('Not Found 找不到这种订阅类型');
                    }
                } else {
                    const path = getscheme(req) + '://' + req.headers.host + req.originalUrl;
                    const ctypes = Object.keys(database.suburl)
                    let ret = {}
                    for (key of ctypes) {
                        const headers = await oss.head('SUB/' + key)
                        ret[key] = getuserinfo(headers)
                        // ret[key] = getuserinfotxt(getuserinfo(headers))
                    }
                    res.render('airportsub', {ret, path, expire: userinfo.expire});
                }
            } else {
                res.send('您的订阅已过期，请联系管理员');
            }
        } else {
            res.status(404).send('Not Found 找不到这个用户');
        }
    } else {
        res.status(400).send('Bad Request 缺少参数');
    }
});

router.get('/cache', async function (req, res, next) {
    const database = await updateDatabase();
    let messages = [];
    // 缓存所有的协程
    let promises = [];
    for (let key in database.suburl) {
        const url = database.suburl[key].url
        const params = database.suburl[key].params
        if (!url) continue
        promises.push(axios.get(url, {params}));
    }
    Promise.all(promises).then(values => {
        promises = [];
        for (let i = 0; i < values.length; i++) {
            const res = values[i];
            const key = Object.keys(database.suburl)[i]
            messages.push({title: 'Download', key, status: res.status})
            const userinfo = res.headers['subscription-userinfo']
            const base64userinfo = Buffer.from(userinfo).toString('base64')
            // 设置强制下载并设置文件名
            headers = {
                'Content-type': 'text/plain; charset=utf-8',
                'content-disposition': `attachment; filename=${key}`,
                'x-oss-persistent-headers': "Subscription-Userinfo:" + base64userinfo
            }
            promises.push(oss.put('SUB/' + key, res.data, headers))
        }
        Promise.all(promises).then(values => {
            for (let i = 0; i < values.length; i++) {
                const res = values[i];
                const key = Object.keys(database.suburl)[i]
                messages.push({title: 'Upload', key, status: res.res.status})
            }
            res.json(messages);
        })
    });
});

router.get('/download', async function (req, res, next) {
    const repos = ['Dreamacro/clash', 'Fndroid/clash_for_windows_pkg', 'Kr328/ClashForAndroid',
        'shadowsocks/shadowsocks-android', 'XTLS/Xray-core', '2dust/v2rayN', 'NetchX/Netch', '2dust/v2rayNG',
        'yichengchen/clashX', 'shadowsocks/shadowsocks-windows',
        'shadowsocksrr/shadowsocksr-csharp', 'FelisCatus/SwitchyOmega']
    const auth = {
        'username': process.env.GITHUB_USERNAME,
        'password': process.env.GITHUB_TOKEN
    }
    const api = 'https://api.github.com/repos/{}/releases/latest'
    const promises = repos.map(repo => axios.get(api.replace('{}', repo), {auth}))
    Promise.all(promises).then(values => {
        let datas = values.map(value => value.data)
        for (let i = 0; i < datas.length; i++) {
            datas[i].repo = repos[i]
            for (asset of datas[i].assets) {
                asset.size = filesize(asset.size, {base: 2, standard: "jedec"})
                asset['fastgit_url'] = asset['browser_download_url'].replace('github.com', 'download.fastgit.org')
                asset['ghproxy_url'] = 'https://mirror.ghproxy.com?q=' + asset['browser_download_url']
            }
        }
        res.render('airportdownload', {datas});
    });

});

module.exports = router;

if (!module.parent) {
    updateDatabase();
}