iceshrimp/src/server/webfinger.ts

import * as mongo from 'mongodb';
import * as Router from 'koa-router';

import config from '../config';
import parseAcct from '../acct/parse';
import User, { IUser } from '../models/user';

// Init router
const router = new Router();

router.get('/.well-known/webfinger', async ctx => {
	if (typeof ctx.query.resource !== 'string') {
		ctx.status = 400;
		return;
	}

	const resourceLower = ctx.query.resource.toLowerCase();
	let acctLower;
	let id;

	if (resourceLower.startsWith(config.url.toLowerCase() + '/@')) {
		acctLower = resourceLower.split('/').pop();
	} else if (resourceLower.startsWith(config.url.toLowerCase() + '/users/')) {
		id = new mongo.ObjectID(resourceLower.split('/').pop());
	} else if (resourceLower.startsWith('acct:')) {
		acctLower = resourceLower.slice('acct:'.length);
	} else {
		acctLower = resourceLower;
	}

	let user: IUser;

	if (acctLower) {
		const parsedAcctLower = parseAcct(acctLower);
		if (![null, config.host.toLowerCase()].includes(parsedAcctLower.host)) {
			ctx.status = 422;
			return;
		}

		user = await User.findOne({
			usernameLower: parsedAcctLower.username,
			host: null
		});
	} else {
		user = await User.findOne({
			_id: id,
			host: null
		});
	}

	if (user === null) {
		ctx.status = 404;
		return;
	}

	ctx.body = {
		subject: `acct:${user.username}@${config.host}`,
		links: [{
			rel: 'self',
			type: 'application/activity+json',
			href: `${config.url}/users/${user._id}`
		}, {
			rel: 'http://webfinger.net/rel/profile-page',
			type: 'text/html',
			href: `${config.url}/@${user.username}`
		}, {
			rel: 'http://ostatus.org/schema/1.0/subscribe',
			template: `${config.url}/authorize-follow?acct={uri}`
		}]
	};
});

export default router;