ADAM / AIAHTML5

/*
 * Copyright (C) 2011 Google Inc. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are
 * met:
 *
 *     * Redistributions of source code must retain the above copyright
 * notice, this list of conditions and the following disclaimer.
 *     * Redistributions in binary form must reproduce the above
 * copyright notice, this list of conditions and the following disclaimer
 * in the documentation and/or other materials provided with the
 * distribution.
 *     * Neither the name of Google Inc. nor the names of its
 * contributors may be used to endorse or promote products derived from
 * this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

WebInspector.ScriptFormatter = function()
{
    this._worker = new Worker("ScriptFormatterWorker.js");
    this._worker.onmessage = this._handleMessage.bind(this);
    this._worker.onerror = this._handleError.bind(this);
    this._tasks = [];
}

WebInspector.ScriptFormatter.locationToPosition = function(lineEndings, lineNumber, columnNumber)
{
    var position = lineNumber ? lineEndings[lineNumber - 1] + 1 : 0;
    return position + columnNumber;
}

WebInspector.ScriptFormatter.positionToLocation = function(lineEndings, position)
{
    var location = {};
    location.lineNumber = lineEndings.upperBound(position - 1);
    if (!location.lineNumber)
        location.columnNumber = position;
    else
        location.columnNumber = position - lineEndings[location.lineNumber - 1] - 1;
    return location;
}

WebInspector.ScriptFormatter.findScriptRanges = function(lineEndings, scripts)
{
    var scriptRanges = [];
    for (var i = 0; i < scripts.length; ++i) {
        var start = { lineNumber: scripts[i].lineOffset, columnNumber: scripts[i].columnOffset };
        start.position = WebInspector.ScriptFormatter.locationToPosition(lineEndings, start.lineNumber, start.columnNumber);
        var endPosition = start.position + scripts[i].length;
        var end = WebInspector.ScriptFormatter.positionToLocation(lineEndings, endPosition);
        end.position = endPosition;
        scriptRanges.push({ start: start, end: end, sourceID: scripts[i].sourceID });
    }
    scriptRanges.sort(function(x, y) { return x.start.position - y.start.position; });
    return scriptRanges;
}

WebInspector.ScriptFormatter.prototype = {
    formatContent: function(content, callback)
    {
        var chunks = this._splitContentIntoChunks(content.text, content.scriptRanges);

        function didFormatChunks()
        {
            var result = this._buildContentFromChunks(chunks);

            var sourceMapping = new WebInspector.SourceMappingForFormattedScript(content.text.lineEndings(), result.text.lineEndings(), result.mapping);
            var formattedScriptRanges = [];
            for (var i = 0; i < content.scriptRanges.length; ++i) {
                var scriptRange = content.scriptRanges[i];
                formattedScriptRange = {};
                formattedScriptRange.start = sourceMapping.originalPositionToFormattedLocation(scriptRange.start.position);
                formattedScriptRange.end = sourceMapping.originalPositionToFormattedLocation(scriptRange.end.position);
                formattedScriptRange.sourceID = scriptRange.sourceID;
                formattedScriptRanges.push(formattedScriptRange);
            }
            callback(new WebInspector.SourceFrameContent(result.text, sourceMapping, formattedScriptRanges));
        }
        this._formatChunks(chunks, 0, didFormatChunks.bind(this));
    },

    _splitContentIntoChunks: function(text, scriptRanges)
    {
        var chunks = [];
        function addChunk(start, end, isScript)
        {
            var chunk = {};
            chunk.start = start;
            chunk.end = end;
            chunk.isScript = isScript;
            chunk.text = text.substring(start, end);
            chunks.push(chunk);
        }
        var currentPosition = 0;
        for (var i = 0; i < scriptRanges.length; ++i) {
            var start = scriptRanges[i].start.position;
            var end = scriptRanges[i].end.position;
            if (currentPosition < start)
                addChunk(currentPosition, start, false);
            addChunk(start, end, true);
            currentPosition = end;
        }
        if (currentPosition < text.length)
            addChunk(currentPosition, text.length, false);
        return chunks;
    },

    _formatChunks: function(chunks, index, callback)
    {
        while(true) {
            if (index === chunks.length) {
                callback();
                return;
            }
            var chunk = chunks[index++];
            if (chunk.isScript)
                break;
        }

        function didFormat(formattedSource, mapping)
        {
            chunk.text = formattedSource;
            chunk.mapping = mapping;
            this._formatChunks(chunks, index, callback);
        }
        this._formatScript(chunk.text, didFormat.bind(this));
    },

    _buildContentFromChunks: function(chunks)
    {
        var text = "";
        var mapping = { original: [], formatted: [] };
        for (var i = 0; i < chunks.length; ++i) {
            var chunk = chunks[i];
            mapping.original.push(chunk.start);
            mapping.formatted.push(text.length);
            if (chunk.isScript) {
                if (text)
                    text += "\n";
                for (var j = 0; j < chunk.mapping.original.length; ++j) {
                    mapping.original.push(chunk.mapping.original[j] + chunk.start);
                    mapping.formatted.push(chunk.mapping.formatted[j] + text.length);
                }
                text += chunk.text;
            } else {
                if (text)
                    text += "\n";
                text += chunk.text;
            }
            mapping.original.push(chunk.end);
            mapping.formatted.push(text.length);
        }
        return { text: text, mapping: mapping };
    },

    _formatScript: function(source, callback)
    {
        this._tasks.push({ source: source, callback: callback });
        this._worker.postMessage(source);
    },

    _handleMessage: function(event)
    {
        var task = this._tasks.shift();
        task.callback(event.data.formattedSource, event.data.mapping);
    },

    _handleError: function(event)
    {
        console.warn("Error in script formatter worker:", event);
        event.preventDefault()
        var task = this._tasks.shift();
        task.callback(task.source, { original: [], formatted: [] });
    }
}


WebInspector.SourceMappingForFormattedScript = function(originalLineEndings, formattedLineEndings, mapping)
{
    WebInspector.SourceMapping.call(this);
    this._originalLineEndings = originalLineEndings;
    this._formattedLineEndings = formattedLineEndings;
    this._mapping = mapping;
}

WebInspector.SourceMappingForFormattedScript.prototype = {
    actualLocationToSourceLocation: function(lineNumber, columnNumber)
    {
        var position = WebInspector.ScriptFormatter.locationToPosition(this._originalLineEndings, lineNumber, columnNumber);
        return this.originalPositionToFormattedLocation(position);
    },

    sourceLocationToActualLocation: function(lineNumber, columnNumber)
    {
        var formattedPosition = WebInspector.ScriptFormatter.locationToPosition(this._formattedLineEndings, lineNumber, columnNumber);
        var position = this._convertPosition(this._mapping.formatted, this._mapping.original, formattedPosition);
        return WebInspector.ScriptFormatter.positionToLocation(this._originalLineEndings, position);
    },

    originalPositionToFormattedLocation: function(position)
    {
        var formattedPosition = this._convertPosition(this._mapping.original, this._mapping.formatted, position);
        var location = WebInspector.ScriptFormatter.positionToLocation(this._formattedLineEndings, formattedPosition);
        location.position = formattedPosition;
        return location;
    },

    _convertPosition: function(positions1, positions2, position)
    {
        var index = positions1.upperBound(position);
        var range1 = positions1[index] - positions1[index - 1];
        var range2 = positions2[index] - positions2[index - 1];
        var position2 = positions2[index - 1];
        if (range1)
            position2 += Math.round((position - positions1[index - 1]) * range2 / range1);
        return position2;
    }
}

WebInspector.SourceMappingForFormattedScript.prototype.__proto__ = WebInspector.SourceMapping.prototype;