photosite/dist/generate.js

301 lines
9.8 KiB
JavaScript
Raw Normal View History

"use strict";
2024-12-09 20:15:27 +00:00
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
2024-01-08 20:10:39 +00:00
require("dotenv/config");
const log_1 = require("./log");
2024-01-08 20:38:27 +00:00
const promises_1 = require("fs/promises");
const path_1 = require("path");
2024-12-05 19:20:28 +00:00
const util_1 = require("util");
2024-12-09 20:15:27 +00:00
const uuid_1 = require("uuid");
const crypto_1 = require("crypto");
const os_1 = require("os");
const moment_1 = __importDefault(require("moment"));
const argparse_1 = require("argparse");
2024-01-08 20:10:39 +00:00
const shell_1 = require("./shell");
const hash_1 = require("./hash");
2024-12-03 04:09:32 +00:00
const files3_1 = require("./files3");
const env_1 = require("./env");
const db_1 = require("./db");
2024-12-09 20:15:27 +00:00
const geocode_1 = require("./geocode");
2024-12-05 19:20:28 +00:00
const sizeOf = (0, util_1.promisify)(require('image-size'));
class Generate {
constructor() {
2024-12-03 04:09:32 +00:00
this.inbox = (0, env_1.envString)('INBOX', '~/Photos/toprocess');
this.photos = (0, env_1.envString)('PHOTOS', '~/Photos/processed');
2024-12-09 20:15:27 +00:00
this.artist = (0, env_1.envString)('ARTIST', 'Unknown');
this.tmp = (0, os_1.tmpdir)();
this.log = (0, log_1.createLog)('generate');
2024-12-09 20:15:27 +00:00
const parser = new argparse_1.ArgumentParser({
description: 'Generate script'
});
parser.add_argument('-s', '--score', { type: 'int', default: 0, help: 'Starting score' });
const args = parser.parse_args();
this.log.info(`Generating site: ${new Date()}`);
2024-12-03 04:09:32 +00:00
this.db = new db_1.DB();
2024-12-09 20:15:27 +00:00
this.s3 = new files3_1.Files3((0, env_1.envString)('S3_BUCKET', 's3bucket'), true);
this.geocode = new geocode_1.Geocode();
2024-01-08 20:38:27 +00:00
this.generate();
2024-12-09 20:15:27 +00:00
this.score = args.score;
2024-01-08 20:38:27 +00:00
}
async generate() {
2024-12-03 04:09:32 +00:00
//check version
//sync
2024-01-08 20:38:27 +00:00
await this.checkInbox();
2024-12-03 04:09:32 +00:00
//validate
}
2024-01-08 20:10:39 +00:00
async checkInbox() {
2024-01-08 20:38:27 +00:00
let inbox;
let images;
2024-12-03 04:09:32 +00:00
let filename;
let meta;
let photo;
2024-12-09 20:15:27 +00:00
let exif;
2024-01-08 20:38:27 +00:00
try {
inbox = await (0, promises_1.realpath)(this.inbox);
}
catch (err) {
this.log.error(err);
return;
}
try {
images = await (0, promises_1.readdir)(inbox);
}
catch (err) {
this.log.error(err);
return;
}
2024-12-03 04:09:32 +00:00
images = images.filter((el) => {
2024-01-08 20:38:27 +00:00
if (el.toLowerCase().indexOf('.jpg') !== -1
|| el.toLowerCase().indexOf('.jpeg') !== -1
|| el.toLowerCase().indexOf('.tif') !== -1
|| el.toLowerCase().indexOf('.tiff') !== -1) {
return true;
}
return false;
});
if (images.length === 0) {
this.log.info(`No new images found`);
return;
}
2024-12-03 04:09:32 +00:00
images = await Promise.all(images.map(async (el) => {
return await (0, promises_1.realpath)((0, path_1.join)(inbox, el));
}));
for (let image of images) {
this.log.info(image);
filename = (0, path_1.basename)(image);
meta = this.parseFilename(filename);
try {
photo = await this.createPhoto(image, meta);
}
catch (err) {
this.log.error(`Error creating photo record metadata`, err);
continue;
}
2024-12-09 20:15:27 +00:00
if (await this.db.existsName(filename)) {
this.log.info(`Image ${filename} already exists`);
if (await this.db.existsHash(photo.hash)) {
this.log.warn(`Image ${name} already exists, moving...`);
await this.move(image);
continue;
}
}
if (await this.db.existsHash(photo.hash)) {
this.log.warn(`Image exists under different name, update?`);
await this.move(image);
continue;
}
try {
await this.db.create(photo);
}
catch (err) {
this.log.error(`Error inserting photo into database`, err);
continue;
}
try {
await this.upload(image);
}
catch (err) {
this.log.error(`Error uploading image`, err);
continue;
}
2024-12-09 20:15:27 +00:00
try {
exif = await this.exif(photo);
}
catch (err) {
this.log.error(`Error building EXIF data`, err);
}
try {
await this.img(image, photo.id, exif);
}
catch (err) {
this.log.error(`Error running img.sh script`, err);
}
await this.move(image);
2024-12-03 04:09:32 +00:00
}
2024-01-08 20:10:39 +00:00
}
2024-12-09 20:15:27 +00:00
//Artist string
//ImageTitle string
//ImageUniqueID string
//ISO int16u[n]
//DateTimeOriginal string (YYYY:MM:DD HH:MM:SS)
//
//GPSLatitude rational64u[3]
//GPSLongitude
async exif(photo) {
const filePath = await this.mktemp('photosite_exif');
const created = moment_1.default.unix(photo.created / 1000).format('YYYY:MM:DD HH:mm:ss');
let exif = `-Artist=${this.artist}
-ImageTitle=${photo.name}
-ImageUniqueId=${photo.id}
-DateTimeOriginal=${created}`;
const iso = photo.filmstock.split(' ').filter(el => this.isOnlyNumbers(el)).map(el => parseInt(el)).filter(el => el > 25);
if (iso.length > 0) {
exif += `
-ISO=${iso}`;
}
if (photo.latitude !== null && photo.longitude !== null) {
exif += `
-GPSLatitude=${photo.latitude}
-GPSLongitude=${photo.longitude}`;
}
try {
await (0, promises_1.writeFile)(filePath, exif, 'utf8');
}
catch (err) {
this.log.error(`Error writing EXIF data`, err);
}
return filePath;
}
async img(file, id, exif) {
const cmd = ['bash', 'scripts/img.sh', file, id, exif];
2024-01-08 20:10:39 +00:00
const shell = new shell_1.Shell(cmd);
try {
await shell.execute();
}
catch (err) {
this.log.error(err);
return;
}
this.log.info(`Processed image file for ${file}`);
}
2024-12-03 04:09:32 +00:00
async getImageDimensions(imagePath) {
let dimensions;
try {
dimensions = await sizeOf(imagePath);
return dimensions;
}
catch (err) {
this.log.error('Error getting image dimensions:', err);
}
}
capitalize(str) {
return (str.substring(0, 1).toUpperCase()) + str.substring(1);
}
formatProperNouns(str) {
let parts = str.split('-');
parts = parts.map(el => this.capitalize(el));
return parts.join(' ');
}
//year
//month
//day
//format
//filmstock
//location
//description
//original
//2024_12_02_35mm_Kodak-Gold-200_Somerville-MA_Walk-towards-Harvard-Square#000061280009.tif
2024-12-03 04:09:32 +00:00
parseFilename(filename) {
const halves = filename.split('#');
2024-12-05 19:20:28 +00:00
const parts = halves[0].split('_');
2024-12-03 04:09:32 +00:00
let meta = {};
for (let i = 0; i < parts.length; i++) {
switch (i) {
case 0:
meta.year = parseInt(parts[i]);
break;
case 1:
meta.month = parseInt(parts[i]);
break;
case 2:
meta.day = parseInt(parts[i]);
break;
case 3:
meta.format = parts[i];
break;
case 4:
meta.filmstock = parts[i].split('-').join(' ');
break;
case 5:
meta.location = parts[i].split('-').join(' ');
break;
case 6:
meta.description = parts[i].split('-').join(' ');
break;
}
}
meta.original = halves[1];
return meta;
}
async createPhoto(image, meta) {
const hash = await hash_1.Hashes.fileHash(image);
const dimensions = await this.getImageDimensions(image);
const now = Date.now();
2024-12-09 20:15:27 +00:00
const latlng = await this.geocode.query(meta.location);
return {
2024-12-09 20:15:27 +00:00
id: (0, uuid_1.v4)(),
name: (0, path_1.basename)(image),
original: meta.original,
hash,
width: dimensions.width,
height: dimensions.height,
format: meta.format,
filmstock: meta.filmstock,
location: meta.location,
2024-12-09 20:15:27 +00:00
latitude: latlng === null ? null : latlng.latitude,
longitude: latlng === null ? null : latlng.longitude,
discovered: now,
updated: now,
2024-12-09 20:15:27 +00:00
created: +new Date(meta.year, meta.month, meta.day),
score: this.score
};
}
async upload(image) {
const name = (0, path_1.basename)(image);
return this.s3.createFromPath(image, name);
}
2024-12-09 20:15:27 +00:00
async move(image) {
const name = (0, path_1.basename)(image);
const dest = (0, path_1.join)(this.photos, name);
try {
//await rename(image, dest);
this.log.info(`Moved image ${name} to outbox`);
}
catch (err) {
this.log.error(`Error moving image`, err);
}
}
async mktemp(prefix = 'tmp') {
const uniqueId = (0, crypto_1.randomBytes)(16).toString('hex');
const tempFilePath = (0, path_1.join)(this.tmp, `${prefix}-${uniqueId}`);
try {
await (0, promises_1.writeFile)(tempFilePath, '', { flag: 'wx' });
return tempFilePath;
}
catch (err) {
if (err.code === 'EEXIST') {
return this.mktemp(prefix);
}
else {
throw err;
}
}
}
isOnlyNumbers(str) {
return /^[0-9]+$/.test(str);
}
}
new Generate();
//# sourceMappingURL=generate.js.map