1
0
mirror of https://github.com/hotomoe/hotomoe synced 2024-12-24 11:38:11 +09:00
hotomoe/src/remote/activitypub/models/person.ts

224 lines
6.1 KiB
TypeScript
Raw Normal View History

2018-04-17 22:20:25 +09:00
import * as mongo from 'mongodb';
2018-04-09 04:08:56 +09:00
import { JSDOM } from 'jsdom';
import { toUnicode } from 'punycode';
import * as debug from 'debug';
import config from '../../../config';
2018-04-09 05:39:18 +09:00
import User, { validateUsername, isValidName, IUser, IRemoteUser } from '../../../models/user';
2018-04-09 04:08:56 +09:00
import webFinger from '../../webfinger';
import Resolver from '../resolver';
import { resolveImage } from './image';
import { isCollectionOrOrderedCollection, IObject, IPerson } from '../type';
const log = debug('misskey:activitypub');
/**
* Personをフェッチします
*
* Misskeyに対象のPersonが登録されていればそれを返します
*/
export async function fetchPerson(value: string | IObject, resolver?: Resolver): Promise<IUser> {
const uri = typeof value == 'string' ? value : value.id;
// URIがこのサーバーを指しているならデータベースからフェッチ
if (uri.startsWith(config.url + '/')) {
2018-04-17 22:20:25 +09:00
const id = new mongo.ObjectID(uri.split('/').pop());
return await User.findOne({ _id: id });
2018-04-09 04:08:56 +09:00
}
//#region このサーバーに既に登録されていたらそれを返す
const exist = await User.findOne({ uri });
if (exist) {
return exist;
}
//#endregion
return null;
}
/**
* Personを作成します
*/
export async function createPerson(value: any, resolver?: Resolver): Promise<IUser> {
if (resolver == null) resolver = new Resolver();
const object = await resolver.resolve(value) as any;
if (
object == null ||
object.type !== 'Person' ||
typeof object.preferredUsername !== 'string' ||
!validateUsername(object.preferredUsername) ||
2018-04-09 05:37:50 +09:00
!isValidName(object.name == '' ? null : object.name)
2018-04-09 04:08:56 +09:00
) {
2018-04-09 05:30:17 +09:00
log(`invalid person: ${JSON.stringify(object, null, 2)}`);
2018-04-09 04:08:56 +09:00
throw new Error('invalid person');
}
const person: IPerson = object;
log(`Creating the Person: ${person.id}`);
const [followersCount = 0, followingCount = 0, notesCount = 0, finger] = await Promise.all([
resolver.resolve(person.followers).then(
resolved => isCollectionOrOrderedCollection(resolved) ? resolved.totalItems : undefined,
() => undefined
),
resolver.resolve(person.following).then(
resolved => isCollectionOrOrderedCollection(resolved) ? resolved.totalItems : undefined,
() => undefined
),
resolver.resolve(person.outbox).then(
resolved => isCollectionOrOrderedCollection(resolved) ? resolved.totalItems : undefined,
() => undefined
),
webFinger(person.id)
]);
const host = toUnicode(finger.subject.replace(/^.*?@/, '')).toLowerCase();
2018-04-09 04:08:56 +09:00
const summaryDOM = JSDOM.fragment(person.summary);
// Create user
const user = await User.insert({
avatarId: null,
bannerId: null,
createdAt: Date.parse(person.published) || null,
description: summaryDOM.textContent,
followersCount,
followingCount,
notesCount,
name: person.name,
driveCapacity: 1024 * 1024 * 8, // 8MiB
username: person.preferredUsername,
usernameLower: person.preferredUsername.toLowerCase(),
host,
publicKey: {
id: person.publicKey.id,
publicKeyPem: person.publicKey.publicKeyPem
},
inbox: person.inbox,
2018-04-15 18:38:40 +09:00
uri: person.id,
url: person.url
2018-04-09 04:08:56 +09:00
}) as IRemoteUser;
//#region アイコンとヘッダー画像をフェッチ
const [avatarId, bannerId] = (await Promise.all([
person.icon,
person.image
].map(img =>
img == null
? Promise.resolve(null)
2018-04-09 05:44:35 +09:00
: resolveImage(user, img)
2018-04-09 04:08:56 +09:00
))).map(file => file != null ? file._id : null);
User.update({ _id: user._id }, { $set: { avatarId, bannerId } });
user.avatarId = avatarId;
user.bannerId = bannerId;
//#endregion
return user;
}
2018-04-17 15:30:58 +09:00
/**
* Personの情報を更新します
*
* Misskeyに対象のPersonが登録されていなければ無視します
*/
export async function updatePerson(value: string | IObject, resolver?: Resolver): Promise<void> {
const uri = typeof value == 'string' ? value : value.id;
// URIがこのサーバーを指しているならスキップ
if (uri.startsWith(config.url + '/')) {
return;
}
//#region このサーバーに既に登録されているか
const exist = await User.findOne({ uri }) as IRemoteUser;
if (exist == null) {
return;
}
//#endregion
if (resolver == null) resolver = new Resolver();
const object = await resolver.resolve(value) as any;
if (
object == null ||
object.type !== 'Person'
) {
log(`invalid person: ${JSON.stringify(object, null, 2)}`);
throw new Error('invalid person');
}
const person: IPerson = object;
log(`Updating the Person: ${person.id}`);
const [followersCount = 0, followingCount = 0, notesCount = 0] = await Promise.all([
resolver.resolve(person.followers).then(
resolved => isCollectionOrOrderedCollection(resolved) ? resolved.totalItems : undefined,
() => undefined
),
resolver.resolve(person.following).then(
resolved => isCollectionOrOrderedCollection(resolved) ? resolved.totalItems : undefined,
() => undefined
),
resolver.resolve(person.outbox).then(
resolved => isCollectionOrOrderedCollection(resolved) ? resolved.totalItems : undefined,
() => undefined
)
]);
const summaryDOM = JSDOM.fragment(person.summary);
// アイコンとヘッダー画像をフェッチ
const [avatarId, bannerId] = (await Promise.all([
person.icon,
person.image
].map(img =>
img == null
? Promise.resolve(null)
: resolveImage(exist, img)
))).map(file => file != null ? file._id : null);
// Update user
await User.update({ _id: exist._id }, {
$set: {
updatedAt: new Date(),
avatarId,
bannerId,
description: summaryDOM.textContent,
followersCount,
followingCount,
notesCount,
name: person.name,
url: person.url
}
});
}
2018-04-09 04:08:56 +09:00
/**
* Personを解決します
*
* Misskeyに対象のPersonが登録されていればそれを返し
* Misskeyに登録しそれを返します
*/
export async function resolvePerson(value: string | IObject, verifier?: string): Promise<IUser> {
const uri = typeof value == 'string' ? value : value.id;
//#region このサーバーに既に登録されていたらそれを返す
const exist = await fetchPerson(uri);
if (exist) {
return exist;
}
//#endregion
// リモートサーバーからフェッチしてきて登録
return await createPerson(value);
}