pigallery2/test/backend/unit/model/sql/TestHelper.ts

import {MediaDimensionEntity} from '../../../../../src/backend/model/database/sql/enitites/MediaEntity';
import {
  CameraMetadataEntity,
  GPSMetadataEntity,
  PhotoEntity,
  PhotoMetadataEntity,
  PositionMetaDataEntity
} from '../../../../../src/backend/model/database/sql/enitites/PhotoEntity';
import {OrientationTypes} from 'ts-exif-parser';
import {DirectoryEntity} from '../../../../../src/backend/model/database/sql/enitites/DirectoryEntity';
import {VideoEntity, VideoMetadataEntity} from '../../../../../src/backend/model/database/sql/enitites/VideoEntity';
import {MediaDimension} from '../../../../../src/common/entities/MediaDTO';
import {
  CameraMetadata,
  FaceRegion,
  GPSMetadata,
  PhotoDTO,
  PhotoMetadata,
  PositionMetaData
} from '../../../../../src/common/entities/PhotoDTO';
import {DirectoryDTO} from '../../../../../src/common/entities/DirectoryDTO';
import {FileDTO} from '../../../../../src/common/entities/FileDTO';
import {DiskMangerWorker} from '../../../../../src/backend/model/threading/DiskMangerWorker';

export class TestHelper {

  public static getDirectoryEntry() {

    const dir = new DirectoryEntity();
    dir.name = 'wars dir';
    dir.path = '.';
    dir.mediaCount = 0;
    dir.lastModified = Date.now();
    dir.lastScanned = null;

    return dir;
  }

  public static getPhotoEntry(dir: DirectoryEntity) {
    const sd = new MediaDimensionEntity();
    sd.height = 200;
    sd.width = 200;
    const gps = new GPSMetadataEntity();
    gps.altitude = 1;
    gps.latitude = 1;
    gps.longitude = 1;
    const pd = new PositionMetaDataEntity();
    pd.city = 'New York';
    pd.country = 'Alderan';
    pd.state = 'Kamino';
    pd.GPSData = gps;
    const cd = new CameraMetadataEntity();
    cd.ISO = 100;
    cd.model = '60D';
    cd.make = 'Canon';
    cd.fStop = 1;
    cd.exposure = 1;
    cd.focalLength = 1;
    cd.lens = 'Lens';
    const m = new PhotoMetadataEntity();
    m.caption = null;
    m.keywords = ['apple'];
    m.cameraData = cd;
    m.positionData = pd;
    m.size = sd;
    m.creationDate = Date.now();
    m.fileSize = 123456789;
    m.orientation = OrientationTypes.TOP_LEFT;
    m.rating = 2;

    // TODO: remove when typeorm is fixed
    m.duration = null;
    m.bitRate = null;


    const d = new PhotoEntity();
    d.name = 'test media.jpg';
    d.directory = dir;
    d.metadata = m;
    dir.mediaCount++;
    return d;
  }

  public static getVideoEntry(dir: DirectoryEntity) {
    const sd = new MediaDimensionEntity();
    sd.height = 200;
    sd.width = 200;

    const m = new VideoMetadataEntity();
    m.caption = null;
    m.keywords = null;
    m.rating = null;
    m.size = sd;
    m.creationDate = Date.now();
    m.fileSize = 123456789;

    m.duration = 10000;
    m.bitRate = 4000;


    const d = new VideoEntity();
    d.name = 'test video.jpg';
    d.directory = dir;
    d.metadata = m;
    return d;
  }

  public static getPhotoEntry1(dir: DirectoryEntity) {
    const p = TestHelper.getPhotoEntry(dir);

    p.metadata.caption = 'Han Solo\'s dice';
    p.metadata.keywords = ['Boba Fett', 'star wars', 'Anakin', 'death star'];
    p.metadata.positionData.city = 'Mos Eisley';
    p.metadata.positionData.country = 'Tatooine';
    p.name = 'sw1';

    p.metadata.faces = [<FaceRegion>{
      box: {height: 10, width: 10, left: 10, top: 10},
      name: 'Boba Fett'
    }, <FaceRegion>{
      box: {height: 10, width: 10, left: 102, top: 102},
      name: 'Luke Skywalker'
    }, <FaceRegion>{
      box: {height: 10, width: 10, left: 103, top: 103},
      name: 'Han Solo'
    }, <FaceRegion>{
      box: {height: 10, width: 10, left: 104, top: 104},
      name: 'Unkle Ben'
    }, <FaceRegion>{
      box: {height: 10, width: 10, left: 105, top: 105},
      name: 'Arvíztűrő Tükörfúrógép'
    }] as any[];
    return p;
  }

  public static getVideoEntry1(dir: DirectoryEntity) {
    const p = TestHelper.getVideoEntry(dir);
    p.name = 'swVideo';
    return p;
  }

  public static getPhotoEntry2(dir: DirectoryEntity) {
    const p = TestHelper.getPhotoEntry(dir);

    p.metadata.keywords = ['Padmé Amidala', 'star wars', 'Natalie Portman', 'death star'];
    p.metadata.positionData.city = 'Derem City';
    p.metadata.positionData.state = 'Research City';
    p.metadata.positionData.country = 'Kamino';
    p.name = 'sw2';
    p.metadata.faces = [<FaceRegion>{
      box: {height: 10, width: 10, left: 10, top: 10},
      name: 'Padmé Amidala'
    }, <FaceRegion>{
      box: {height: 10, width: 10, left: 101, top: 101},
      name: 'Anakin Skywalker'
    }, <FaceRegion>{
      box: {height: 10, width: 10, left: 101, top: 101},
      name: 'Obivan Kenobi'
    }] as any[];
    return p;
  }


  public static getRandomizedDirectoryEntry(parent: DirectoryDTO = null, forceStr: string = null) {

    const dir: DirectoryDTO = {
      id: null,
      name: DiskMangerWorker.dirName(forceStr || Math.random().toString(36).substring(7)),
      path: DiskMangerWorker.pathFromParent({path: '', name: '.'}),
      mediaCount: 0,
      directories: [],
      metaFile: [],
      media: [],
      lastModified: Date.now(),
      lastScanned: null,
      parent: null
    };
    if (parent !== null) {
      dir.path = DiskMangerWorker.pathFromParent(parent);
      parent.directories.push(dir);
    }
    return dir;
  }


  public static getRandomizedGPXEntry(dir: DirectoryDTO, forceStr: string = null): FileDTO {
    const d: FileDTO = {
      id: null,
      name: forceStr + '_' + Math.random().toString(36).substring(7) + '.gpx',
      directory: dir
    };
    dir.metaFile.push(d);
    return d;
  }


  public static getRandomizedFace(media: PhotoDTO, forceStr: string = null) {
    const rndStr = () => {
      return forceStr + '_' + Math.random().toString(36).substring(7);
    };

    const rndInt = (max = 5000) => {
      return Math.floor(Math.random() * max);
    };

    const f: FaceRegion = {
      name: rndStr() + '.jpg',
      box: {
        left: rndInt(),
        top: rndInt(),
        width: rndInt(),
        height: rndInt()
      }
    };
    media.metadata.faces = (media.metadata.faces || []);
    media.metadata.faces.push(f);
    return f;
  }

  public static getRandomizedPhotoEntry(dir: DirectoryDTO, forceStr: string = null, faces: number = 2): PhotoDTO {


    const rndStr = () => {
      return forceStr + '_' + Math.random().toString(36).substring(7);
    };

    const rndInt = (max = 5000) => {
      return Math.floor(Math.random() * max);
    };

    const sd: MediaDimension = {
      height: rndInt(),
      width: rndInt(),
    };

    const gps: GPSMetadata = {
      altitude: rndInt(1000),
      latitude: rndInt(1000),
      longitude: rndInt(1000)
    };
    const pd: PositionMetaData = {
      city: rndStr(),
      country: rndStr(),
      state: rndStr(),
      GPSData: gps
    };
    const cd: CameraMetadata = {
      ISO: rndInt(500),
      model: rndStr(),
      make: rndStr(),
      fStop: rndInt(10),
      exposure: rndInt(10),
      focalLength: rndInt(10),
      lens: rndStr()
    };
    const m: PhotoMetadata = {
      keywords: [rndStr(), rndStr()],
      cameraData: cd,
      positionData: pd,
      size: sd,
      creationDate: Date.now(),
      fileSize: rndInt(10000),
      orientation: OrientationTypes.TOP_LEFT,
      caption: rndStr(),
      rating: <any>rndInt(5),
    };


    const d: PhotoDTO = {
      id: null,
      name: rndStr() + '.jpg',
      directory: dir,
      metadata: m,
      readyThumbnails: [],
      readyIcon: false
    };

    for (let i = 0; i < faces; i++) {
      this.getRandomizedFace(d, 'Person ' + i);
    }

    dir.media.push(d);
    return d;
  }


}