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