feat: facial recognition (#2180)

server/libs/infra/src/entities/asset-face.entity.ts

@@ -0,0 +1,25 @@
+import { Column, Entity, ManyToOne, PrimaryColumn } from 'typeorm';
+import { AssetEntity } from './asset.entity';
+import { PersonEntity } from './person.entity';
+
+@Entity('asset_faces')
+export class AssetFaceEntity {
+  @PrimaryColumn()
+  assetId!: string;
+
+  @PrimaryColumn()
+  personId!: string;
+
+  @Column({
+    type: 'float4',
+    array: true,
+    nullable: true,
+  })
+  embedding!: number[] | null;
+
+  @ManyToOne(() => AssetEntity, (asset) => asset.faces, { onDelete: 'CASCADE', onUpdate: 'CASCADE' })
+  asset!: AssetEntity;
+
+  @ManyToOne(() => PersonEntity, (person) => person.faces, { onDelete: 'CASCADE', onUpdate: 'CASCADE' })
+  person!: PersonEntity;
+}

server/libs/infra/src/entities/asset.entity.ts

@@ -7,12 +7,14 @@ import {
   JoinTable,
   ManyToMany,
   ManyToOne,
+  OneToMany,
   OneToOne,
   PrimaryGeneratedColumn,
   Unique,
   UpdateDateColumn,
 } from 'typeorm';
 import { AlbumEntity } from './album.entity';
+import { AssetFaceEntity } from './asset-face.entity';
 import { ExifEntity } from './exif.entity';
 import { SharedLinkEntity } from './shared-link.entity';
 import { SmartInfoEntity } from './smart-info.entity';
@@ -109,6 +111,9 @@ export class AssetEntity {
 
   @ManyToMany(() => AlbumEntity, (album) => album.assets, { onDelete: 'CASCADE', onUpdate: 'CASCADE' })
   albums?: AlbumEntity[];
+
+  @OneToMany(() => AssetFaceEntity, (assetFace) => assetFace.asset)
+  faces!: AssetFaceEntity[];
 }
 
 export enum AssetType {

server/libs/infra/src/entities/index.ts

@@ -1,18 +1,22 @@
 import { AlbumEntity } from './album.entity';
 import { APIKeyEntity } from './api-key.entity';
+import { AssetFaceEntity } from './asset-face.entity';
 import { AssetEntity } from './asset.entity';
 import { PartnerEntity } from './partner.entity';
+import { PersonEntity } from './person.entity';
 import { SharedLinkEntity } from './shared-link.entity';
 import { SmartInfoEntity } from './smart-info.entity';
 import { SystemConfigEntity } from './system-config.entity';
-import { UserEntity } from './user.entity';
 import { UserTokenEntity } from './user-token.entity';
+import { UserEntity } from './user.entity';
 
 export * from './album.entity';
 export * from './api-key.entity';
+export * from './asset-face.entity';
 export * from './asset.entity';
 export * from './exif.entity';
 export * from './partner.entity';
+export * from './person.entity';
 export * from './shared-link.entity';
 export * from './smart-info.entity';
 export * from './system-config.entity';
@@ -24,7 +28,9 @@ export const databaseEntities = [
   AlbumEntity,
   APIKeyEntity,
   AssetEntity,
+  AssetFaceEntity,
   PartnerEntity,
+  PersonEntity,
   SharedLinkEntity,
   SmartInfoEntity,
   SystemConfigEntity,

server/libs/infra/src/entities/person.entity.ts

@@ -0,0 +1,38 @@
+import {
+  Column,
+  CreateDateColumn,
+  Entity,
+  ManyToOne,
+  OneToMany,
+  PrimaryGeneratedColumn,
+  UpdateDateColumn,
+} from 'typeorm';
+import { AssetFaceEntity } from './asset-face.entity';
+import { UserEntity } from './user.entity';
+
+@Entity('person')
+export class PersonEntity {
+  @PrimaryGeneratedColumn('uuid')
+  id!: string;
+
+  @CreateDateColumn({ type: 'timestamptz' })
+  createdAt!: Date;
+
+  @UpdateDateColumn({ type: 'timestamptz' })
+  updatedAt!: Date;
+
+  @Column()
+  ownerId!: string;
+
+  @ManyToOne(() => UserEntity, { onDelete: 'CASCADE', onUpdate: 'CASCADE', nullable: false })
+  owner!: UserEntity;
+
+  @Column({ default: '' })
+  name!: string;
+
+  @Column({ default: '' })
+  thumbnailPath!: string;
+
+  @OneToMany(() => AssetFaceEntity, (assetFace) => assetFace.person)
+  faces!: AssetFaceEntity[];
+}