Ghost/core/server/themes/Storage.js

'use strict';

var fs = require('fs-extra'),
    os = require('os'),
    path = require('path'),
    Promise = require('bluebird'),
    config = require('../config'),
    globalUtils = require('../utils'),
    LocalFileStorage = require('../adapters/storage/LocalFileStorage'),
    remove = Promise.promisify(fs.remove);

/**
 * @TODO: combine with loader.js?
 */
class ThemeStorage extends LocalFileStorage {
    constructor() {
        super();

        this.storagePath = config.getContentPath('themes');
    }

    getTargetDir() {
        return this.storagePath;
    }

    serve(options) {
        var self = this;

        return function downloadTheme(req, res, next) {
            var themeName = options.name,
                themePath = path.join(self.storagePath, themeName),
                zipName = themeName + '.zip',
                // store this in a unique temporary folder
                zipBasePath = path.join(os.tmpdir(), globalUtils.uid(10)),
                zipPath = path.join(zipBasePath, zipName),
                stream;

            Promise.promisify(fs.ensureDir)(zipBasePath)
                .then(function () {
                    return Promise.promisify(globalUtils.zipFolder)(themePath, zipPath);
                })
                .then(function (length) {
                    res.set({
                        'Content-disposition': 'attachment; filename={themeName}.zip'.replace('{themeName}', themeName),
                        'Content-Type': 'application/zip',
                        'Content-Length': length
                    });

                    stream = fs.createReadStream(zipPath);
                    stream.pipe(res);
                })
                .catch(function (err) {
                    next(err);
                })
                .finally(function () {
                    remove(zipBasePath);
                });
        };
    }

    delete(fileName) {
        return remove(path.join(this.storagePath, fileName));
    }
}

module.exports = ThemeStorage;