良い感じに

src/models/mute.ts

@@ -1,7 +1,12 @@
 import * as mongo from 'mongodb';
 import db from '../db/mongodb';
+import isObjectId from '../misc/is-objectid';
+const deepcopy = require('deepcopy');
+import { pack as packUser } from './user';
 
 const Mute = db.get<IMute>('mute');
+Mute.createIndex('muterId');
+Mute.createIndex('muteeId');
 Mute.createIndex(['muterId', 'muteeId'], { unique: true });
 export default Mute;
 
@@ -11,3 +16,39 @@ export interface IMute {
 	muterId: mongo.ObjectID;
 	muteeId: mongo.ObjectID;
 }
+
+export const packMany = async (
+	mutes: (string | mongo.ObjectID | IMute)[],
+	me?: string | mongo.ObjectID | IUser
+) => {
+	return (await Promise.all(mutes.map(x => pack(x, me)))).filter(x => x != null);
+};
+
+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);
+});