]>
Commit | Line | Data |
---|---|---|
1 | import * as WebFinger from 'webfinger.js' | |
2 | import { WebFingerData } from '../../shared' | |
3 | import { ActorModel } from '../models/activitypub/actor' | |
4 | import { isTestInstance } from './core-utils' | |
5 | import { isActivityPubUrlValid } from './custom-validators/activitypub/misc' | |
6 | import { WEBSERVER } from '../initializers/constants' | |
7 | import { MActorFull } from '../typings/models' | |
8 | ||
9 | const webfinger = new WebFinger({ | |
10 | webfist_fallback: false, | |
11 | tls_only: isTestInstance(), | |
12 | uri_fallback: false, | |
13 | request_timeout: 3000 | |
14 | }) | |
15 | ||
16 | async function loadActorUrlOrGetFromWebfinger (uriArg: string) { | |
17 | // Handle strings like @toto@example.com | |
18 | const uri = uriArg.startsWith('@') ? uriArg.slice(1) : uriArg | |
19 | ||
20 | const [ name, host ] = uri.split('@') | |
21 | let actor: MActorFull | |
22 | ||
23 | if (!host || host === WEBSERVER.HOST) { | |
24 | actor = await ActorModel.loadLocalByName(name) | |
25 | } else { | |
26 | actor = await ActorModel.loadByNameAndHost(name, host) | |
27 | } | |
28 | ||
29 | if (actor) return actor.url | |
30 | ||
31 | return getUrlFromWebfinger(uri) | |
32 | } | |
33 | ||
34 | async function getUrlFromWebfinger (uri: string) { | |
35 | const webfingerData: WebFingerData = await webfingerLookup(uri) | |
36 | return getLinkOrThrow(webfingerData) | |
37 | } | |
38 | ||
39 | // --------------------------------------------------------------------------- | |
40 | ||
41 | export { | |
42 | getUrlFromWebfinger, | |
43 | loadActorUrlOrGetFromWebfinger | |
44 | } | |
45 | ||
46 | // --------------------------------------------------------------------------- | |
47 | ||
48 | function getLinkOrThrow (webfingerData: WebFingerData) { | |
49 | if (Array.isArray(webfingerData.links) === false) throw new Error('WebFinger links is not an array.') | |
50 | ||
51 | const selfLink = webfingerData.links.find(l => l.rel === 'self') | |
52 | if (selfLink === undefined || isActivityPubUrlValid(selfLink.href) === false) { | |
53 | throw new Error('Cannot find self link or href is not a valid URL.') | |
54 | } | |
55 | ||
56 | return selfLink.href | |
57 | } | |
58 | ||
59 | function webfingerLookup (nameWithHost: string) { | |
60 | return new Promise<WebFingerData>((res, rej) => { | |
61 | webfinger.lookup(nameWithHost, (err, p) => { | |
62 | if (err) return rej(err) | |
63 | ||
64 | return res(p.object) | |
65 | }) | |
66 | }) | |
67 | } |