misskey/src/models/blocking.ts

55 lines
1.3 KiB
TypeScript
Raw Normal View History

import * as mongo from 'mongodb';
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';
const Blocking = db.get<IBlocking>('blocking');
2018-10-31 03:16:13 +01:00
Blocking.createIndex('blockerId');
Blocking.createIndex('blockeeId');
Blocking.createIndex(['blockerId', 'blockeeId'], { unique: true });
export default Blocking;
export type IBlocking = {
_id: mongo.ObjectID;
createdAt: Date;
blockeeId: mongo.ObjectID;
blockerId: mongo.ObjectID;
};
2018-10-31 03:16:13 +01:00
export const packMany = async (
blockings: (string | mongo.ObjectID | IBlocking)[],
me?: string | mongo.ObjectID | IUser
) => {
2018-10-31 03:20:54 +01:00
return (await Promise.all(blockings.map(x => pack(x, me))));
2018-10-31 03:16:13 +01:00
};
export const pack = (
blocking: any,
me?: any
) => new Promise<any>(async (resolve, reject) => {
let _blocking: any;
// Populate the blocking if 'blocking' is ID
if (isObjectId(blocking)) {
_blocking = await Blocking.findOne({
_id: blocking
});
} else if (typeof blocking === 'string') {
_blocking = await Blocking.findOne({
_id: new mongo.ObjectID(blocking)
});
} else {
_blocking = deepcopy(blocking);
}
// Rename _id to id
_blocking.id = _blocking._id;
delete _blocking._id;
// Populate blockee
_blocking.blockee = await packUser(_blocking.blockeeId, me);
resolve(_blocking);
});