97 lines
2.7 KiB
JavaScript
97 lines
2.7 KiB
JavaScript
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 });
|
|
};
|