Completed
Push — master ( 4bea56...b5efe9 )
by Equim
01:05
created

app.js (2 issues)

1
/* TODO
2
 * 完善异常处理
3
 * 建立用户名-Cookie数据库
4
 */
5
6
'use strict';
7
8
var express    = require('express'),
9
    superagent = require('superagent'),
10
    cheerio    = require('cheerio'),
11
    escaper    = require('true-html-escape'),
12
    colors     = require('colors'),
13
    program    = require('commander'),
14
    Date       = require('./lib/Date.js'),
15
    access     = require('./lib/access.js');
16
17
program
18
    .option('-h, --help')
19
    .option('-v, --version')
20
    .option('-p, --port [n]', parseInt)
21
    .option('-f, --fullLog')
22
    .parse(process.argv);
23
24
// 别问我为什么这里逻辑这么奇怪……测试的结果确实是这样的啊hhh
25
if (!program.help || !program.version) {
26
    console.log(('CSUEMS API v2.0.3').rainbow);
27
    console.log(('by The Liberators').rainbow);
28
    if (!program.help) {
29
        console.log('Preparation:');
30
        console.log('  \\\\This section is WIP\\\\');
31
        console.log('\nUsage:');
32
        console.log('  npm start [-- <options...>]');
33
        console.log('\nOptions:');
34
        console.log('  -h, --help          print this message and exit.');
35
        console.log('  -v, --version       print the version and exit.');
36
        console.log('  -f, --fullLog       enable full log, by default only errors are logged.');
37
        console.log('  -p, --port [value]  specify a port to listen, 2333 by default.');
38
        console.log('\nExamples:');
39
        console.log('  $ npm start -p 43715                  # listening to 43715');
40
        console.log('  $ forever start app.js                # deploy with forever as daemon (root access recommended)');
41
        console.log('  $ pm2 start -i 0 -n "csuapi" app.js   # deploy with pm2 as daemon  (root access recommended)');
42
    }
43
    process.exit(0);
44
}
45
46
const timeStamp = () => new Date().format('[MM-dd hh:mm:ss] '),
47
      // 以系统时间获取当前学期
48
      getSem = () => {
49
          let now = new Date();
50
          let month = now.getMonth();
51
          let year = now.getFullYear();
52
          if (month === 0) {
53
              return (year - 1) + '-' + year + '-1';
54
          } else if (month <= 6) {
55
              return (year - 1) + '-' + year + '-2';
56
          } else {
57
              return year + '-' + (year + 1) + '-1';
58
          }
59
      },
60
      port = program.port || 2333,
61
      fullLog = program.fullLog;
62
63
var app = express();
64
65
// 查成绩API,通过GET传入用户名和密码
66
app.get('/grades', function (req, res, next) {
67
    if (!req.query.id || !req.query.pwd || (req.query.sem && !(/^20\d{2}-20\d{2}-[1-2]$/).test(req.query.sem))) {
68
        res.send({ error: "参数不正确" });
69
        return;
70
    }
71
    if (fullLog) {
72
        var start = new Date();
73
        console.log((timeStamp() + 'Started to query the grades: ').cyan + req.query.id.yellow);
74
    }
75
    access.login(req.query.id, req.query.pwd, res, function (headers, ires) {
76
        fullLog && console.log((timeStamp() + 'Successfully logged in.').green);
77
78
        var ret = {};
79
        var $ = cheerio.load(ires.text);
80
81
        ret.name = escaper.unescape($('.block1text').html()).match(/姓名:.+</)[0].substring(3).replace(/</, '');
82
        ret.id = req.query.id;
83
84
        // 实际上xnxq01id为空的时候和GET这个URL的效果是一样的,都是查询所有学期
85
        superagent
86
            .post('http://csujwc.its.csu.edu.cn/jsxsd/kscj/yscjcx_list')
87
            .set(headers)
88
            .type('form')
89
            .send({
90
                xnxq01id: req.query.sem
91
            })
92
            .end(function (err, iires) {
93
                if (err) {
94
                    console.log((timeStamp() + 'Failed to get grades page\n' + err.stack).red);
95
                    res.send({ error: '无法进入成绩页面' });
96
                    return next(err);
97
                }
98
                fullLog && console.log((timeStamp() + 'Successfully entered grades page.').green);
99
100
                $ = cheerio.load(iires.text);
101
                
102
                ret.grades = {};
103
                ret.failed = {};
104
105
                // 获取成绩列表
106
                $('#dataList tr').each(function (index) {
107
                    if (index === 0) {
108
                        return;
109
                    }
110
                    let element = $(this).find('td');
111
                    let title = escaper.unescape(element.eq(3).text().match(/].+$/)[0].substring(1));
112
113
                    let item = {};
114
                    item.sem = escaper.unescape(element.eq(2).text());
115
                    item.regular = escaper.unescape(element.eq(4).text());
116
                    item.exam = escaper.unescape(element.eq(5).text());
117
                    item.overall = escaper.unescape(element.eq(6).text());
118
                    if (req.query.details) {
119
                        item.id = escaper.unescape(element.eq(3).text().match(/\[.+\]/)[0].replace(/\[|\]/g, ''));
120
                        item.attr = escaper.unescape(element.eq(8).text());
121
                        item.genre = escaper.unescape(element.eq(9).text());
122
                        item.credit = escaper.unescape(element.eq(7).text());
123
                    }
124
125
                    // 如果有补考记录,则以最高分的为准
126
                    if (title in ret.grades) {
127
                        if (item.overall < ret.grades[title].overall) {
128
                            return;
129
                        }
130
                        if (!element.eq(6).css('color')) {
131
                            delete ret.failed[title];
132
                        }
133
                    } else {
134
                        if (element.eq(6).css('color')) {
135
                            ret.failed[title] = item;
136
                        }
137
                    }
138
139
                    ret.grades[title] = item;
140
                });
141
142
                ret['subject-count'] = Object.keys(ret.grades).length;
143
                ret['failed-count'] = Object.keys(ret.failed).length;
144
145
                access.logout(headers, res, function() {
146
                    // 返回JSON
147
                    res.send(JSON.stringify(ret));
148
                    fullLog && console.log((timeStamp() +
149
                        'Successfully logged out: ').green +
150
                        req.query.id.yellow +
151
                        (' (processed in ' +
152
                            (new Date() - start) + 'ms)'
153
                        ).green);
154
                });
0 ignored issues
show
There is no return statement in this branch, but you do return something in other branches. Did you maybe miss it? If you do not want to return anything, consider adding return undefined; explicitly.
Loading history...
155
            });
156
    });
157
});
158
159
// 查考试API,通过GET传入用户名和密码
160
app.get('/exams', function (req, res, next) {
161
    if (!req.query.id || !req.query.pwd || (req.query.sem && !(/^20\d{2}-20\d{2}-[1-2]$/).test(req.query.sem))) {
162
        res.send({ error: "参数不正确" });
163
        return;
164
    }
165
    if (fullLog) {
166
        var start = new Date();
167
        console.log((timeStamp() + 'Started to query the exams: ').cyan + req.query.id.yellow);
168
    }
169
    access.login(req.query.id, req.query.pwd, res, function (headers, ires) {
170
        fullLog && console.log((timeStamp() + 'Successfully logged in.').green);
171
172
        var ret = {};
173
        var $ = cheerio.load(ires.text);
174
175
        ret.name = escaper.unescape($('.block1text').html()).match(/姓名:.+</)[0].replace('<', '').substring(3);
176
        ret.id = req.query.id;
177
        ret.sem = req.query.sem || getSem();
178
179
        superagent
180
            .post('http://csujwc.its.csu.edu.cn/jsxsd/xsks/xsksap_list')
181
            .set(headers)
182
            .type('form')
183
            .send({
184
                xqlbmc: '',
185
                xnxqid: ret.sem,
186
                xqlb: ''
187
            })
188
            .end(function (err, iires) {
189
                if (err) {
190
                    console.log((timeStamp() + 'Failed to reach exams page\n' + err.stack).red);
191
                    res.send({ error: '获取成绩失败' });
192
                    return next(err);
193
                }
194
                fullLog && console.log((timeStamp() + 'Successfully entered exams page.').green);
195
196
                $ = cheerio.load(iires.text);
197
198
                ret.exams = {};
199
                ret['exams-count'] = 0;
200
201
                $('#dataList tr').each(function (index) {
202
                    if (index === 0) {
203
                        return;
204
                    }
205
                    let element = $(this).find('td');
206
                    let title = escaper.unescape(element.eq(3).text());
207
208
                    let item = {};
209
                    item.time = escaper.unescape(element.eq(4).text());
210
                    item.location = escaper.unescape(element.eq(5).text());
211
                    item.seat = escaper.unescape(element.eq(6).text());
212
213
                    ret.exams[title] = item;
214
                    ret['exams-count']++;
215
                });
216
217
                access.logout(headers, res, function() {
218
                    res.send(JSON.stringify(ret));
219
                    fullLog && console.log((timeStamp() + 'Successfully logged out: ').green + req.query.id.yellow + (' (processed in ' + (new Date() - start) + 'ms)').green);
220
                });
0 ignored issues
show
There is no return statement in this branch, but you do return something in other branches. Did you maybe miss it? If you do not want to return anything, consider adding return undefined; explicitly.
Loading history...
221
            });
222
    });
223
});
224
225
app.listen(port);
226
console.log((timeStamp() +
227
    'The server is now running on port ' +
228
    port + '. Full logging is ' +
229
    (fullLog ? 'enabled.' : 'disabled.')
230
    ).green);
231