mediawiki-extensions-VisualEditor/modules/parser/mediawiki.HTML5TreeBuilder.node.js

/* Front-end/Wrapper for a particular tree builder, in this case the
 * parser/tree builder from the node 'html5' module. Feed it tokens using
 * processToken, and it will build you a DOM tree retrievable using .document
 * or .body(). */

var events = require('events'),
	HTML5 = require('./html5/index');

FauxHTML5 = {};


FauxHTML5.TreeBuilder = function ( ) {
	// The parser we are going to emit our tokens to
	this.parser = new HTML5.Parser();

	// Sets up the parser
	this.parser.parse(this);

	// implicitly start a new document
	this.processToken({type: 'TAG', name: 'body'});
};

// Inherit from EventEmitter
FauxHTML5.TreeBuilder.prototype = new events.EventEmitter();
FauxHTML5.TreeBuilder.prototype.constructor = FauxHTML5.TreeBuilder;

/**
 * Register for (token) 'chunk' and 'end' events from a token emitter,
 * normally the TokenTransformDispatcher.
 */
FauxHTML5.TreeBuilder.prototype.listenForTokensFrom = function ( emitter ) {
	emitter.addListener('chunk', this.onChunk.bind( this ) );
	emitter.addListener('end', this.onEnd.bind( this ) );
};

FauxHTML5.TreeBuilder.prototype.onChunk = function ( tokens ) {
	//console.log( 'chunk: ' + JSON.stringify( tokens, null, 2 ) );
	for (var i = 0, length = tokens.length; i < length; i++) {
		this.processToken(tokens[i]);
	}
};

FauxHTML5.TreeBuilder.prototype.onEnd = function ( ) {
	//console.log('Fauxhtml5 onEnd');
	// FIXME HACK: For some reason the end token is not processed sometimes,
	// which normally fixes the body reference up.
	var document = this.parser.document;
	document.body = document.getElementsByTagName('body')[0];

	//console.log( 'onEnd: ' + document.body.innerHTML );

	this.emit( 'document', document );

	// XXX: more clean up to allow reuse.
	this.parser.setup();
	this.processToken({type: 'TAG', name: 'body'});
};

FauxHTML5.TreeBuilder.prototype._att = function (maybeAttribs) {
	var atts = [];
	if ( maybeAttribs && $.isArray( maybeAttribs ) ) {
		for(var i = 0, length = maybeAttribs.length; i < length; i++) {
			var att = maybeAttribs[i];
			atts.push({nodeName: att.k, nodeValue: att.v});
		}
	}
	return atts;
};

// Adapt the token format to internal HTML tree builder format, call the actual
// html tree builder by emitting the token.
FauxHTML5.TreeBuilder.prototype.processToken = function (token) {

	if ( token.constructor === String ) {
		this.emit('token', {type: 'Characters', data: token});
	} else {
		switch (token.type) {
			case "TAG":
				this.emit('token', {type: 'StartTag', 
					name: token.name, 
					data: this._att(token.attribs)});
				break;
			case "ENDTAG":
				this.emit('token', {type: 'EndTag', 
					name: token.name, 
					data: this._att(token.attribs)});
				break;
			case "SELFCLOSINGTAG":
				this.emit('token', {type: 'StartTag', 
					name: token.name, 
					data: this._att(token.attribs)});
				if ( HTML5.VOID_ELEMENTS.indexOf( token.name.toLowerCase() ) < 0 ) {
					// VOID_ELEMENTS are automagically treated as self-closing by
					// the tree builder
					this.emit('token', {type: 'EndTag', 
						name: token.name, 
						data: this._att(token.attribs)});
				}
				break;
			case "COMMENT":
				this.emit('token', {type: 'Comment', 
					data: token.value});
				break;
			case "END":
				this.emit('end');
				this.emit('token', { type: 'EOF' } );
				this.document = this.parser.document;
				if ( ! this.document.body ) {
					// HACK: This should not be needed really.
					this.document.body = this.parser.document.getElementsByTagName('body')[0];
				}
				// Emit the document to consumers
				//this.emit('document', this.document);
				break;
			case "NEWLINE":
				//this.emit('end');
				//this.emit('token', {type: 'Characters', data: "\n"});
				break;
			default:
				console.log("Unhandled token: " + JSON.stringify(token));
				break;
		}
	}
};



if (typeof module == "object") {
	module.exports.FauxHTML5 = FauxHTML5;
}