mirror of https://github.com/TryGhost/Ghost.git
66 lines
2.0 KiB
JavaScript
66 lines
2.0 KiB
JavaScript
const fs = require('fs-extra');
|
|
const os = require('os');
|
|
const path = require('path');
|
|
const config = require('../../../shared/config');
|
|
const security = require('../../../server/lib/security');
|
|
const {compress} = require('@tryghost/zip');
|
|
const LocalFileStorage = require('../../../server/adapters/storage/LocalFileStorage');
|
|
|
|
/**
|
|
* @TODO: combine with loader.js?
|
|
*/
|
|
class ThemeStorage extends LocalFileStorage {
|
|
constructor() {
|
|
super();
|
|
|
|
this.storagePath = config.getContentPath('themes');
|
|
}
|
|
|
|
getTargetDir() {
|
|
return this.storagePath;
|
|
}
|
|
|
|
serve(options) {
|
|
const self = this;
|
|
|
|
return function downloadTheme(req, res, next) {
|
|
const themeName = options.name;
|
|
const themePath = path.join(self.storagePath, themeName);
|
|
const zipName = themeName + '.zip';
|
|
|
|
// store this in a unique temporary folder
|
|
const zipBasePath = path.join(os.tmpdir(), security.identifier.uid(10));
|
|
|
|
const zipPath = path.join(zipBasePath, zipName);
|
|
let stream;
|
|
|
|
fs.ensureDir(zipBasePath)
|
|
.then(function () {
|
|
return compress(themePath, zipPath);
|
|
})
|
|
.then(function (result) {
|
|
res.set({
|
|
'Content-disposition': 'attachment; filename={themeName}.zip'.replace('{themeName}', themeName),
|
|
'Content-Type': 'application/zip',
|
|
'Content-Length': result.size
|
|
});
|
|
|
|
stream = fs.createReadStream(zipPath);
|
|
stream.pipe(res);
|
|
})
|
|
.catch(function (err) {
|
|
next(err);
|
|
})
|
|
.finally(function () {
|
|
return fs.remove(zipBasePath);
|
|
});
|
|
};
|
|
}
|
|
|
|
delete(fileName) {
|
|
return fs.remove(path.join(this.storagePath, fileName));
|
|
}
|
|
}
|
|
|
|
module.exports = ThemeStorage;
|