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