Sharkey/src/models/note.ts

391 lines
8.6 KiB
TypeScript
Raw Normal View History

2017-09-07 22:13:01 +03:00
import * as mongo from 'mongodb';
2018-06-18 03:54:53 +03:00
const deepcopy = require('deepcopy');
2018-02-02 01:06:01 +02:00
import rap from '@prezzemolo/rap';
2018-03-29 14:32:18 +03:00
import db from '../db/mongodb';
2018-10-16 05:38:09 +03:00
import isObjectId from '../misc/is-objectid';
import { length } from 'stringz';
2018-02-02 01:06:01 +02:00
import { IUser, pack as packUser } from './user';
import { pack as packApp } from './app';
import PollVote from './poll-vote';
import Reaction from './note-reaction';
import { packMany as packFileMany, IDriveFile } from './drive-file';
import Favorite from './favorite';
2018-04-28 22:30:51 +03:00
import Following from './following';
import Emoji from './emoji';
2018-02-02 01:06:01 +02:00
2018-04-07 20:30:37 +03:00
const Note = db.get<INote>('notes');
Note.createIndex('uri', { sparse: true, unique: true });
Note.createIndex('userId');
Note.createIndex('mentions');
Note.createIndex('visibleUserIds');
2018-11-23 16:12:28 +02:00
Note.createIndex('replyId');
2018-06-16 09:23:03 +03:00
Note.createIndex('tagsLower');
2018-10-25 01:04:15 +03:00
Note.createIndex('_user.host');
2018-09-25 15:09:38 +03:00
Note.createIndex('_files._id');
2018-09-05 13:32:46 +03:00
Note.createIndex('_files.contentType');
2018-10-25 01:04:15 +03:00
Note.createIndex({ createdAt: -1 });
Note.createIndex({ score: -1 }, { sparse: true });
2018-04-07 20:30:37 +03:00
export default Note;
2017-03-01 20:16:39 +02:00
2018-03-30 05:24:07 +03:00
export function isValidCw(text: string): boolean {
return length(text.trim()) <= 100;
2018-03-30 05:24:07 +03:00
}
2018-04-07 20:30:37 +03:00
export type INote = {
2017-09-07 22:13:01 +03:00
_id: mongo.ObjectID;
2018-03-29 08:48:47 +03:00
createdAt: Date;
deletedAt: Date;
2018-09-05 13:32:46 +03:00
fileIds: mongo.ObjectID[];
2018-03-29 08:48:47 +03:00
replyId: mongo.ObjectID;
2018-04-07 20:30:37 +03:00
renoteId: mongo.ObjectID;
2018-06-18 03:54:53 +03:00
poll: {
choices: Array<{
id: number;
}>
};
2017-09-07 22:13:01 +03:00
text: string;
2018-04-01 12:07:29 +03:00
tags: string[];
2018-06-16 09:23:03 +03:00
tagsLower: string[];
emojis: string[];
2018-03-30 05:24:07 +03:00
cw: string;
2018-03-29 08:48:47 +03:00
userId: mongo.ObjectID;
appId: mongo.ObjectID;
viaMobile: boolean;
localOnly: boolean;
2018-04-07 20:30:37 +03:00
renoteCount: number;
2018-03-29 08:48:47 +03:00
repliesCount: number;
reactionCounts: any;
mentions: mongo.ObjectID[];
2018-06-12 23:11:55 +03:00
mentionedRemoteUsers: Array<{
uri: string;
username: string;
host: string;
}>;
2018-04-28 11:25:56 +03:00
/**
* public ...
* home ... ()
* followers ...
2018-04-28 22:30:51 +03:00
* specified ... visibleUserIds
2018-04-28 11:25:56 +03:00
*/
2018-12-28 19:55:46 +02:00
visibility: 'public' | 'home' | 'followers' | 'specified';
2018-04-28 22:30:51 +03:00
visibleUserIds: mongo.ObjectID[];
2018-04-28 11:25:56 +03:00
2018-03-05 01:44:37 +02:00
geo: {
2018-03-29 09:23:15 +03:00
coordinates: number[];
2018-03-05 01:44:37 +02:00
altitude: number;
accuracy: number;
altitudeAccuracy: number;
heading: number;
speed: number;
};
2018-10-25 01:04:15 +03:00
2018-04-03 17:45:13 +03:00
uri: string;
2018-04-05 22:04:50 +03:00
2018-10-25 01:04:15 +03:00
/**
* 稿
*/
score: number;
2018-04-19 06:43:25 +03:00
// 非正規化
2018-04-05 22:04:50 +03:00
_reply?: {
userId: mongo.ObjectID;
};
2018-04-07 20:30:37 +03:00
_renote?: {
2018-04-05 22:04:50 +03:00
userId: mongo.ObjectID;
};
_user: {
host: string;
2018-04-19 06:43:25 +03:00
inbox?: string;
2018-04-05 22:04:50 +03:00
};
2018-09-05 13:32:46 +03:00
_files?: IDriveFile[];
2017-09-07 22:13:01 +03:00
};
2018-02-02 01:06:01 +02:00
2018-09-09 20:43:16 +03:00
export const hideNote = async (packedNote: any, meId: mongo.ObjectID) => {
let hide = false;
2018-12-28 19:55:46 +02:00
// visibility が private かつ投稿者のIDが自分のIDではなかったら非表示(後方互換性のため)
2018-09-09 20:43:16 +03:00
if (packedNote.visibility == 'private' && (meId == null || !meId.equals(packedNote.userId))) {
hide = true;
}
// visibility が specified かつ自分が指定されていなかったら非表示
if (packedNote.visibility == 'specified') {
if (meId == null) {
hide = true;
} else if (meId.equals(packedNote.userId)) {
hide = false;
} else {
// 指定されているかどうか
2018-09-17 20:13:42 +03:00
const specified = packedNote.visibleUserIds.some((id: any) => meId.equals(id));
2018-09-09 20:43:16 +03:00
if (specified) {
hide = false;
} else {
hide = true;
}
}
}
// visibility が followers かつ自分が投稿者のフォロワーでなかったら非表示
if (packedNote.visibility == 'followers') {
if (meId == null) {
hide = true;
} else if (meId.equals(packedNote.userId)) {
hide = false;
} else {
// フォロワーかどうか
const following = await Following.findOne({
followeeId: packedNote.userId,
followerId: meId
});
if (following == null) {
hide = true;
} else {
hide = false;
}
}
}
if (hide) {
packedNote.fileIds = [];
packedNote.files = [];
packedNote.text = null;
packedNote.poll = null;
packedNote.cw = null;
packedNote.tags = [];
packedNote.geo = null;
packedNote.isHidden = true;
}
};
2018-10-31 04:22:49 +02:00
export const packMany = (
notes: (string | mongo.ObjectID | INote)[],
me?: string | mongo.ObjectID | IUser,
options?: {
detail?: boolean;
skipHide?: boolean;
}
) => {
2018-10-31 04:22:49 +02:00
return Promise.all(notes.map(n => pack(n, me, options)));
};
2018-02-02 01:06:01 +02:00
/**
2018-04-07 20:30:37 +03:00
* Pack a note for API response
2018-02-02 01:06:01 +02:00
*
2018-04-07 20:30:37 +03:00
* @param note target
2018-02-02 01:06:01 +02:00
* @param me? serializee
* @param options? serialize options
* @return response
*/
export const pack = async (
2018-04-07 20:30:37 +03:00
note: string | mongo.ObjectID | INote,
2018-02-02 01:06:01 +02:00
me?: string | mongo.ObjectID | IUser,
options?: {
2018-09-09 20:43:16 +03:00
detail?: boolean;
skipHide?: boolean;
2018-02-02 01:06:01 +02:00
}
) => {
2018-04-29 01:01:47 +03:00
const opts = Object.assign({
2018-09-09 20:43:16 +03:00
detail: true,
skipHide: false
2018-04-29 01:01:47 +03:00
}, options);
2018-02-02 01:06:01 +02:00
// Me
const meId: mongo.ObjectID = me
2018-10-16 05:38:09 +03:00
? isObjectId(me)
2018-02-02 01:06:01 +02:00
? me as mongo.ObjectID
: typeof me === 'string'
? new mongo.ObjectID(me)
: (me as IUser)._id
: null;
2018-04-07 20:30:37 +03:00
let _note: any;
2018-02-02 01:06:01 +02:00
2018-04-07 20:30:37 +03:00
// Populate the note if 'note' is ID
2018-10-16 05:38:09 +03:00
if (isObjectId(note)) {
2018-04-07 20:30:37 +03:00
_note = await Note.findOne({
_id: note
2018-02-02 01:06:01 +02:00
});
2018-04-07 20:30:37 +03:00
} else if (typeof note === 'string') {
_note = await Note.findOne({
_id: new mongo.ObjectID(note)
2018-02-02 01:06:01 +02:00
});
} else {
2018-04-07 20:30:37 +03:00
_note = deepcopy(note);
2018-02-02 01:06:01 +02:00
}
2018-10-10 20:19:21 +03:00
// (データベースの欠損などで)投稿がデータベース上に見つからなかったとき
if (_note == null) {
2018-10-10 20:19:21 +03:00
console.warn(`[DAMAGED DB] (missing) pkg: note :: ${note}`);
return null;
}
2018-02-02 01:06:01 +02:00
2018-04-07 20:30:37 +03:00
const id = _note._id;
2018-02-02 01:06:01 +02:00
// _note._userを消す前か、_note.userを解決した後でないとホストがわからない
if (_note._user) {
const host = _note._user.host;
// 互換性のため。(古いMisskeyではNoteにemojisが無い)
if (_note.emojis == null) {
_note.emojis = Emoji.find({
host: host
}, {
fields: { _id: false }
});
} else {
_note.emojis = Emoji.find({
name: { $in: _note.emojis },
host: host
}, {
fields: { _id: false }
});
}
}
2018-02-02 01:06:01 +02:00
// Rename _id to id
2018-04-07 20:30:37 +03:00
_note.id = _note._id;
delete _note._id;
2018-02-02 01:06:01 +02:00
2018-10-07 19:25:34 +03:00
delete _note.prev;
delete _note.next;
delete _note.tagsLower;
2018-10-25 01:04:15 +03:00
delete _note.score;
2018-04-29 01:01:47 +03:00
delete _note._user;
delete _note._reply;
2018-09-19 08:18:34 +03:00
delete _note._renote;
delete _note._files;
2018-11-23 16:12:28 +02:00
delete _note._replyIds;
2018-04-07 20:30:37 +03:00
if (_note.geo) delete _note.geo.type;
2018-02-02 01:06:01 +02:00
// Populate user
2018-04-07 20:30:37 +03:00
_note.user = packUser(_note.userId, meId);
2018-02-02 01:06:01 +02:00
// Populate app
2018-04-07 20:30:37 +03:00
if (_note.appId) {
_note.app = packApp(_note.appId);
2018-02-02 01:06:01 +02:00
}
2018-09-05 13:32:46 +03:00
// Populate files
_note.files = packFileMany(_note.fileIds || []);
2018-02-02 01:06:01 +02:00
2018-12-28 19:04:29 +02:00
// Some counts
_note.renoteCount = _note.renoteCount || 0;
_note.repliesCount = _note.repliesCount || 0;
_note.reactionCounts = _note.reactionCounts || {};
2018-09-05 13:32:46 +03:00
// 後方互換性のため
_note.mediaIds = _note.fileIds;
_note.media = _note.files;
2018-04-07 20:30:37 +03:00
// When requested a detailed note data
2018-02-02 01:06:01 +02:00
if (opts.detail) {
2018-04-07 20:30:37 +03:00
if (_note.replyId) {
// Populate reply to note
_note.reply = pack(_note.replyId, meId, {
2018-02-02 01:06:01 +02:00
detail: false
});
}
2018-04-07 20:30:37 +03:00
if (_note.renoteId) {
// Populate renote
_note.renote = pack(_note.renoteId, meId, {
detail: _note.text == null
2018-02-02 01:06:01 +02:00
});
}
// Poll
2018-09-09 20:43:16 +03:00
if (meId && _note.poll) {
2018-06-12 12:54:36 +03:00
_note.poll = (async poll => {
2018-04-15 00:34:55 +03:00
const vote = await PollVote
2018-02-02 01:06:01 +02:00
.findOne({
2018-03-29 08:48:47 +03:00
userId: meId,
2018-04-07 20:30:37 +03:00
noteId: id
2018-02-02 01:06:01 +02:00
});
if (vote != null) {
const myChoice = poll.choices
2018-06-18 03:54:53 +03:00
.filter((c: any) => c.id == vote.choice)[0];
2018-02-02 01:06:01 +02:00
2018-03-29 08:48:47 +03:00
myChoice.isVoted = true;
2018-02-02 01:06:01 +02:00
}
return poll;
2018-04-07 20:30:37 +03:00
})(_note.poll);
2018-02-02 01:06:01 +02:00
}
if (meId) {
// Fetch my reaction
2018-04-07 20:30:37 +03:00
_note.myReaction = (async () => {
2018-02-02 01:06:01 +02:00
const reaction = await Reaction
.findOne({
2018-03-29 08:48:47 +03:00
userId: meId,
2018-04-07 20:30:37 +03:00
noteId: id,
2018-03-29 08:48:47 +03:00
deletedAt: { $exists: false }
2018-02-02 01:06:01 +02:00
});
if (reaction) {
return reaction.reaction;
}
return null;
})();
// isFavorited
_note.isFavorited = (async () => {
const favorite = await Favorite
.count({
userId: meId,
noteId: id
}, {
limit: 1
});
return favorite === 1;
})();
2018-02-02 01:06:01 +02:00
}
}
2018-04-07 20:30:37 +03:00
// resolve promises in _note object
_note = await rap(_note);
2018-02-02 01:06:01 +02:00
2018-10-10 20:11:12 +03:00
//#region (データベースの欠損などで)参照しているデータがデータベース上に見つからなかったとき
if (_note.user == null) {
2018-10-10 20:19:21 +03:00
console.warn(`[DAMAGED DB] (missing) pkg: note -> user :: ${_note.id} (user ${_note.userId})`);
return null;
}
2018-10-11 11:00:22 +03:00
if (opts.detail) {
if (_note.replyId != null && _note.reply == null) {
console.warn(`[DAMAGED DB] (missing) pkg: note -> reply :: ${_note.id} (reply ${_note.replyId})`);
return null;
}
2018-10-10 20:11:12 +03:00
2018-10-11 11:00:22 +03:00
if (_note.renoteId != null && _note.renote == null) {
console.warn(`[DAMAGED DB] (missing) pkg: note -> renote :: ${_note.id} (renote ${_note.renoteId})`);
return null;
}
2018-10-10 20:11:12 +03:00
}
//#endregion
2018-05-21 05:08:08 +03:00
if (_note.user.isCat && _note.text) {
_note.text = (_note.text
// ja-JP
.replace(/な/g, 'にゃ').replace(/ナ/g, 'ニャ').replace(/ナ/g, 'ニャ')
// ko-KR
.replace(/[나-낳]/g, (match: string) => String.fromCharCode(
match.codePointAt(0) + '냐'.charCodeAt(0) - '나'.charCodeAt(0)
))
);
2018-05-21 05:08:08 +03:00
}
2018-09-09 20:43:16 +03:00
if (!opts.skipHide) {
await hideNote(_note, meId);
2018-04-28 22:51:19 +03:00
}
2018-04-07 20:30:37 +03:00
return _note;
2018-02-02 01:06:01 +02:00
};