Hydro/hydro/handler/problem.js

const fs = require('fs');
const paginate = require('../lib/paginate');
const validator = require('../lib/validator');
const file = require('../model/file');
const problem = require('../model/problem');
const record = require('../model/record');
const user = require('../model/user');
const solution = require('../model/solution');
const system = require('../model/system');
const bus = require('../service/bus');
const {
    Route, Connection, Handler, ConnectionHandler,
} = require('../service/server');
const {
    NoProblemError, ProblemDataNotFoundError, BadRequestError,
    SolutionNotFoundError,
} = require('../error');
const {
    PERM_VIEW_PROBLEM, PERM_VIEW_PROBLEM_HIDDEN, PERM_SUBMIT_PROBLEM,
    PERM_CREATE_PROBLEM, PERM_READ_PROBLEM_DATA, PERM_EDIT_PROBLEM,
    PERM_JUDGE, PERM_VIEW_PROBLEM_SOLUTION, PERM_CREATE_PROBLEM_SOLUTION,
    PERM_EDIT_PROBLEM_SOLUTION, PERM_DELETE_PROBLEM_SOLUTION, PERM_EDIT_PROBLEM_SOLUTION_REPLY,
    PERM_REPLY_PROBLEM_SOLUTION, PERM_LOGGEDIN,
} = require('../permission');

class ProblemHandler extends Handler {
    async _prepare() {
        this.checkPerm(PERM_VIEW_PROBLEM);
    }

    async get({ domainId, page = 1, category = '' }) {
        this.response.template = 'problem_main.html';
        const q = {};
        let psdict = {};
        if (category) q.$or = [{ category }, { tag: category }];
        if (!this.user.hasPerm(PERM_VIEW_PROBLEM_HIDDEN)) q.hidden = false;
        const [pdocs, ppcount, pcount] = await paginate(
            problem.getMulti(domainId, q).sort({ pid: 1 }),
            page,
            await system.get('PROBLEM_PER_PAGE'),
        );
        if (this.user.hasPerm(PERM_LOGGEDIN)) {
            psdict = await problem.getListStatus(
                domainId, this.user._id, pdocs.map((pdoc) => pdoc.docId),
            );
        }
        const path = [
            ['Hydro', '/'],
            ['problem_main', null],
        ];
        this.response.body = {
            path, page, pcount, ppcount, pdocs, psdict, category,
        };
    }

    async cleanup() {
        if (this.response.template === 'problem_main.html' && this.preferJson) {
            const {
                path, page, pcount, ppcount, pdocs, psdict, category,
            } = this.response.body;
            this.response.body = {
                title: this.renderTitle(category),
                fragments: (await Promise.all([
                    this.renderHTML('partials/problem_list.html', {
                        page, ppcount, pcount, pdocs, psdict,
                    }),
                    this.renderHTML('partials/problem_stat.html', { pcount }),
                    this.renderHTML('partials/problem_lucky.html', { category }),
                    this.renderHTML('partials/path.html', { path }),
                ])).map((i) => ({ html: i })),
                raw: {
                    path, page, pcount, ppcount, pdocs, psdict, category,
                },
            };
        }
    }
}

class ProblemCategoryHandler extends ProblemHandler {
    async get({ domainId, page = 1, category }) {
        this.response.template = 'problem_main.html';
        const q = {
            $or: [{ category }, { tag: category }],
        };
        let psdict = {};
        if (!this.user.hasPerm(PERM_VIEW_PROBLEM_HIDDEN)) q.hidden = false;
        const [pdocs, ppcount, pcount] = await paginate(
            problem.getMulti(domainId, q).sort({ pid: 1 }),
            page,
            await system.get('PROBLEM_PER_PAGE'),
        );
        if (this.user.hasPerm(PERM_LOGGEDIN)) {
            psdict = await problem.getListStatus(
                domainId, this.user._id, pdocs.map((pdoc) => pdoc.docId),
            );
        }
        const path = [
            ['Hydro', '/'],
            ['problem_main', '/p'],
            [category, null, true],
        ];
        this.response.body = {
            path, page, pcount, ppcount, pdocs, psdict, category,
        };
    }
}

class ProblemRandomHandler extends ProblemHandler {
    async get({ domainId }) {
        const q = {};
        if (!this.user.hasPerm(PERM_VIEW_PROBLEM_HIDDEN)) q.hidden = false;
        const pid = await problem.random(domainId, q);
        if (!pid) throw new NoProblemError();
        this.response.body = { pid };
        this.response.redirect = `/p/${pid}`;
    }
}

class ProblemDetailHandler extends ProblemHandler {
    async _prepare({ domainId, pid }) {
        this.response.template = 'problem_detail.html';
        this.uid = this.user._id;
        this.pid = pid;
        if (pid) {
            this.pdoc = await problem.get(domainId, pid, this.uid);
            if (this.pdoc.hidden && this.pdoc.owner !== this.uid) {
                this.checkPerm(PERM_VIEW_PROBLEM_HIDDEN);
            }
            if (this.pdoc) this.udoc = await user.getById(domainId, this.pdoc.owner);
        }
        this.response.body = {
            pdoc: this.pdoc,
            udoc: this.udoc,
            title: (this.pdoc || {}).title || '',
        };
    }

    async get() {
        this.response.body.path = [
            ['Hydro', '/'],
            ['problem_main', '/p'],
            [this.pdoc.title, null, true],
        ];
    }
}

class ProblemSubmitHandler extends ProblemDetailHandler {
    async prepare() {
        this.checkPerm(PERM_SUBMIT_PROBLEM);
    }

    async get({ domainId }) {
        this.response.template = 'problem_submit.html';
        const rdocs = await record.getUserInProblemMulti(domainId, this.uid, this.pdoc.docId)
            .sort({ _id: -1 })
            .limit(10)
            .toArray();
        this.response.body = {
            path: [
                ['Hydro', '/'],
                ['problem_main', '/p'],
                [this.pdoc.title, `/p/${this.pid}`, true],
                ['problem_submit', null],
            ],
            pdoc: this.pdoc,
            udoc: this.udoc,
            rdocs,
            title: this.pdoc.title,
        };
    }

    async post({ domainId }) {
        const { lang, code } = this.request.body;
        const rid = await record.add(domainId, {
            uid: this.uid, lang, code, pid: this.pdoc.docId,
        });
        await Promise.all([
            record.judge(domainId, rid),
            user.incDomain(domainId, this.user._id, 'nSubmit'),
        ]);
        this.response.body = { rid };
        this.response.redirect = `/r/${rid}`;
    }
}

class ProblemPretestHandler extends ProblemDetailHandler {
    async post({
        domainId, lang, code, input,
    }) {
        this.limitRate('add_record', 60, 100);
        const rid = await record.add(domainId, {
            uid: this.uid, lang, code, pid: this.pdoc.docId, input,
        });
        await record.judge(domainId, rid);
        this.response.body = { rid };
    }
}

class ProblemPretestConnectionHandler extends ConnectionHandler {
    async prepare({ domainId, pid }) {
        this.pid = pid.toString();
        this.domainId = domainId;
        bus.subscribe(['record_change'], this.onRecordChange);
    }

    async onRecordChange(data) {
        const rdoc = data.value;
        if (
            rdoc.uid !== this.user._id
            || rdoc.pid.toString() !== this.pid
            || rdoc.domainId !== this.domainId
        ) return;
        if (rdoc.tid) return;
        this.send({ rdoc });
    }

    async cleanup() {
        bus.unsubscribe(['record_change'], this.onRecordChange);
    }
}

class ProblemStatisticsHandler extends ProblemDetailHandler {
    async get({ domainId }) {
        const udoc = await user.getById(domainId, this.pdoc.owner);
        const path = [
            ['problem_main', '/p'],
            [this.pdoc.title, `/p/${this.pdoc.pid}`, true],
            ['problem_statistics', null],
        ];
        this.response.template = 'problem_statistics.html';
        this.response.body = { pdoc: this.pdoc, udoc, path };
    }
}

class ProblemManageHandler extends ProblemDetailHandler {
    async prepare() {
        if (this.pdoc.owner !== this.uid) this.checkPerm(PERM_EDIT_PROBLEM);
    }
}

class ProblemSettingsHandler extends ProblemManageHandler {
    async get() {
        this.response.template = 'problem_settings.html';
        this.response.body.path = [
            ['Hydro', '/'],
            ['problem_main', '/p'],
            [this.pdoc.title, `/p/${this.pid}`, true],
            ['problem_settings', null],
        ];
    }

    async post() {
        // TODO(masnn)
        this.back();
    }
}

class ProblemEditHandler extends ProblemManageHandler {
    async get({ pid }) {
        this.response.template = 'problem_edit.html';
        this.response.body.path = [
            ['Hydro', '/'],
            ['problem_main', '/p'],
            [this.pdoc.title, `/p/${pid}`, true],
            ['problem_edit', null],
        ];
        this.response.body.page_name = 'problem_edit';
    }

    async post({ domainId, title, content }) {
        const pid = validator.checkPid(this.request.body.pid);
        const pdoc = await problem.get(domainId, this.params.pid);
        await problem.edit(domainId, pdoc.docId, { title, content, pid });
        this.response.redirect = `/p/${pid}`;
    }
}

class ProblemDataUploadHandler extends ProblemManageHandler {
    async prepare() {
        this.response.template = 'problem_upload.html';
    }

    async get() {
        if (this.pdoc.data && typeof this.pdoc.data === 'object') {
            const f = await file.getMeta(this.pdoc.data);
            this.md5 = f.md5;
        }
        this.response.body.md5 = this.md5;
    }

    async post({ domainId }) {
        if (!this.request.files.file) throw new BadRequestError();
        const r = fs.createReadStream(this.request.files.file.path);
        await problem.setTestdata(domainId, this.pdoc.docId, r);
        if (this.pdoc.data && typeof this.pdoc.data === 'object') {
            const f = await file.getMeta(this.pdoc.data);
            this.md5 = f.md5;
        }
        this.response.body.md5 = this.md5;
    }
}

class ProblemDataDownloadHandler extends ProblemDetailHandler {
    async get({ pid }) {
        if (this.uid !== this.pdoc.owner) this.checkPerm([PERM_READ_PROBLEM_DATA, PERM_JUDGE]);
        if (!this.pdoc.data) throw new ProblemDataNotFoundError(pid);
        else if (typeof this.pdoc.data === 'string') [, this.response.redirect] = this.pdoc.data.split('from:');
        this.response.redirect = file.url(this.pdoc.data, this.pdoc.title);
    }
}

class ProblemSolutionHandler extends ProblemDetailHandler {
    async get({ domainId, page = 1 }) {
        this.response.template = 'problem_solution.html';
        this.checkPerm(PERM_VIEW_PROBLEM_SOLUTION);
        const [psdocs, pcount, pscount] = await paginate(
            solution.getMulti(domainId, this.pdoc.docId),
            page,
            await system.get('SOLUTION_PER_PAGE'),
        );
        const uids = [this.pdoc.owner];
        const docids = [];
        for (const psdoc of psdocs) {
            docids.push(psdoc.docId);
            uids.push(psdoc.owner);
            if (psdoc.reply.length) {
                for (const psrdoc of psdoc.reply) uids.push(psrdoc.owner);
            }
        }
        const udict = await user.getList(uids);
        const pssdict = solution.getListStatus(domainId, docids, this.uid);
        const path = [
            ['problem_main', '/p'],
            [this.pdoc.title, `/p/${this.pdoc.pid}`, true],
            ['problem_solution', null],
        ];
        this.response.body = {
            path, psdocs, page, pcount, pscount, udict, pssdict, pdoc: this.pdoc,
        };
    }

    async post({ domainId, psid }) {
        if (psid) this.psdoc = await solution.get(domainId, psid);
    }

    async postSubmit({ domainId, content }) {
        this.checkPerm(PERM_CREATE_PROBLEM_SOLUTION);
        await solution.add(domainId, this.pdoc.docId, this.uid, content);
        this.back();
    }

    async postEditSolution({ domainId, content }) {
        if (this.psdoc.owner !== this.uid) this.checkPerm(PERM_EDIT_PROBLEM_SOLUTION);
        this.psdoc = await solution.edit(domainId, this.psdoc.docId, content);
        this.ctx.body.psdoc = this.psdoc;
        this.back();
    }

    async postDeleteSolution({ domainId }) {
        if (this.psdoc.owner !== this.uid) this.checkPerm(PERM_DELETE_PROBLEM_SOLUTION);
        await solution.del(domainId, this.psdoc.docId);
        this.back();
    }

    async postReply({ domainId, psid, content }) {
        this.checkPerm(PERM_REPLY_PROBLEM_SOLUTION);
        const psdoc = await solution.get(domainId, psid);
        await solution.reply(domainId, psdoc.docId, this.uid, content);
    }

    async postEditReply({
        domainId, content, psid, psrid,
    }) {
        const [psdoc, psrdoc] = await solution.getReply(domainId, psid, psrid);
        if ((!psdoc) || psdoc.pid !== this.pdoc.docId) throw new SolutionNotFoundError(psid);
        if (psrdoc.owner !== this.uid) this.checkPerm(PERM_EDIT_PROBLEM_SOLUTION_REPLY);
        await solution.editReply(domainId, psid, psrid, content);
    }

    async postDeleteReply({ domainId, psid, psrid }) {
        const [psdoc, psrdoc] = await solution.getReply(domainId, psid, psrid);
        if ((!psdoc) || psdoc.pid !== this.pdoc.docId) throw new SolutionNotFoundError(psid);
        if (psrdoc.owner !== this.uid) this.checkPerm(PERM_EDIT_PROBLEM_SOLUTION_REPLY);
        await solution.delReply(domainId, psid, psrid);
        this.back();
    }

    async postUpvote({ domainId }) {
        const [psdoc, pssdoc] = await solution.vote(domainId, this.psdoc.docId, this.uid, 1);
        this.response.body = { vote: psdoc.vote, user_vote: pssdoc.vote };
        this.back();
    }

    async postDownvote({ domainId }) {
        const [psdoc, pssdoc] = await solution.vote(domainId, this.psdoc.docId, this.uid, -1);
        this.response.body = { vote: psdoc.vote, user_vote: pssdoc.vote };
        this.back();
    }
}

class ProblemSolutionRawHandler extends ProblemDetailHandler {
    async get({ domainId, psid }) {
        this.checkPerm(PERM_VIEW_PROBLEM_SOLUTION);
        const psdoc = await solution.get(domainId, psid);
        this.response.type = 'text/markdown';
        this.response.body = psdoc.content;
    }
}

class ProblemSolutionReplyRawHandler extends ProblemDetailHandler {
    async get({ domainId, psid }) {
        this.checkPerm(PERM_VIEW_PROBLEM_SOLUTION);
        const [psdoc, psrdoc] = await solution.getReply(domainId, psid);
        if ((!psdoc) || psdoc.pid !== this.pdoc.docId) throw new SolutionNotFoundError(psid);
        this.response.type = 'text/markdown';
        this.response.body = psrdoc.content;
    }
}

class ProblemCreateHandler extends Handler {
    async get() {
        this.response.template = 'problem_edit.html';
        this.checkPerm(PERM_CREATE_PROBLEM);
        this.response.body = {
            path: [
                ['Hydro', '/'],
                ['problem_main', '/p'],
                ['problem_create', null],
            ],
            page_name: 'problem_create',
        };
    }

    async post({
        domainId, title, pid, content, hidden,
    }) {
        pid = await problem.add(domainId, title, content, this.user._id, {
            pid, hidden,
        });
        this.response.body = { pid };
        this.response.redirect = `/p/${pid}/settings`;
    }
}

async function apply() {
    Route('/p', ProblemHandler);
    Route('/p/category/:category', ProblemCategoryHandler);
    Route('/problem/random', ProblemRandomHandler);
    Route('/p/:pid', ProblemDetailHandler);
    Route('/p/:pid/submit', ProblemSubmitHandler);
    Route('/p/:pid/pretest', ProblemPretestHandler);
    Route('/p/:pid/settings', ProblemSettingsHandler);
    Route('/p/:pid/statistics', ProblemStatisticsHandler);
    Route('/p/:pid/edit', ProblemEditHandler);
    Route('/p/:pid/upload', ProblemDataUploadHandler);
    Route('/p/:pid/data', ProblemDataDownloadHandler);
    Route('/p/:pid/solution', ProblemSolutionHandler);
    Route('/p/:pid/solution/:psid/raw', ProblemSolutionRawHandler);
    Route('/p/:pid/solution/:psid/:psrid/raw', ProblemSolutionReplyRawHandler);
    Route('/problem/create', ProblemCreateHandler);
    Connection('/p/:pid/pretest-conn', ProblemPretestConnectionHandler);
}

global.Hydro.handler.problem = module.exports = apply;