import { toUnicode, toASCII } from 'punycode'; import User, { IUser, IRemoteUser } from '../models/user'; import webFinger from './webfinger'; import config from '../config'; import { createPerson, updatePerson } from './activitypub/models/person'; import { URL } from 'url'; import { remoteLogger } from './logger'; import chalk from 'chalk'; const logger = remoteLogger.createSubLogger('resolve-user'); export default async (username: string, _host: string, option?: any, resync?: boolean): Promise => { const usernameLower = username.toLowerCase(); if (_host == null) { logger.info(`return local user: ${usernameLower}`); return await User.findOne({ usernameLower, host: null }); } const configHostAscii = toASCII(config.host).toLowerCase(); const configHost = toUnicode(configHostAscii); const hostAscii = toASCII(_host).toLowerCase(); const host = toUnicode(hostAscii); if (configHost == host) { logger.info(`return local user: ${usernameLower}`); return await User.findOne({ usernameLower, host: null }); } const user = await User.findOne({ usernameLower, host }, option); const acctLower = `${usernameLower}@${hostAscii}`; if (user === null) { const self = await resolveSelf(acctLower); logger.succ(`return new remote user: ${chalk.magenta(acctLower)}`); return await createPerson(self.href); } if (resync) { logger.info(`try resync: ${acctLower}`); const self = await resolveSelf(acctLower); if ((user as IRemoteUser).uri !== self.href) { // if uri mismatch, Fix (user@host <=> AP's Person id(IRemoteUser.uri)) mapping. logger.info(`uri missmatch: ${acctLower}`); logger.info(`recovery missmatch uri for (username=${username}, host=${host}) from ${(user as IRemoteUser).uri} to ${self.href}`); // validate uri const uri = new URL(self.href); if (uri.hostname !== hostAscii) { throw new Error(`Invalied uri`); } await User.update({ usernameLower, host: host }, { $set: { uri: self.href } }); } else { logger.info(`uri is fine: ${acctLower}`); } await updatePerson(self.href); logger.info(`return resynced remote user: ${acctLower}`); return await User.findOne({ uri: self.href }); } logger.info(`return existing remote user: ${acctLower}`); return user; }; async function resolveSelf(acctLower: string) { logger.info(`WebFinger for ${chalk.yellow(acctLower)}`); const finger = await webFinger(acctLower); const self = finger.links.find(link => link.rel && link.rel.toLowerCase() === 'self'); if (!self) { throw new Error('self link not found'); } return self; }