misskey/packages/backend/src/core/FederatedInstanceService.ts

import { Inject, Injectable } from '@nestjs/common';
import type { InstancesRepository } from '@/models/index.js';
import type { Instance } from '@/models/entities/Instance.js';
import { KVCache } from '@/misc/cache.js';
import { IdService } from '@/core/IdService.js';
import { DI } from '@/di-symbols.js';
import { UtilityService } from '@/core/UtilityService.js';
import { bindThis } from '@/decorators.js';

@Injectable()
export class FederatedInstanceService {
	private cache: KVCache<Instance>;

	constructor(
		@Inject(DI.instancesRepository)
		private instancesRepository: InstancesRepository,

		private utilityService: UtilityService,
		private idService: IdService,
	) {
		this.cache = new KVCache<Instance>(1000 * 60 * 60);
	}

	@bindThis
	public async fetch(host: string): Promise<Instance> {
		host = this.utilityService.toPuny(host);
	
		const cached = this.cache.get(host);
		if (cached) return cached;
	
		const index = await this.instancesRepository.findOneBy({ host });
	
		if (index == null) {
			const i = await this.instancesRepository.insert({
				id: this.idService.genId(),
				host,
				firstRetrievedAt: new Date(),
			}).then(x => this.instancesRepository.findOneByOrFail(x.identifiers[0]));
	
			this.cache.set(host, i);
			return i;
		} else {
			this.cache.set(host, index);
			return index;
		}
	}

	@bindThis
	public async updateCachePartial(host: string, data: Partial<Instance>): Promise<void> {
		host = this.utilityService.toPuny(host);
	
		const cached = this.cache.get(host);
		if (cached == null) return;
	
		this.cache.set(host, {
			...cached,
			...data,
		});
	}
}