/* * This file is part of SYZOJ. * * Copyright (c) 2016 Menci * * 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 . */ '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'); 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('
').join('
').split('
').join('
') .split('').join('
') .split('
').join('
').split('
').join(''); } 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)); }, 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); 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]+?」/, ''); } };