From cf195b8e9213d05cb0ae1fa289e3f305c53e6bad Mon Sep 17 00:00:00 2001
From: Akihiko Odaki <nekomanma@pixiv.co.jp>
Date: Mon, 2 Apr 2018 19:37:00 +0900
Subject: [PATCH] Improve WebFinger verification

---
 src/remote/webfinger.ts | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/src/remote/webfinger.ts b/src/remote/webfinger.ts
index 4c0304e3f..c84beb099 100644
--- a/src/remote/webfinger.ts
+++ b/src/remote/webfinger.ts
@@ -20,14 +20,19 @@ export default async function resolve(query, verifier?: string): Promise<IWebFin
 
 		res(result.object);
 	})) as IWebFinger;
+	const subject = finger.subject.toLowerCase().replace(/^acct:/, '');
 
-	if (verifier) {
-		if (finger.subject.toLowerCase().replace(/^acct:/, '') !== verifier) {
-			throw 'WebFinger verfification failed';
+	if (typeof verifier === 'string') {
+		if (subject !== verifier) {
+			throw new Error;
 		}
 
 		return finger;
 	}
 
-	return resolve(finger.subject, finger.subject.toLowerCase());
+	if (typeof subject === 'string') {
+		return resolve(subject, subject);
+	}
+
+	throw new Error;
 }