feat: asset face v3 (#31591)

2026-09-16 16:27:01 -04:00
parent 810454a343
commit 54e7ffa4e6
12 changed files with 336 additions and 39 deletions
@@ -228,3 +228,187 @@ describe(SyncEntityType.AssetFaceV2, () => {
    await ctx.assertSyncIsComplete(auth, [SyncRequestType.AssetFacesV2]);
  });
 });
+
+describe(SyncEntityType.AssetFaceV3, () => {
+  it('should detect and sync the first asset face', async () => {
+    const { auth, ctx } = await setup();
+    const { asset } = await ctx.newAsset({ ownerId: auth.user.id });
+    const { person } = await ctx.newPerson({ ownerId: auth.user.id });
+    const { assetFace } = await ctx.newAssetFace({ assetId: asset.id, personGroupId: person.personGroupId });
+
+    const response = await ctx.syncStream(auth, [SyncRequestType.AssetFacesV3]);
+    expect(response).toEqual([
+      {
+        ack: expect.any(String),
+        data: {
+          id: assetFace.id,
+          assetId: asset.id,
+          personId: person.personGroupId,
+          imageWidth: assetFace.imageWidth,
+          imageHeight: assetFace.imageHeight,
+          boundingBoxX1: assetFace.boundingBoxX1,
+          boundingBoxY1: assetFace.boundingBoxY1,
+          boundingBoxX2: assetFace.boundingBoxX2,
+          boundingBoxY2: assetFace.boundingBoxY2,
+          sourceType: assetFace.sourceType,
+          deletedAt: null,
+          isVisible: true,
+        },
+        type: SyncEntityType.AssetFaceV3,
+      },
+      expect.objectContaining({ type: SyncEntityType.SyncCompleteV1 }),
+    ]);
+
+    await ctx.syncAckAll(auth, response);
+    await ctx.assertSyncIsComplete(auth, [SyncRequestType.AssetFacesV3]);
+  });
+
+  it('should sync an asset face without a person', async () => {
+    const { auth, ctx } = await setup();
+    const { asset } = await ctx.newAsset({ ownerId: auth.user.id });
+    const { assetFace } = await ctx.newAssetFace({ assetId: asset.id });
+
+    const response = await ctx.syncStream(auth, [SyncRequestType.AssetFacesV3]);
+    expect(response).toEqual([
+      {
+        ack: expect.any(String),
+        data: expect.objectContaining({ id: assetFace.id, assetId: asset.id, personId: null }),
+        type: SyncEntityType.AssetFaceV3,
+      },
+      expect.objectContaining({ type: SyncEntityType.SyncCompleteV1 }),
+    ]);
+  });
+
+  it('should sync an asset face belonging to another user in the same cluster group', async () => {
+    const { auth, user, ctx } = await setup();
+    const { user: user2 } = await ctx.newUser({ clusterGroupId: user.clusterGroupId });
+    const { asset } = await ctx.newAsset({ ownerId: user2.id });
+    const { person } = await ctx.newPerson({ ownerId: user2.id });
+    const { assetFace } = await ctx.newAssetFace({ assetId: asset.id, personGroupId: person.personGroupId });
+
+    const response = await ctx.syncStream(auth, [SyncRequestType.AssetFacesV3]);
+    expect(response).toEqual([
+      {
+        ack: expect.any(String),
+        data: expect.objectContaining({ id: assetFace.id, assetId: asset.id, personId: person.personGroupId }),
+        type: SyncEntityType.AssetFaceV3,
+      },
+      expect.objectContaining({ type: SyncEntityType.SyncCompleteV1 }),
+    ]);
+
+    await ctx.syncAckAll(auth, response);
+    await ctx.assertSyncIsComplete(auth, [SyncRequestType.AssetFacesV3]);
+  });
+
+  it('should detect and sync an updated asset face for another user in the same cluster group', async () => {
+    const { auth, user, ctx } = await setup();
+    const personRepo = ctx.get(PersonRepository);
+    const { user: user2 } = await ctx.newUser({ clusterGroupId: user.clusterGroupId });
+    const { asset } = await ctx.newAsset({ ownerId: user2.id });
+    const { person } = await ctx.newPerson({ ownerId: user2.id });
+    const { assetFace } = await ctx.newAssetFace({ assetId: asset.id, personGroupId: person.personGroupId });
+
+    const response = await ctx.syncStream(auth, [SyncRequestType.AssetFacesV3]);
+    expect(response).toEqual([
+      expect.objectContaining({ type: SyncEntityType.AssetFaceV3 }),
+      expect.objectContaining({ type: SyncEntityType.SyncCompleteV1 }),
+    ]);
+    await ctx.syncAckAll(auth, response);
+    await ctx.assertSyncIsComplete(auth, [SyncRequestType.AssetFacesV3]);
+
+    await personRepo.softDeleteAssetFaces(assetFace.id);
+
+    expect(await ctx.syncStream(auth, [SyncRequestType.AssetFacesV3])).toEqual([
+      {
+        ack: expect.any(String),
+        data: expect.objectContaining({ id: assetFace.id, deletedAt: expect.any(String) }),
+        type: SyncEntityType.AssetFaceV3,
+      },
+      expect.objectContaining({ type: SyncEntityType.SyncCompleteV1 }),
+    ]);
+  });
+
+  it('should detect and sync a deleted asset face', async () => {
+    const { auth, ctx } = await setup();
+    const personRepo = ctx.get(PersonRepository);
+    const { asset } = await ctx.newAsset({ ownerId: auth.user.id });
+    const { person } = await ctx.newPerson({ ownerId: auth.user.id });
+    const { assetFace } = await ctx.newAssetFace({ assetId: asset.id, personGroupId: person.personGroupId });
+    await personRepo.deleteAssetFace(assetFace.id);
+
+    const response = await ctx.syncStream(auth, [SyncRequestType.AssetFacesV3]);
+    expect(response).toEqual([
+      {
+        ack: expect.any(String),
+        data: {
+          assetFaceId: assetFace.id,
+        },
+        type: SyncEntityType.AssetFaceDeleteV1,
+      },
+      expect.objectContaining({ type: SyncEntityType.SyncCompleteV1 }),
+    ]);
+
+    await ctx.syncAckAll(auth, response);
+    await ctx.assertSyncIsComplete(auth, [SyncRequestType.AssetFacesV3]);
+  });
+
+  it('should detect and sync a deleted asset face belonging to another user in the same cluster group', async () => {
+    const { auth, user, ctx } = await setup();
+    const personRepo = ctx.get(PersonRepository);
+    const { user: user2 } = await ctx.newUser({ clusterGroupId: user.clusterGroupId });
+    const { asset } = await ctx.newAsset({ ownerId: user2.id });
+    const { person } = await ctx.newPerson({ ownerId: user2.id });
+    const { assetFace } = await ctx.newAssetFace({ assetId: asset.id, personGroupId: person.personGroupId });
+    await personRepo.deleteAssetFace(assetFace.id);
+
+    const response = await ctx.syncStream(auth, [SyncRequestType.AssetFacesV3]);
+    expect(response).toEqual([
+      {
+        ack: expect.any(String),
+        data: {
+          assetFaceId: assetFace.id,
+        },
+        type: SyncEntityType.AssetFaceDeleteV1,
+      },
+      expect.objectContaining({ type: SyncEntityType.SyncCompleteV1 }),
+    ]);
+  });
+
+  it('should not sync an asset face or asset face delete for a user in a different cluster group', async () => {
+    const { auth, ctx } = await setup();
+    const personRepo = ctx.get(PersonRepository);
+    const { user: user2 } = await ctx.newUser();
+    const { session } = await ctx.newSession({ userId: user2.id });
+    const { asset } = await ctx.newAsset({ ownerId: user2.id });
+    const { person } = await ctx.newPerson({ ownerId: user2.id });
+    const { assetFace } = await ctx.newAssetFace({ assetId: asset.id, personGroupId: person.personGroupId });
+    const auth2 = factory.auth({ session, user: user2 });
+
+    expect(await ctx.syncStream(auth2, [SyncRequestType.AssetFacesV3])).toEqual([
+      expect.objectContaining({ type: SyncEntityType.AssetFaceV3 }),
+      expect.objectContaining({ type: SyncEntityType.SyncCompleteV1 }),
+    ]);
+    await ctx.assertSyncIsComplete(auth, [SyncRequestType.AssetFacesV3]);
+
+    await personRepo.deleteAssetFace(assetFace.id);
+
+    expect(await ctx.syncStream(auth2, [SyncRequestType.AssetFacesV3])).toEqual([
+      expect.objectContaining({ type: SyncEntityType.AssetFaceDeleteV1 }),
+      expect.objectContaining({ type: SyncEntityType.SyncCompleteV1 }),
+    ]);
+    await ctx.assertSyncIsComplete(auth, [SyncRequestType.AssetFacesV3]);
+  });
+
+  it('should detect and sync a deleted asset face without a person', async () => {
+    const { auth, ctx } = await setup();
+    const personRepo = ctx.get(PersonRepository);
+    const { asset } = await ctx.newAsset({ ownerId: auth.user.id });
+    const { assetFace } = await ctx.newAssetFace({ assetId: asset.id });
+    await personRepo.deleteAssetFace(assetFace.id);
+
+    expect(await ctx.syncStream(auth, [SyncRequestType.AssetFacesV3])).toEqual([
+      { ack: expect.any(String), data: { assetFaceId: assetFace.id }, type: SyncEntityType.AssetFaceDeleteV1 },
+      expect.objectContaining({ type: SyncEntityType.SyncCompleteV1 }),
+    ]);
+  });
+});