mediawiki-extensions-Visual.../modules/es/serializers/es.WikitextSerializer.js

203 lines
5.6 KiB
JavaScript

/**
* Serializes a WikiDom plain object into a Wikitext string.
*
* @class
* @constructor
* @param options {Object} List of options for serialization
*/
es.WikitextSerializer = function( options ) {
this.options = $.extend( {
// defaults
}, options || {} );
};
/* Static Methods */
/**
* Get a serialized version of data.
*
* @static
* @method
* @param {Object} data Data to serialize
* @param {Object} options Options to use, @see {es.WikitextSerializer} for details
* @returns {String} Serialized version of data
*/
es.WikitextSerializer.stringify = function( data, options ) {
return ( new es.WikitextSerializer( options ) ).document( data );
};
es.WikitextSerializer.getHtmlAttributes = function( attributes ) {
var htmlAttributes = {},
count = 0;
for ( var key in attributes ) {
if ( key.indexOf( 'html/' ) === 0 ) {
htmlAttributes[key.substr( 5 )] = attributes[key];
count++;
}
}
return count ? htmlAttributes : null;
};
/* Methods */
es.WikitextSerializer.prototype.document = function( node, rawFirstParagraph ) {
var lines = [];
for ( var i = 0, length = node.children.length; i < length; i++ ) {
var childNode = node.children[i];
if ( childNode.type in this ) {
// Special case for paragraphs which have particular spacing needs
if ( childNode.type === 'paragraph' ) {
lines.push( this.paragraph( childNode, rawFirstParagraph && i === 0 ) );
if ( i + 1 < length /* && node.children[i + 1].type === 'paragraph' */ ) {
lines.push( '' );
}
} else {
lines.push( this[childNode.type].call( this, childNode ) );
}
}
}
return lines.join( '\n' );
};
es.WikitextSerializer.prototype.comment = function( node ) {
return '<!--' + node.text + '-->';
};
es.WikitextSerializer.prototype.horizontalRule = function( node ) {
return '----';
};
es.WikitextSerializer.prototype.heading = function( node ) {
var symbols = es.repeatString( '=', node.attributes.level );
return symbols + this.content( node.content ) + symbols;
};
es.WikitextSerializer.prototype.paragraph = function( node ) {
return this.content( node.content );
};
es.WikitextSerializer.prototype.list = function( node ) {
var symbolTable = {
'bullet': '*',
'number': '#',
'term': ';',
'description': ':'
};
function convertStyles( styles ) {
var symbols = '';
for ( var i = 0, length = styles.length; i < length; i++ ) {
symbols += styles[i] in symbolTable ? symbolTable[styles[i]] : '';
}
return symbols;
}
var lines = [];
for ( var i = 0, length = node.children.length; i < length; i++ ) {
var childNode = node.children[i];
lines.push(
convertStyles( childNode.attributes.styles ) + ' ' +
this.content( childNode.content )
);
}
return lines.join( '\n' );
};
es.WikitextSerializer.prototype.table = function( node ) {
var lines = [],
attributes = es.WikitextSerializer.getHtmlAttributes( node.attributes );
if ( attributes ) {
attributes = es.Html.makeAttributeList( attributes );
}
lines.push( '{|' + attributes );
for ( var i = 0, length = node.children.length; i < length; i++ ) {
lines.push( this.tableRow( node.children[i], i === 0 ) );
}
lines.push( '|}' );
return lines.join( '\n' );
};
es.WikitextSerializer.prototype.tableRow = function( node, first ) {
var lines = [],
attributes = es.WikitextSerializer.getHtmlAttributes( node.attributes );
if ( attributes ) {
attributes = es.Html.makeAttributeList( attributes );
}
if ( !first || attributes ) {
lines.push( '|-' + attributes );
}
for ( var i = 0, length = node.children.length; i < length; i++ ) {
lines.push( this.tableCell( node.children[i] ) );
}
return lines.join( '\n' );
};
es.WikitextSerializer.prototype.tableCell = function( node ) {
var symbolTable = {
'tableHeading': '!',
'tableCell': '|'
};
var attributes = es.WikitextSerializer.getHtmlAttributes( node.attributes );
if ( attributes ) {
attributes = es.Html.makeAttributeList( attributes ) + '|';
}
return symbolTable[node.type] + attributes + this.document( node, true );
};
es.WikitextSerializer.prototype.transclusion = function( node ) {
var title = [];
if ( node.namespace === 'Main' ) {
title.push( '' );
} else if ( node.namespace !== 'Template' ) {
title.push( node.namespace );
}
title.push( node.title );
return '{{' + title.join( ':' ) + '}}';
};
es.WikitextSerializer.prototype.parameter = function( node ) {
return '{{{' + node.name + '}}}';
};
es.WikitextSerializer.prototype.content = function( node ) {
if ( 'annotations' in node && node.annotations.length ) {
var annotationSerializer = new es.AnnotationSerializer(),
tagTable = {
'textStyle/strong': 'strong',
'textStyle/emphasize': 'em',
'textStyle/big': 'big',
'textStyle/small': 'small',
'textStyle/superScript': 'sup',
'textStyle/subScript': 'sub'
},
markupTable = {
'textStyle/bold': "'''",
'textStyle/italic': "''"
};
for ( var i = 0, length = node.annotations.length; i < length; i++ ) {
var annotation = node.annotations[i];
if ( annotation.type in tagTable ) {
annotationSerializer.addTags( annotation.range, tagTable[annotation.type] );
} else if ( annotation.type in markupTable ) {
annotationSerializer.add(
annotation.range, markupTable[annotation.type], markupTable[annotation.type]
);
} else {
switch ( annotation.type ) {
case 'link/external':
annotationSerializer.add(
annotation.range, '[' + annotation.data.href + ' ', ']'
);
break;
case 'link/internal':
annotationSerializer.add(
annotation.range, '[[' + annotation.data.title + '|', ']]'
);
break;
}
}
}
return annotationSerializer.render( node.text );
} else {
return node.text;
}
};