pigallery2/src/backend/model/database/sql/PersonManager.ts

import {SQLConnection} from './SQLConnection';
import {PersonEntry} from './enitites/PersonEntry';
import {FaceRegionEntry} from './enitites/FaceRegionEntry';
import {PersonDTO} from '../../../../common/entities/PersonDTO';
import {ISQLPersonManager} from './IPersonManager';


export class PersonManager implements ISQLPersonManager {
  // samplePhotos: { [key: string]: PhotoDTO } = {};
  persons: PersonEntry[] = null;

  async updatePerson(name: string, partialPerson: PersonDTO): Promise<PersonEntry> {
    const connection = await SQLConnection.getConnection();
    const repository = connection.getRepository(PersonEntry);
    const person = await repository.createQueryBuilder('person')
      .limit(1)
      .where('person.name LIKE :name COLLATE utf8_general_ci', {name: name}).getOne();


    if (typeof partialPerson.name !== 'undefined') {
      person.name = partialPerson.name;
    }
    if (typeof partialPerson.isFavourite !== 'undefined') {
      person.isFavourite = partialPerson.isFavourite;
    }
    await repository.save(person);

    await this.loadAll();

    return person;
  }


  private async loadAll(): Promise<void> {
    const connection = await SQLConnection.getConnection();
    const personRepository = connection.getRepository(PersonEntry);
    this.persons = await personRepository.find({
      relations: ['sampleRegion',
        'sampleRegion.media',
        'sampleRegion.media.directory']
    });
  }

  public async getAll(): Promise<PersonEntry[]> {
    if (this.persons === null) {
      await this.loadAll();
    }
    return this.persons;
  }


  /**
   * Used for statistic
   */
  public async countFaces(): Promise<number> {
    const connection = await SQLConnection.getConnection();
    return await connection.getRepository(FaceRegionEntry)
      .createQueryBuilder('faceRegion')
      .getCount();
  }

  public async get(name: string): Promise<PersonEntry> {
    if (this.persons === null) {
      await this.loadAll();
    }
    return this.persons.find(p => p.name === name);
  }


  public async saveAll(names: string[]): Promise<void> {
    const toSave: { name: string }[] = [];
    const connection = await SQLConnection.getConnection();
    const personRepository = connection.getRepository(PersonEntry);
    await this.loadAll();

    for (let i = 0; i < names.length; i++) {

      const person = this.persons.find(p => p.name === names[i]);
      if (!person) {
        toSave.push({name: names[i]});
      }
    }

    if (toSave.length > 0) {
      for (let i = 0; i < toSave.length / 200; i++) {
        await personRepository.insert(toSave.slice(i * 200, (i + 1) * 200));
      }
      await this.loadAll();
    }

  }


  public async onGalleryIndexUpdate() {
    await this.updateCounts();
    await this.updateSamplePhotos();
  }


  private async updateCounts() {
    const connection = await SQLConnection.getConnection();
    await connection.query('update person_entry set count = ' +
      ' (select COUNT(1) from face_region_entry where face_region_entry.personId = person_entry.id)');

    // remove persons without photo
    await connection.getRepository(PersonEntry)
      .createQueryBuilder()
      .where('count = 0')
      .delete()
      .execute();
  }

  private async updateSamplePhotos() {
    const connection = await SQLConnection.getConnection();
    await connection.query('update person_entry set sampleRegionId = ' +
      '(Select face_region_entry.id from  media_entity ' +
      'left join face_region_entry on media_entity.id = face_region_entry.mediaId ' +
      'where face_region_entry.personId=person_entry.id ' +
      'order by media_entity.metadataCreationdate desc ' +
      'limit 1)');

  }

}