文件结构更改

packages/hydrooj/src/model/message.ts

@@ -0,0 +1,85 @@
+import { ObjectID } from 'mongodb';
+import * as user from './user';
+import { Mdoc } from '../interface';
+import * as db from '../service/db';
+import * as bus from '../service/bus';
+
+const coll = db.collection('message');
+
+export const FLAG_UNREAD = 1;
+export const FLAG_ALERT = 2;
+
+export async function send(
+    from: number, to: number,
+    content: string, flag: number = FLAG_UNREAD,
+): Promise<Mdoc> {
+    const res = await coll.insertOne({
+        from, to, content, flag,
+    });
+    const mdoc = {
+        from, to, content, _id: res.insertedId, flag,
+    };
+    if (from !== to) {
+        // ENHANCE domainId?
+        const udoc = await user.getById('system', to);
+        bus.publish(`user_message-${to}`, { mdoc, udoc });
+    }
+    return mdoc;
+}
+
+export async function get(_id: ObjectID): Promise<Mdoc | null> {
+    return await coll.findOne({ _id });
+}
+
+export async function getByUser(uid: number): Promise<Mdoc[]> {
+    return await coll.find({ $or: [{ from: uid }, { to: uid }] }).sort('_id', 1).toArray();
+}
+
+export async function getMany(query: any, sort: any, page: number, limit: number): Promise<Mdoc[]> {
+    return await coll.find(query).sort(sort)
+        .skip((page - 1) * limit).limit(limit)
+        .toArray();
+}
+
+export async function setFlag(messageId: ObjectID, flag: number): Promise<Mdoc | null> {
+    const result = await coll.findOneAndUpdate(
+        { _id: messageId },
+        { $bit: { flag: { xor: flag } } },
+        { returnOriginal: false },
+    );
+    return result.value;
+}
+
+export async function del(_id: ObjectID) {
+    return await coll.deleteOne({ _id });
+}
+
+export function count(query: any) {
+    return coll.find(query).count();
+}
+
+export function getMulti(uid: number) {
+    return coll.find({ $or: [{ from: uid }, { to: uid }] });
+}
+
+function ensureIndexes() {
+    return Promise.all([
+        coll.createIndex({ to: 1, _id: -1 }),
+        coll.createIndex({ from: 1, _id: -1 }),
+    ]);
+}
+
+global.Hydro.postInit.push(ensureIndexes);
+global.Hydro.model.message = {
+    FLAG_UNREAD,
+    FLAG_ALERT,
+
+    count,
+    get,
+    getByUser,
+    del,
+    setFlag,
+    getMany,
+    getMulti,
+    send,
+};