55 lines
1.7 KiB
TypeScript
55 lines
1.7 KiB
TypeScript
import { Generated, NonAttribute } from 'kysely-typeorm';
|
|
import { AssetEntity } from 'src/entities/asset.entity';
|
|
import { FaceSearchEntity } from 'src/entities/face-search.entity';
|
|
import { PersonEntity } from 'src/entities/person.entity';
|
|
import { SourceType } from 'src/enum';
|
|
import { Column, Entity, Index, ManyToOne, OneToOne, PrimaryGeneratedColumn } from 'typeorm';
|
|
|
|
@Entity('asset_faces', { synchronize: false })
|
|
@Index('IDX_asset_faces_assetId_personId', ['assetId', 'personId'])
|
|
@Index(['personId', 'assetId'])
|
|
export class AssetFaceEntity {
|
|
@PrimaryGeneratedColumn('uuid')
|
|
id!: Generated<string>;
|
|
|
|
@Column()
|
|
assetId!: string;
|
|
|
|
@Column({ nullable: true, type: 'uuid' })
|
|
personId!: string | null;
|
|
|
|
@OneToOne(() => FaceSearchEntity, (faceSearchEntity) => faceSearchEntity.face, { cascade: ['insert'] })
|
|
faceSearch!: NonAttribute<FaceSearchEntity>;
|
|
|
|
@Column({ default: 0, type: 'int' })
|
|
imageWidth!: Generated<number>;
|
|
|
|
@Column({ default: 0, type: 'int' })
|
|
imageHeight!: Generated<number>;
|
|
|
|
@Column({ default: 0, type: 'int' })
|
|
boundingBoxX1!: Generated<number>;
|
|
|
|
@Column({ default: 0, type: 'int' })
|
|
boundingBoxY1!: Generated<number>;
|
|
|
|
@Column({ default: 0, type: 'int' })
|
|
boundingBoxX2!: Generated<number>;
|
|
|
|
@Column({ default: 0, type: 'int' })
|
|
boundingBoxY2!: Generated<number>;
|
|
|
|
@Column({ default: SourceType.MACHINE_LEARNING, type: 'enum', enum: SourceType })
|
|
sourceType!: Generated<SourceType>;
|
|
|
|
@ManyToOne(() => AssetEntity, (asset) => asset.faces, { onDelete: 'CASCADE', onUpdate: 'CASCADE' })
|
|
asset!: NonAttribute<AssetEntity>;
|
|
|
|
@ManyToOne(() => PersonEntity, (person) => person.faces, {
|
|
onDelete: 'SET NULL',
|
|
onUpdate: 'CASCADE',
|
|
nullable: true,
|
|
})
|
|
person!: NonAttribute<PersonEntity | null>;
|
|
}
|