diff --git a/src/common/repositories/TotpCodesRepository.ts b/src/common/repositories/TotpCodesRepository.ts index 8d9d9c33..8d1ebf48 100644 --- a/src/common/repositories/TotpCodesRepository.ts +++ b/src/common/repositories/TotpCodesRepository.ts @@ -67,4 +67,11 @@ export default class TotpCodesRepository extends BaseRepository { }, }); } + + /** + * Delete many totp codes + */ + public async deleteMany(query: Prisma.TotpCodesDeleteManyArgs) { + return this.model.deleteMany({ ...query }); + } } diff --git a/src/services/common/CronService/CronService.ts b/src/services/common/CronService/CronService.ts index 4b46e962..94584f54 100644 --- a/src/services/common/CronService/CronService.ts +++ b/src/services/common/CronService/CronService.ts @@ -6,6 +6,7 @@ import IdNotService from "../IdNotService/IdNotService"; import { PrismaClient } from "@prisma/client"; import NotificationBuilder from "@Common/notifications/NotificationBuilder"; import EmailBuilder from "@Common/emails/EmailBuilder"; +import TotpService from "@Services/customer/TotpService/TotpService"; // import { PrismaClient } from "@prisma/client"; @Service() @@ -16,6 +17,7 @@ export default class CronService { private idNotService: IdNotService, private notificationBuilder: NotificationBuilder, private emailBuilder: EmailBuilder, + private totpService: TotpService, ) {} public async sendMails() { @@ -86,8 +88,8 @@ export default class CronService { // Once a day at midnight try { const prisma = new PrismaClient(); - const expiringDocuments15Days: [{uid: string, expiration_date: Date}] = await prisma.$queryRaw - `SELECT distinct o.uid, f.created_at as expiration_date + const expiringDocuments15Days: [{ uid: string; expiration_date: Date }] = + await prisma.$queryRaw`SELECT distinct o.uid, f.created_at as expiration_date FROM documents d JOIN files f ON d.uid=f.document_uid JOIN office_folders o ON o.uid=d.folder_uid @@ -95,11 +97,24 @@ export default class CronService { AND d.document_status = 'DEPOSITED' AND current_date = (DATE(f.created_at) + interval '3 months' - interval '2 days');`; - expiringDocuments15Days.forEach(expiringFolder => { + expiringDocuments15Days.forEach((expiringFolder) => { this.notificationBuilder.sendDocumentExpiringSoonNotification(expiringFolder.uid, expiringFolder.expiration_date); }); + } catch (e) { + console.error(e); + } + }); + // Start job + if (!cronJob.running) { + cronJob.start(); + } + } - + public async cleanExpiredTotpCodes() { + const cronJob = new CronJob("0 0 * * *", async () => { + // Once a day at midnight + try { + await this.totpService.cleanExpiredTotpCodes(); } catch (e) { console.error(e); } diff --git a/src/services/customer/TotpService/TotpService.ts b/src/services/customer/TotpService/TotpService.ts new file mode 100644 index 00000000..ed8a9b20 --- /dev/null +++ b/src/services/customer/TotpService/TotpService.ts @@ -0,0 +1,51 @@ +import { Prisma, TotpCodes } from "@prisma/client"; +import TotpCodesRepository from "@Repositories/TotpCodesRepository"; +import BaseService from "@Services/BaseService"; +import { Service } from "typedi"; + +@Service() +export default class TotpService extends BaseService { + constructor(private totpCodesRepository: TotpCodesRepository) { + super(); + } + + /** + * @description : Get all totp codes + * @throws {Error} If totp codes cannot be get + */ + public async get(query: Prisma.TotpCodesFindManyArgs): Promise { + return this.totpCodesRepository.findMany(query); + } + + /** + * @description : Get one totp code + * @throws {Error} If totp code cannot be get + */ + public async getOne(query: Prisma.TotpCodesFindFirstArgs): Promise { + return this.totpCodesRepository.findOne(query); + } + + /** + * @description : Delete many totp codes + * @throws {Error} If totp code cannot be deleted + */ + public async deleteMany(query: Prisma.TotpCodesDeleteManyArgs): Promise { + return this.totpCodesRepository.deleteMany(query); + } + + /** + * @description : Delete every totp code expired for more than 30 days + * @throws {Error} If totp codes cannot be deleted + */ + public async cleanExpiredTotpCodes() { + const query: Prisma.TotpCodesDeleteManyArgs = { + where: { + expire_at: { + lte: new Date(Date.now() - 30 * 24 * 60 * 60 * 1000), + }, + }, + }; + + return this.totpCodesRepository.deleteMany(query); + } +}