Equim-chan / csu-ems-api

const

    express    = require('express'),

    superagent = require('superagent'),

    cheerio    = require('cheerio'),

    program    = require('commander'),

    moment     = require('moment'),

    co         = require('co'),

    thunkify   = require('thunkify'),

    access     = require('./lib/access.js');

program

    .option('-h, --help')

    .option('-v, --version')

    .option('-f, --fullLog')

    .parse(process.argv);

// 别问我为什么这里逻辑这么奇怪……测试的结果确实是这样的啊hhh

if (!program.help || !program.version) {

    console.log(

`${`CSUEMS API v${require('./package').version}

by The Liberators`.rainbow}${program.help ? '' :

`

Preparation:

  ${'(This section is WIP)'.grey}

Usage:

  npm start [-- <options...>]

Options:

  -h, --help          print this message and exit.

  -v, --version       print the version and exit.

  -f, --fullLog       enable full log, by default only errors are logged.

  -p, --port [value]  specify a port to listen, process.env.PORT || 2333 by default.

Examples:

  $ npm start -- -p 43715               # listening to 43715

  $ forever start app.js                # deploy with forever as daemon (root access recommended)

  $ pm2 start -i 0 -n "csuapi" app.js   # deploy with pm2 as daemon  (root access recommended)`}`);

    process.exit(0);

}

superagent.Request.prototype.endThunk = thunkify(superagent.Request.prototype.end);

const

    logging = log => console.log(`${moment().format('[[]YY-MM-DD HH:mm:ss[]]')} ${log}`),

    fullLogging = log => program.fullLog && logging(log),

    getSem = () => {

        // 以系统时间获取当前学期

        let now = new Date();

        let month = now.getMonth();

        let year = now.getFullYear();

        if (month <= 6) {

            return `${year - 1}-${year}-${month > 0 ? 2 : 1}`;

        } else {

            return `${year}-${year + 1}-1`;

        }

    },

//  wait = ms => callback => setTimeout(callback, ms),

    port = program.port || process.env.PORT || 2333,

    app = express();

// 获取文档

app.get('/doc', (req, res) => res.sendFile(`${__dirname}/doc/API.html`));

// 查成绩API，通过GET传入用户名和密码

    if (!req.query.id || !req.query.pwd || (req.query.sem && !(/^20\d{2}-20\d{2}-[1-2]$/).test(req.query.sem))) {

        res.status(404).send({ error: "参数不正确" });

        return;

    }

    let start = new Date();

    fullLogging('Started to query the grades: '.cyan + req.query.id.yellow);

    let headers;

    try {

        headers = yield access.login(req.query.id, req.query.pwd, res);

    } catch (err) { return; }

    fullLogging('Successfully logged in.'.green);

    let ires;

    try {

        // 实际上xnxq01id为空的时候和GET这个URL的效果是一样的，都是查询所有学期

        ires = yield superagent

            .post('http://csujwc.its.csu.edu.cn/jsxsd/kscj/yscjcx_list')

            .set(headers)

            .type('form')

            .send({ xnxq01id: req.query.sem })

            .endThunk();

    } catch (err) {

        logging(` Failed to get grades page\n${err.stack}`.red);

        res.status(404).send({ error: '无法进入成绩页面' });

        return;

    } finally {

        // 直接异步进行?

    fullLogging('Successfully entered grades page.'.green);

    let $ = cheerio.load(ires.text);

    let top = $('#Top1_divLoginName').text();

    let result = {

        name: top.match(/\s.+\(/)[0].replace(/\s|\(/g, ''),

        id: top.match(/\(.+\)/)[0].replace(/\(|\)/g, ''),

        grades: {},

        'subject-count': 0,

        failed: {},

        'failed-count': 0,

    };

    // 获取成绩列表

    $('#dataList tr').each(function (index) {

        let element = $(this).find('td');

        let title = element.eq(3).text().match(/].+$/)[0].substring(1);

        let item = {

            sem: element.eq(2).text(),

            reg: element.eq(4).text(),

            exam: element.eq(5).text(),

            overall: element.eq(6).text()

        };

        if (req.query.details) {

            item.id = element.eq(3).text().match(/\[.+\]/)[0].replace(/\[|\]/g, '');

            item.attr = element.eq(8).text();

            item.genre = element.eq(9).text();

            item.credit = element.eq(7).text();

        }

        // 如果有补考记录，则以最高分的为准

        // 这段代码是拿可读性换了时间复杂度……

        if (title in result.grades) {

            // 暂不考虑NaN

            if (item.overall < result.grades[title].overall) {

                return;

            }

            // 如果新的成绩不是挂科的

            if (!element.eq(6).css('color')) {

                delete result.failed[title];

            }

        } else if (element.eq(6).css('color')) {

            result.failed[title] = item;

        }

        result.grades[title] = item;

    });

    result['subject-count'] = Object.keys(result.grades).length;

    result['failed-count'] = Object.keys(result.failed).length;

    res.send(JSON.stringify(result));

    fullLogging(`Successfully responded. (req -> res processed in ${new Date() - start} ms)`.green);

}));

// 查考试API，通过GET传入用户名和密码

    if (!req.query.id || !req.query.pwd || (req.query.sem && !(/^20\d{2}-20\d{2}-[1-2]$/).test(req.query.sem))) {

        res.status(404).send({ error: "参数不正确" });

        return;

    }

    let start = new Date();

    fullLogging('Started to query the exams: '.cyan + req.query.id.yellow);

    let headers;

    try {

        headers = yield access.login(req.query.id, req.query.pwd, res);

    } catch (err) { return; }

    fullLogging('Successfully logged in.'.green);

    let ires;

    let _sem = req.query.sem || getSem();

    try {

        ires = yield superagent

            .post('http://csujwc.its.csu.edu.cn/jsxsd/xsks/xsksap_list')

            .set(headers)

            .type('form')

            .send({

                xqlbmc: '',

                xnxqid: _sem,

                xqlb: ''

            })

            .endThunk();

    } catch (err) {

        logging(` Failed to reach exams page\n${err.stack}`.red);

        res.status(404).send({ error: '无法进入考试页面' });

        return;

    } finally {

        co(function* () {

            yield access.logout(headers);

            fullLogging('Successfully logged out: '.green + req.query.id.yellow);

        });

    }

    fullLogging('Successfully entered exams page.'.green);

    let $ = cheerio.load(ires.text);

    let top = $('#Top1_divLoginName').text();

    let result = {

        name: top.match(/\s.+\(/)[0].replace(/\s|\(/g, ''),

        id: top.match(/\(.+\)/)[0].replace(/\(|\)/g, ''),

        sem: _sem,

        exams: {},

        'exams-count': 0,

    };

    $('#dataList tr').each(function (index) {

        let element = $(this).find('td');

        let title = element.eq(3).text();

        let item = {

            time: element.eq(4).text(),

            location: element.eq(5).text(),

            seat: element.eq(6).text()

        };

        result.exams[title] = item;

        result['exams-count']++;

    });

    res.send(JSON.stringify(result));

    fullLogging(`Successfully responded. (req -> res processed in ${new Date() - start} ms)`.green);

}));

app.listen(port, () => {

    logging(` The API is now running on port ${port}. Full logging is ${program.fullLog ? 'enabled' : 'disabled'}`.green);