Fix #1438

src/server/webfinger.ts

@@ -1,8 +1,9 @@
+import * as mongo from 'mongodb';
 import * as Router from 'koa-router';
 
 import config from '../config';
 import parseAcct from '../acct/parse';
-import User from '../models/user';
+import User, { IUser } from '../models/user';
 
 // Init router
 const router = new Router();
@@ -14,27 +15,38 @@ router.get('/.well-known/webfinger', async ctx => {
 	}
 
 	const resourceLower = ctx.query.resource.toLowerCase();
-	const webPrefix = config.url.toLowerCase() + '/@';
 	let acctLower;
+	let id;
 
-	if (resourceLower.startsWith(webPrefix)) {
+	if (resourceLower.startsWith(config.url.toLowerCase() + '/@')) {
-		acctLower = resourceLower.slice(webPrefix.length);
+		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;
 		}
 
-	const user = await User.findOne({
+		user = await User.findOne({
 			usernameLower: parsedAcctLower.username,
 			host: null
 		});
+	} else {
+		user = await User.findOne({
+			_id: id,
+			host: null
+		});
+	}
 
 	if (user === null) {
 		ctx.status = 404;