rudeshark.net/src/models/note.ts

242 lines
4.5 KiB
TypeScript
Raw Normal View History

2017-09-07 21:13:01 +02:00
import * as mongo from 'mongodb';
2018-02-02 00:06:01 +01:00
import deepcopy = require('deepcopy');
import rap from '@prezzemolo/rap';
2018-03-29 13:32:18 +02:00
import db from '../db/mongodb';
2018-02-02 00:06:01 +01:00
import { IUser, pack as packUser } from './user';
import { pack as packApp } from './app';
import { pack as packChannel } from './channel';
import Vote from './poll-vote';
2018-04-07 19:30:37 +02:00
import Reaction from './note-reaction';
2018-02-02 00:06:01 +01:00
import { pack as packFile } from './drive-file';
2018-04-07 19:30:37 +02:00
const Note = db.get<INote>('notes');
2017-01-17 01:12:33 +01:00
2018-04-07 19:30:37 +02:00
Note.createIndex('uri', { sparse: true, unique: true });
2018-04-03 16:45:13 +02:00
2018-04-07 19:30:37 +02:00
export default Note;
2017-03-01 19:16:39 +01:00
export function isValidText(text: string): boolean {
return text.length <= 1000 && text.trim() != '';
}
2017-09-07 21:13:01 +02:00
2018-03-30 04:24:07 +02:00
export function isValidCw(text: string): boolean {
return text.length <= 100 && text.trim() != '';
}
2018-04-07 19:30:37 +02:00
export type INote = {
2017-09-07 21:13:01 +02:00
_id: mongo.ObjectID;
2018-03-29 07:48:47 +02:00
channelId: mongo.ObjectID;
createdAt: Date;
deletedAt: Date;
2018-03-29 07:48:47 +02:00
mediaIds: mongo.ObjectID[];
replyId: mongo.ObjectID;
2018-04-07 19:30:37 +02:00
renoteId: mongo.ObjectID;
2018-02-04 06:52:33 +01:00
poll: any; // todo
2017-09-07 21:13:01 +02:00
text: string;
2018-04-01 11:07:29 +02:00
tags: string[];
2018-03-31 12:53:30 +02:00
textHtml: string;
2018-03-30 04:24:07 +02:00
cw: string;
2018-03-29 07:48:47 +02:00
userId: mongo.ObjectID;
appId: mongo.ObjectID;
viaMobile: boolean;
2018-04-07 19:30:37 +02:00
renoteCount: number;
2018-03-29 07:48:47 +02:00
repliesCount: number;
reactionCounts: any;
mentions: mongo.ObjectID[];
2018-04-02 07:18:01 +02:00
visibility: 'public' | 'unlisted' | 'private' | 'direct';
2018-03-05 00:44:37 +01:00
geo: {
2018-03-29 08:23:15 +02:00
coordinates: number[];
2018-03-05 00:44:37 +01:00
altitude: number;
accuracy: number;
altitudeAccuracy: number;
heading: number;
speed: number;
};
2018-04-03 16:45:13 +02:00
uri: string;
2018-04-05 21:04:50 +02:00
_reply?: {
userId: mongo.ObjectID;
};
2018-04-07 19:30:37 +02:00
_renote?: {
2018-04-05 21:04:50 +02:00
userId: mongo.ObjectID;
};
_user: {
host: string;
hostLower: string;
account: {
inbox?: string;
};
};
2017-09-07 21:13:01 +02:00
};
2018-02-02 00:06:01 +01:00
/**
2018-04-07 19:30:37 +02:00
* Pack a note for API response
2018-02-02 00:06:01 +01:00
*
2018-04-07 19:30:37 +02:00
* @param note target
2018-02-02 00:06:01 +01:00
* @param me? serializee
* @param options? serialize options
* @return response
*/
export const pack = async (
2018-04-07 19:30:37 +02:00
note: string | mongo.ObjectID | INote,
2018-02-02 00:06:01 +01:00
me?: string | mongo.ObjectID | IUser,
options?: {
detail: boolean
}
) => {
const opts = options || {
detail: true,
};
// Me
const meId: mongo.ObjectID = me
? mongo.ObjectID.prototype.isPrototypeOf(me)
? me as mongo.ObjectID
: typeof me === 'string'
? new mongo.ObjectID(me)
: (me as IUser)._id
: null;
2018-04-07 19:30:37 +02:00
let _note: any;
2018-02-02 00:06:01 +01:00
2018-04-07 19:30:37 +02:00
// Populate the note if 'note' is ID
if (mongo.ObjectID.prototype.isPrototypeOf(note)) {
_note = await Note.findOne({
_id: note
2018-02-02 00:06:01 +01:00
});
2018-04-07 19:30:37 +02:00
} else if (typeof note === 'string') {
_note = await Note.findOne({
_id: new mongo.ObjectID(note)
2018-02-02 00:06:01 +01:00
});
} else {
2018-04-07 19:30:37 +02:00
_note = deepcopy(note);
2018-02-02 00:06:01 +01:00
}
2018-04-07 19:30:37 +02:00
if (!_note) throw 'invalid note arg.';
2018-02-02 00:06:01 +01:00
2018-04-07 19:30:37 +02:00
const id = _note._id;
2018-02-02 00:06:01 +01:00
// Rename _id to id
2018-04-07 19:30:37 +02:00
_note.id = _note._id;
delete _note._id;
2018-02-02 00:06:01 +01:00
2018-04-07 19:30:37 +02:00
delete _note.mentions;
if (_note.geo) delete _note.geo.type;
2018-02-02 00:06:01 +01:00
// Populate user
2018-04-07 19:30:37 +02:00
_note.user = packUser(_note.userId, meId);
2018-02-02 00:06:01 +01:00
// Populate app
2018-04-07 19:30:37 +02:00
if (_note.appId) {
_note.app = packApp(_note.appId);
2018-02-02 00:06:01 +01:00
}
// Populate channel
2018-04-07 19:30:37 +02:00
if (_note.channelId) {
_note.channel = packChannel(_note.channelId);
2018-02-02 00:06:01 +01:00
}
// Populate media
2018-04-07 19:30:37 +02:00
if (_note.mediaIds) {
_note.media = Promise.all(_note.mediaIds.map(fileId =>
2018-02-02 00:06:01 +01:00
packFile(fileId)
));
}
2018-04-07 19:30:37 +02:00
// When requested a detailed note data
2018-02-02 00:06:01 +01:00
if (opts.detail) {
2018-04-07 19:30:37 +02:00
// Get previous note info
_note.prev = (async () => {
const prev = await Note.findOne({
userId: _note.userId,
2018-02-02 00:06:01 +01:00
_id: {
$lt: id
}
}, {
fields: {
_id: true
},
sort: {
_id: -1
}
});
return prev ? prev._id : null;
})();
2018-04-07 19:30:37 +02:00
// Get next note info
_note.next = (async () => {
const next = await Note.findOne({
userId: _note.userId,
2018-02-02 00:06:01 +01:00
_id: {
$gt: id
}
}, {
fields: {
_id: true
},
sort: {
_id: 1
}
});
return next ? next._id : null;
})();
2018-04-07 19:30:37 +02:00
if (_note.replyId) {
// Populate reply to note
_note.reply = pack(_note.replyId, meId, {
2018-02-02 00:06:01 +01:00
detail: false
});
}
2018-04-07 19:30:37 +02:00
if (_note.renoteId) {
// Populate renote
_note.renote = pack(_note.renoteId, meId, {
detail: _note.text == null
2018-02-02 00:06:01 +01:00
});
}
// Poll
2018-04-07 19:30:37 +02:00
if (meId && _note.poll) {
_note.poll = (async (poll) => {
2018-02-02 00:06:01 +01:00
const vote = await Vote
.findOne({
2018-03-29 07:48:47 +02:00
userId: meId,
2018-04-07 19:30:37 +02:00
noteId: id
2018-02-02 00:06:01 +01:00
});
if (vote != null) {
const myChoice = poll.choices
.filter(c => c.id == vote.choice)[0];
2018-03-29 07:48:47 +02:00
myChoice.isVoted = true;
2018-02-02 00:06:01 +01:00
}
return poll;
2018-04-07 19:30:37 +02:00
})(_note.poll);
2018-02-02 00:06:01 +01:00
}
// Fetch my reaction
if (meId) {
2018-04-07 19:30:37 +02:00
_note.myReaction = (async () => {
2018-02-02 00:06:01 +01:00
const reaction = await Reaction
.findOne({
2018-03-29 07:48:47 +02:00
userId: meId,
2018-04-07 19:30:37 +02:00
noteId: id,
2018-03-29 07:48:47 +02:00
deletedAt: { $exists: false }
2018-02-02 00:06:01 +01:00
});
if (reaction) {
return reaction.reaction;
}
return null;
})();
}
}
2018-04-07 19:30:37 +02:00
// resolve promises in _note object
_note = await rap(_note);
2018-02-02 00:06:01 +01:00
2018-04-07 19:30:37 +02:00
return _note;
2018-02-02 00:06:01 +01:00
};