mirror of https://github.com/immich-app/immich.git
Merge bb0ee28486 into 4cbce072be
Diogo Correia
GitHub
commit
9dd5e84496
@ -0,0 +1,69 @@
|
||||
import { Injectable } from '@nestjs/common';
|
||||
import { Insertable, Kysely, sql } from 'kysely';
|
||||
import _ from 'lodash';
|
||||
import { DateTime } from 'luxon';
|
||||
import { InjectKysely } from 'nestjs-kysely';
|
||||
import { DummyValue, GenerateSql } from 'src/decorators';
|
||||
import { DB } from 'src/schema';
|
||||
import { DownloadRequestTable } from 'src/schema/tables/download-request.table';
|
||||
|
||||
@Injectable()
|
||||
export class DownloadRequestRepository {
|
||||
constructor(@InjectKysely() private db: Kysely<DB>) {}
|
||||
|
||||
cleanup() {
|
||||
return this.db
|
||||
.deleteFrom('download_request')
|
||||
.where('download_request.expiresAt', '<=', DateTime.now().toJSDate())
|
||||
.returning(['id'])
|
||||
.execute();
|
||||
}
|
||||
|
||||
@GenerateSql({ params: [DummyValue.UUID] })
|
||||
get(id: string) {
|
||||
return this.db
|
||||
.selectFrom('download_request')
|
||||
.selectAll('download_request')
|
||||
.where((eb) =>
|
||||
eb.and([eb('download_request.id', '=', id), eb('download_request.expiresAt', '>', DateTime.now().toJSDate())]),
|
||||
)
|
||||
.leftJoin('download_request_asset', 'download_request_asset.downloadRequestId', 'download_request.id')
|
||||
.select((eb) =>
|
||||
eb.fn
|
||||
.coalesce(eb.fn.jsonAgg('download_request_asset.assetId'), sql`'[]'`)
|
||||
.$castTo<string[]>()
|
||||
.as('assetIds'),
|
||||
)
|
||||
.groupBy('download_request.id')
|
||||
.executeTakeFirstOrThrow();
|
||||
}
|
||||
|
||||
async create(entity: Insertable<DownloadRequestTable> & { assetIds?: string[] }) {
|
||||
const { id } = await this.db
|
||||
.insertInto('download_request')
|
||||
.values(_.omit(entity, 'assetIds'))
|
||||
.returningAll()
|
||||
.executeTakeFirstOrThrow();
|
||||
|
||||
if (entity.assetIds && entity.assetIds.length > 0) {
|
||||
await this.db
|
||||
.insertInto('download_request_asset')
|
||||
.values(entity.assetIds!.map((assetId) => ({ assetId, downloadRequestId: id })))
|
||||
.execute();
|
||||
}
|
||||
|
||||
return this.getDownloadRequest(id);
|
||||
}
|
||||
|
||||
async remove(id: string): Promise<void> {
|
||||
await this.db.deleteFrom('download_request').where('download_request.id', '=', id).execute();
|
||||
}
|
||||
|
||||
private getDownloadRequest(id: string) {
|
||||
return this.db
|
||||
.selectFrom('download_request')
|
||||
.selectAll('download_request')
|
||||
.where('download_request.id', '=', id)
|
||||
.executeTakeFirstOrThrow();
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,23 @@
|
||||
import { Kysely, sql } from 'kysely';
|
||||
|
||||
export async function up(db: Kysely<any>): Promise<void> {
|
||||
await sql`CREATE TABLE "download_request" (
|
||||
"id" uuid NOT NULL DEFAULT uuid_generate_v4(),
|
||||
"expiresAt" timestamp with time zone NOT NULL,
|
||||
CONSTRAINT "download_request_pkey" PRIMARY KEY ("id")
|
||||
);`.execute(db);
|
||||
await sql`CREATE TABLE "download_request_asset" (
|
||||
"assetId" uuid NOT NULL,
|
||||
"downloadRequestId" uuid NOT NULL,
|
||||
CONSTRAINT "download_request_asset_assetId_fkey" FOREIGN KEY ("assetId") REFERENCES "asset" ("id") ON UPDATE CASCADE ON DELETE CASCADE,
|
||||
CONSTRAINT "download_request_asset_downloadRequestId_fkey" FOREIGN KEY ("downloadRequestId") REFERENCES "download_request" ("id") ON UPDATE CASCADE ON DELETE CASCADE,
|
||||
CONSTRAINT "download_request_asset_pkey" PRIMARY KEY ("assetId", "downloadRequestId")
|
||||
);`.execute(db);
|
||||
await sql`CREATE INDEX "download_request_asset_assetId_idx" ON "download_request_asset" ("assetId");`.execute(db);
|
||||
await sql`CREATE INDEX "download_request_asset_downloadRequestId_idx" ON "download_request_asset" ("downloadRequestId");`.execute(db);
|
||||
}
|
||||
|
||||
export async function down(db: Kysely<any>): Promise<void> {
|
||||
await sql`DROP TABLE "download_request_asset";`.execute(db);
|
||||
await sql`DROP TABLE "download_request";`.execute(db);
|
||||
}
|
||||
@ -0,0 +1,12 @@
|
||||
import { AssetTable } from 'src/schema/tables/asset.table';
|
||||
import { DownloadRequestTable } from 'src/schema/tables/download-request.table';
|
||||
import { ForeignKeyColumn, Table } from 'src/sql-tools';
|
||||
|
||||
@Table('download_request_asset')
|
||||
export class DownloadRequestAssetTable {
|
||||
@ForeignKeyColumn(() => AssetTable, { onUpdate: 'CASCADE', onDelete: 'CASCADE', primary: true })
|
||||
assetId!: string;
|
||||
|
||||
@ForeignKeyColumn(() => DownloadRequestTable, { onUpdate: 'CASCADE', onDelete: 'CASCADE', primary: true })
|
||||
downloadRequestId!: string;
|
||||
}
|
||||
@ -0,0 +1,10 @@
|
||||
import { Column, Generated, PrimaryGeneratedColumn, Table, Timestamp } from 'src/sql-tools';
|
||||
|
||||
@Table('download_request')
|
||||
export class DownloadRequestTable {
|
||||
@PrimaryGeneratedColumn()
|
||||
id!: Generated<string>;
|
||||
|
||||
@Column({ type: 'timestamp with time zone' })
|
||||
expiresAt!: Timestamp;
|
||||
}
|
||||
Loading…
Reference in New Issue