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
Best Practice
introduced
by
![]() |
|||
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
|
|||
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 |