Array.prototype.forEachAsync = Array.prototype.mapAsync = async function (fn) { return Promise.all(this.map(fn)); }; Array.prototype.filterAsync = async function (fn) { let a = await this.mapAsync(fn); return this.filter((x, i) => a[i]); }; global.ErrorMessage = class ErrorMessage { constructor(message, nextUrls, details) { this.message = message; this.nextUrls = nextUrls || {}; this.details = details; } }; let Promise = require('bluebird'); let path = require('path'); let fs = Promise.promisifyAll(require('fs-extra')); let util = require('util'); let markdownRenderer = require('./libs/markdown'); let moment = require('moment'); let url = require('url'); let querystring = require('querystring'); let gravatar = require('gravatar'); let filesize = require('file-size'); let AsyncLock = require('async-lock'); module.exports = { resolvePath(s) { let a = Array.from(arguments); a.unshift(__dirname); return path.resolve.apply(null, a); }, markdown(obj, keys, noReplaceUI) { let replaceUI = s => { if (noReplaceUI) return s; s = s.split('
').join('
').split('
').join('
') .split('').join('
') .split('
').join('
').split('
').join(''); let cheerio = require('cheerio'); let $ = cheerio.load(''); let body = $('body'); body.html(s); let a = $('img:only-child'); for (let img of Array.from(a)) { if (!img.prev && !img.next) { $(img).css('display', 'block'); $(img).css('margin', '0 auto'); } } return body.html(); }; return new Promise((resolve, reject) => { if (!keys) { if (!obj || !obj.trim()) resolve(""); else markdownRenderer(obj, s => { resolve(replaceUI(s)); }); } else { let res = obj, cnt = keys.length; for (let key of keys) { markdownRenderer(res[key], (s) => { res[key] = replaceUI(s); if (!--cnt) resolve(res); }); } } }); }, formatDate(ts, format) { if (ts == null) { return "Unknown"; } let m = moment(ts * 1000); m.locale('eu'); return m.format(format || 'L H:mm:ss'); }, formatTime(x) { let sgn = x < 0 ? '-' : ''; x = Math.abs(x); function toStringWithPad(x) { x = parseInt(x); if (x < 10) return '0' + x.toString(); else return x.toString(); } return sgn + util.format('%s:%s:%s', toStringWithPad(x / 3600), toStringWithPad(x / 60 % 60), toStringWithPad(x % 60)); }, formatSize(x, precision) { if (typeof x !== 'number') return '0 B'; let unit = 'B', units = ['K', 'M', 'G', 'T']; for (let i in units) if (x > 1024) x /= 1024, unit = units[i]; var fixed = x === Math.round(x) ? x.toString() : x.toFixed(precision); return fixed + ' ' + unit; }, getFormattedCodeKey(code, lang) { if (syzoj.languages[lang].format) { return syzoj.languages[lang].format + '\n' + syzoj.utils.md5(code); } return null; }, judgeServer(suffix) { return JSON.stringify(url.resolve(syzoj.config.judge_server_addr, suffix)); }, parseDate(s) { return parseInt(+new Date(s) / 1000); }, getCurrentDate(removeTime) { let d = new Date; if (removeTime) { d.setHours(0); d.setMinutes(0); d.setSeconds(0); d.setMilliseconds(0); } return parseInt(+d / 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; }, highlight(code, lang) { return new Promise((resolve, reject) => { require('./libs/highlight')(code, lang, res => { resolve(res); }); }); }, gravatar(email, size) { return gravatar.url(email, { s: size, d: 'mm' }).replace('//www.gravatar.com/avatar', syzoj.config.gravatar_url); }, async parseTestdata(dir, submitAnswer) { if (!await syzoj.utils.isDir(dir)) return null; try { // Get list of *files* let list = await (await fs.readdirAsync(dir)).filterAsync(async x => await syzoj.utils.isFile(path.join(dir, x))); let res = []; if (!list.includes('data.yml')) { res[0] = {}; res[0].cases = []; for (let file of list) { let parsedName = path.parse(file); if (parsedName.ext === '.in') { if (list.includes(`${parsedName.name}.out`)) { let o = { input: file, output: `${parsedName.name}.out` }; if (submitAnswer) o.answer = `${parsedName.name}.out`; res[0].cases.push(o); } if (list.includes(`${parsedName.name}.ans`)) { let o = { input: file, output: `${parsedName.name}.ans` }; if (submitAnswer) o.answer = `${parsedName.name}.out`; res[0].cases.push(o); } } } res[0].type = 'sum'; res[0].score = 100; res[0].cases.forEach((e) => { e.key = (e.input.match(/\d+/g) || []).map((x) => parseInt(x)).concat(e.input); }); res[0].cases.sort((a, b) => { for (let i = 0; i < Math.max(a.key.length, b.key.length); ++i) { if (a.key[i] == undefined) return -1; if (b.key[i] == undefined) return +1; if (a.key[i] !== b.key[i]) return (a.key[i] < b.key[i] ? -1 : +1); } return 0; }); res.spj = list.some(s => s.startsWith('spj_')); } else { let config = require('js-yaml').load((await fs.readFileAsync(dir + '/data.yml'))); let input = config.inputFile, output = config.outputFile, answer = config.userOutput; res = config.subtasks.map(st => ({ score: st.score, type: st.type, cases: st.cases.map(c => { function getFileName(template, id, mustExist) { let s = template.split('#').join(String(id)); if (mustExist && !list.includes(s)) throw `找不到文件 ${s}`; return s; } let o = {}; if (input) o.input = getFileName(input, c, true); if (output) o.output = getFileName(output, c, true); if (answer) o.answer = getFileName(answer, c, false); return o; }) })); res = res.filter(x => x.cases && x.cases.length !== 0); res.spj = !!config.specialJudge; } return res; } catch (e) { console.log(e); return { error: e }; } }, 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, toSQL: () => { if (!pageCnt) return ''; else return ` LIMIT ${(currPage - 1) * perPage},${perPage}` } }; }, removeTitleTag(s) { return s.replace(/「[\S\s]+?」/, ''); }, md5(data) { let crypto = require('crypto'); let md5 = crypto.createHash('md5'); md5.update(data); return md5.digest('hex'); }, isValidUsername(s) { return /^[a-zA-Z0-9\-\_]+$/.test(s); }, locks: [], lock(key, cb) { let s = JSON.stringify(key); if (!this.locks[s]) this.locks[s] = new AsyncLock(); return this.locks[s].acquire(s, cb); }, encrypt(buffer, password) { if (typeof buffer === 'string') buffer = Buffer.from(buffer); let crypto = require('crypto'); let cipher = crypto.createCipher('aes-256-ctr', password); return Buffer.concat([cipher.update(buffer), cipher.final()]); }, decrypt(buffer, password) { let crypto = require('crypto'); let decipher = crypto.createDecipher('aes-256-ctr', password); return Buffer.concat([decipher.update(buffer), decipher.final()]); }, async isFile(path) { try { return (await fs.statAsync(path)).isFile(); } catch (e) { return false; } }, async isDir(path) { try { return (await fs.statAsync(path)).isDirectory(); } catch (e) { return false; } }, async saveConfig() { let fs = require('fs-extra'); fs.writeFileAsync(syzoj.configDir, JSON.stringify(syzoj.config, null, 2)); }, withTimeoutRetry(func) { let attemptCount = 0; return new Promise((resolve, reject) => { function attempt() { if (attemptCount++) console.log(`syzoj.utils.withTimeout(): attemptCount = ${attemptCount}`); Promise.method(func)().timeout(5000) .then(resolve) .catch(Promise.TimeoutError, attempt) .catch(reject); } attempt(); }); } };