114 lines
2.7 KiB
TypeScript
114 lines
2.7 KiB
TypeScript
/*
|
|
* SPDX-FileCopyrightText: syuilo and misskey-project
|
|
* SPDX-License-Identifier: AGPL-3.0-only
|
|
*/
|
|
|
|
import { Inject, Injectable } from '@nestjs/common';
|
|
import { IdService } from '@/core/IdService.js';
|
|
import type { SwSubscriptionsRepository } from '@/models/_.js';
|
|
import { Endpoint } from '@/server/api/endpoint-base.js';
|
|
import { MetaService } from '@/core/MetaService.js';
|
|
import { DI } from '@/di-symbols.js';
|
|
import { PushNotificationService } from '@/core/PushNotificationService.js';
|
|
|
|
export const meta = {
|
|
tags: ['account'],
|
|
|
|
requireCredential: true,
|
|
secure: true,
|
|
|
|
description: 'Register to receive push notifications.',
|
|
|
|
res: {
|
|
type: 'object',
|
|
optional: false, nullable: false,
|
|
properties: {
|
|
state: {
|
|
type: 'string',
|
|
optional: true, nullable: false,
|
|
enum: ['already-subscribed', 'subscribed'],
|
|
},
|
|
key: {
|
|
type: 'string',
|
|
optional: false, nullable: true,
|
|
},
|
|
userId: {
|
|
type: 'string',
|
|
optional: false, nullable: false,
|
|
},
|
|
endpoint: {
|
|
type: 'string',
|
|
optional: false, nullable: false,
|
|
},
|
|
sendReadMessage: {
|
|
type: 'boolean',
|
|
optional: false, nullable: false,
|
|
},
|
|
},
|
|
},
|
|
} as const;
|
|
|
|
export const paramDef = {
|
|
type: 'object',
|
|
properties: {
|
|
endpoint: { type: 'string' },
|
|
auth: { type: 'string' },
|
|
publickey: { type: 'string' },
|
|
sendReadMessage: { type: 'boolean', default: false },
|
|
},
|
|
required: ['endpoint', 'auth', 'publickey'],
|
|
} as const;
|
|
|
|
@Injectable()
|
|
export default class extends Endpoint<typeof meta, typeof paramDef> { // eslint-disable-line import/no-default-export
|
|
constructor(
|
|
@Inject(DI.swSubscriptionsRepository)
|
|
private swSubscriptionsRepository: SwSubscriptionsRepository,
|
|
|
|
private idService: IdService,
|
|
private metaService: MetaService,
|
|
private pushNotificationService: PushNotificationService,
|
|
) {
|
|
super(meta, paramDef, async (ps, me) => {
|
|
// if already subscribed
|
|
const exist = await this.swSubscriptionsRepository.findOneBy({
|
|
userId: me.id,
|
|
endpoint: ps.endpoint,
|
|
auth: ps.auth,
|
|
publickey: ps.publickey,
|
|
});
|
|
|
|
const instance = await this.metaService.fetch(true);
|
|
|
|
if (exist != null) {
|
|
return {
|
|
state: 'already-subscribed' as const,
|
|
key: instance.swPublicKey,
|
|
userId: me.id,
|
|
endpoint: exist.endpoint,
|
|
sendReadMessage: exist.sendReadMessage,
|
|
};
|
|
}
|
|
|
|
await this.swSubscriptionsRepository.insert({
|
|
id: this.idService.gen(),
|
|
userId: me.id,
|
|
endpoint: ps.endpoint,
|
|
auth: ps.auth,
|
|
publickey: ps.publickey,
|
|
sendReadMessage: ps.sendReadMessage,
|
|
});
|
|
|
|
this.pushNotificationService.refreshCache(me.id);
|
|
|
|
return {
|
|
state: 'subscribed' as const,
|
|
key: instance.swPublicKey,
|
|
userId: me.id,
|
|
endpoint: ps.endpoint,
|
|
sendReadMessage: ps.sendReadMessage,
|
|
};
|
|
});
|
|
}
|
|
}
|