impress-2020/api/uploadLayerImage.js

98 lines
2.7 KiB
JavaScript
Raw Normal View History

const AWS = require("aws-sdk");
const Jimp = require("jimp");
const connectToDb = require("../src/server/db.js");
const s3 = new AWS.S3({
accessKeyId: process.env["DTI_AWS_ACCESS_KEY_ID"],
secretAccessKey: process.env["DTI_AWS_SECRET_ACCESS_KEY"],
});
async function upload(bucket, key, imageData) {
await s3
.putObject({
Bucket: bucket,
Key: key,
Body: imageData,
ContentType: "image/png",
ACL: "public-read",
})
.promise();
}
async function resize(imageData, size) {
const image = await Jimp.read(imageData);
const resizedImage = image.resize(size, size);
const resizedImageData = await resizedImage.getBufferAsync("image/png");
return resizedImageData;
}
async function processImage(assetType, remoteId, size, imageData, now) {
if (size !== 600) {
imageData = await resize(imageData, size);
}
const paddedId = String(remoteId).padStart(12, "0");
const id1 = paddedId.slice(0, 3);
const id2 = paddedId.slice(3, 6);
const id3 = paddedId.slice(6, 9);
const key = `${assetType}/${id1}/${id2}/${id3}/${remoteId}/${size}x${size}.png`;
upload("impress-asset-images", key, imageData);
const when = Number(now);
return `https://impress-asset-images.s3.amazonaws.com/${key}?v2-${when}`;
}
export default async (req, res) => {
if (req.headers["dti-support-secret"] !== process.env["SUPPORT_SECRET"]) {
res.status(401).send(`Support secret is incorrect. Try setting up again?`);
return;
}
let imageData = Buffer.alloc(0);
await new Promise((resolve) => {
req.on("data", (chunk) => {
imageData = Buffer.concat([imageData, chunk]);
});
req.on("end", () => {
resolve();
});
});
const db = await connectToDb();
const { layerId } = req.query;
const [layerRows] = await db.execute(
`SELECT * FROM swf_assets WHERE id = ?`,
[layerId]
);
const layer = layerRows[0];
if (!layer) {
res.status(404).send(`Layer not found`);
}
const now = new Date();
const { remote_id: remoteId, type: assetType } = layer;
const [imageUrl600, imageUrl300, imageUrl150] = await Promise.all([
processImage(assetType, remoteId, 600, imageData, now),
processImage(assetType, remoteId, 300, imageData, now),
processImage(assetType, remoteId, 150, imageData, now),
]);
const nowTimestamp = now.toISOString().slice(0, 19).replace("T", " ");
const [result] = await db.execute(
`UPDATE swf_assets SET image_manual = 1, converted_at = ?
WHERE type = ? AND remote_id = ? LIMIT 1`,
[nowTimestamp, assetType, remoteId]
);
if (result.affectedRows !== 1) {
res
.status(500)
.send(`expected 1 affected row but found ${result.affectedRows}`);
}
res.status(200).send({ imageUrl600, imageUrl300, imageUrl150 });
};