barkey/packages/backend/src/core/WebhookService.ts

98 lines
2.5 KiB
TypeScript
Raw Normal View History

/*
* SPDX-FileCopyrightText: syuilo and other misskey contributors
* SPDX-License-Identifier: AGPL-3.0-only
*/
2022-09-17 18:27:08 +00:00
import { Inject, Injectable } from '@nestjs/common';
2023-04-14 04:50:05 +00:00
import * as Redis from 'ioredis';
import type { WebhooksRepository } from '@/models/_.js';
import type { MiWebhook } from '@/models/Webhook.js';
2022-09-17 18:27:08 +00:00
import { DI } from '@/di-symbols.js';
import { bindThis } from '@/decorators.js';
2023-09-29 02:29:54 +00:00
import type { GlobalEvents } from '@/core/GlobalEventService.js';
import type { OnApplicationShutdown } from '@nestjs/common';
2022-09-17 18:27:08 +00:00
@Injectable()
export class WebhookService implements OnApplicationShutdown {
2022-09-18 18:11:50 +00:00
private webhooksFetched = false;
private webhooks: MiWebhook[] = [];
2022-09-17 18:27:08 +00:00
constructor(
@Inject(DI.redisForSub)
private redisForSub: Redis.Redis,
2022-09-17 18:27:08 +00:00
@Inject(DI.webhooksRepository)
private webhooksRepository: WebhooksRepository,
) {
//this.onMessage = this.onMessage.bind(this);
this.redisForSub.on('message', this.onMessage);
2022-09-17 18:27:08 +00:00
}
@bindThis
2022-09-17 18:27:08 +00:00
public async getActiveWebhooks() {
2022-09-18 18:11:50 +00:00
if (!this.webhooksFetched) {
this.webhooks = await this.webhooksRepository.findBy({
2022-09-17 18:27:08 +00:00
active: true,
});
2022-09-18 18:11:50 +00:00
this.webhooksFetched = true;
2022-09-17 18:27:08 +00:00
}
2022-09-18 18:11:50 +00:00
return this.webhooks;
2022-09-17 18:27:08 +00:00
}
@bindThis
2022-09-23 22:12:11 +00:00
private async onMessage(_: string, data: string): Promise<void> {
2022-09-17 18:27:08 +00:00
const obj = JSON.parse(data);
if (obj.channel === 'internal') {
2023-09-29 02:29:54 +00:00
const { type, body } = obj.message as GlobalEvents['internal']['payload'];
2022-09-17 18:27:08 +00:00
switch (type) {
case 'webhookCreated':
if (body.active) {
2023-02-01 11:15:11 +00:00
this.webhooks.push({
...body,
createdAt: new Date(body.createdAt),
2023-02-28 07:46:25 +00:00
latestSentAt: body.latestSentAt ? new Date(body.latestSentAt) : null,
2023-02-01 11:15:11 +00:00
});
2022-09-17 18:27:08 +00:00
}
break;
case 'webhookUpdated':
if (body.active) {
2022-09-18 18:11:50 +00:00
const i = this.webhooks.findIndex(a => a.id === body.id);
2022-09-17 18:27:08 +00:00
if (i > -1) {
2023-02-01 11:15:11 +00:00
this.webhooks[i] = {
...body,
createdAt: new Date(body.createdAt),
2023-02-28 07:46:25 +00:00
latestSentAt: body.latestSentAt ? new Date(body.latestSentAt) : null,
2023-02-01 11:15:11 +00:00
};
2022-09-17 18:27:08 +00:00
} else {
2023-02-01 11:15:11 +00:00
this.webhooks.push({
...body,
createdAt: new Date(body.createdAt),
2023-02-28 07:46:25 +00:00
latestSentAt: body.latestSentAt ? new Date(body.latestSentAt) : null,
2023-02-01 11:15:11 +00:00
});
2022-09-17 18:27:08 +00:00
}
} else {
2022-09-18 18:11:50 +00:00
this.webhooks = this.webhooks.filter(a => a.id !== body.id);
2022-09-17 18:27:08 +00:00
}
break;
case 'webhookDeleted':
2022-09-18 18:11:50 +00:00
this.webhooks = this.webhooks.filter(a => a.id !== body.id);
2022-09-17 18:27:08 +00:00
break;
default:
break;
}
}
}
@bindThis
2023-05-29 04:21:26 +00:00
public dispose(): void {
this.redisForSub.off('message', this.onMessage);
2022-09-17 18:27:08 +00:00
}
2023-05-29 04:21:26 +00:00
@bindThis
public onApplicationShutdown(signal?: string | undefined): void {
this.dispose();
}
2022-09-17 18:27:08 +00:00
}