pigallery2/backend/model/DiskManger.ts

import * as fs from "fs";
import * as path from "path";
import * as mime from "mime";
import * as sizeOf from "image-size";
import {Directory} from "../../common/entities/Directory";
import {Photo} from "../../common/entities/Photo";

export class DiskManager {
    public static scanDirectory(relativeDirectoryName, cb:(error:any, result:Directory) => void) {
        console.log("DiskManager: scanDirectory");
        let directoryName = path.basename(relativeDirectoryName);
        let directoryParent = path.join(path.dirname(relativeDirectoryName), "/");
        let absoluteDirectoryName = path.join(__dirname, "/../../demo/images", relativeDirectoryName);

        let directory = new Directory(1, directoryName, directoryParent, new Date(), [], []);

        fs.readdir(absoluteDirectoryName, function (err, list) {

            if (err) {
                return cb(err, null);
            }


            for (let i = 0; i < list.length; i++) {
                let file = list[i];
                let fullFilePath = path.resolve(absoluteDirectoryName, file);
                if (fs.statSync(fullFilePath).isDirectory()) {
                    directory.directories.push(new Directory(2, file, relativeDirectoryName, new Date(), [], []));
                }

                if (DiskManager.isImage(fullFilePath)) {
                    let dimensions = sizeOf(fullFilePath);
                    directory.photos.push(new Photo(1, file, dimensions.width, dimensions.height));
                }
            }

            return cb(err, directory);

        });
    }

    private static isImage(fullPath) {
        let imageMimeTypes = [
            'image/bmp',
            'image/gif',
            'image/jpeg',
            'image/png',
            'image/pjpeg',
            'image/tiff',
            'image/webp',
            'image/x-tiff',
            'image/x-windows-bmp'
        ];

        var extension = mime.lookup(fullPath);

        if (imageMimeTypes.indexOf(extension) !== -1) {
            return true;
        }

        return false;
    }
}