mirror of
https://iceshrimp.dev/iceshrimp/iceshrimp
synced 2024-12-23 11:08:13 +09:00
129 lines
3.4 KiB
TypeScript
129 lines
3.4 KiB
TypeScript
import * as Router from 'koa-router';
|
|
|
|
import config from '../config';
|
|
import parseAcct from '../misc/acct/parse';
|
|
import Acct from '../misc/acct/type';
|
|
import { links } from './nodeinfo';
|
|
import { escapeAttribute, escapeValue } from '../prelude/xml';
|
|
import { Users } from '../models';
|
|
import { User } from '../models/entities/user';
|
|
|
|
// Init router
|
|
const router = new Router();
|
|
|
|
const XRD = (...x: { element: string, value?: string, attributes?: Record<string, string> }[]) =>
|
|
`<?xml version="1.0" encoding="UTF-8"?><XRD xmlns="http://docs.oasis-open.org/ns/xri/xrd-1.0">${x.map(({ element, value, attributes }) =>
|
|
`<${
|
|
Object.entries(typeof attributes === 'object' && attributes || {}).reduce((a, [k, v]) => `${a} ${k}="${escapeAttribute(v)}"`, element)
|
|
}${
|
|
typeof value === 'string' ? `>${escapeValue(value)}</${element}` : '/'
|
|
}>`).reduce((a, c) => a + c, '')}</XRD>`;
|
|
|
|
const webFingerPath = '/.well-known/webfinger';
|
|
const jrd = 'application/jrd+json';
|
|
const xrd = 'application/xrd+xml';
|
|
|
|
router.get('/.well-known/host-meta', async ctx => {
|
|
ctx.set('Content-Type', xrd);
|
|
ctx.body = XRD({ element: 'Link', attributes: {
|
|
type: xrd,
|
|
template: `${config.url}${webFingerPath}?resource={uri}`
|
|
}});
|
|
});
|
|
|
|
router.get('/.well-known/host-meta.json', async ctx => {
|
|
ctx.set('Content-Type', jrd);
|
|
ctx.body = {
|
|
links: [{
|
|
rel: 'lrdd',
|
|
type: jrd,
|
|
template: `${config.url}${webFingerPath}?resource={uri}`
|
|
}]
|
|
};
|
|
});
|
|
|
|
router.get('/.well-known/nodeinfo', async ctx => {
|
|
ctx.body = { links };
|
|
});
|
|
|
|
router.get(webFingerPath, async ctx => {
|
|
const fromId = (id: User['id']): Record<string, any> => ({
|
|
id,
|
|
host: null
|
|
});
|
|
|
|
const generateQuery = (resource: string) =>
|
|
resource.startsWith(`${config.url.toLowerCase()}/users/`) ?
|
|
fromId(resource.split('/').pop()!) :
|
|
fromAcct(parseAcct(
|
|
resource.startsWith(`${config.url.toLowerCase()}/@`) ? resource.split('/').pop()! :
|
|
resource.startsWith('acct:') ? resource.slice('acct:'.length) :
|
|
resource));
|
|
|
|
const fromAcct = (acct: Acct): Record<string, any> | number =>
|
|
!acct.host || acct.host === config.host.toLowerCase() ? {
|
|
usernameLower: acct.username,
|
|
host: null
|
|
} : 422;
|
|
|
|
if (typeof ctx.query.resource !== 'string') {
|
|
ctx.status = 400;
|
|
return;
|
|
}
|
|
|
|
const query = generateQuery(ctx.query.resource.toLowerCase());
|
|
|
|
if (typeof query === 'number') {
|
|
ctx.status = query;
|
|
return;
|
|
}
|
|
|
|
const user = await Users.findOne(query);
|
|
|
|
if (user == null) {
|
|
ctx.status = 404;
|
|
return;
|
|
}
|
|
|
|
const subject = `acct:${user.username}@${config.host}`;
|
|
const self = {
|
|
rel: 'self',
|
|
type: 'application/activity+json',
|
|
href: `${config.url}/users/${user.id}`
|
|
};
|
|
const profilePage = {
|
|
rel: 'http://webfinger.net/rel/profile-page',
|
|
type: 'text/html',
|
|
href: `${config.url}/@${user.username}`
|
|
};
|
|
const subscribe = {
|
|
rel: 'http://ostatus.org/schema/1.0/subscribe',
|
|
template: `${config.url}/authorize-follow?acct={uri}`
|
|
};
|
|
|
|
if (ctx.accepts(jrd, xrd) === xrd) {
|
|
ctx.body = XRD(
|
|
{ element: 'Subject', value: subject },
|
|
{ element: 'Link', attributes: self },
|
|
{ element: 'Link', attributes: profilePage },
|
|
{ element: 'Link', attributes: subscribe });
|
|
ctx.type = xrd;
|
|
} else {
|
|
ctx.body = {
|
|
subject,
|
|
links: [self, profilePage, subscribe]
|
|
};
|
|
ctx.type = jrd;
|
|
}
|
|
|
|
ctx.vary('Accept');
|
|
ctx.set('Cache-Control', 'public, max-age=180');
|
|
});
|
|
|
|
// Return 404 for other .well-known
|
|
router.all('/.well-known/*', async ctx => {
|
|
ctx.status = 404;
|
|
});
|
|
|
|
export default router;
|