misskey/src/models/mute.ts

57 lines
1.2 KiB
TypeScript
Raw Normal View History

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