pigallery2/backend/model/DiskManger.ts

///<reference path="exif.d.ts"/>
import * as path from "path";
import {DirectoryDTO} from "../../common/entities/DirectoryDTO";
import {ProjectPath} from "../ProjectPath";
import {Logger} from "../Logger";
import {diskManagerTask, DiskManagerTask} from "./DiskMangerTask";
import {Config} from "../../common/config/private/Config";

const Pool = require('threads').Pool;
const pool = new Pool();

const LOG_TAG = "[DiskManager]";


pool.run(diskManagerTask);

export class DiskManager {
  public static scanDirectory(relativeDirectoryName: string, cb: (error: any, result: DirectoryDTO) => void) {
    Logger.silly(LOG_TAG, "scanning directory:", relativeDirectoryName);
    let directoryName = path.basename(relativeDirectoryName);
    let directoryParent = path.join(path.dirname(relativeDirectoryName), path.sep);
    let absoluteDirectoryName = path.join(ProjectPath.ImageFolder, relativeDirectoryName);

    let input = <DiskManagerTask.PoolInput>{
      relativeDirectoryName,
      directoryName,
      directoryParent,
      absoluteDirectoryName
    };

    let done = (error: any, result: DirectoryDTO) => {
      if (error || !result) {
        return cb(error, result);
      }

      let addDirs = (dir: DirectoryDTO) => {
        dir.photos.forEach((ph) => {
          ph.directory = dir;
        });
        dir.directories.forEach((d) => {
          addDirs(d);
        });
      };
      addDirs(result);
      return cb(error, result);
    };

    let error = (error) => {
      return cb(error, null);
    };


    if (Config.Server.enableThreading == true) {
      pool.send(input).on('done', done).on('error', error);
    } else {
      try {
        diskManagerTask(input, done);
      } catch (err) {
        error(err);
      }
    }
  }

}