From aeb8b66375335e8c9d6cb9cb0d8d7da3d8b79628 Mon Sep 17 00:00:00 2001
From: Leonardo Bishop <me@leonardobishop.com>
Date: Fri, 15 Mar 2024 20:23:41 +0000
Subject: Initial commit

---
 src/storage/service/storage-cron.service.ts | 34 +++++++++++++++++++++++++++++
 src/storage/service/storage.service.ts      | 21 ++++++++++++++++++
 2 files changed, 55 insertions(+)
 create mode 100644 src/storage/service/storage-cron.service.ts
 create mode 100644 src/storage/service/storage.service.ts

(limited to 'src/storage/service')

diff --git a/src/storage/service/storage-cron.service.ts b/src/storage/service/storage-cron.service.ts
new file mode 100644
index 0000000..ede6ce9
--- /dev/null
+++ b/src/storage/service/storage-cron.service.ts
@@ -0,0 +1,34 @@
+import { Injectable, Logger } from '@nestjs/common';
+import { StoredFile } from '../entity/stored-file.entity';
+import { LessThan, Repository } from 'typeorm';
+import { InjectRepository } from '@nestjs/typeorm';
+import { Cron } from '@nestjs/schedule';
+
+@Injectable()
+export class StorageCronService {
+  private readonly logger = new Logger(StorageCronService.name);
+
+  constructor(
+    @InjectRepository(StoredFile)
+    private storedFileRepository: Repository<StoredFile>,
+  ) {}
+
+  @Cron('0 30 * * * *')
+  async deleteExpired() {
+    const filesToDelete = await this.storedFileRepository.find({
+      relations: {
+        fileMetadata: true,
+      },
+      where: {
+        fileMetadata: {
+          validUntil: LessThan(new Date(Date.now())),
+        },
+      },
+    });
+
+    if (filesToDelete.length > 0) {
+      await this.storedFileRepository.remove(filesToDelete);
+      this.logger.log(`Deleted ${filesToDelete.length} expired files`);
+    }
+  }
+}
diff --git a/src/storage/service/storage.service.ts b/src/storage/service/storage.service.ts
new file mode 100644
index 0000000..6bc6b11
--- /dev/null
+++ b/src/storage/service/storage.service.ts
@@ -0,0 +1,21 @@
+import { Injectable } from '@nestjs/common';
+import { FileMetadata } from 'src/file/entity/file.entity';
+import { StoredFile } from '../entity/stored-file.entity';
+import { Repository } from 'typeorm';
+import { InjectRepository } from '@nestjs/typeorm';
+
+@Injectable()
+export class StorageService {
+  constructor(
+    @InjectRepository(StoredFile)
+    private storedFileRepository: Repository<StoredFile>,
+  ) {}
+
+  async createFile(b64Contents: string, fileMetadata: FileMetadata): Promise<StoredFile> {
+    const newFile = new StoredFile();
+    newFile.b64Contents = b64Contents;
+    newFile.fileMetadata = fileMetadata;
+
+    return this.storedFileRepository.save(newFile);
+  }
+}
-- 
cgit v1.2.3-70-g09d2