misskey/src/models/mute.ts

55 lines
1.2 KiB
TypeScript
Raw Normal View History

2018-03-29 05:48:47 +00:00
import * as mongo from 'mongodb';
2018-03-29 11:32:18 +00:00
import db from '../db/mongodb';
2018-10-31 02:16:13 +00:00
import isObjectId from '../misc/is-objectid';
const deepcopy = require('deepcopy');
2018-10-31 02:20:54 +00:00
import { pack as packUser, IUser } from './user';
2017-12-21 19:50:50 +00:00
2018-03-29 05:48:47 +00:00
const Mute = db.get<IMute>('mute');
2018-10-31 02:16:13 +00:00
Mute.createIndex('muterId');
Mute.createIndex('muteeId');
Mute.createIndex(['muterId', 'muteeId'], { unique: true });
2018-03-29 05:48:47 +00:00
export default Mute;
export interface IMute {
_id: mongo.ObjectID;
createdAt: Date;
muterId: mongo.ObjectID;
muteeId: mongo.ObjectID;
}
2018-10-31 02:16:13 +00:00
2018-10-31 02:22:49 +00:00
export const packMany = (
2018-10-31 02:16:13 +00:00
mutes: (string | mongo.ObjectID | IMute)[],
me?: string | mongo.ObjectID | IUser
) => {
2018-10-31 02:22:49 +00:00
return Promise.all(mutes.map(x => pack(x, me)));
2018-10-31 02:16:13 +00:00
};
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);
});