54 lines
1.2 KiB
TypeScript
54 lines
1.2 KiB
TypeScript
import * as mongo from 'mongodb';
|
|
import db from '../db/mongodb';
|
|
import isObjectId from '../misc/is-objectid';
|
|
const deepcopy = require('deepcopy');
|
|
import { pack as packUser, IUser } from './user';
|
|
|
|
const Mute = db.get<IMute>('mute');
|
|
Mute.createIndex('muterId');
|
|
Mute.createIndex('muteeId');
|
|
Mute.createIndex(['muterId', 'muteeId'], { unique: true });
|
|
export default Mute;
|
|
|
|
export interface IMute {
|
|
_id: mongo.ObjectID;
|
|
createdAt: Date;
|
|
muterId: mongo.ObjectID;
|
|
muteeId: mongo.ObjectID;
|
|
}
|
|
|
|
export const packMany = (
|
|
mutes: (string | mongo.ObjectID | IMute)[],
|
|
me?: string | mongo.ObjectID | IUser
|
|
) => {
|
|
return Promise.all(mutes.map(x => pack(x, me)));
|
|
};
|
|
|
|
export const pack = (
|
|
mute: any,
|
|
me?: any
|
|
) => new Promise<any>(async (resolve, reject) => {
|
|
let _mute: any;
|
|
|
|
// Populate the mute if 'mute' is ID
|
|
if (isObjectId(mute)) {
|
|
_mute = await Mute.findOne({
|
|
_id: mute
|
|
});
|
|
} else if (typeof mute === 'string') {
|
|
_mute = await Mute.findOne({
|
|
_id: new mongo.ObjectID(mute)
|
|
});
|
|
} else {
|
|
_mute = deepcopy(mute);
|
|
}
|
|
|
|
// Rename _id to id
|
|
_mute.id = _mute._id;
|
|
delete _mute._id;
|
|
|
|
// Populate mutee
|
|
_mute.mutee = await packUser(_mute.muteeId, me);
|
|
|
|
resolve(_mute);
|
|
});
|