123 lines
2.6 KiB
JavaScript
123 lines
2.6 KiB
JavaScript
const { S3 } = require("@aws-sdk/client-s3");
|
|
const { PrismaClient } = require("@prisma/client");
|
|
const { existsSync } = require("fs");
|
|
const util = require("util");
|
|
|
|
const prisma = new PrismaClient();
|
|
|
|
const STORAGE_FOLDER = process.env.STORAGE_FOLDER || "data";
|
|
|
|
const s3Client =
|
|
process.env.SPACES_ENDPOINT &&
|
|
process.env.SPACES_REGION &&
|
|
process.env.SPACES_KEY &&
|
|
process.env.SPACES_SECRET
|
|
? new S3({
|
|
forcePathStyle: false,
|
|
endpoint: process.env.SPACES_ENDPOINT,
|
|
region: process.env.SPACES_REGION,
|
|
credentials: {
|
|
accessKeyId: process.env.SPACES_KEY,
|
|
secretAccessKey: process.env.SPACES_SECRET,
|
|
},
|
|
})
|
|
: undefined;
|
|
|
|
async function checkFileExistence(path) {
|
|
if (s3Client) {
|
|
const bucketParams = {
|
|
Bucket: process.env.SPACES_BUCKET_NAME,
|
|
Key: path,
|
|
};
|
|
|
|
try {
|
|
const headObjectAsync = util.promisify(
|
|
s3Client.headObject.bind(s3Client)
|
|
);
|
|
|
|
try {
|
|
await headObjectAsync(bucketParams);
|
|
return true;
|
|
} catch (err) {
|
|
return false;
|
|
}
|
|
} catch (err) {
|
|
console.log("Error:", err);
|
|
|
|
return false;
|
|
}
|
|
} else {
|
|
try {
|
|
if (existsSync(STORAGE_FOLDER + "/" + path)) {
|
|
return true;
|
|
} else return false;
|
|
} catch (err) {
|
|
console.log(err);
|
|
}
|
|
}
|
|
}
|
|
|
|
// Avatars
|
|
async function migrateToV2() {
|
|
const users = await prisma.user.findMany();
|
|
|
|
for (let user of users) {
|
|
const path = `uploads/avatar/${user.id}.jpg`;
|
|
|
|
const res = await checkFileExistence(path);
|
|
|
|
if (res) {
|
|
await prisma.user.update({
|
|
where: { id: user.id },
|
|
data: { image: path },
|
|
});
|
|
console.log(`${user.id}`);
|
|
} else {
|
|
console.log(`${user.id}`);
|
|
}
|
|
}
|
|
|
|
const links = await prisma.link.findMany();
|
|
|
|
// PDFs
|
|
for (let link of links) {
|
|
const path = `archives/${link.collectionId}/${link.id}.pdf`;
|
|
|
|
const res = await checkFileExistence(path);
|
|
|
|
if (res) {
|
|
await prisma.link.update({
|
|
where: { id: link.id },
|
|
data: { pdfPath: path },
|
|
});
|
|
console.log(`${link.id}`);
|
|
} else {
|
|
console.log(`${link.id}`);
|
|
}
|
|
}
|
|
|
|
// Screenshots
|
|
for (let link of links) {
|
|
const path = `archives/${link.collectionId}/${link.id}.png`;
|
|
|
|
const res = await checkFileExistence(path);
|
|
|
|
if (res) {
|
|
await prisma.link.update({
|
|
where: { id: link.id },
|
|
data: { screenshotPath: path },
|
|
});
|
|
console.log(`${link.id}`);
|
|
} else {
|
|
console.log(`${link.id}`);
|
|
}
|
|
}
|
|
|
|
await prisma.$disconnect();
|
|
}
|
|
|
|
migrateToV2().catch((e) => {
|
|
console.error(e);
|
|
process.exit(1);
|
|
});
|