marko/runtime/RenderResult.js

var domInsert = require('./dom-insert');

function checkAddedToDOM(result, method) {
    if (!result.$__widgets) {
        throw Error('Not added to DOM');
    }
}

function getWidgetDefs(result) {
    var widgetDefs = result.$__widgets;

    if (widgetDefs.length === 0) {
        throw Error('No widget');
    }
    return widgetDefs;
}

function RenderResult(out) {
   this.out = this.$__out = out;
   this.$__widgets = null;
}

module.exports = RenderResult;

var proto = RenderResult.prototype = {
    getWidget: function() {
        checkAddedToDOM(this, 'getWidget');

        var rerenderWidgetInfo = this.$__out.global.$w;
        var rerenderWidget = rerenderWidgetInfo && rerenderWidgetInfo[0];
        if (rerenderWidget) {
            return rerenderWidget;
        }

        return getWidgetDefs(this)[0].$__widget;
    },
    getWidgets: function(selector) {
        checkAddedToDOM(this, 'getWidgets');

        var widgetDefs = getWidgetDefs(this);

        var widgets = [];
        var i;

        for (i = 0; i < widgetDefs.length; i++) {
            var widget = widgetDefs[i].$__widget;
            if (!selector || selector(widget)) {
                widgets.push(widget);
            }
        }

        return widgets;
    },

    afterInsert: function(doc) {
        var out = this.$__out;
        var widgetsContext = out.global.widgets;
        if (widgetsContext) {
            this.$__widgets = widgetsContext.$__widgets;
            widgetsContext.$__initWidgets(doc);
        }

        return this;
    },
    getNode: function(doc) {
        return this.$__out.$__getNode(doc);
    },
    getOutput: function() {
        return this.$__out.$__getOutput();
    },
    toString: function() {
        return this.$__out.toString();
    },
    toJSON: function() {
        return this.$__out.$__getOutput();
    },
    document: typeof document !== 'undefined' && document
};

// Add all of the following DOM methods to Widget.prototype:
// - appendTo(referenceEl)
// - replace(referenceEl)
// - replaceChildrenOf(referenceEl)
// - insertBefore(referenceEl)
// - insertAfter(referenceEl)
// - prependTo(referenceEl)
domInsert(
    proto,
    function getEl(renderResult, referenceEl) {
        return renderResult.getNode(referenceEl.ownerDocument);
    },
    function afterInsert(renderResult, referenceEl) {
        return renderResult.afterInsert(referenceEl.ownerDocument);
    });