piskel/cli/export-png.js

const fs = require('fs');

function onPageEvaluate(window, options, piskel) {
    console.log("\nPiskel name: " + piskel.descriptor.name);

    // Setup piskelController
    var piskelController = new pskl.controller.piskel.PiskelController(piskel);

    pskl.app.piskelController = piskelController;

    piskelController.init();

    // Apply crop if enabled
    if (options.crop) {
        // Mock selection manager to avoid errors during crop
        pskl.app.selectionManager = {};

        // Setup crop tool
        var crop = new pskl.tools.transform.Crop();

        // Perform crop
        crop.applyTransformation();

        // Get cropped piskel
        piskel = piskelController.getPiskel();
    }

    // Mock exportController to provide zoom value based on cli args
    // and to avoid errors and/or unnecessary bootstrapping
    var exportController = {
        getExportZoom: function () {
            var zoom = options.zoom;

            if (options.scaledWidth) {
                zoom = options.scaledWidth / piskel.getWidth();
            } else if (options.scaledHeight) {
                zoom = options.scaledHeight / piskel.getHeight();
            }

            return zoom;
        }
    };

    // Setup pngExportController
    var pngExportController = new pskl.controller.settings.exportimage.PngExportController(piskelController, exportController);

    // Mock getColumns and getRows to use values from cli arguments
    pngExportController.getColumns_ = function () {
        if (options.columns) return options.columns;

        if (options.rows) {
            return Math.ceil(piskelController.getFrameCount() / pngExportController.getRows_());
        } else {
            return pngExportController.getBestFit_();
        }
    };

    pngExportController.getRows_ = function () {
        if (options.columns && !options.rows) {
            return Math.ceil(piskelController.getFrameCount() / pngExportController.getColumns_());
        }

        return options.rows;
    };

    // Render to output canvas
    var canvas;

    if (options.frame > -1) {
        // Render a single frame
        canvas = piskelController.renderFrameAt(options.frame, true);

        var zoom = exportController.getExportZoom();

        if (zoom != 1) {
            // Scale rendered frame
            canvas = pskl.utils.ImageResizer.resize(canvas, canvas.width * zoom, canvas.height * zoom, false);
        }
    } else {
        // Render the sprite sheet
        canvas = pngExportController.createPngSpritesheet_();
    }

    // Add output canvas to DOM
    window.document.body.appendChild(canvas);

    // Prepare return data
    const returnData = {
        width: canvas.width,
        height: canvas.height
    };

    // Wait a tick for things to wrap up
    setTimeout(function ()  {
        // Exit and pass data to parent process
        window.callPhantom(returnData);
    }, 0);
}

function onPageExit(page, options, data) {
    // Set clip for output image
    if (data.width && data.height) {
        page.clipRect = { top: 0, left: 0, width: data.width, height: data.height };
    }

    console.log("\n" + 'Generated file(s):');

    const dest = options.dest.replace('.png', '') + '.png';

    // Render page to the output image
    page.render(dest);

    console.log(" " + dest);

    if (options.dataUri) {
        const dataUriPath = options.dest + '.datauri';

        const dataUri = 'data:image/png;base64,' + page.renderBase64('PNG');

        // Write data-uri to file
        fs.write(dataUriPath, dataUri, 'w');

        console.log(" " + dataUriPath);
    }
}

module.exports = {
    onPageEvaluate: onPageEvaluate,
    onPageExit: onPageExit
};