ketr.photos/util/check.js

"use strict";

const fs = require("fs"),
  Promise = require("bluebird"),
  config = require("config");

if (process.argv.length <= 2) {
  console.log("usage: node check file|id [file|id] ... ");
  process.exit(-1);
}
const items = process.argv.splice(2);

const picturesPath = config.get("picturesPath").replace(/\/$/, "") + "/";

for (let i = 0; i < items.length; i++) {
  if (parseInt(items[i]) == items[i]) {
    items[i] = {
      type: "id",
      value: items[i]
    };
  } else {
    items[i] = {
      type: "file",
      value: items[i]
    }
  }
}

const exists = function(path) {
  return stat(path).then(function(stats) {
    return stats;
  }).catch(function() {
    return null;
  });
}

const stat = function (_path) {
  if (_path.indexOf(picturesPath.replace(/\/$/, "")) == 0) {
    _path = _path.substring(picturesPath.length);
  }

  let path = picturesPath + _path;

  return new Promise(function (resolve, reject) {
    fs.stat(path, function (error, stats) {
      if (error) {
        return reject(error);
      }
      return resolve(stats);
    });
  });
}

const imageExtensions = [ "jpg", "jpeg", "png", "gif", "nef" ],
  imageRegExp = new RegExp("\.((" + imageExtensions.join(")|(") + "))$", "i"),
  videoExtensions = [ "mov", "avi", "mp4", "webm" ],
  videoRegExp = new RegExp("\.((" + videoExtensions.join(")|(") + "))$", "i");

function getAssetInfoFromDisk(asset, filepath) {
  return exists(filepath).then(function(stats) {
    asset.filepath = filepath;
    if (stats) {
      asset.stats = stats;
    }

    /* Check if entry exists in DB */
    let parts = /^(.*)\/([^/]+)$/.exec(filepath), path, filename;
    if (parts) {
      path = parts[1] + "/";
      filename = parts[2];
    } else {
      path = "";
      filename = filepath;
    }
    
    return photoDB.sequelize.query("SELECT photos.*,(albums.path || photos.filename) AS filepath " +
      "FROM photos LEFT JOIN albums ON albums.id=photos.albumId " +
      "WHERE albums.path=:path AND photos.filename=:filename", {
      replacements: {
        path: path,
        filename: filename
      },
      type: photoDB.Sequelize.QueryTypes.SELECT,
      raw: true
    }).then(function(photos) {
      if (photos.length == 1) {
        Object.assign(asset, photos[0]);
      }
    });
  }).then(function() {
    return asset;
  });
}

function getAssetInfoFromDB(asset, id) {
  return photoDB.sequelize.query("SELECT photos.*,(albums.path || photos.filename) AS filepath " +
    "FROM photos LEFT JOIN albums ON albums.id=photos.albumId " +
    "WHERE photos.id=:id", {
    replacements: {
      id: id
    },
    type: photoDB.Sequelize.QueryTypes.SELECT,
    raw: true
  }).then(function(photos) {
    if (photos.length == 1) {
      Object.assign(asset, photos[0]);
    }
  }).then(function() {
    if (!asset.filepath) {
      return;
    }
    
    return exists(asset.filepath).then(function(stats) {
      if (stats) {
        asset.stats = stats;
      }
    });
  }).then(function() {
    return asset;
  });
}

let photoDB;

require("../server/db/photos").then(function(db) {
  photoDB = db;

  return Promise.mapSeries(items, function(item) {
    const asset = {};

    return Promise.resolve().then(function() {
      switch (item.type) {
      case "file":
        return getAssetInfoFromDisk(asset, item.value);
        break;

      case "id":
        return getAssetInfoFromDB(asset, item.value);
        break;
      }
    }).then(function(asset) {
      if (!asset.stats && !asset.id) {
        console.log("Item " + item.value + " exists neither on disk nor in the DB.");
        return;
      }

      if (!asset.id) {
        console.log("Item " + asset.filepath + " does not exist in the DB.");
      }

      if (!asset.stats) {
        console.log("Item " + asset.id + " does not exist on disk.");
      } else {
        if (asset.modified > asset.stats.mtime) {
          asset.needsUpdate = true;
        }
      }

      if (asset.needsUpdate) {
        console.log("DB needs to be updated for " + item.value);
      }
    })
  });
});