core: allow lock topic

packages/hydrooj/src/error.ts

@@ -98,6 +98,7 @@ export const DomainJoinForbiddenError = Err('DomainJoinForbiddenError', Forbidde
 export const DomainJoinAlreadyMemberError = Err('DomainJoinAlreadyMemberError', ForbiddenError, 'Failed to join the domain. You are already a member.');
 export const InvalidJoinInvitationCodeError = Err('InvalidJoinInvitationCodeError', ForbiddenError, 'The invitation code you provided is invalid.');
 export const CurrentPasswordError = Err('CurrentPasswordError', ForbiddenError, "Current password doesn't match.");
+export const DiscussionLockedError = Err('DiscussionLockedError', ForbiddenError, 'The discussion is locked, you can not reply anymore.');
 
 export const UserNotFoundError = Err('UserNotFoundError', NotFoundError, 'User {0} not found.');
 export const NoProblemError = Err('NoProblemError', NotFoundError, 'No problem.');

packages/hydrooj/src/handler/discussion.ts

@@ -1,6 +1,9 @@
 import { isSafeInteger } from 'lodash';
 import { ObjectID } from 'mongodb';
-import { DiscussionNotFoundError, DocumentNotFoundError, PermissionError } from '../error';
+import {
+    DiscussionLockedError, DiscussionNotFoundError, DocumentNotFoundError,
+    PermissionError,
+} from '../error';
 import { DiscussionDoc, DiscussionReplyDoc, DiscussionTailReplyDoc } from '../interface';
 import paginate from '../lib/paginate';
 import { PERM, PRIV } from '../model/builtin';
@@ -207,6 +210,14 @@ class DiscussionDetailHandler extends DiscussionHandler {
         this.checkPriv(PRIV.PRIV_USER_PROFILE);
     }
 
+    @param('did', Types.ObjectID)
+    @param('lock', Types.Boolean)
+    async postSetLock(domainId: string, did: ObjectID, lock: boolean) {
+        if (!this.user.own(this.ddoc)) this.checkPerm(PERM.PERM_LOCK_DISCUSSION);
+        await discussion.edit(domainId, did, { lock });
+        this.back();
+    }
+
     @param('type', Types.Range(['did', 'drid']))
     @param('id', Types.ObjectID)
     @param('emoji', Types.Emoji)
@@ -223,6 +234,7 @@ class DiscussionDetailHandler extends DiscussionHandler {
     @param('content', Types.Content)
     async postReply(domainId: string, did: ObjectID, content: string) {
         this.checkPerm(PERM.PERM_REPLY_DISCUSSION);
+        if (this.ddoc.lock) throw new DiscussionLockedError(domainId, did);
         await this.limitRate('add_discussion', 3600, 60);
         const targets = new Set(Array.from(content.matchAll(/@\[\]\(\/user\/(\d+)\)/g)).map((i) => +i[1]));
         const uids = Object.keys(await user.getList(domainId, Array.from(targets))).map((i) => +i);
@@ -241,6 +253,7 @@ class DiscussionDetailHandler extends DiscussionHandler {
     @param('content', Types.Content)
     async postTailReply(domainId: string, drid: ObjectID, content: string) {
         this.checkPerm(PERM.PERM_REPLY_DISCUSSION);
+        if (this.ddoc.lock) throw new DiscussionLockedError(domainId, this.ddoc.docId);
         await this.limitRate('add_discussion', 3600, 60);
         const targets = new Set(Array.from(content.matchAll(/@\[\]\(\/user\/(\d+)\)/g)).map((i) => +i[1]));
         const uids = Object.keys(await user.getList(domainId, Array.from(targets))).map((i) => +i);
@@ -355,7 +368,9 @@ class DiscussionEditHandler extends DiscussionHandler {
         if (!this.user.hasPerm(PERM.PERM_HIGHLIGHT_DISCUSSION)) highlight = this.ddoc.highlight;
         if (!this.user.hasPerm(PERM.PERM_PIN_DISCUSSION)) pin = this.ddoc.pin;
         await Promise.all([
-            discussion.edit(domainId, did, title, content, highlight, pin),
+            discussion.edit(domainId, did, {
+                title, content, highlight, pin,
+            }),
             oplog.log(this, 'discussion.edit', this.ddoc),
         ]);
         this.response.body = { did };

packages/hydrooj/src/interface.ts

@@ -435,6 +435,7 @@ declare module './model/discussion' {
         react: Record<string, number>;
         sort: number;
         lastRCount: number;
+        lock?: boolean;
     }
 }
 

packages/hydrooj/src/model/builtin.ts

@@ -58,6 +58,7 @@ export const PERM = {
     PERM_DELETE_DISCUSSION_REPLY: 1n << 38n,
     PERM_DELETE_DISCUSSION_REPLY_SELF: 1n << 39n,
     PERM_DELETE_DISCUSSION_REPLY_SELF_DISCUSSION: 1n << 40n,
+    PERM_LOCK_DISCUSSION: 64n,
 
     // Contest
     PERM_VIEW_CONTEST: 1n << 41n,
@@ -131,6 +132,7 @@ export const PERMS = [
     Permission('perm_discussion', PERM.PERM_PIN_DISCUSSION, 'Pin discussions'),
     Permission('perm_discussion', PERM.PERM_EDIT_DISCUSSION, 'Edit discussions'),
     Permission('perm_discussion', PERM.PERM_EDIT_DISCUSSION_SELF, 'Edit own discussions'),
+    Permission('perm_discussion', PERM.PERM_LOCK_DISCUSSION, 'Lock discussions'),
     Permission('perm_discussion', PERM.PERM_DELETE_DISCUSSION, 'Delete discussions'),
     Permission('perm_discussion', PERM.PERM_DELETE_DISCUSSION_SELF, 'Delete own discussions'),
     Permission('perm_discussion', PERM.PERM_REPLY_DISCUSSION, 'Reply discussions'),

packages/hydrooj/src/model/discussion.ts

@@ -57,6 +57,7 @@ export const PROJECTION_LIST: DiscussionDoc.Field[] = [
 ];
 export const PROJECTION_PUBLIC: DiscussionDoc.Field[] = [
     ...PROJECTION_LIST, 'content', 'history', 'react', 'maintainer',
+    'lock',
 ];
 
 export const typeDisplay = {
@@ -103,14 +104,8 @@ export async function get<T extends DiscussionDoc.Field>(
     return await document.get(domainId, document.TYPE_DISCUSSION, did, projection);
 }
 
-export function edit(
-    domainId: string, did: ObjectID,
-    title: string, content: string, highlight: boolean, pin: boolean,
-): Promise<DiscussionDoc | null> {
-    const payload = {
-        title, content, highlight, pin,
-    };
-    return document.set(domainId, document.TYPE_DISCUSSION, did, payload);
+export function edit(domainId: string, did: ObjectID, $set: Partial<DiscussionDoc>) {
+    return document.set(domainId, document.TYPE_DISCUSSION, did, $set);
 }
 
 export function inc(

packages/hydrooj/src/model/message.ts

@@ -26,7 +26,7 @@ class MessageModel {
             from, to, content, _id: res.insertedId, flag,
         };
         if (from !== to) bus.broadcast('user/message', to, mdoc);
-        await user.inc(to, 'unreadMsg', 1);
+        if (flag & MessageModel.FLAG_UNREAD) await user.inc(to, 'unreadMsg', 1);
         return mdoc;
     }