8
0

buildviewconverter.js 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469
  1. /**
  2. * @license Copyright (c) 2003-2017, CKSource - Frederico Knabben. All rights reserved.
  3. * For licensing, see LICENSE.md.
  4. */
  5. import buildViewConverter from '../../src/conversion/buildviewconverter';
  6. import ModelSchema from '../../src/model/schema';
  7. import ModelDocument from '../../src/model/document';
  8. import ModelElement from '../../src/model/element';
  9. import ModelTextProxy from '../../src/model/textproxy';
  10. import ModelRange from '../../src/model/range';
  11. import ModelWalker from '../../src/model/treewalker';
  12. import ViewDocumentFragment from '../../src/view/documentfragment';
  13. import ViewContainerElement from '../../src/view/containerelement';
  14. import ViewAttributeElement from '../../src/view/attributeelement';
  15. import ViewText from '../../src/view/text';
  16. import ViewMatcher from '../../src/view/matcher';
  17. import ViewConversionDispatcher from '../../src/conversion/viewconversiondispatcher';
  18. import CKEditorError from '@ckeditor/ckeditor5-utils/src/ckeditorerror';
  19. import { convertToModelFragment, convertText } from '../../src/conversion/view-to-model-converters';
  20. function modelAttributesToString( item ) {
  21. let result = '';
  22. for ( let attr of item.getAttributes() ) {
  23. result += ' ' + attr[ 0 ] + '="' + attr[ 1 ] + '"';
  24. }
  25. return result;
  26. }
  27. function modelToString( item ) {
  28. let result = '';
  29. if ( item instanceof ModelTextProxy ) {
  30. let attributes = modelAttributesToString( item );
  31. result = attributes ? '<$text' + attributes + '>' + item.data + '</$text>' : item.data;
  32. } else {
  33. let walker = new ModelWalker( { boundaries: ModelRange.createIn( item ), shallow: true } );
  34. for ( let value of walker ) {
  35. result += modelToString( value.item );
  36. }
  37. if ( item instanceof ModelElement ) {
  38. let attributes = modelAttributesToString( item );
  39. result = '<' + item.name + attributes + '>' + result + '</' + item.name + '>';
  40. }
  41. }
  42. return result;
  43. }
  44. const textAttributes = [ undefined, 'linkHref', 'linkTitle', 'bold', 'italic', 'style' ];
  45. const pAttributes = [ undefined, 'class', 'important', 'theme', 'decorated', 'size' ];
  46. describe( 'View converter builder', () => {
  47. let dispatcher, modelDoc, modelRoot, schema, objWithContext;
  48. beforeEach( () => {
  49. // `additionalData` parameter for `.convert` calls.
  50. objWithContext = { context: [ '$root' ] };
  51. schema = new ModelSchema();
  52. schema.registerItem( 'paragraph', '$block' );
  53. schema.registerItem( 'div', '$block' );
  54. schema.registerItem( 'customP', 'paragraph' );
  55. schema.registerItem( 'image', '$inline' );
  56. schema.registerItem( 'span', '$inline' );
  57. schema.registerItem( 'MEGATRON', '$inline' ); // Yes, folks, we are building MEGATRON.
  58. schema.registerItem( 'abcd', '$inline' );
  59. schema.allow( { name: '$inline', attributes: textAttributes, inside: '$root' } );
  60. schema.allow( { name: 'image', attributes: [ 'src' ], inside: '$root' } );
  61. schema.allow( { name: 'image', attributes: [ 'src' ], inside: '$block' } );
  62. schema.allow( { name: '$text', inside: '$inline' } );
  63. schema.allow( { name: '$text', attributes: textAttributes, inside: '$block' } );
  64. schema.allow( { name: '$text', attributes: textAttributes, inside: '$root' } );
  65. schema.allow( { name: 'paragraph', attributes: pAttributes, inside: '$root' } );
  66. schema.allow( { name: 'span', attributes: [ 'transformer' ], inside: '$root' } );
  67. schema.allow( { name: 'div', attributes: [ 'class' ], inside: '$root' } );
  68. dispatcher = new ViewConversionDispatcher( { schema } );
  69. dispatcher.on( 'text', convertText() );
  70. modelDoc = new ModelDocument();
  71. modelRoot = modelDoc.createRoot();
  72. } );
  73. it( 'should convert from view element to model element', () => {
  74. buildViewConverter().for( dispatcher ).fromElement( 'p' ).toElement( 'paragraph' );
  75. const result = dispatcher.convert( new ViewContainerElement( 'p', null, new ViewText( 'foo' ) ), objWithContext );
  76. modelRoot.appendChildren( result );
  77. expect( modelToString( result ) ).to.equal( '<paragraph>foo</paragraph>' );
  78. } );
  79. it( 'should convert from view element to model element using creator function', () => {
  80. buildViewConverter().for( dispatcher )
  81. .fromElement( 'img' )
  82. .toElement( ( viewElement ) => new ModelElement( 'image', { src: viewElement.getAttribute( 'src' ) } ) );
  83. const result = dispatcher.convert( new ViewContainerElement( 'img', { src: 'foo.jpg' } ), objWithContext );
  84. modelRoot.appendChildren( result );
  85. expect( modelToString( result ) ).to.equal( '<image src="foo.jpg"></image>' );
  86. } );
  87. it( 'should convert from view element to model attribute', () => {
  88. buildViewConverter().for( dispatcher ).fromElement( 'strong' ).toAttribute( 'bold', true );
  89. const result = dispatcher.convert( new ViewAttributeElement( 'strong', null, new ViewText( 'foo' ) ), objWithContext );
  90. modelRoot.appendChildren( result );
  91. // Have to check root because result is a ModelText.
  92. expect( modelToString( modelRoot ) ).to.equal( '<$root><$text bold="true">foo</$text></$root>' );
  93. } );
  94. it( 'should convert from view element to model attributes using creator function', () => {
  95. buildViewConverter().for( dispatcher )
  96. .fromElement( 'a' )
  97. .toAttribute( ( viewElement ) => ( { key: 'linkHref', value: viewElement.getAttribute( 'href' ) } ) );
  98. const result = dispatcher.convert( new ViewAttributeElement( 'a', { href: 'foo.html' }, new ViewText( 'foo' ) ), objWithContext );
  99. modelRoot.appendChildren( result );
  100. // Have to check root because result is a ModelText.
  101. expect( modelToString( modelRoot ) ).to.equal( '<$root><$text linkHref="foo.html">foo</$text></$root>' );
  102. } );
  103. it( 'should convert from view attribute to model attribute', () => {
  104. buildViewConverter().for( dispatcher ).fromElement( 'p' ).toElement( 'paragraph' );
  105. buildViewConverter().for( dispatcher )
  106. .fromAttribute( 'class' )
  107. .toAttribute( ( viewElement ) => ( { key: 'class', value: viewElement.getAttribute( 'class' ) } ) );
  108. const result = dispatcher.convert( new ViewContainerElement( 'p', { class: 'myClass' }, new ViewText( 'foo' ) ), objWithContext );
  109. modelRoot.appendChildren( result );
  110. expect( modelToString( result ) ).to.equal( '<paragraph class="myClass">foo</paragraph>' );
  111. } );
  112. it( 'should convert from view attribute and key to model attribute', () => {
  113. dispatcher.on( 'documentFragment', convertToModelFragment() );
  114. buildViewConverter().for( dispatcher ).fromElement( 'p' ).toElement( 'paragraph' );
  115. buildViewConverter().for( dispatcher ).fromAttribute( 'class', 'important' ).toAttribute( 'important', true );
  116. buildViewConverter().for( dispatcher ).fromAttribute( 'class', 'theme-nice' ).toAttribute( 'theme', 'nice' );
  117. const viewStructure = new ViewDocumentFragment( [
  118. new ViewContainerElement( 'p', { class: 'important' }, new ViewText( 'foo' ) ),
  119. new ViewContainerElement( 'p', { class: 'important theme-nice' }, new ViewText( 'bar' ) )
  120. ] );
  121. const result = dispatcher.convert( viewStructure, objWithContext );
  122. expect( modelToString( result ) )
  123. .to.equal( '<paragraph important="true">foo</paragraph><paragraph important="true" theme="nice">bar</paragraph>' );
  124. } );
  125. it( 'should convert from multiple view entities to model attribute', () => {
  126. buildViewConverter().for( dispatcher ).fromElement( 'p' ).toElement( 'paragraph' );
  127. buildViewConverter().for( dispatcher )
  128. .fromElement( 'strong' )
  129. .fromElement( 'b' )
  130. .fromAttribute( 'class', 'bold' )
  131. .fromAttribute( 'style', { 'font-weight': 'bold' } )
  132. .toAttribute( 'bold', true );
  133. const viewElement = new ViewContainerElement( 'p', null, [
  134. new ViewAttributeElement( 'strong', null, new ViewText( 'aaa' ) ),
  135. new ViewAttributeElement( 'b', null, new ViewText( 'bbb' ) ),
  136. new ViewContainerElement( 'span', { class: 'bold' }, new ViewText( 'ccc' ) ),
  137. new ViewContainerElement( 'span', { style: 'font-weight:bold; font-size:20px' }, new ViewText( 'ddd' ) )
  138. ] );
  139. const result = dispatcher.convert( viewElement, objWithContext );
  140. modelRoot.appendChildren( result );
  141. expect( modelToString( result ) ).to.equal( '<paragraph><$text bold="true">aaabbbcccddd</$text></paragraph>' );
  142. } );
  143. it( 'should convert from pattern to marker', () => {
  144. buildViewConverter().for( dispatcher ).from( { attribute: { 'marker-name': 'search' } } ).toMarker();
  145. const viewElement = new ViewAttributeElement( 'span', { 'marker-name': 'search' } );
  146. const result = dispatcher.convert( viewElement, objWithContext );
  147. modelRoot.appendChildren( result );
  148. expect( modelToString( result ) ).to.equal( '<$marker marker-name="search"></$marker>' );
  149. } );
  150. it( 'should convert from element to marker using creator function', () => {
  151. buildViewConverter().for( dispatcher ).fromElement( 'marker' ).toMarker( ( data ) => {
  152. return new ModelElement( '$marker', { 'marker-name': data.getAttribute( 'class' ) } );
  153. } );
  154. const element = new ViewAttributeElement( 'marker', { class: 'search' } );
  155. const result = dispatcher.convert( element, objWithContext );
  156. modelRoot.appendChildren( result );
  157. expect( modelToString( result ) ).to.equal( '<$marker marker-name="search"></$marker>' );
  158. } );
  159. it( 'should convert from multiple view entities to marker', () => {
  160. buildViewConverter().for( dispatcher ).fromElement( 'p' ).toElement( 'paragraph' );
  161. buildViewConverter().for( dispatcher )
  162. .fromElement( 'marker' )
  163. .from( { attribute: { 'marker-name': 'search' } } )
  164. .toMarker();
  165. const viewElement = new ViewContainerElement( 'p', null, [
  166. new ViewAttributeElement( 'marker', { 'marker-name': 'comment' } ),
  167. new ViewAttributeElement( 'span', { 'marker-name': 'search' } )
  168. ] );
  169. const result = dispatcher.convert( viewElement, objWithContext );
  170. modelRoot.appendChildren( result );
  171. expect( modelToString( result ) ).to.equal(
  172. '<paragraph><$marker marker-name="comment"></$marker><$marker marker-name="search"></$marker></paragraph>'
  173. );
  174. } );
  175. it( 'should do nothing when there is no element matching to marker pattern', () => {
  176. buildViewConverter().for( dispatcher ).from( { class: 'color' } ).toMarker();
  177. const element = new ViewAttributeElement( 'span' );
  178. expect( dispatcher.convert( element, objWithContext ) ).to.null;
  179. } );
  180. it( 'should throw an error when view element in not valid to convert to marker', () => {
  181. buildViewConverter().for( dispatcher ).fromElement( 'marker' ).toMarker();
  182. const element = new ViewAttributeElement( 'marker', { class: 'search' } );
  183. expect( () => {
  184. dispatcher.convert( element, objWithContext );
  185. } ).to.throw( CKEditorError, /^build-view-converter-invalid-marker/ );
  186. } );
  187. it( 'should throw an error when model element returned by creator has not valid name', () => {
  188. buildViewConverter().for( dispatcher ).fromElement( 'marker' ).toMarker( () => {
  189. return new ModelElement( 'element', { 'marker-name': 'search' } );
  190. } );
  191. const element = new ViewAttributeElement( 'marker', { 'marker-name': 'search' } );
  192. expect( () => {
  193. dispatcher.convert( element, objWithContext );
  194. } ).to.throw( CKEditorError, /^build-view-converter-invalid-marker/ );
  195. } );
  196. it( 'should throw an error when model element returned by creator has not valid marker-name attribute', () => {
  197. buildViewConverter().for( dispatcher ).fromElement( 'marker' ).toMarker( () => {
  198. return new ModelElement( '$marker', { 'foo': 'search' } );
  199. } );
  200. const element = new ViewAttributeElement( 'marker', { 'marker-name': 'search' } );
  201. expect( () => {
  202. dispatcher.convert( element, objWithContext );
  203. } ).to.throw( CKEditorError, /^build-view-converter-invalid-marker/ );
  204. } );
  205. it( 'should convert from pattern to model element', () => {
  206. buildViewConverter().for( dispatcher ).from(
  207. { name: 'span', class: 'megatron', attribute: { head: 'megatron', body: 'megatron', legs: 'megatron' } }
  208. ).toElement( 'MEGATRON' );
  209. // Adding callbacks later so they are called later. MEGATRON callback is more important.
  210. buildViewConverter().for( dispatcher ).fromElement( 'span' ).toElement( 'span' );
  211. buildViewConverter().for( dispatcher ).fromElement( 'p' ).toElement( 'paragraph' );
  212. let result;
  213. // Not quite megatron.
  214. result = dispatcher.convert( new ViewContainerElement( 'span', { class: 'megatron' }, new ViewText( 'foo' ) ), objWithContext );
  215. modelRoot.appendChildren( result );
  216. expect( modelToString( result ) ).to.equal( '<span>foo</span>' );
  217. // Almost a megatron. Missing a head.
  218. result = dispatcher.convert(
  219. new ViewContainerElement( 'span', { class: 'megatron', body: 'megatron', legs: 'megatron' }, new ViewText( 'foo' ) ),
  220. objWithContext
  221. );
  222. modelRoot.appendChildren( result );
  223. expect( modelToString( result ) ).to.equal( '<span>foo</span>' );
  224. // This would be a megatron but is a paragraph.
  225. result = dispatcher.convert(
  226. new ViewContainerElement(
  227. 'p',
  228. { class: 'megatron', body: 'megatron', legs: 'megatron', head: 'megatron' },
  229. new ViewText( 'foo' )
  230. ),
  231. objWithContext
  232. );
  233. modelRoot.appendChildren( result );
  234. expect( modelToString( result ) ).to.equal( '<paragraph>foo</paragraph>' );
  235. // At last we have a megatron!
  236. result = dispatcher.convert(
  237. new ViewContainerElement(
  238. 'span',
  239. { class: 'megatron', body: 'megatron', legs: 'megatron', head: 'megatron' },
  240. new ViewText( 'foo' )
  241. ),
  242. objWithContext
  243. );
  244. modelRoot.appendChildren( result );
  245. expect( modelToString( result ) ).to.equal( '<MEGATRON>foo</MEGATRON>' );
  246. } );
  247. it( 'should convert from pattern to model attribute', () => {
  248. buildViewConverter().for( dispatcher ).fromElement( 'span' ).toElement( 'span' );
  249. // This time without name so default span converter will convert children.
  250. buildViewConverter().for( dispatcher )
  251. .from( { class: 'megatron', attribute: { head: 'megatron', body: 'megatron', legs: 'megatron' } } )
  252. .toAttribute( 'transformer', 'megatron' );
  253. let viewElement = new ViewContainerElement(
  254. 'span',
  255. { class: 'megatron', body: 'megatron', legs: 'megatron', head: 'megatron' },
  256. new ViewText( 'foo' )
  257. );
  258. let result = dispatcher.convert( viewElement, objWithContext );
  259. modelRoot.appendChildren( result );
  260. expect( modelToString( result ) ).to.equal( '<span transformer="megatron">foo</span>' );
  261. } );
  262. it( 'should set different priorities for `toElement` and `toAttribute` conversion', () => {
  263. buildViewConverter().for( dispatcher )
  264. .fromAttribute( 'class' )
  265. .toAttribute( ( viewElement ) => ( { key: 'class', value: viewElement.getAttribute( 'class' ) } ) );
  266. buildViewConverter().for( dispatcher ).fromElement( 'p' ).toElement( 'paragraph' );
  267. let result = dispatcher.convert( new ViewContainerElement( 'p', { class: 'myClass' }, new ViewText( 'foo' ) ), objWithContext );
  268. modelRoot.appendChildren( result );
  269. // Element converter was fired first even though attribute converter was added first.
  270. expect( modelToString( result ) ).to.equal( '<paragraph class="myClass">foo</paragraph>' );
  271. } );
  272. it( 'should overwrite default priorities for converters', () => {
  273. buildViewConverter().for( dispatcher ).fromElement( 'p' ).toElement( 'paragraph' );
  274. buildViewConverter().for( dispatcher )
  275. .fromAttribute( 'class' )
  276. .toAttribute( ( viewElement ) => ( { key: 'class', value: viewElement.getAttribute( 'class' ) } ) );
  277. let result;
  278. result = dispatcher.convert( new ViewContainerElement( 'p', { class: 'myClass' }, new ViewText( 'foo' ) ), objWithContext );
  279. modelRoot.appendChildren( result );
  280. expect( modelToString( result ) ).to.equal( '<paragraph class="myClass">foo</paragraph>' );
  281. buildViewConverter().for( dispatcher )
  282. .from( { name: 'p', class: 'myClass' } ).withPriority( 'high' )
  283. .toElement( 'customP' );
  284. result = dispatcher.convert( new ViewContainerElement( 'p', { class: 'myClass' }, new ViewText( 'foo' ) ), objWithContext );
  285. modelRoot.appendChildren( result );
  286. expect( modelToString( result ) ).to.equal( '<customP>foo</customP>' );
  287. } );
  288. it( 'should overwrite default consumed values', () => {
  289. // Converter (1).
  290. buildViewConverter().for( dispatcher ).fromElement( 'p' ).toElement( 'paragraph' );
  291. // Converter (2).
  292. buildViewConverter().for( dispatcher )
  293. .from( { name: 'p', class: 'decorated' } ).consuming( { class: 'decorated' } )
  294. .toAttribute( 'decorated', true );
  295. // Converter (3).
  296. buildViewConverter().for( dispatcher )
  297. .fromAttribute( 'class', 'small' ).consuming( { class: 'small' } )
  298. .toAttribute( 'size', 'small' );
  299. const viewElement = new ViewContainerElement( 'p', { class: 'decorated small' }, new ViewText( 'foo' ) );
  300. const result = dispatcher.convert( viewElement, objWithContext );
  301. modelRoot.appendChildren( result );
  302. // P element and it's children got converted by the converter (1) and the converter (1) got fired
  303. // because P name was not consumed in converter (2). Converter (3) could consume class="small" because
  304. // only class="decorated" was consumed in converter (2).
  305. expect( modelToString( result ) ).to.equal( '<paragraph decorated="true" size="small">foo</paragraph>' );
  306. } );
  307. it( 'should convert from matcher instance to model', () => {
  308. // Universal class converter, synonymous to .fromAttribute( 'class' ).
  309. buildViewConverter().for( dispatcher )
  310. .from( new ViewMatcher( { class: /.*/ } ) )
  311. .toAttribute( ( viewElement ) => ( { key: 'class', value: viewElement.getAttribute( 'class' ) } ) );
  312. // Universal element converter.
  313. buildViewConverter().for( dispatcher )
  314. .from( new ViewMatcher( { name: /.*/ } ) )
  315. .toElement( ( viewElement ) => new ModelElement( viewElement.name ) );
  316. let viewStructure = new ViewContainerElement( 'div', { class: 'myClass' }, [
  317. new ViewContainerElement( 'abcd', null, new ViewText( 'foo' ) )
  318. ] );
  319. let result = dispatcher.convert( viewStructure, objWithContext );
  320. modelRoot.appendChildren( result );
  321. expect( modelToString( result ) ).to.equal( '<div class="myClass"><abcd>foo</abcd></div>' );
  322. } );
  323. it( 'should filter out structure that is wrong with schema - elements', () => {
  324. buildViewConverter().for( dispatcher ).fromElement( 'div' ).toElement( 'div' );
  325. buildViewConverter().for( dispatcher ).fromElement( 'p' ).toElement( 'paragraph' );
  326. schema.disallow( { name: 'div', inside: '$root' } );
  327. dispatcher.on( 'element', convertToModelFragment(), { priority: 'lowest' } );
  328. let viewElement = new ViewContainerElement( 'div', null,
  329. new ViewContainerElement( 'p', null,
  330. new ViewText( 'foo' )
  331. )
  332. );
  333. let result = dispatcher.convert( viewElement, objWithContext );
  334. expect( modelToString( result ) ).to.equal( '<paragraph>foo</paragraph>' );
  335. } );
  336. it( 'should filter out structure that is wrong with schema - attributes', () => {
  337. buildViewConverter().for( dispatcher ).fromElement( 'p' ).toElement( 'paragraph' );
  338. buildViewConverter().for( dispatcher ).fromElement( 'strong' ).toAttribute( 'bold', true );
  339. schema.disallow( { name: '$text', attributes: 'bold', inside: 'paragraph' } );
  340. dispatcher.on( 'element', convertToModelFragment(), { priority: 'lowest' } );
  341. let viewElement = new ViewContainerElement( 'p', null,
  342. new ViewAttributeElement( 'strong', null,
  343. new ViewText( 'foo' )
  344. )
  345. );
  346. let result = dispatcher.convert( viewElement, objWithContext );
  347. expect( modelToString( result ) ).to.equal( '<paragraph>foo</paragraph>' );
  348. } );
  349. } );