pigallery2/backend/model/mongoose/MongoGalleryManager.ts

import * as fs from 'fs';
import * as path from 'path';
import * as mime from 'mime';
import * as sizeOf  from 'image-size';
import {Schema} from "mongoose";

import {Directory} from "../../../common/entities/Directory";
import {Photo} from "../../../common/entities/Photo";
import {IGalleryManager} from "../IGalleryManager";
import {DatabaseManager} from "./DatabaseManager";
import {DiskManager} from "../DiskManger";
import {Utils} from "../../../common/Utils";

export class MongoGalleryManager implements IGalleryManager{
    private DirectoryModel;
    private PhotoModel;

    constructor(){
        this.DirectoryModel = DatabaseManager.getInstance().getModel('directory',{
            name:String,
            path:String,
            lastUpdate:Date,
            directories: [{
                type: Schema.Types.ObjectId,
                ref: 'directory'
            }],
            photos: [{
                type: Schema.Types.ObjectId,
                ref: 'photo'
            }]
        });
        this.PhotoModel = DatabaseManager.getInstance().getModel('photo',{
            name:String,
            width:Number,
            height:Number
        });
    }

    public listDirectory(relativeDirectoryName, cb:(error: any,result:Directory) => void){
        let directoryName = path.basename(relativeDirectoryName);
        let directoryParent = path.join( path.dirname(relativeDirectoryName),"/");

        this.DirectoryModel.findOne({name:directoryName, path: directoryParent}).populate('photos').populate('directories').exec( (err,res) =>{
            if(err || !res){
                return  this.indexDirectory(relativeDirectoryName,cb);
            }
            return cb(err,res);
        });

    }

    public indexDirectory(relativeDirectoryName, cb:(error: any,result:Directory) => void){
        DiskManager.scanDirectory(relativeDirectoryName,(err,scannedDirectory)=>{
            let arr = [];
            scannedDirectory.directories.forEach((value) => {
                let dir = new this.DirectoryModel(value);
                Utils.setKeys(dir,value);
                dir.save();
                arr.push(dir);
            });
            scannedDirectory.directories = arr;
            arr = [];
            scannedDirectory.photos.forEach((value) => {
                let p = new this.PhotoModel(value);
                Utils.setKeys(p,value);
                p.save();
                arr.push(p);
            });

            scannedDirectory.photos = arr;
            this.DirectoryModel.create(scannedDirectory,(err)=>{
                return cb(err,scannedDirectory);
            });

        });
    }



}