misskey/src/models/favorite.ts

65 lines
1.5 KiB
TypeScript
Raw Normal View History

2018-03-29 07:48:47 +02:00
import * as mongo from 'mongodb';
2018-06-18 02:54:53 +02:00
const deepcopy = require('deepcopy');
2018-03-29 13:32:18 +02:00
import db from '../db/mongodb';
2018-10-16 04:38:09 +02:00
import isObjectId from '../misc/is-objectid';
2018-04-20 06:31:43 +02:00
import { pack as packNote } from './note';
2017-01-17 01:12:33 +01:00
2018-04-11 20:46:32 +02:00
const Favorite = db.get<IFavorite>('favorites');
2018-10-29 13:53:40 +01:00
Favorite.createIndex('userId');
2018-04-20 05:38:31 +02:00
Favorite.createIndex(['userId', 'noteId'], { unique: true });
2018-04-11 20:46:32 +02:00
export default Favorite;
2018-03-29 07:48:47 +02:00
export type IFavorite = {
_id: mongo.ObjectID;
createdAt: Date;
userId: mongo.ObjectID;
2018-04-07 19:30:37 +02:00
noteId: mongo.ObjectID;
2018-03-29 07:48:47 +02:00
};
2018-04-11 20:46:32 +02:00
2018-10-31 03:22:49 +01:00
export const packMany = (
2018-10-04 06:33:59 +02:00
favorites: any[],
me: any
) => {
2018-10-31 03:22:49 +01:00
return Promise.all(favorites.map(f => pack(f, me)));
2018-10-04 06:33:59 +02:00
};
2018-04-20 06:31:43 +02:00
/**
* Pack a favorite for API response
*/
export const pack = (
favorite: any,
me: any
) => new Promise<any>(async (resolve, reject) => {
let _favorite: any;
// Populate the favorite if 'favorite' is ID
2018-10-16 04:38:09 +02:00
if (isObjectId(favorite)) {
2018-04-20 06:31:43 +02:00
_favorite = await Favorite.findOne({
_id: favorite
});
} else if (typeof favorite === 'string') {
_favorite = await Favorite.findOne({
_id: new mongo.ObjectID(favorite)
});
} else {
_favorite = deepcopy(favorite);
}
// Rename _id to id
_favorite.id = _favorite._id;
delete _favorite._id;
// Populate note
_favorite.note = await packNote(_favorite.noteId, me, {
detail: true
});
2018-04-20 06:31:43 +02:00
2018-10-04 06:33:59 +02:00
// (データベースの不具合などで)投稿が見つからなかったら
if (_favorite.note == null) {
2018-10-10 19:19:21 +02:00
console.warn(`[DAMAGED DB] (missing) pkg: favorite -> note :: ${_favorite.id} (note ${_favorite.noteId})`);
2018-10-04 06:33:59 +02:00
return resolve(null);
}
2018-04-20 06:31:43 +02:00
resolve(_favorite);
});