misskey/src/common/remote/activitypub/resolver.ts

98 lines
2.2 KiB
TypeScript
Raw Normal View History

2018-03-31 12:55:00 +02:00
import RemoteUserObject from '../../../models/remote-user-object';
import { IObject } from './type';
const request = require('request-promise-native');
type IResult = {
resolver: Resolver;
object: IObject;
};
async function resolveUnrequestedOne(this: Resolver, value) {
if (typeof value !== 'string') {
return { resolver: this, object: value };
}
const resolver = new Resolver(this.requesting);
resolver.requesting.add(value);
const object = await request({
url: value,
headers: {
Accept: 'application/activity+json, application/ld+json'
},
json: true
});
if (object === null || (
Array.isArray(object['@context']) ?
!object['@context'].includes('https://www.w3.org/ns/activitystreams') :
object['@context'] !== 'https://www.w3.org/ns/activitystreams'
)) {
2018-04-01 14:24:25 +02:00
throw new Error();
2018-03-31 12:55:00 +02:00
}
return { resolver, object };
}
async function resolveCollection(this: Resolver, value) {
if (Array.isArray(value)) {
return value;
}
const resolved = typeof value === 'string' ?
await resolveUnrequestedOne.call(this, value) :
value;
switch (resolved.type) {
case 'Collection':
return resolved.items;
case 'OrderedCollection':
return resolved.orderedItems;
default:
return [resolved];
}
}
export default class Resolver {
2018-04-01 14:24:25 +02:00
private requesting: Set<string>;
2018-03-31 12:55:00 +02:00
constructor(iterable?: Iterable<string>) {
this.requesting = new Set(iterable);
}
2018-04-01 14:24:25 +02:00
public async resolve(value): Promise<Array<Promise<IResult>>> {
2018-03-31 12:55:00 +02:00
const collection = await resolveCollection.call(this, value);
return collection
.filter(element => !this.requesting.has(element))
.map(resolveUnrequestedOne.bind(this));
}
2018-04-01 14:24:25 +02:00
public resolveOne(value) {
2018-03-31 12:55:00 +02:00
if (this.requesting.has(value)) {
2018-04-01 14:24:25 +02:00
throw new Error();
2018-03-31 12:55:00 +02:00
}
return resolveUnrequestedOne.call(this, value);
}
2018-04-01 14:24:25 +02:00
public async resolveRemoteUserObjects(value) {
2018-03-31 12:55:00 +02:00
const collection = await resolveCollection.call(this, value);
return collection.filter(element => !this.requesting.has(element)).map(element => {
if (typeof element === 'string') {
const object = RemoteUserObject.findOne({ uri: element });
if (object !== null) {
return object;
}
}
return resolveUnrequestedOne.call(this, element);
});
}
}