generate-code.js

const namespace = require('./constants').vueOptionsNamespace;

const splitRE = /\r?\n/g;

module.exports = function generateCode(
    scriptResult,
    templateResult,
    stylesResult,
    customBlocksResult,
    isFunctional,
) {
    let output = '';
    let renderFnStartLine;
    let renderFnEndLine;

    if (scriptResult) {
        output += `${scriptResult.code};\n`;
    } else {
        output +=
            `Object.defineProperty(exports, "__esModule", {\n` +
            `  value: true\n` +
            `});\n` +
            'module.exports.default = {};\n';
    }

    output +=
        `var ${namespace} = typeof exports.default === 'function' ` +
        `? exports.default.options ` +
        `: exports.default\n`;

    if (templateResult) {
        renderFnStartLine = output.split(splitRE).length;
        templateResult.code = templateResult.code.replace(
            'var _c = _vm._self._c || _h',
            '/* istanbul ignore next */\nvar _c = _vm._self._c || _h',
        );
        output += `${templateResult.code}\n`;

        renderFnEndLine = output.split(splitRE).length;

        output +=
            `__options__.render = render\n` +
            `${namespace}.staticRenderFns = staticRenderFns\n`;

        if (isFunctional) {
            output += `${namespace}.functional = true\n`;
            output += `${namespace}._compiled = true\n`;
        }
    }

    if (stylesResult) {
        const styleStr = stylesResult
            .map(
                ({ code, moduleName }) => `if(!this['${moduleName}']) {\n` +
                    `  this['${moduleName}'] = {};\n` +
                    `}\n` +
                    `this['${moduleName}'] = Object.assign(\n` +
                    `this['${moduleName}'], ${code});\n`,
            )
            .join('');
        if (isFunctional) {
            output +=
                `;(function() {\n` +
                `  var originalRender = ${namespace}.render\n` +
                `  var styleFn = function () { ${styleStr} }\n` +
                `  ${namespace}.render = function renderWithStyleInjection (h, context) {\n` +
                `    styleFn.call(context)\n` +
                `    return originalRender(h, context)\n` +
                `  }\n` +
                `})()\n`;
        } else {
            output +=
                `;(function() {\n` +
                `  var beforeCreate = ${namespace}.beforeCreate\n` +
                `  var styleFn = function () { ${styleStr} }\n` +
                `  ${namespace}.beforeCreate = beforeCreate ? [].concat(beforeCreate, styleFn) : [styleFn]\n` +
                `})()\n`;
        }
    }

    if (customBlocksResult) {
        output += `;\n ${customBlocksResult}`;
    }

    return {
        code: output,
        renderFnStartLine,
        renderFnEndLine,
    };
};