mirror of
				https://github.com/KevinMidboe/immich.git
				synced 2025-10-29 17:40:28 +00:00 
			
		
		
		
	chore(server): Use access core for person permissions (#4138)
* use access core for all person methods * minor fixes, feedback * reorder assignments * remove unnecessary permission requirement * unify naming of tests * reorder variables
This commit is contained in:
		@@ -23,6 +23,10 @@ export enum Permission {
 | 
			
		||||
 | 
			
		||||
  LIBRARY_READ = 'library.read',
 | 
			
		||||
  LIBRARY_DOWNLOAD = 'library.download',
 | 
			
		||||
 | 
			
		||||
  PERSON_READ = 'person.read',
 | 
			
		||||
  PERSON_WRITE = 'person.write',
 | 
			
		||||
  PERSON_MERGE = 'person.merge',
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
export class AccessCore {
 | 
			
		||||
@@ -167,6 +171,15 @@ export class AccessCore {
 | 
			
		||||
      case Permission.LIBRARY_DOWNLOAD:
 | 
			
		||||
        return authUser.id === id;
 | 
			
		||||
 | 
			
		||||
      case Permission.PERSON_READ:
 | 
			
		||||
        return this.repository.person.hasOwnerAccess(authUser.id, id);
 | 
			
		||||
 | 
			
		||||
      case Permission.PERSON_WRITE:
 | 
			
		||||
        return this.repository.person.hasOwnerAccess(authUser.id, id);
 | 
			
		||||
 | 
			
		||||
      case Permission.PERSON_MERGE:
 | 
			
		||||
        return this.repository.person.hasOwnerAccess(authUser.id, id);
 | 
			
		||||
 | 
			
		||||
      default:
 | 
			
		||||
        return false;
 | 
			
		||||
    }
 | 
			
		||||
 
 | 
			
		||||
@@ -17,4 +17,8 @@ export interface IAccessRepository {
 | 
			
		||||
  library: {
 | 
			
		||||
    hasPartnerAccess(userId: string, partnerId: string): Promise<boolean>;
 | 
			
		||||
  };
 | 
			
		||||
 | 
			
		||||
  person: {
 | 
			
		||||
    hasOwnerAccess(userId: string, personId: string): Promise<boolean>;
 | 
			
		||||
  };
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -17,9 +17,9 @@ export interface IPersonRepository {
 | 
			
		||||
  getAllWithoutThumbnail(): Promise<PersonEntity[]>;
 | 
			
		||||
  getAllForUser(userId: string, options: PersonSearchOptions): Promise<PersonEntity[]>;
 | 
			
		||||
  getAllWithoutFaces(): Promise<PersonEntity[]>;
 | 
			
		||||
  getById(userId: string, personId: string): Promise<PersonEntity | null>;
 | 
			
		||||
  getById(personId: string): Promise<PersonEntity | null>;
 | 
			
		||||
 | 
			
		||||
  getAssets(userId: string, personId: string): Promise<AssetEntity[]>;
 | 
			
		||||
  getAssets(personId: string): Promise<AssetEntity[]>;
 | 
			
		||||
  prepareReassignFaces(data: UpdateFacesData): Promise<string[]>;
 | 
			
		||||
  reassignFaces(data: UpdateFacesData): Promise<number>;
 | 
			
		||||
 | 
			
		||||
 
 | 
			
		||||
@@ -1,8 +1,10 @@
 | 
			
		||||
import { BadRequestException, NotFoundException } from '@nestjs/common';
 | 
			
		||||
import {
 | 
			
		||||
  IAccessRepositoryMock,
 | 
			
		||||
  assetStub,
 | 
			
		||||
  authStub,
 | 
			
		||||
  faceStub,
 | 
			
		||||
  newAccessRepositoryMock,
 | 
			
		||||
  newJobRepositoryMock,
 | 
			
		||||
  newPersonRepositoryMock,
 | 
			
		||||
  newStorageRepositoryMock,
 | 
			
		||||
@@ -26,18 +28,20 @@ const responseDto: PersonResponseDto = {
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
describe(PersonService.name, () => {
 | 
			
		||||
  let sut: PersonService;
 | 
			
		||||
  let personMock: jest.Mocked<IPersonRepository>;
 | 
			
		||||
  let accessMock: IAccessRepositoryMock;
 | 
			
		||||
  let configMock: jest.Mocked<ISystemConfigRepository>;
 | 
			
		||||
  let storageMock: jest.Mocked<IStorageRepository>;
 | 
			
		||||
  let jobMock: jest.Mocked<IJobRepository>;
 | 
			
		||||
  let personMock: jest.Mocked<IPersonRepository>;
 | 
			
		||||
  let storageMock: jest.Mocked<IStorageRepository>;
 | 
			
		||||
  let sut: PersonService;
 | 
			
		||||
 | 
			
		||||
  beforeEach(async () => {
 | 
			
		||||
    accessMock = newAccessRepositoryMock();
 | 
			
		||||
    personMock = newPersonRepositoryMock();
 | 
			
		||||
    storageMock = newStorageRepositoryMock();
 | 
			
		||||
    configMock = newSystemConfigRepositoryMock();
 | 
			
		||||
    jobMock = newJobRepositoryMock();
 | 
			
		||||
    sut = new PersonService(personMock, configMock, storageMock, jobMock);
 | 
			
		||||
    sut = new PersonService(accessMock, personMock, configMock, storageMock, jobMock);
 | 
			
		||||
  });
 | 
			
		||||
 | 
			
		||||
  it('should be defined', () => {
 | 
			
		||||
@@ -93,74 +97,124 @@ describe(PersonService.name, () => {
 | 
			
		||||
  });
 | 
			
		||||
 | 
			
		||||
  describe('getById', () => {
 | 
			
		||||
    it('should require person.read permission', async () => {
 | 
			
		||||
      personMock.getById.mockResolvedValue(personStub.withName);
 | 
			
		||||
      accessMock.person.hasOwnerAccess.mockResolvedValue(false);
 | 
			
		||||
      await expect(sut.getById(authStub.admin, 'person-1')).rejects.toBeInstanceOf(BadRequestException);
 | 
			
		||||
      expect(accessMock.person.hasOwnerAccess).toHaveBeenCalledWith(authStub.admin.id, 'person-1');
 | 
			
		||||
    });
 | 
			
		||||
 | 
			
		||||
    it('should throw a bad request when person is not found', async () => {
 | 
			
		||||
      personMock.getById.mockResolvedValue(null);
 | 
			
		||||
      accessMock.person.hasOwnerAccess.mockResolvedValue(true);
 | 
			
		||||
      await expect(sut.getById(authStub.admin, 'person-1')).rejects.toBeInstanceOf(BadRequestException);
 | 
			
		||||
      expect(accessMock.person.hasOwnerAccess).toHaveBeenCalledWith(authStub.admin.id, 'person-1');
 | 
			
		||||
    });
 | 
			
		||||
 | 
			
		||||
    it('should get a person by id', async () => {
 | 
			
		||||
      personMock.getById.mockResolvedValue(personStub.withName);
 | 
			
		||||
      accessMock.person.hasOwnerAccess.mockResolvedValue(true);
 | 
			
		||||
      await expect(sut.getById(authStub.admin, 'person-1')).resolves.toEqual(responseDto);
 | 
			
		||||
      expect(personMock.getById).toHaveBeenCalledWith(authStub.admin.id, 'person-1');
 | 
			
		||||
      expect(personMock.getById).toHaveBeenCalledWith('person-1');
 | 
			
		||||
      expect(accessMock.person.hasOwnerAccess).toHaveBeenCalledWith(authStub.admin.id, 'person-1');
 | 
			
		||||
    });
 | 
			
		||||
  });
 | 
			
		||||
 | 
			
		||||
  describe('getThumbnail', () => {
 | 
			
		||||
    it('should require person.read permission', async () => {
 | 
			
		||||
      personMock.getById.mockResolvedValue(personStub.noName);
 | 
			
		||||
      accessMock.person.hasOwnerAccess.mockResolvedValue(false);
 | 
			
		||||
      await expect(sut.getThumbnail(authStub.admin, 'person-1')).rejects.toBeInstanceOf(BadRequestException);
 | 
			
		||||
      expect(storageMock.createReadStream).not.toHaveBeenCalled();
 | 
			
		||||
      expect(accessMock.person.hasOwnerAccess).toHaveBeenCalledWith(authStub.admin.id, 'person-1');
 | 
			
		||||
    });
 | 
			
		||||
 | 
			
		||||
    it('should throw an error when personId is invalid', async () => {
 | 
			
		||||
      personMock.getById.mockResolvedValue(null);
 | 
			
		||||
      accessMock.person.hasOwnerAccess.mockResolvedValue(true);
 | 
			
		||||
      await expect(sut.getThumbnail(authStub.admin, 'person-1')).rejects.toBeInstanceOf(NotFoundException);
 | 
			
		||||
      expect(storageMock.createReadStream).not.toHaveBeenCalled();
 | 
			
		||||
      expect(accessMock.person.hasOwnerAccess).toHaveBeenCalledWith(authStub.admin.id, 'person-1');
 | 
			
		||||
    });
 | 
			
		||||
 | 
			
		||||
    it('should throw an error when person has no thumbnail', async () => {
 | 
			
		||||
      personMock.getById.mockResolvedValue(personStub.noThumbnail);
 | 
			
		||||
      accessMock.person.hasOwnerAccess.mockResolvedValue(true);
 | 
			
		||||
      await expect(sut.getThumbnail(authStub.admin, 'person-1')).rejects.toBeInstanceOf(NotFoundException);
 | 
			
		||||
      expect(storageMock.createReadStream).not.toHaveBeenCalled();
 | 
			
		||||
      expect(accessMock.person.hasOwnerAccess).toHaveBeenCalledWith(authStub.admin.id, 'person-1');
 | 
			
		||||
    });
 | 
			
		||||
 | 
			
		||||
    it('should serve the thumbnail', async () => {
 | 
			
		||||
      personMock.getById.mockResolvedValue(personStub.noName);
 | 
			
		||||
      accessMock.person.hasOwnerAccess.mockResolvedValue(true);
 | 
			
		||||
      await sut.getThumbnail(authStub.admin, 'person-1');
 | 
			
		||||
      expect(storageMock.createReadStream).toHaveBeenCalledWith('/path/to/thumbnail.jpg', 'image/jpeg');
 | 
			
		||||
      expect(accessMock.person.hasOwnerAccess).toHaveBeenCalledWith(authStub.admin.id, 'person-1');
 | 
			
		||||
    });
 | 
			
		||||
  });
 | 
			
		||||
 | 
			
		||||
  describe('getAssets', () => {
 | 
			
		||||
    it('should require person.read permission', async () => {
 | 
			
		||||
      personMock.getAssets.mockResolvedValue([assetStub.image, assetStub.video]);
 | 
			
		||||
      accessMock.person.hasOwnerAccess.mockResolvedValue(false);
 | 
			
		||||
      await expect(sut.getAssets(authStub.admin, 'person-1')).rejects.toBeInstanceOf(BadRequestException);
 | 
			
		||||
      expect(personMock.getAssets).not.toHaveBeenCalled();
 | 
			
		||||
      expect(accessMock.person.hasOwnerAccess).toHaveBeenCalledWith(authStub.admin.id, 'person-1');
 | 
			
		||||
    });
 | 
			
		||||
 | 
			
		||||
    it("should return a person's assets", async () => {
 | 
			
		||||
      personMock.getAssets.mockResolvedValue([assetStub.image, assetStub.video]);
 | 
			
		||||
      accessMock.person.hasOwnerAccess.mockResolvedValue(true);
 | 
			
		||||
      await sut.getAssets(authStub.admin, 'person-1');
 | 
			
		||||
      expect(personMock.getAssets).toHaveBeenCalledWith('admin_id', 'person-1');
 | 
			
		||||
      expect(personMock.getAssets).toHaveBeenCalledWith('person-1');
 | 
			
		||||
      expect(accessMock.person.hasOwnerAccess).toHaveBeenCalledWith(authStub.admin.id, 'person-1');
 | 
			
		||||
    });
 | 
			
		||||
  });
 | 
			
		||||
 | 
			
		||||
  describe('update', () => {
 | 
			
		||||
    it('should throw an error when personId is invalid', async () => {
 | 
			
		||||
      personMock.getById.mockResolvedValue(null);
 | 
			
		||||
    it('should require person.write permission', async () => {
 | 
			
		||||
      personMock.getById.mockResolvedValue(personStub.noName);
 | 
			
		||||
      accessMock.person.hasOwnerAccess.mockResolvedValue(false);
 | 
			
		||||
      await expect(sut.update(authStub.admin, 'person-1', { name: 'Person 1' })).rejects.toBeInstanceOf(
 | 
			
		||||
        BadRequestException,
 | 
			
		||||
      );
 | 
			
		||||
      expect(personMock.update).not.toHaveBeenCalled();
 | 
			
		||||
      expect(accessMock.person.hasOwnerAccess).toHaveBeenCalledWith(authStub.admin.id, 'person-1');
 | 
			
		||||
    });
 | 
			
		||||
 | 
			
		||||
    it('should throw an error when personId is invalid', async () => {
 | 
			
		||||
      personMock.getById.mockResolvedValue(null);
 | 
			
		||||
      accessMock.person.hasOwnerAccess.mockResolvedValue(true);
 | 
			
		||||
      await expect(sut.update(authStub.admin, 'person-1', { name: 'Person 1' })).rejects.toBeInstanceOf(
 | 
			
		||||
        BadRequestException,
 | 
			
		||||
      );
 | 
			
		||||
      expect(personMock.update).not.toHaveBeenCalled();
 | 
			
		||||
      expect(accessMock.person.hasOwnerAccess).toHaveBeenCalledWith(authStub.admin.id, 'person-1');
 | 
			
		||||
    });
 | 
			
		||||
 | 
			
		||||
    it("should update a person's name", async () => {
 | 
			
		||||
      personMock.getById.mockResolvedValue(personStub.noName);
 | 
			
		||||
      personMock.update.mockResolvedValue(personStub.withName);
 | 
			
		||||
      personMock.getAssets.mockResolvedValue([assetStub.image]);
 | 
			
		||||
      accessMock.person.hasOwnerAccess.mockResolvedValue(true);
 | 
			
		||||
 | 
			
		||||
      await expect(sut.update(authStub.admin, 'person-1', { name: 'Person 1' })).resolves.toEqual(responseDto);
 | 
			
		||||
 | 
			
		||||
      expect(personMock.getById).toHaveBeenCalledWith('admin_id', 'person-1');
 | 
			
		||||
      expect(personMock.getById).toHaveBeenCalledWith('person-1');
 | 
			
		||||
      expect(personMock.update).toHaveBeenCalledWith({ id: 'person-1', name: 'Person 1' });
 | 
			
		||||
      expect(jobMock.queue).toHaveBeenCalledWith({
 | 
			
		||||
        name: JobName.SEARCH_INDEX_ASSET,
 | 
			
		||||
        data: { ids: [assetStub.image.id] },
 | 
			
		||||
      });
 | 
			
		||||
      expect(accessMock.person.hasOwnerAccess).toHaveBeenCalledWith(authStub.admin.id, 'person-1');
 | 
			
		||||
    });
 | 
			
		||||
 | 
			
		||||
    it("should update a person's date of birth", async () => {
 | 
			
		||||
      personMock.getById.mockResolvedValue(personStub.noBirthDate);
 | 
			
		||||
      personMock.update.mockResolvedValue(personStub.withBirthDate);
 | 
			
		||||
      personMock.getAssets.mockResolvedValue([assetStub.image]);
 | 
			
		||||
      accessMock.person.hasOwnerAccess.mockResolvedValue(true);
 | 
			
		||||
 | 
			
		||||
      await expect(sut.update(authStub.admin, 'person-1', { birthDate: new Date('1976-06-30') })).resolves.toEqual({
 | 
			
		||||
        id: 'person-1',
 | 
			
		||||
@@ -170,35 +224,39 @@ describe(PersonService.name, () => {
 | 
			
		||||
        isHidden: false,
 | 
			
		||||
      });
 | 
			
		||||
 | 
			
		||||
      expect(personMock.getById).toHaveBeenCalledWith('admin_id', 'person-1');
 | 
			
		||||
      expect(personMock.getById).toHaveBeenCalledWith('person-1');
 | 
			
		||||
      expect(personMock.update).toHaveBeenCalledWith({ id: 'person-1', birthDate: new Date('1976-06-30') });
 | 
			
		||||
      expect(jobMock.queue).not.toHaveBeenCalled();
 | 
			
		||||
      expect(accessMock.person.hasOwnerAccess).toHaveBeenCalledWith(authStub.admin.id, 'person-1');
 | 
			
		||||
    });
 | 
			
		||||
 | 
			
		||||
    it('should update a person visibility', async () => {
 | 
			
		||||
      personMock.getById.mockResolvedValue(personStub.hidden);
 | 
			
		||||
      personMock.update.mockResolvedValue(personStub.withName);
 | 
			
		||||
      personMock.getAssets.mockResolvedValue([assetStub.image]);
 | 
			
		||||
      accessMock.person.hasOwnerAccess.mockResolvedValue(true);
 | 
			
		||||
 | 
			
		||||
      await expect(sut.update(authStub.admin, 'person-1', { isHidden: false })).resolves.toEqual(responseDto);
 | 
			
		||||
 | 
			
		||||
      expect(personMock.getById).toHaveBeenCalledWith('admin_id', 'person-1');
 | 
			
		||||
      expect(personMock.getById).toHaveBeenCalledWith('person-1');
 | 
			
		||||
      expect(personMock.update).toHaveBeenCalledWith({ id: 'person-1', isHidden: false });
 | 
			
		||||
      expect(jobMock.queue).toHaveBeenCalledWith({
 | 
			
		||||
        name: JobName.SEARCH_INDEX_ASSET,
 | 
			
		||||
        data: { ids: [assetStub.image.id] },
 | 
			
		||||
      });
 | 
			
		||||
      expect(accessMock.person.hasOwnerAccess).toHaveBeenCalledWith(authStub.admin.id, 'person-1');
 | 
			
		||||
    });
 | 
			
		||||
 | 
			
		||||
    it("should update a person's thumbnailPath", async () => {
 | 
			
		||||
      personMock.getById.mockResolvedValue(personStub.withName);
 | 
			
		||||
      personMock.getFaceById.mockResolvedValue(faceStub.face1);
 | 
			
		||||
      accessMock.person.hasOwnerAccess.mockResolvedValue(true);
 | 
			
		||||
 | 
			
		||||
      await expect(
 | 
			
		||||
        sut.update(authStub.admin, 'person-1', { featureFaceAssetId: faceStub.face1.assetId }),
 | 
			
		||||
      ).resolves.toEqual(responseDto);
 | 
			
		||||
 | 
			
		||||
      expect(personMock.getById).toHaveBeenCalledWith('admin_id', 'person-1');
 | 
			
		||||
      expect(personMock.getById).toHaveBeenCalledWith('person-1');
 | 
			
		||||
      expect(personMock.getFaceById).toHaveBeenCalledWith({
 | 
			
		||||
        assetId: faceStub.face1.assetId,
 | 
			
		||||
        personId: 'person-1',
 | 
			
		||||
@@ -218,25 +276,31 @@ describe(PersonService.name, () => {
 | 
			
		||||
          imageWidth: faceStub.face1.imageWidth,
 | 
			
		||||
        },
 | 
			
		||||
      });
 | 
			
		||||
      expect(accessMock.person.hasOwnerAccess).toHaveBeenCalledWith(authStub.admin.id, 'person-1');
 | 
			
		||||
    });
 | 
			
		||||
 | 
			
		||||
    it('should throw an error when the face feature assetId is invalid', async () => {
 | 
			
		||||
      personMock.getById.mockResolvedValue(personStub.withName);
 | 
			
		||||
      accessMock.person.hasOwnerAccess.mockResolvedValue(true);
 | 
			
		||||
 | 
			
		||||
      await expect(sut.update(authStub.admin, 'person-1', { featureFaceAssetId: '-1' })).rejects.toThrow(
 | 
			
		||||
        BadRequestException,
 | 
			
		||||
      );
 | 
			
		||||
      expect(personMock.update).not.toHaveBeenCalled();
 | 
			
		||||
      expect(accessMock.person.hasOwnerAccess).toHaveBeenCalledWith(authStub.admin.id, 'person-1');
 | 
			
		||||
    });
 | 
			
		||||
  });
 | 
			
		||||
 | 
			
		||||
  describe('updateAll', () => {
 | 
			
		||||
    it('should throw an error when personId is invalid', async () => {
 | 
			
		||||
      personMock.getById.mockResolvedValue(null);
 | 
			
		||||
      accessMock.person.hasOwnerAccess.mockResolvedValue(true);
 | 
			
		||||
 | 
			
		||||
      await expect(
 | 
			
		||||
        sut.updatePeople(authStub.admin, { people: [{ id: 'person-1', name: 'Person 1' }] }),
 | 
			
		||||
      ).resolves.toEqual([{ error: BulkIdErrorReason.UNKNOWN, id: 'person-1', success: false }]);
 | 
			
		||||
      expect(personMock.update).not.toHaveBeenCalled();
 | 
			
		||||
      expect(accessMock.person.hasOwnerAccess).toHaveBeenCalledWith(authStub.admin.id, 'person-1');
 | 
			
		||||
    });
 | 
			
		||||
  });
 | 
			
		||||
 | 
			
		||||
@@ -255,11 +319,31 @@ describe(PersonService.name, () => {
 | 
			
		||||
  });
 | 
			
		||||
 | 
			
		||||
  describe('mergePerson', () => {
 | 
			
		||||
    it('should require person.write and person.merge permission', async () => {
 | 
			
		||||
      personMock.getById.mockResolvedValueOnce(personStub.primaryPerson);
 | 
			
		||||
      personMock.getById.mockResolvedValueOnce(personStub.mergePerson);
 | 
			
		||||
      personMock.prepareReassignFaces.mockResolvedValue([]);
 | 
			
		||||
      personMock.delete.mockResolvedValue(personStub.mergePerson);
 | 
			
		||||
      accessMock.person.hasOwnerAccess.mockResolvedValue(false);
 | 
			
		||||
 | 
			
		||||
      await expect(sut.mergePerson(authStub.admin, 'person-1', { ids: ['person-2'] })).rejects.toBeInstanceOf(
 | 
			
		||||
        BadRequestException,
 | 
			
		||||
      );
 | 
			
		||||
 | 
			
		||||
      expect(personMock.prepareReassignFaces).not.toHaveBeenCalled();
 | 
			
		||||
 | 
			
		||||
      expect(personMock.reassignFaces).not.toHaveBeenCalled();
 | 
			
		||||
 | 
			
		||||
      expect(personMock.delete).not.toHaveBeenCalled();
 | 
			
		||||
      expect(accessMock.person.hasOwnerAccess).toHaveBeenCalledWith(authStub.admin.id, 'person-1');
 | 
			
		||||
    });
 | 
			
		||||
 | 
			
		||||
    it('should merge two people', async () => {
 | 
			
		||||
      personMock.getById.mockResolvedValueOnce(personStub.primaryPerson);
 | 
			
		||||
      personMock.getById.mockResolvedValueOnce(personStub.mergePerson);
 | 
			
		||||
      personMock.prepareReassignFaces.mockResolvedValue([]);
 | 
			
		||||
      personMock.delete.mockResolvedValue(personStub.mergePerson);
 | 
			
		||||
      accessMock.person.hasOwnerAccess.mockResolvedValue(true);
 | 
			
		||||
 | 
			
		||||
      await expect(sut.mergePerson(authStub.admin, 'person-1', { ids: ['person-2'] })).resolves.toEqual([
 | 
			
		||||
        { id: 'person-2', success: true },
 | 
			
		||||
@@ -276,12 +360,14 @@ describe(PersonService.name, () => {
 | 
			
		||||
      });
 | 
			
		||||
 | 
			
		||||
      expect(personMock.delete).toHaveBeenCalledWith(personStub.mergePerson);
 | 
			
		||||
      expect(accessMock.person.hasOwnerAccess).toHaveBeenCalledWith(authStub.admin.id, 'person-1');
 | 
			
		||||
    });
 | 
			
		||||
 | 
			
		||||
    it('should delete conflicting faces before merging', async () => {
 | 
			
		||||
      personMock.getById.mockResolvedValue(personStub.primaryPerson);
 | 
			
		||||
      personMock.getById.mockResolvedValue(personStub.mergePerson);
 | 
			
		||||
      personMock.prepareReassignFaces.mockResolvedValue([assetStub.image.id]);
 | 
			
		||||
      accessMock.person.hasOwnerAccess.mockResolvedValue(true);
 | 
			
		||||
 | 
			
		||||
      await expect(sut.mergePerson(authStub.admin, 'person-1', { ids: ['person-2'] })).resolves.toEqual([
 | 
			
		||||
        { id: 'person-2', success: true },
 | 
			
		||||
@@ -296,21 +382,25 @@ describe(PersonService.name, () => {
 | 
			
		||||
        name: JobName.SEARCH_REMOVE_FACE,
 | 
			
		||||
        data: { assetId: assetStub.image.id, personId: personStub.mergePerson.id },
 | 
			
		||||
      });
 | 
			
		||||
      expect(accessMock.person.hasOwnerAccess).toHaveBeenCalledWith(authStub.admin.id, 'person-1');
 | 
			
		||||
    });
 | 
			
		||||
 | 
			
		||||
    it('should throw an error when the primary person is not found', async () => {
 | 
			
		||||
      personMock.getById.mockResolvedValue(null);
 | 
			
		||||
      accessMock.person.hasOwnerAccess.mockResolvedValue(true);
 | 
			
		||||
 | 
			
		||||
      await expect(sut.mergePerson(authStub.admin, 'person-1', { ids: ['person-2'] })).rejects.toBeInstanceOf(
 | 
			
		||||
        BadRequestException,
 | 
			
		||||
      );
 | 
			
		||||
 | 
			
		||||
      expect(personMock.delete).not.toHaveBeenCalled();
 | 
			
		||||
      expect(accessMock.person.hasOwnerAccess).toHaveBeenCalledWith(authStub.admin.id, 'person-1');
 | 
			
		||||
    });
 | 
			
		||||
 | 
			
		||||
    it('should handle invalid merge ids', async () => {
 | 
			
		||||
      personMock.getById.mockResolvedValueOnce(personStub.primaryPerson);
 | 
			
		||||
      personMock.getById.mockResolvedValueOnce(null);
 | 
			
		||||
      accessMock.person.hasOwnerAccess.mockResolvedValue(true);
 | 
			
		||||
 | 
			
		||||
      await expect(sut.mergePerson(authStub.admin, 'person-1', { ids: ['person-2'] })).resolves.toEqual([
 | 
			
		||||
        { id: 'person-2', success: false, error: BulkIdErrorReason.NOT_FOUND },
 | 
			
		||||
@@ -319,6 +409,7 @@ describe(PersonService.name, () => {
 | 
			
		||||
      expect(personMock.prepareReassignFaces).not.toHaveBeenCalled();
 | 
			
		||||
      expect(personMock.reassignFaces).not.toHaveBeenCalled();
 | 
			
		||||
      expect(personMock.delete).not.toHaveBeenCalled();
 | 
			
		||||
      expect(accessMock.person.hasOwnerAccess).toHaveBeenCalledWith(authStub.admin.id, 'person-1');
 | 
			
		||||
    });
 | 
			
		||||
 | 
			
		||||
    it('should handle an error reassigning faces', async () => {
 | 
			
		||||
@@ -326,12 +417,14 @@ describe(PersonService.name, () => {
 | 
			
		||||
      personMock.getById.mockResolvedValue(personStub.mergePerson);
 | 
			
		||||
      personMock.prepareReassignFaces.mockResolvedValue([assetStub.image.id]);
 | 
			
		||||
      personMock.reassignFaces.mockRejectedValue(new Error('update failed'));
 | 
			
		||||
      accessMock.person.hasOwnerAccess.mockResolvedValue(true);
 | 
			
		||||
 | 
			
		||||
      await expect(sut.mergePerson(authStub.admin, 'person-1', { ids: ['person-2'] })).resolves.toEqual([
 | 
			
		||||
        { id: 'person-2', success: false, error: BulkIdErrorReason.UNKNOWN },
 | 
			
		||||
      ]);
 | 
			
		||||
 | 
			
		||||
      expect(personMock.delete).not.toHaveBeenCalled();
 | 
			
		||||
      expect(accessMock.person.hasOwnerAccess).toHaveBeenCalledWith(authStub.admin.id, 'person-1');
 | 
			
		||||
    });
 | 
			
		||||
  });
 | 
			
		||||
});
 | 
			
		||||
 
 | 
			
		||||
@@ -1,4 +1,5 @@
 | 
			
		||||
import { BadRequestException, Inject, Injectable, Logger, NotFoundException } from '@nestjs/common';
 | 
			
		||||
import { AccessCore, IAccessRepository, Permission } from '../access';
 | 
			
		||||
import { AssetResponseDto, BulkIdErrorReason, BulkIdResponseDto, mapAsset } from '../asset';
 | 
			
		||||
import { AuthUserDto } from '../auth';
 | 
			
		||||
import { mimeTypes } from '../domain.constant';
 | 
			
		||||
@@ -18,15 +19,18 @@ import { IPersonRepository, UpdateFacesData } from './person.repository';
 | 
			
		||||
 | 
			
		||||
@Injectable()
 | 
			
		||||
export class PersonService {
 | 
			
		||||
  private access: AccessCore;
 | 
			
		||||
  private configCore: SystemConfigCore;
 | 
			
		||||
  readonly logger = new Logger(PersonService.name);
 | 
			
		||||
 | 
			
		||||
  constructor(
 | 
			
		||||
    @Inject(IAccessRepository) private accessRepository: IAccessRepository,
 | 
			
		||||
    @Inject(IPersonRepository) private repository: IPersonRepository,
 | 
			
		||||
    @Inject(ISystemConfigRepository) configRepository: ISystemConfigRepository,
 | 
			
		||||
    @Inject(IStorageRepository) private storageRepository: IStorageRepository,
 | 
			
		||||
    @Inject(IJobRepository) private jobRepository: IJobRepository,
 | 
			
		||||
  ) {
 | 
			
		||||
    this.access = new AccessCore(accessRepository);
 | 
			
		||||
    this.configCore = new SystemConfigCore(configRepository);
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
@@ -48,12 +52,14 @@ export class PersonService {
 | 
			
		||||
    };
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  getById(authUser: AuthUserDto, id: string): Promise<PersonResponseDto> {
 | 
			
		||||
    return this.findOrFail(authUser, id).then(mapPerson);
 | 
			
		||||
  async getById(authUser: AuthUserDto, id: string): Promise<PersonResponseDto> {
 | 
			
		||||
    await this.access.requirePermission(authUser, Permission.PERSON_READ, id);
 | 
			
		||||
    return this.findOrFail(id).then(mapPerson);
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  async getThumbnail(authUser: AuthUserDto, id: string): Promise<ImmichReadStream> {
 | 
			
		||||
    const person = await this.repository.getById(authUser.id, id);
 | 
			
		||||
    await this.access.requirePermission(authUser, Permission.PERSON_READ, id);
 | 
			
		||||
    const person = await this.repository.getById(id);
 | 
			
		||||
    if (!person || !person.thumbnailPath) {
 | 
			
		||||
      throw new NotFoundException();
 | 
			
		||||
    }
 | 
			
		||||
@@ -62,17 +68,19 @@ export class PersonService {
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  async getAssets(authUser: AuthUserDto, id: string): Promise<AssetResponseDto[]> {
 | 
			
		||||
    const assets = await this.repository.getAssets(authUser.id, id);
 | 
			
		||||
    await this.access.requirePermission(authUser, Permission.PERSON_READ, id);
 | 
			
		||||
    const assets = await this.repository.getAssets(id);
 | 
			
		||||
    return assets.map(mapAsset);
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  async update(authUser: AuthUserDto, id: string, dto: PersonUpdateDto): Promise<PersonResponseDto> {
 | 
			
		||||
    let person = await this.findOrFail(authUser, id);
 | 
			
		||||
    await this.access.requirePermission(authUser, Permission.PERSON_WRITE, id);
 | 
			
		||||
    let person = await this.findOrFail(id);
 | 
			
		||||
 | 
			
		||||
    if (dto.name !== undefined || dto.birthDate !== undefined || dto.isHidden !== undefined) {
 | 
			
		||||
      person = await this.repository.update({ id, name: dto.name, birthDate: dto.birthDate, isHidden: dto.isHidden });
 | 
			
		||||
      if (this.needsSearchIndexUpdate(dto)) {
 | 
			
		||||
        const assets = await this.repository.getAssets(authUser.id, id);
 | 
			
		||||
        const assets = await this.repository.getAssets(id);
 | 
			
		||||
        const ids = assets.map((asset) => asset.id);
 | 
			
		||||
        await this.jobRepository.queue({ name: JobName.SEARCH_INDEX_ASSET, data: { ids } });
 | 
			
		||||
      }
 | 
			
		||||
@@ -141,14 +149,22 @@ export class PersonService {
 | 
			
		||||
 | 
			
		||||
  async mergePerson(authUser: AuthUserDto, id: string, dto: MergePersonDto): Promise<BulkIdResponseDto[]> {
 | 
			
		||||
    const mergeIds = dto.ids;
 | 
			
		||||
    const primaryPerson = await this.findOrFail(authUser, id);
 | 
			
		||||
    await this.access.requirePermission(authUser, Permission.PERSON_WRITE, id);
 | 
			
		||||
    const primaryPerson = await this.findOrFail(id);
 | 
			
		||||
    const primaryName = primaryPerson.name || primaryPerson.id;
 | 
			
		||||
 | 
			
		||||
    const results: BulkIdResponseDto[] = [];
 | 
			
		||||
 | 
			
		||||
    for (const mergeId of mergeIds) {
 | 
			
		||||
      const hasPermission = await this.access.hasPermission(authUser, Permission.PERSON_MERGE, mergeId);
 | 
			
		||||
 | 
			
		||||
      if (!hasPermission) {
 | 
			
		||||
        results.push({ id: mergeId, success: false, error: BulkIdErrorReason.NO_PERMISSION });
 | 
			
		||||
        continue;
 | 
			
		||||
      }
 | 
			
		||||
 | 
			
		||||
      try {
 | 
			
		||||
        const mergePerson = await this.repository.getById(authUser.id, mergeId);
 | 
			
		||||
        const mergePerson = await this.repository.getById(mergeId);
 | 
			
		||||
        if (!mergePerson) {
 | 
			
		||||
          results.push({ id: mergeId, success: false, error: BulkIdErrorReason.NOT_FOUND });
 | 
			
		||||
          continue;
 | 
			
		||||
@@ -188,8 +204,8 @@ export class PersonService {
 | 
			
		||||
    return dto.name !== undefined || dto.isHidden !== undefined;
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  private async findOrFail(authUser: AuthUserDto, id: string) {
 | 
			
		||||
    const person = await this.repository.getById(authUser.id, id);
 | 
			
		||||
  private async findOrFail(id: string) {
 | 
			
		||||
    const person = await this.repository.getById(id);
 | 
			
		||||
    if (!person) {
 | 
			
		||||
      throw new BadRequestException('Person not found');
 | 
			
		||||
    }
 | 
			
		||||
 
 | 
			
		||||
@@ -1,13 +1,14 @@
 | 
			
		||||
import { IAccessRepository } from '@app/domain';
 | 
			
		||||
import { InjectRepository } from '@nestjs/typeorm';
 | 
			
		||||
import { Repository } from 'typeorm';
 | 
			
		||||
import { AlbumEntity, AssetEntity, PartnerEntity, SharedLinkEntity } from '../entities';
 | 
			
		||||
import { AlbumEntity, AssetEntity, PartnerEntity, PersonEntity, SharedLinkEntity } from '../entities';
 | 
			
		||||
 | 
			
		||||
export class AccessRepository implements IAccessRepository {
 | 
			
		||||
  constructor(
 | 
			
		||||
    @InjectRepository(AssetEntity) private assetRepository: Repository<AssetEntity>,
 | 
			
		||||
    @InjectRepository(AlbumEntity) private albumRepository: Repository<AlbumEntity>,
 | 
			
		||||
    @InjectRepository(PartnerEntity) private partnerRepository: Repository<PartnerEntity>,
 | 
			
		||||
    @InjectRepository(PersonEntity) private personRepository: Repository<PersonEntity>,
 | 
			
		||||
    @InjectRepository(SharedLinkEntity) private sharedLinkRepository: Repository<SharedLinkEntity>,
 | 
			
		||||
  ) {}
 | 
			
		||||
 | 
			
		||||
@@ -156,4 +157,15 @@ export class AccessRepository implements IAccessRepository {
 | 
			
		||||
      });
 | 
			
		||||
    },
 | 
			
		||||
  };
 | 
			
		||||
 | 
			
		||||
  person = {
 | 
			
		||||
    hasOwnerAccess: (userId: string, personId: string): Promise<boolean> => {
 | 
			
		||||
      return this.personRepository.exist({
 | 
			
		||||
        where: {
 | 
			
		||||
          id: personId,
 | 
			
		||||
          ownerId: userId,
 | 
			
		||||
        },
 | 
			
		||||
      });
 | 
			
		||||
    },
 | 
			
		||||
  };
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -86,14 +86,13 @@ export class PersonRepository implements IPersonRepository {
 | 
			
		||||
      .getMany();
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  getById(ownerId: string, personId: string): Promise<PersonEntity | null> {
 | 
			
		||||
    return this.personRepository.findOne({ where: { id: personId, ownerId } });
 | 
			
		||||
  getById(personId: string): Promise<PersonEntity | null> {
 | 
			
		||||
    return this.personRepository.findOne({ where: { id: personId } });
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  getAssets(ownerId: string, personId: string): Promise<AssetEntity[]> {
 | 
			
		||||
  getAssets(personId: string): Promise<AssetEntity[]> {
 | 
			
		||||
    return this.assetRepository.find({
 | 
			
		||||
      where: {
 | 
			
		||||
        ownerId,
 | 
			
		||||
        faces: {
 | 
			
		||||
          personId,
 | 
			
		||||
        },
 | 
			
		||||
 
 | 
			
		||||
@@ -4,6 +4,7 @@ export interface IAccessRepositoryMock {
 | 
			
		||||
  asset: jest.Mocked<IAccessRepository['asset']>;
 | 
			
		||||
  album: jest.Mocked<IAccessRepository['album']>;
 | 
			
		||||
  library: jest.Mocked<IAccessRepository['library']>;
 | 
			
		||||
  person: jest.Mocked<IAccessRepository['person']>;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
export const newAccessRepositoryMock = (): IAccessRepositoryMock => {
 | 
			
		||||
@@ -24,5 +25,9 @@ export const newAccessRepositoryMock = (): IAccessRepositoryMock => {
 | 
			
		||||
    library: {
 | 
			
		||||
      hasPartnerAccess: jest.fn(),
 | 
			
		||||
    },
 | 
			
		||||
 | 
			
		||||
    person: {
 | 
			
		||||
      hasOwnerAccess: jest.fn(),
 | 
			
		||||
    },
 | 
			
		||||
  };
 | 
			
		||||
};
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user