Sharkey/src/tools/move-drive-files.ts

76 lines
2.1 KiB
TypeScript
Raw Normal View History

2018-10-08 20:14:03 +03:00
import * as Minio from 'minio';
import * as uuid from 'uuid';
2018-10-08 23:46:21 +03:00
const sequential = require('promise-sequential');
2018-10-08 20:14:03 +03:00
import DriveFile, { DriveFileChunk, getDriveFileBucket } from '../models/drive-file';
import DriveFileThumbnail, { DriveFileThumbnailChunk } from '../models/drive-file-thumbnail';
import config from '../config';
DriveFile.find({
$or: [{
withoutChunks: { $exists: false }
}, {
withoutChunks: false
}]
}, {
fields: {
_id: true
}
2018-10-08 20:14:03 +03:00
}).then(async files => {
console.log(`there is ${files.length} files`);
2018-10-08 23:46:21 +03:00
await sequential(files.map(file => async () => {
file = await DriveFile.findOne({ _id: file._id });
2018-10-08 20:14:03 +03:00
const minio = new Minio.Client(config.drive.config);
const keyDir = `${config.drive.prefix}/${uuid.v4()}`;
const key = `${keyDir}/${name}`;
const thumbnailKeyDir = `${config.drive.prefix}/${uuid.v4()}`;
const thumbnailKey = `${thumbnailKeyDir}/${name}.thumbnail.jpg`;
const baseUrl = config.drive.baseUrl
|| `${ config.drive.config.useSSL ? 'https' : 'http' }://${ config.drive.config.endPoint }${ config.drive.config.port ? `:${config.drive.config.port}` : '' }/${ config.drive.bucket }`;
const bucket = await getDriveFileBucket();
const readable = bucket.openDownloadStream(file._id);
await minio.putObject(config.drive.bucket, key, readable, file.length, {
'Content-Type': file.contentType,
'Cache-Control': 'max-age=31536000, immutable'
});
await DriveFile.findOneAndUpdate({ _id: file._id }, {
$set: {
'metadata.withoutChunks': true,
'metadata.storage': 'minio',
'metadata.storageProps': {
key: key,
thumbnailKey: thumbnailKey
},
'metadata.url': `${ baseUrl }/${ keyDir }/${ encodeURIComponent(name) }`,
}
});
// チャンクをすべて削除
await DriveFileChunk.remove({
files_id: file._id
});
//#region サムネイルもあれば削除
const thumbnail = await DriveFileThumbnail.findOne({
'metadata.originalId': file._id
});
if (thumbnail) {
await DriveFileThumbnailChunk.remove({
files_id: thumbnail._id
});
await DriveFileThumbnail.remove({ _id: thumbnail._id });
}
//#endregion
2018-10-08 23:46:21 +03:00
console.log('done', file._id);
}));
2018-10-08 20:14:03 +03:00
});