Equim-chan / csu-ems-api

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

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

app.get(/^\/g(?:|rades)$/, co.wrap(function *(req, res) {

    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);

    } catch (err) {

        logging(err.eqMessage.inner.red);

        res.status(404).send({ error: err.eqMessage.public });

        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 {

        // 直接异步进行?

        co(function *() {

            try {

                yield access.logout(headers);

            } catch (err) {

                logging(err.eqMessage.inner.red);

                return;

            }

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

        });

    }

    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) {

        if (index === 0) {

            return;

        }

        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();

        }

        // 如果有补考记录，则以最高分的为准(暂不考虑NaN)

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

            return;

        }

        result.grades[title] = item;

        // 挂科判定

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

            result.failed[title] = item;

        } else {

            delete result.failed[title];

        }

    });

    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传入用户名和密码

app.get(/^\/e(?:|xams)$/, co.wrap(function *(req, res) {

}));

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

app.get(/^\/e(?:|xams)$/, co.wrap(function *(req, res) {

    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);

    } catch (err) {

        logging(err.eqMessage.inner.red);

        res.status(404).send({ error: err.eqMessage.public });

        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 *() {

            try {

                yield access.logout(headers);

            } catch (err) {

                logging(err.eqMessage.inner.red);

            }

            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) {

        if (index === 0) {

            return;

        }

        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);