Hydro/hydro/handler/record.js

const
    { constants } = require('../options'),
    { PERM_READ_RECORD_CODE, PERM_VIEW_CONTEST_HIDDEN_SCOREBOARD,
        PERM_REJUDGE, PERM_VIEW_PROBLEM_HIDDEN } = require('../permission'),
    problem = require('../model/problem'),
    record = require('../model/record'),
    contest = require('../model/contest'),
    user = require('../model/user'),
    bus = require('../service/bus'),
    queue = require('../service/queue'),
    { Route, Handler, Connection, ConnectionHandler } = require('../service/server');

class RecordListHandler extends Handler {
    async get({ page = 1 }) {
        this.response.template = 'record_main.html';
        let q = {};
        let rdocs = await record.getMany(q, { _id: -1 }, page, constants.RECORD_PER_PAGE);
        let pdict = {}, udict = {};
        for (let rdoc of rdocs) {
            udict[rdoc.uid] = await user.getById(rdoc.uid);
            pdict[rdoc.pid] = await problem.get(rdoc.pid, this.user._id);
        }
        this.response.body = {
            path: [
                ['Hydro', '/'],
                ['record_main', null]
            ],
            page, rdocs, pdict, udict
        };
    }
}
class RecordDetailHandler extends Handler {
    async get({ rid }) {
        this.response.template = 'record_detail.html';
        let rdoc = await record.get(rid);
        if (rdoc.hidden) this.checkPerm(PERM_VIEW_CONTEST_HIDDEN_SCOREBOARD);
        if (rdoc.uid != this.user.uid && !this.user.hasPerm(PERM_READ_RECORD_CODE)) rdoc.code = null;
        this.response.body = {
            path: [
                ['Hydro', '/'],
                ['record_detail', null]
            ],
            rdoc, show_status: true
        };
    }
}
class RecordRejudgeHandler extends Handler {
    async post({ rid }) {
        this.checkPerm(PERM_REJUDGE);
        let rdoc = await record.get(rid);
        if (rdoc) {
            await record.reset(rid);
            await queue.push('judge', rid);
        }
        this.back();
    }
}
class RecordConnectionHandler extends ConnectionHandler {
    async prepare() {
        bus.subscribe(['record_change'], this.onRecordChange);
    }
    async message(msg) {
        for (let rid of msg.rids) {
            let rdoc = await record.get(rid);
            await this.onRecordChange({ value: rdoc });
        }
    }
    async cleanup() {
        bus.unsubscribe(['record_change'], this.onRecordChange);
    }
    async onRecordChange(data) {
        let rdoc = data.value;
        if (rdoc.tid && rdoc.tid.toString() != this.tid) return;
        let [udoc, pdoc] = await Promise.all([user.getById(rdoc.uid), problem.getById(rdoc.pid)]);
        if (pdoc.hidden && !this.user.hasPerm(PERM_VIEW_PROBLEM_HIDDEN)) pdoc = null;
        this.send({ html: await this.renderHTML('record_main_tr.html', { rdoc, udoc, pdoc }) });
    }
}
class RecordDetailConnectionHandler extends contest.ContestHandlerMixin(ConnectionHandler) {
    async prepare({ rid }) {
        let rdoc = await record.get(rid);
        if (rdoc.tid) {
            let tdoc = await contest.get(rdoc.tid);
            if (!this.canShowRecord(tdoc)) {
                this.close();
                return;
            }
        }
        this.rid = rid;
        bus.subscribe(['record_change'], this.onRecordChange);
        this.onRecordChange({ value: rdoc });
    }
    async onRecordChange(data) {
        let rdoc = data.value;
        if (rdoc._id.toString() != this.rid) return;
        this.send({
            status_html: await this.renderHTML('record_detail_status.html', { rdoc }),
            summary_html: await this.renderHTML('record_detail_summary.html', { rdoc })
        });
    }
    async cleanup() {
        bus.unsubscribe(['record_change'], this.onRecordChange);
    }
}

Route('/r', RecordListHandler);
Route('/r/:rid', RecordDetailHandler);
Route('/r/:rid/rejudge', RecordRejudgeHandler);
Connection('/record-conn', RecordConnectionHandler);
Connection('/record-detail-conn', RecordDetailConnectionHandler);