mediawiki-extensions-VisualEditor/modules/ve2/ui/tools/ve.ui.ListButtonTool.js

/**
 * Creates an ve.ui.ListButtonTool object.
 *
 * @class
 * @constructor
 * @extends {ve.ui.ButtonTool}
 * @param {ve.ui.Toolbar} toolbar
 * @param {String} name
 */
 ve.ui.ListButtonTool = function( toolbar, name, title, data ) {
	// Inheritance
	ve.ui.ButtonTool.call( this, toolbar, name, title );

	// Properties
	this.data = data;
	this.nodes = [];
};

/* Methods */

ve.ui.ListButtonTool.prototype.list = function( nodes, style ) {
	var surface = this.toolbar.surfaceView,
		model = surface.getModel(),
		doc = model.getDocument(),
		selection = model.getSelection(),
		siblings = doc.selectNodes( selection, 'siblings'),
		outerRange = null,
		tx;

	if ( siblings.length > 1 ){
		outerRange = new ve.Range(
			siblings[0].nodeOuterRange.from,
			siblings[siblings.length-1].nodeOuterRange.to
		);
	} else {
		outerRange = ( siblings[0].node['parent'].getOuterRange() );
	}

	tx = ve.dm.Transaction.newFromWrap(
		doc,
		outerRange,
		[],
		[ { 'type': 'list', 'attributes': { 'style': style } } ],
		[],
		[ { 'type': 'listItem' } ]
	);
	model.change ( tx );
	return ;
	
	siblings = doc.selectNodes(selection, 'siblings');
	outerRange = new ve.Range(
		siblings[0].nodeOuterRange.from,
		siblings[siblings.length-1].nodeOuterRange.to
	);
	model.setSelection( outerRange );
	surface.showSelection( model.getSelection() );
};

ve.ui.ListButtonTool.prototype.unlist = function( node ){
	var surface = this.toolbar.surfaceView,
		model = surface.getModel(),
		doc = model.getDocument(),
		selection = model.getSelection(),
		siblings = doc.selectNodes( selection, 'siblings'),
		listNode,
		tx;

	if ( siblings.length === 0 ) {
		return ;
	}
	
	listNode = siblings[0].node;
	// Get the parent list node
	while ( listNode && listNode.getType() !== 'list' ) {
		listNode = listNode.getParent();
	}

	tx = ve.dm.Transaction.newFromWrap(
		doc,
		listNode.getRange(),
		[ { 'type': 'list' } ],
		[],
		[ { 'type': 'listItem' } ],
		[]
	);

	model.change( tx );

};

ve.ui.ListButtonTool.prototype.onClick = function() {
	this.toolbar.surfaceView.model.breakpoint();
	if ( !this.$.hasClass( 'es-toolbarButtonTool-down' ) ) {
		this.list( this.nodes, this.name );
	} else {
		this.unlist( this.nodes );
	}
	this.toolbar.surfaceView.model.breakpoint();
};

ve.ui.ListButtonTool.prototype.updateState = function( annotations, nodes ) {

	function areListItemsOfStyle( nodes, style ) {
		var parent;
		for ( var i=0; i < nodes.length; i++ ) {
			parent = nodes[i].getParent();
			if ( !parent || parent.getType() !== 'listItem' ) {
				return false;
			}
			if (
				parent.getType() === 'listItem' //&&
				//parent.getAttribute('styles') === style
			) {
				console.log ('styles', parent.getAttribute('styles'));
				//return true;
			}

		}
		return true;
	}
	console.log (this.name);
	if ( areListItemsOfStyle( nodes, this.name ) ) {
		this.$.addClass( 'es-toolbarButtonTool-down' );
	} else {
		this.$.removeClass( 'es-toolbarButtonTool-down' );
	}

	/*
	 * XXX: Disabled for now because lists work differently now (they are structured, not flat)
	 *
	function areListItemsOfStyle( nodes, style ) {
		var parent, styles;
		for( var i = 0; i < nodes.length; i++ ) {
			parent = nodes[i].getParent();
			if ( parent.getType() !== 'listItem' ) {
				return false;
			}
			styles = parent.getElementAttribute( 'styles' );
			if ( styles[ styles.length - 1] !== style ) {
				return false;
			}
		}
		return true;
	}

	this.nodes = nodes;

	if ( areListItemsOfStyle( this.nodes, this.name ) ) {
		this.$.addClass( 'es-toolbarButtonTool-down' );
	} else {
		this.$.removeClass( 'es-toolbarButtonTool-down' );
	}
	*/
};

/* Registration */

ve.ui.Tool.tools.number = {
	'constructor': ve.ui.ListButtonTool,
	'name': 'number',
	'title': mw.msg( 'visualeditor-listbutton-number-tooltip' ),
};

ve.ui.Tool.tools.bullet = {
	'constructor': ve.ui.ListButtonTool,
	'name': 'bullet',
	'title': mw.msg( 'visualeditor-listbutton-bullet-tooltip' ),
};

/* Inheritance */

ve.extendClass( ve.ui.ListButtonTool, ve.ui.ButtonTool );