You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
44 lines
1.3 KiB
JavaScript
44 lines
1.3 KiB
JavaScript
'use strict';
|
|
|
|
const crypto = require('crypto');
|
|
|
|
const db = require('../../database');
|
|
const batch = require('../../batch');
|
|
|
|
const md5 = filename => crypto.createHash('md5').update(filename).digest('hex');
|
|
|
|
module.exports = {
|
|
name: 'Fix paths in user uploads sorted sets',
|
|
timestamp: Date.UTC(2022, 1, 10),
|
|
method: async function () {
|
|
const { progress } = this;
|
|
|
|
await batch.processSortedSet('users:joindate', async (uids) => {
|
|
progress.incr(uids.length);
|
|
|
|
await Promise.all(uids.map(async (uid) => {
|
|
const key = `uid:${uid}:uploads`;
|
|
// Rename the paths within
|
|
let uploads = await db.getSortedSetRangeWithScores(key, 0, -1);
|
|
if (uploads.length) {
|
|
// Don't process those that have already the right format
|
|
uploads = uploads.filter(upload => upload.value.startsWith('/files/'));
|
|
|
|
await db.sortedSetRemove(key, uploads.map(upload => upload.value));
|
|
await db.sortedSetAdd(
|
|
key,
|
|
uploads.map(upload => upload.score),
|
|
uploads.map(upload => upload.value.slice(1))
|
|
);
|
|
// Add uid to the upload's hash object
|
|
uploads = await db.getSortedSetMembers(key);
|
|
await db.setObjectBulk(uploads.map(relativePath => [`upload:${md5(relativePath)}`, { uid: uid }]));
|
|
}
|
|
}));
|
|
}, {
|
|
batch: 500,
|
|
progress: progress,
|
|
});
|
|
},
|
|
};
|