You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
222 lines
6.5 KiB
222 lines
6.5 KiB
/* |
|
* This file is part of SYZOJ. |
|
* |
|
* Copyright (c) 2016 Menci <huanghaorui301@gmail.com> |
|
* |
|
* SYZOJ is free software: you can redistribute it and/or modify |
|
* it under the terms of the GNU Affero General Public License as |
|
* published by the Free Software Foundation, either version 3 of the |
|
* License, or (at your option) any later version. |
|
* |
|
* SYZOJ is distributed in the hope that it will be useful, |
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
* GNU Affero General Public License for more details. |
|
* |
|
* You should have received a copy of the GNU Affero General Public |
|
* License along with SYZOJ. If not, see <http://www.gnu.org/licenses/>. |
|
*/ |
|
|
|
'use strict'; |
|
|
|
Array.prototype.forEachAsync = Array.prototype.mapAsync = async function (fn) { |
|
return Promise.all(this.map(fn)); |
|
}; |
|
|
|
let path = require('path'); |
|
let util = require('util'); |
|
let renderer = require('moemark-renderer'); |
|
let moment = require('moment'); |
|
let url = require('url'); |
|
let querystring = require('querystring'); |
|
let pygmentize = require('pygmentize-bundled-cached'); |
|
let gravatar = require('gravatar'); |
|
let AdmZip = require('adm-zip'); |
|
let filesize = require('file-size'); |
|
|
|
function escapeHTML(s) { |
|
// Code from http://stackoverflow.com/questions/5251520/how-do-i-escape-some-html-in-javascript/5251551 |
|
return s.replace(/[^0-9A-Za-z ]/g, (c) => { |
|
return "&#" + c.charCodeAt(0) + ";"; |
|
}); |
|
} |
|
|
|
function highlightPygmentize(code, lang, cb) { |
|
code = code.split('\t').join(' '); |
|
pygmentize({ |
|
lang: lang, |
|
format: 'html', |
|
options: { |
|
nowrap: true, |
|
classprefix: 'pl-' |
|
} |
|
}, code, (err, res) => { |
|
if (err || res.toString() === 'undefined') { |
|
cb(escapeHTML(code)); |
|
} else { |
|
cb(res); |
|
} |
|
}); |
|
} |
|
|
|
renderer.config.highlight = highlightPygmentize; |
|
|
|
module.exports = { |
|
resolvePath(s) { |
|
let a = Array.from(arguments); |
|
a.unshift(__dirname); |
|
return path.resolve.apply(null, a); |
|
}, |
|
markdown(obj, keys) { |
|
let replaceUI = s => { |
|
return s.split('<pre>').join('<div class="ui existing segment"><pre style="margin-top: 0; margin-bottom: 0; ">').split('</pre>').join('</pre></div>') |
|
.split('<table>').join('<table class="ui table">') |
|
.split('<blockquote>').join('<div class="ui message">').split('</blockquote>').join('</div>'); |
|
} |
|
return new Promise((resolve, reject) => { |
|
if (!keys) { |
|
if (!obj || !obj.trim()) resolve(""); |
|
else renderer(obj, s => { |
|
resolve(replaceUI(s)); |
|
}); |
|
} else { |
|
let res = obj, cnt = keys.length; |
|
for (let key of keys) { |
|
renderer(res[key], (s) => { |
|
res[key] = replaceUI(s); |
|
if (!--cnt) resolve(res); |
|
}); |
|
} |
|
} |
|
}); |
|
}, |
|
formatDate(ts, format) { |
|
let m = moment(ts * 1000); |
|
m.locale('zh-cn'); |
|
return m.format(format || 'L H:mm:ss'); |
|
}, |
|
formatTime(x) { |
|
function toStringWithPad(x) { |
|
x = parseInt(x); |
|
if (x < 10) return '0' + x.toString(); |
|
else return x.toString(); |
|
} |
|
return util.format('%s:%s:%s', toStringWithPad(x / 3600), toStringWithPad(x / 60 % 60), toStringWithPad(x % 60)); |
|
}, |
|
formatSize(x) { |
|
let res = filesize(x, { fixed: 1 }).calculate(); |
|
if (res.result === parseInt(res.result)) res.fixed = res.result.toString(); |
|
if (res.suffix === 'Bytes') res.suffix = 'B'; |
|
else res.suffix = res.suffix.replace('iB', ''); |
|
return res.fixed + ' ' + res.suffix; |
|
}, |
|
parseDate(s) { |
|
return parseInt(+new Date(s) / 1000); |
|
}, |
|
getCurrentDate() { |
|
return parseInt(+new Date / 1000); |
|
}, |
|
makeUrl(req_params, form) { |
|
let res = ''; |
|
if (!req_params) res = '/'; |
|
else if (req_params.originalUrl) { |
|
let u = url.parse(req_params.originalUrl); |
|
res = u.pathname; |
|
} else { |
|
if (!Array.isArray(req_params)) req_params = [req_params]; |
|
for (let param of req_params) res += '/' + param; |
|
} |
|
let encoded = querystring.encode(form); |
|
if (encoded) res += '?' + encoded; |
|
return res; |
|
}, |
|
escapeHTML: escapeHTML, |
|
highlight(code, lang) { |
|
return new Promise((resolve, reject) => { |
|
highlightPygmentize(code, lang, res => { |
|
resolve(res); |
|
}); |
|
}); |
|
}, |
|
gravatar(email, size) { |
|
return gravatar.url(email, { s: size, d: 'mm' }).replace('www', 'cn'); |
|
}, |
|
parseTestData(filename) { |
|
let zip = new AdmZip(filename); |
|
let list = zip.getEntries().filter(e => !e.isDirectory).map(e => e.entryName); |
|
if (!list.includes('data_rule.txt')) { |
|
let testcases = []; |
|
for (let file of list) { |
|
let parsedName = path.parse(file); |
|
if (parsedName.ext === '.in') { |
|
if (list.includes(`${parsedName.name}.out`)) { |
|
testcases.push({ |
|
input: file, |
|
output: `${parsedName.name}.out` |
|
}); |
|
} |
|
|
|
if (list.includes(`${parsedName.name}.ans`)) { |
|
testcases.push({ |
|
input: file, |
|
output: `${parsedName.name}.ans` |
|
}); |
|
} |
|
} |
|
} |
|
|
|
testcases.sort((a, b) => { |
|
function getLastInteger(s) { |
|
let re = /(\d+)\D*$/; |
|
let x = re.exec(s); |
|
if (x) return parseInt(x[1]); |
|
else return -1; |
|
} |
|
|
|
return getLastInteger(a.input) - getLastInteger(b.input); |
|
}); |
|
|
|
return testcases; |
|
} |
|
let lines = zip.readAsText('data_rule.txt').split('\r').join('').split('\n'); |
|
|
|
if (lines.length < 3) throw 'Invalid data_rule.txt'; |
|
|
|
let numbers = lines[0].split(' ').filter(x => x); |
|
let input = lines[1]; |
|
let output = lines[2]; |
|
|
|
let res = []; |
|
for (let i of numbers) { |
|
res[i] = {}; |
|
res[i].input = input.replace('#', i); |
|
res[i].output = output.replace('#', i); |
|
|
|
if (!list.includes(res[i].input)) throw `Can't find file ${res[i].input}`; |
|
if (!list.includes(res[i].output)) throw `Can't find file ${res[i].output}`; |
|
} |
|
|
|
return res.filter(x => x); |
|
}, |
|
ansiToHTML(s) { |
|
let Convert = require('ansi-to-html'); |
|
let convert = new Convert({ escapeXML: true }); |
|
return convert.toHtml(s); |
|
}, |
|
paginate(count, currPage, perPage) { |
|
currPage = parseInt(currPage); |
|
if (!currPage || currPage < 1) currPage = 1; |
|
|
|
let pageCnt = Math.ceil(count / perPage); |
|
if (currPage > pageCnt) currPage = pageCnt; |
|
|
|
return { |
|
currPage: currPage, |
|
perPage: perPage, |
|
pageCnt: pageCnt |
|
}; |
|
}, |
|
removeTitleTag(s) { |
|
return s.replace(/「[\S\s]+?」/, ''); |
|
} |
|
};
|
|
|