sharkey/packages/backend/src/core/InstanceActorService.ts

45 lines
1.2 KiB
TypeScript
Raw Normal View History

2022-09-18 03:27:08 +09:00
import { Inject, Injectable } from '@nestjs/common';
import { IsNull } from 'typeorm';
2023-02-13 15:50:22 +09:00
import type { LocalUser } from '@/models/entities/User.js';
2022-09-21 05:33:11 +09:00
import type { UsersRepository } from '@/models/index.js';
2023-04-04 17:32:09 +09:00
import { MemoryCache } from '@/misc/cache.js';
2022-09-18 03:27:08 +09:00
import { DI } from '@/di-symbols.js';
2022-12-04 10:16:03 +09:00
import { CreateSystemUserService } from '@/core/CreateSystemUserService.js';
2022-12-04 17:05:32 +09:00
import { bindThis } from '@/decorators.js';
2022-09-18 03:27:08 +09:00
const ACTOR_USERNAME = 'instance.actor' as const;
@Injectable()
export class InstanceActorService {
2023-04-04 17:32:09 +09:00
private cache: MemoryCache<LocalUser>;
2022-09-18 03:27:08 +09:00
constructor(
@Inject(DI.usersRepository)
private usersRepository: UsersRepository,
private createSystemUserService: CreateSystemUserService,
) {
2023-04-04 17:32:09 +09:00
this.cache = new MemoryCache<LocalUser>(Infinity);
2022-09-18 03:27:08 +09:00
}
@bindThis
2023-02-13 15:50:22 +09:00
public async getInstanceActor(): Promise<LocalUser> {
2023-04-04 17:32:09 +09:00
const cached = this.cache.get();
2022-09-18 03:27:08 +09:00
if (cached) return cached;
const user = await this.usersRepository.findOneBy({
host: IsNull(),
username: ACTOR_USERNAME,
2023-02-13 15:50:22 +09:00
}) as LocalUser | undefined;
2022-09-18 03:27:08 +09:00
if (user) {
2023-04-04 17:32:09 +09:00
this.cache.set(user);
2022-09-18 03:27:08 +09:00
return user;
} else {
2023-02-13 15:50:22 +09:00
const created = await this.createSystemUserService.createSystemUser(ACTOR_USERNAME) as LocalUser;
2023-04-04 17:32:09 +09:00
this.cache.set(created);
2022-09-18 03:27:08 +09:00
return created;
}
}
}