mediawiki-extensions-VisualEditor/modules/ve/ce/nodes/ve.ce.HeadingNode.js

/**
 * Creates an ve.ce.HeadingNode object.
 * 
 * @class
 * @constructor
 * @extends {ve.ce.LeafNode}
 * @param {ve.dm.HeadingNode} model Heading model to view
 */
ve.ce.HeadingNode = function( model ) {
	// Inheritance
	ve.ce.LeafNode.call( this, model, $( '<h' + model.getElementAttribute( 'level' ) + '>' ) );

	// Properties
	this.currentLevelHash = null;

	// DOM Changes
	this.$.addClass( 'es-headingView' );

	// Events
	var _this = this;
	this.model.on( 'update', function() {
		_this.setClasses();
	} );

	// Initialization
	this.setClasses();
};

/* Methods */

ve.ce.HeadingNode.prototype.setClasses = function() {
	var level = this.model.getElementAttribute( 'level' );
	if ( level !== this.currentLevelHash ) {
		this.currentLevelHash = level;
		var classes = this.$.attr( 'class' );
		this.$
			// Remove any existing level classes
			.attr( 'class', classes.replace( / ?es-headingView-level[0-9]+/, '' ) )
			// Add a new level class
			.addClass( 'es-headingView-level' + level );
	}
};

/* Registration */

ve.ce.DocumentNode.splitRules.heading = {
	'self': true,
	'children': null
};

/* Inheritance */

ve.extendClass( ve.ce.HeadingNode, ve.ce.LeafNode );