mirror of
https://github.com/xuthus83/pigallery2.git
synced 2024-11-03 21:04:03 +08:00
436 lines
12 KiB
TypeScript
436 lines
12 KiB
TypeScript
import {
|
|
CameraMetadataEntity,
|
|
GPSMetadataEntity,
|
|
MediaDimensionEntity,
|
|
PositionMetaDataEntity
|
|
} from '../src/backend/model/database/enitites/MediaEntity';
|
|
import {PhotoEntity, PhotoMetadataEntity} from '../src/backend/model/database/enitites/PhotoEntity';
|
|
import {DirectoryEntity} from '../src/backend/model/database/enitites/DirectoryEntity';
|
|
import {VideoEntity, VideoMetadataEntity} from '../src/backend/model/database/enitites/VideoEntity';
|
|
import {MediaDimension, MediaDTO} from '../src/common/entities/MediaDTO';
|
|
import {
|
|
CameraMetadata,
|
|
CoverPhotoDTO,
|
|
FaceRegion,
|
|
GPSMetadata,
|
|
PhotoDTO,
|
|
PhotoMetadata,
|
|
PositionMetaData
|
|
} from '../src/common/entities/PhotoDTO';
|
|
import {DirectoryBaseDTO, DirectoryPathDTO} from '../src/common/entities/DirectoryDTO';
|
|
import {FileDTO} from '../src/common/entities/FileDTO';
|
|
import {DiskManager} from '../src/backend/model/fileaccess/DiskManager';
|
|
|
|
export class TestHelper {
|
|
|
|
static creationCounter = 0;
|
|
|
|
public static getDirectoryEntry(parent: DirectoryBaseDTO = null, name = 'wars dir'): DirectoryEntity {
|
|
|
|
const dir = new DirectoryEntity();
|
|
dir.name = name;
|
|
dir.path = DiskManager.pathFromParent({path: '', name: '.'});
|
|
dir.mediaCount = 0;
|
|
dir.youngestMedia = 10;
|
|
dir.oldestMedia = 1000;
|
|
dir.directories = [];
|
|
dir.metaFile = [];
|
|
dir.media = [];
|
|
dir.lastModified = 1656069687773;
|
|
dir.lastScanned = 1656069687773;
|
|
// dir.parent = null;
|
|
if (parent !== null) {
|
|
dir.path = DiskManager.pathFromParent(parent);
|
|
parent.directories.push(dir);
|
|
}
|
|
return dir;
|
|
}
|
|
|
|
|
|
public static getBasePhotoEntry(dir: DirectoryPathDTO, name = 'base media.jpg'): PhotoEntity {
|
|
const sd = new MediaDimensionEntity();
|
|
sd.height = 400;
|
|
sd.width = 200;
|
|
const m = new PhotoMetadataEntity();
|
|
m.caption = null;
|
|
m.size = sd;
|
|
m.creationDate = 1656069387772;
|
|
m.fileSize = 123456789;
|
|
// m.rating = 0; no rating by default
|
|
|
|
// TODO: remove when typeorm is fixed
|
|
m.duration = null;
|
|
m.bitRate = null;
|
|
|
|
|
|
const d = new PhotoEntity();
|
|
d.name = name;
|
|
d.directory = (dir as any);
|
|
if ((dir as DirectoryBaseDTO).media) {
|
|
(dir as DirectoryBaseDTO).media.push(d);
|
|
(dir as DirectoryBaseDTO).mediaCount++;
|
|
}
|
|
d.metadata = m;
|
|
return d;
|
|
}
|
|
|
|
public static getPhotoEntry(dir: DirectoryPathDTO): PhotoEntity {
|
|
const sd = new MediaDimensionEntity();
|
|
sd.height = 400;
|
|
sd.width = 200;
|
|
const gps = new GPSMetadataEntity();
|
|
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 = 1656069387772;
|
|
m.fileSize = 123456789;
|
|
// m.rating = 0; no rating by default
|
|
|
|
// TODO: remove when typeorm is fixed
|
|
m.duration = null;
|
|
m.bitRate = null;
|
|
|
|
|
|
const d = new PhotoEntity();
|
|
d.name = 'test media.jpg';
|
|
d.directory = (dir as any);
|
|
if ((dir as DirectoryBaseDTO).media) {
|
|
(dir as DirectoryBaseDTO).media.push(d);
|
|
(dir as DirectoryBaseDTO).mediaCount++;
|
|
}
|
|
d.metadata = m;
|
|
return d;
|
|
}
|
|
|
|
public static getVideoEntry(dir: DirectoryPathDTO): VideoEntity {
|
|
const sd = new MediaDimensionEntity();
|
|
sd.height = 200;
|
|
sd.width = 300;
|
|
|
|
const m = new VideoMetadataEntity();
|
|
m.caption = null;
|
|
m.keywords = null;
|
|
m.rating = null;
|
|
m.size = sd;
|
|
m.creationDate = 1656069387771;
|
|
m.fileSize = 123456789;
|
|
|
|
m.duration = 10000;
|
|
m.bitRate = 4000;
|
|
|
|
|
|
const d = new VideoEntity();
|
|
d.name = 'test video.mp4';
|
|
d.directory = (dir as any);
|
|
if ((dir as DirectoryBaseDTO).media) {
|
|
(dir as DirectoryBaseDTO).media.push(d);
|
|
(dir as DirectoryBaseDTO).mediaCount++;
|
|
}
|
|
d.metadata = m;
|
|
return d;
|
|
}
|
|
|
|
public static getGPXEntry(dir: DirectoryPathDTO): FileDTO {
|
|
const d: FileDTO = {
|
|
id: null,
|
|
name: 'saturdayRun.gpx',
|
|
directory: dir
|
|
};
|
|
if ((dir as DirectoryBaseDTO).metaFile) {
|
|
(dir as DirectoryBaseDTO).metaFile.push(d);
|
|
}
|
|
return d;
|
|
}
|
|
|
|
public static getVideoEntry1(dir: DirectoryBaseDTO): VideoEntity {
|
|
const p = TestHelper.getVideoEntry(dir);
|
|
p.name = 'swVideo.mp4';
|
|
return p;
|
|
}
|
|
|
|
public static getPhotoEntry1(dir: DirectoryPathDTO): PhotoEntity {
|
|
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.jpg';
|
|
p.metadata.positionData.GPSData.latitude = 10;
|
|
p.metadata.positionData.GPSData.longitude = 10;
|
|
p.metadata.creationDate = 1656069387772 - 1000;
|
|
p.metadata.rating = 1;
|
|
p.metadata.size.height = 1000;
|
|
p.metadata.size.width = 1000;
|
|
|
|
p.metadata.faces = [{
|
|
box: {height: 10, width: 10, left: 10, top: 10},
|
|
name: 'Boba Fett'
|
|
} as FaceRegion, {
|
|
box: {height: 10, width: 10, left: 102, top: 102},
|
|
name: 'Luke Skywalker'
|
|
} as FaceRegion, {
|
|
box: {height: 10, width: 10, left: 103, top: 103},
|
|
name: 'Han Solo'
|
|
} as FaceRegion, {
|
|
box: {height: 10, width: 10, left: 104, top: 104},
|
|
name: 'Unkle Ben'
|
|
} as FaceRegion, {
|
|
box: {height: 10, width: 10, left: 105, top: 105},
|
|
name: 'Arvíztűrő Tükörfúrógép'
|
|
} as FaceRegion, {
|
|
box: {height: 10, width: 10, left: 201, top: 201},
|
|
name: 'R2-D2'
|
|
} as FaceRegion] as any[];
|
|
return p;
|
|
}
|
|
|
|
public static getPhotoEntry2(dir: DirectoryPathDTO): PhotoEntity {
|
|
const p = TestHelper.getPhotoEntry(dir);
|
|
|
|
p.metadata.caption = 'Light saber';
|
|
p.metadata.keywords = ['Padmé Amidala', 'star wars', 'Natalie Portman', 'death star', 'wookiee'];
|
|
p.metadata.positionData.city = 'Derem City';
|
|
p.metadata.positionData.state = 'Research City';
|
|
p.metadata.positionData.country = 'Kamino';
|
|
p.name = 'sw2.jpg';
|
|
p.metadata.positionData.GPSData.latitude = -10;
|
|
p.metadata.positionData.GPSData.longitude = -10;
|
|
p.metadata.creationDate = 1656069387772 - 2000;
|
|
p.metadata.rating = 2;
|
|
p.metadata.size.height = 2000;
|
|
p.metadata.size.width = 1000;
|
|
|
|
p.metadata.faces = [{
|
|
box: {height: 10, width: 10, left: 10, top: 10},
|
|
name: 'Padmé Amidala'
|
|
} as FaceRegion, {
|
|
box: {height: 10, width: 10, left: 101, top: 101},
|
|
name: 'Anakin Skywalker'
|
|
} as FaceRegion, {
|
|
box: {height: 10, width: 10, left: 101, top: 101},
|
|
name: 'Obivan Kenobi'
|
|
} as FaceRegion, {
|
|
box: {height: 10, width: 10, left: 201, top: 201},
|
|
name: 'R2-D2'
|
|
} as FaceRegion] as any[];
|
|
return p;
|
|
}
|
|
|
|
public static getPhotoEntry3(dir: DirectoryPathDTO): PhotoEntity {
|
|
const p = TestHelper.getPhotoEntry(dir);
|
|
|
|
p.metadata.caption = 'Amber stone';
|
|
p.metadata.keywords = ['star wars', 'wookiees'];
|
|
p.metadata.positionData.city = 'Castilon';
|
|
p.metadata.positionData.state = 'Devaron';
|
|
p.metadata.positionData.country = 'Ajan Kloss';
|
|
p.name = 'sw3.jpg';
|
|
p.metadata.positionData.GPSData.latitude = 10;
|
|
p.metadata.positionData.GPSData.longitude = 15;
|
|
p.metadata.creationDate = 1656069387772 - 3000;
|
|
p.metadata.rating = 3;
|
|
p.metadata.size.height = 1000;
|
|
p.metadata.size.width = 2000;
|
|
p.metadata.faces = [{
|
|
box: {height: 10, width: 10, left: 10, top: 10},
|
|
name: 'Kylo Ren'
|
|
} as FaceRegion, {
|
|
box: {height: 10, width: 10, left: 101, top: 101},
|
|
name: 'Leia Organa'
|
|
} as FaceRegion, {
|
|
box: {height: 10, width: 10, left: 103, top: 103},
|
|
name: 'Han Solo'
|
|
} as FaceRegion] as any[];
|
|
return p;
|
|
}
|
|
|
|
public static getPhotoEntry4(dir: DirectoryPathDTO): PhotoEntity {
|
|
const p = TestHelper.getPhotoEntry(dir);
|
|
|
|
p.metadata.caption = 'Millennium falcon';
|
|
p.metadata.keywords = ['star wars', 'ewoks'];
|
|
p.metadata.positionData.city = 'Tipoca City';
|
|
p.metadata.positionData.state = 'Exegol';
|
|
p.metadata.positionData.country = 'Jedha';
|
|
p.name = 'sw4.jpg';
|
|
p.metadata.positionData.GPSData.latitude = 15;
|
|
p.metadata.positionData.GPSData.longitude = 10;
|
|
p.metadata.creationDate = 1656069387772 - 4000;
|
|
p.metadata.size.height = 3000;
|
|
p.metadata.size.width = 2000;
|
|
|
|
p.metadata.faces = [{
|
|
box: {height: 10, width: 10, left: 10, top: 10},
|
|
name: 'Kylo Ren'
|
|
} as FaceRegion, {
|
|
box: {height: 10, width: 10, left: 101, top: 101},
|
|
name: 'Anakin Skywalker'
|
|
} as FaceRegion, {
|
|
box: {height: 10, width: 10, left: 101, top: 101},
|
|
name: 'Obivan Kenobi'
|
|
} as FaceRegion, {
|
|
box: {height: 10, width: 10, left: 201, top: 201},
|
|
name: 'R2-D2'
|
|
} as FaceRegion] as any[];
|
|
|
|
return p;
|
|
}
|
|
|
|
public static getRandomizedDirectoryEntry(parent: DirectoryBaseDTO = null, forceStr: string = null): DirectoryBaseDTO<MediaDTO> {
|
|
|
|
const dir: DirectoryBaseDTO = {
|
|
id: null,
|
|
name: DiskManager.dirName(forceStr || Math.random().toString(36).substring(7)),
|
|
path: DiskManager.pathFromParent({path: '', name: '.'}),
|
|
mediaCount: 0,
|
|
youngestMedia: 10,
|
|
oldestMedia: 1000,
|
|
directories: [],
|
|
metaFile: [],
|
|
cover: null,
|
|
validCover: false,
|
|
media: [],
|
|
lastModified: Date.now(),
|
|
lastScanned: null,
|
|
parent
|
|
};
|
|
if (parent !== null) {
|
|
dir.path = DiskManager.pathFromParent(parent);
|
|
parent.directories.push(dir);
|
|
}
|
|
return dir;
|
|
}
|
|
|
|
public static getRandomizedGPXEntry(dir: DirectoryBaseDTO, 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): FaceRegion {
|
|
const rndStr = (): string => {
|
|
return forceStr + '_' + Math.random().toString(36).substring(7);
|
|
};
|
|
|
|
const rndInt = (max = 5000): number => {
|
|
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: DirectoryBaseDTO, forceStr: string = null, faces = 2): PhotoDTO {
|
|
|
|
|
|
const rndStr = (): string => {
|
|
return forceStr + '_' + Math.random().toString(36).substring(7);
|
|
};
|
|
|
|
const rndInt = (max = 5000): number => {
|
|
return Math.floor(Math.random() * max);
|
|
};
|
|
|
|
const sd: MediaDimension = {
|
|
height: rndInt(),
|
|
width: rndInt(),
|
|
};
|
|
|
|
const gps: GPSMetadata = {
|
|
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() + ++TestHelper.creationCounter,
|
|
fileSize: rndInt(10000),
|
|
caption: rndStr(),
|
|
rating: rndInt(5) as any,
|
|
};
|
|
|
|
|
|
const p: PhotoDTO = {
|
|
id: null,
|
|
name: rndStr() + '.jpg',
|
|
directory: dir,
|
|
metadata: m
|
|
};
|
|
|
|
for (let i = 0; i < faces; i++) {
|
|
this.getRandomizedFace(p, 'Person ' + i);
|
|
}
|
|
|
|
dir.media.push(p);
|
|
TestHelper.updateCover(dir);
|
|
return p;
|
|
}
|
|
|
|
static updateCover(dir: DirectoryBaseDTO): void {
|
|
if (dir.media.length > 0) {
|
|
dir.cover = dir.media.sort((a, b): number => b.metadata.creationDate - a.metadata.creationDate)[0];
|
|
} else {
|
|
const filtered = dir.directories.filter((d): CoverPhotoDTO => d.cover).map((d): CoverPhotoDTO => d.cover);
|
|
if (filtered.length > 0) {
|
|
dir.cover = filtered.sort((a, b): number => b.metadata.creationDate - a.metadata.creationDate)[0];
|
|
}
|
|
}
|
|
if (dir.parent) {
|
|
TestHelper.updateCover(dir.parent);
|
|
}
|
|
|
|
}
|
|
|
|
|
|
}
|