forked from FoundKeyGang/FoundKey
113 lines
3.4 KiB
TypeScript
113 lines
3.4 KiB
TypeScript
import { URL } from 'node:url';
|
|
import chalk from 'chalk';
|
|
import { IsNull } from 'typeorm';
|
|
import { DAY } from '@/const.js';
|
|
import config from '@/config/index.js';
|
|
import { isSelfHost, toPuny } from '@/misc/convert-host.js';
|
|
import { User, IRemoteUser } from '@/models/entities/user.js';
|
|
import { Users } from '@/models/index.js';
|
|
import webFinger from './webfinger.js';
|
|
import { createPerson, updatePerson } from './activitypub/models/person.js';
|
|
import { remoteLogger } from './logger.js';
|
|
|
|
const logger = remoteLogger.createSubLogger('resolve-user');
|
|
|
|
export async function resolveUser(username: string, idnHost: string | null): Promise<User> {
|
|
const usernameLower = username.toLowerCase();
|
|
|
|
if (idnHost == null) {
|
|
logger.info(`return local user: ${usernameLower}`);
|
|
return await Users.findOneBy({ usernameLower, host: IsNull() }).then(u => {
|
|
if (u == null) {
|
|
throw new Error('user not found');
|
|
} else {
|
|
return u;
|
|
}
|
|
});
|
|
}
|
|
|
|
if (isSelfHost(idnHost)) {
|
|
logger.info(`return local user: ${usernameLower}`);
|
|
return await Users.findOneBy({ usernameLower, host: IsNull() }).then(u => {
|
|
if (u == null) {
|
|
throw new Error('user not found');
|
|
} else {
|
|
return u;
|
|
}
|
|
});
|
|
}
|
|
|
|
// `idnHost` can not be null here because that would have branched off with `isSelfHost`.
|
|
const host = toPuny(idnHost!);
|
|
const user = await Users.findOneBy({ usernameLower, host }) as IRemoteUser | null;
|
|
|
|
const acctLower = `${usernameLower}@${host}`;
|
|
|
|
if (user == null) {
|
|
const self = await resolveSelf(acctLower);
|
|
|
|
logger.succ(`return new remote user: ${chalk.magenta(acctLower)}`);
|
|
return await createPerson(self.href);
|
|
}
|
|
|
|
// If user information is out of date, start over with webfinger
|
|
if (user.lastFetchedAt == null || Date.now() - user.lastFetchedAt.getTime() > DAY) {
|
|
// Prevent race conditions
|
|
await Users.update(user.id, {
|
|
lastFetchedAt: new Date(),
|
|
});
|
|
|
|
logger.info(`try resync: ${acctLower}`);
|
|
const self = await resolveSelf(acctLower);
|
|
|
|
if (user.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.uri} to ${self.href}`);
|
|
|
|
// validate uri
|
|
const uri = new URL(self.href);
|
|
if (uri.hostname !== host) {
|
|
throw new Error('Invalid uri');
|
|
}
|
|
|
|
await Users.update({
|
|
usernameLower,
|
|
host,
|
|
}, {
|
|
uri: self.href,
|
|
});
|
|
} else {
|
|
logger.info(`uri is fine: ${acctLower}`);
|
|
}
|
|
|
|
await updatePerson(self.href);
|
|
|
|
logger.info(`return resynced remote user: ${acctLower}`);
|
|
return await Users.findOneBy({ uri: self.href }).then(u => {
|
|
if (u == null) {
|
|
throw new Error('user not found');
|
|
} else {
|
|
return u;
|
|
}
|
|
});
|
|
}
|
|
|
|
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).catch(e => {
|
|
logger.error(`Failed to WebFinger for ${chalk.yellow(acctLower)}: ${ e.statusCode || e.message }`);
|
|
throw new Error(`Failed to WebFinger for ${acctLower}: ${ e.statusCode || e.message }`);
|
|
});
|
|
const self = finger.links.find(link => link.rel != null && link.rel.toLowerCase() === 'self');
|
|
if (!self) {
|
|
logger.error(`Failed to WebFinger for ${chalk.yellow(acctLower)}: self link not found`);
|
|
throw new Error('self link not found');
|
|
}
|
|
return self;
|
|
}
|