backend/src/AppModule.ts

@@ -3,6 +3,7 @@ import { ConfigModule, ConfigService, registerAs } from '@nestjs/config';
 import { TypeOrmModule } from '@nestjs/typeorm';
 import { getAppConfig, getPostgresConfig } from './config';
 import { HealthCheckModule } from './modules/healthCheck/HealthCheckModule';
+import { UploadModule } from './modules/upload/UploadModule';
 import { Config } from './types/Config';
 
 @Module({
@@ -15,7 +16,8 @@ import { Config } from './types/Config';
             useFactory: (configService: ConfigService) => configService.get('postgres') as Config['postgres'],
             inject: [ConfigService]
         }),
-        HealthCheckModule
+        HealthCheckModule,
+        UploadModule
     ]
 })
 export class AppModule {}

backend/src/database/migrations/1791230839246-add-upload-table.ts

@@ -0,0 +1,33 @@
+import { MigrationInterface, QueryRunner } from 'typeorm';
+
+export class AddUploadTable1791230839246 implements MigrationInterface {
+    name = 'AddUploadTable1791230839246';
+
+    public async up(queryRunner: QueryRunner): Promise<void> {
+        await queryRunner.query(
+            `CREATE TABLE "upload_zk" ("id" uuid NOT NULL DEFAULT uuid_generate_v4(), "passwordProtected" boolean NOT NULL DEFAULT false, "wrappedDek" bytea, "wrapIv" bytea, "salt" bytea, "uploadId" uuid, CONSTRAINT "REL_690bb5d62da9aed77980cb1106" UNIQUE ("uploadId"), CONSTRAINT "PK_8e9c09159715a88387bc8a1fa4a" PRIMARY KEY ("id"))`
+        );
+        await queryRunner.query(`CREATE TYPE "public"."uploads_mode_enum" AS ENUM('zk', 'hotlink')`);
+        await queryRunner.query(
+            `CREATE TABLE "uploads" ("id" uuid NOT NULL DEFAULT uuid_generate_v4(), "mode" "public"."uploads_mode_enum" NOT NULL, "storagePath" text NOT NULL, "iv" bytea NOT NULL, "maxViews" integer, "viewCount" integer NOT NULL DEFAULT '0', "lastViewedAt" TIMESTAMP WITH TIME ZONE, "deleteTokenHash" character varying NOT NULL, "expiresAt" TIMESTAMP WITH TIME ZONE, "createdAt" TIMESTAMP WITH TIME ZONE NOT NULL DEFAULT now(), CONSTRAINT "PK_d1781d1eedd7459314f60f39bd3" PRIMARY KEY ("id"))`
+        );
+        await queryRunner.query(
+            `CREATE TABLE "upload_hotlink" ("id" uuid NOT NULL DEFAULT uuid_generate_v4(), "mimeType" character varying NOT NULL, "uploadId" uuid, CONSTRAINT "REL_05156f9756799f6aecfcc14c10" UNIQUE ("uploadId"), CONSTRAINT "PK_abe0f3d297ad4c63773135691a4" PRIMARY KEY ("id"))`
+        );
+        await queryRunner.query(
+            `ALTER TABLE "upload_zk" ADD CONSTRAINT "FK_690bb5d62da9aed77980cb11067" FOREIGN KEY ("uploadId") REFERENCES "uploads"("id") ON DELETE CASCADE ON UPDATE NO ACTION`
+        );
+        await queryRunner.query(
+            `ALTER TABLE "upload_hotlink" ADD CONSTRAINT "FK_05156f9756799f6aecfcc14c108" FOREIGN KEY ("uploadId") REFERENCES "uploads"("id") ON DELETE CASCADE ON UPDATE NO ACTION`
+        );
+    }
+
+    public async down(queryRunner: QueryRunner): Promise<void> {
+        await queryRunner.query(`ALTER TABLE "upload_hotlink" DROP CONSTRAINT "FK_05156f9756799f6aecfcc14c108"`);
+        await queryRunner.query(`ALTER TABLE "upload_zk" DROP CONSTRAINT "FK_690bb5d62da9aed77980cb11067"`);
+        await queryRunner.query(`DROP TABLE "upload_hotlink"`);
+        await queryRunner.query(`DROP TABLE "uploads"`);
+        await queryRunner.query(`DROP TYPE "public"."uploads_mode_enum"`);
+        await queryRunner.query(`DROP TABLE "upload_zk"`);
+    }
+}

backend/src/modules/upload/UploadModule.ts

@@ -0,0 +1,10 @@
+import { Module } from '@nestjs/common';
+import { TypeOrmModule } from '@nestjs/typeorm';
+import { Upload } from './entities/Upload';
+import { UploadHotlink } from './entities/UploadHotlink';
+import { UploadZk } from './entities/UploadZk';
+
+@Module({
+    imports: [TypeOrmModule.forFeature([Upload, UploadZk, UploadHotlink])]
+})
+export class UploadModule {}

backend/src/modules/upload/entities/Upload.ts

@@ -0,0 +1,43 @@
+import { Column, CreateDateColumn, Entity, OneToOne, PrimaryGeneratedColumn } from 'typeorm';
+import { UploadMode } from '../types/UploadMode';
+import { UploadHotlink } from './UploadHotlink';
+import { UploadZk } from './UploadZk';
+
+@Entity('uploads')
+export class Upload {
+    @PrimaryGeneratedColumn('uuid')
+    id: string;
+
+    @Column({ type: 'enum', enum: UploadMode })
+    mode: UploadMode;
+
+    @Column({ type: 'text' })
+    storagePath: string;
+
+    @Column({ type: 'bytea' })
+    iv: Buffer;
+
+    @Column({ type: 'int', nullable: true })
+    maxViews: number | null;
+
+    @Column({ type: 'int', default: 0 })
+    viewCount: number;
+
+    @Column()
+    deleteTokenHash: string;
+
+    @OneToOne(() => UploadZk, zk => zk.upload)
+    zk: UploadZk | null;
+
+    @OneToOne(() => UploadHotlink, hotlink => hotlink.upload)
+    hotlink: UploadHotlink | null;
+
+    @Column({ type: 'timestamptz', nullable: true })
+    lastViewedAt: Date | null;
+
+    @Column({ type: 'timestamptz', nullable: true })
+    expiresAt: Date | null;
+
+    @CreateDateColumn({ type: 'timestamptz' })
+    createdAt: Date;
+}

backend/src/modules/upload/entities/UploadHotlink.ts

@@ -0,0 +1,15 @@
+import { Column, Entity, JoinColumn, OneToOne, PrimaryGeneratedColumn } from 'typeorm';
+import { Upload } from './Upload';
+
+@Entity('upload_hotlink')
+export class UploadHotlink {
+    @PrimaryGeneratedColumn('uuid')
+    id: string;
+
+    @Column()
+    mimeType: string;
+
+    @OneToOne(() => Upload, upload => upload.hotlink, { onDelete: 'CASCADE' })
+    @JoinColumn()
+    upload: Upload;
+}

backend/src/modules/upload/entities/UploadZk.ts

@@ -0,0 +1,24 @@
+import { Column, Entity, JoinColumn, OneToOne, PrimaryGeneratedColumn } from 'typeorm';
+import { Upload } from './Upload';
+
+@Entity('upload_zk')
+export class UploadZk {
+    @PrimaryGeneratedColumn('uuid')
+    id: string;
+
+    @Column({ type: 'boolean', default: false })
+    passwordProtected: boolean;
+
+    @Column({ type: 'bytea', nullable: true })
+    wrappedDek: Buffer | null;
+
+    @Column({ type: 'bytea', nullable: true })
+    wrapIv: Buffer | null;
+
+    @Column({ type: 'bytea', nullable: true })
+    salt: Buffer | null;
+
+    @OneToOne(() => Upload, upload => upload.zk, { onDelete: 'CASCADE' })
+    @JoinColumn()
+    upload: Upload;
+}

backend/src/modules/upload/types/UploadMode.ts

@@ -0,0 +1,4 @@
+export enum UploadMode {
+    Zk = 'zk',
+    Hotlink = 'hotlink'
+}