8
0

buildviewconverter.js 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473
  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: { 'data-name': 'search' } } ).toMarker();
  145. const viewElement = new ViewAttributeElement( 'span', { 'data-name': 'search' } );
  146. const result = dispatcher.convert( viewElement, objWithContext );
  147. modelRoot.appendChildren( result );
  148. expect( modelToString( result ) ).to.equal( '<$marker data-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', { 'data-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 data-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. .from( { attribute: { 'foo': 'marker' } } )
  163. .from( { attribute: { 'bar': 'marker' } } )
  164. .from( { attribute: { 'foo': 'marker', 'bar': 'marker' } } )
  165. .toMarker();
  166. const viewElement = new ViewContainerElement( 'p', null, [
  167. new ViewAttributeElement( 'span', { 'foo': 'marker', 'data-name': 'marker1' } ),
  168. new ViewAttributeElement( 'span', { 'bar': 'marker', 'data-name': 'marker2' } ),
  169. new ViewAttributeElement( 'span', { 'foo': 'marker', 'bar': 'marker', 'data-name': 'marker3' } )
  170. ] );
  171. const result = dispatcher.convert( viewElement, objWithContext );
  172. modelRoot.appendChildren( result );
  173. expect( modelToString( result ) ).to.equal(
  174. '<paragraph>' +
  175. '<$marker data-name="marker1"></$marker><$marker data-name="marker2"></$marker><$marker data-name="marker3"></$marker>' +
  176. '</paragraph>'
  177. );
  178. } );
  179. it( 'should do nothing when there is no element matching to marker pattern', () => {
  180. buildViewConverter().for( dispatcher ).from( { class: 'color' } ).toMarker();
  181. const element = new ViewAttributeElement( 'span' );
  182. expect( dispatcher.convert( element, objWithContext ) ).to.null;
  183. } );
  184. it( 'should throw an error when view element in not valid to convert to marker', () => {
  185. buildViewConverter().for( dispatcher ).fromElement( 'marker' ).toMarker();
  186. const element = new ViewAttributeElement( 'marker', { class: 'search' } );
  187. expect( () => {
  188. dispatcher.convert( element, objWithContext );
  189. } ).to.throw( CKEditorError, /^build-view-converter-invalid-marker/ );
  190. } );
  191. it( 'should throw an error when model element returned by creator has not valid name', () => {
  192. buildViewConverter().for( dispatcher ).fromElement( 'marker' ).toMarker( () => {
  193. return new ModelElement( 'element', { 'data-name': 'search' } );
  194. } );
  195. const element = new ViewAttributeElement( 'marker', { 'data-name': 'search' } );
  196. expect( () => {
  197. dispatcher.convert( element, objWithContext );
  198. } ).to.throw( CKEditorError, /^build-view-converter-invalid-marker/ );
  199. } );
  200. it( 'should throw an error when model element returned by creator has not valid data-name attribute', () => {
  201. buildViewConverter().for( dispatcher ).fromElement( 'marker' ).toMarker( () => {
  202. return new ModelElement( '$marker', { 'foo': 'search' } );
  203. } );
  204. const element = new ViewAttributeElement( 'marker', { 'data-name': 'search' } );
  205. expect( () => {
  206. dispatcher.convert( element, objWithContext );
  207. } ).to.throw( CKEditorError, /^build-view-converter-invalid-marker/ );
  208. } );
  209. it( 'should convert from pattern to model element', () => {
  210. buildViewConverter().for( dispatcher ).from(
  211. { name: 'span', class: 'megatron', attribute: { head: 'megatron', body: 'megatron', legs: 'megatron' } }
  212. ).toElement( 'MEGATRON' );
  213. // Adding callbacks later so they are called later. MEGATRON callback is more important.
  214. buildViewConverter().for( dispatcher ).fromElement( 'span' ).toElement( 'span' );
  215. buildViewConverter().for( dispatcher ).fromElement( 'p' ).toElement( 'paragraph' );
  216. let result;
  217. // Not quite megatron.
  218. result = dispatcher.convert( new ViewContainerElement( 'span', { class: 'megatron' }, new ViewText( 'foo' ) ), objWithContext );
  219. modelRoot.appendChildren( result );
  220. expect( modelToString( result ) ).to.equal( '<span>foo</span>' );
  221. // Almost a megatron. Missing a head.
  222. result = dispatcher.convert(
  223. new ViewContainerElement( 'span', { class: 'megatron', body: 'megatron', legs: 'megatron' }, new ViewText( 'foo' ) ),
  224. objWithContext
  225. );
  226. modelRoot.appendChildren( result );
  227. expect( modelToString( result ) ).to.equal( '<span>foo</span>' );
  228. // This would be a megatron but is a paragraph.
  229. result = dispatcher.convert(
  230. new ViewContainerElement(
  231. 'p',
  232. { class: 'megatron', body: 'megatron', legs: 'megatron', head: 'megatron' },
  233. new ViewText( 'foo' )
  234. ),
  235. objWithContext
  236. );
  237. modelRoot.appendChildren( result );
  238. expect( modelToString( result ) ).to.equal( '<paragraph>foo</paragraph>' );
  239. // At last we have a megatron!
  240. result = dispatcher.convert(
  241. new ViewContainerElement(
  242. 'span',
  243. { class: 'megatron', body: 'megatron', legs: 'megatron', head: 'megatron' },
  244. new ViewText( 'foo' )
  245. ),
  246. objWithContext
  247. );
  248. modelRoot.appendChildren( result );
  249. expect( modelToString( result ) ).to.equal( '<MEGATRON>foo</MEGATRON>' );
  250. } );
  251. it( 'should convert from pattern to model attribute', () => {
  252. buildViewConverter().for( dispatcher ).fromElement( 'span' ).toElement( 'span' );
  253. // This time without name so default span converter will convert children.
  254. buildViewConverter().for( dispatcher )
  255. .from( { class: 'megatron', attribute: { head: 'megatron', body: 'megatron', legs: 'megatron' } } )
  256. .toAttribute( 'transformer', 'megatron' );
  257. let viewElement = new ViewContainerElement(
  258. 'span',
  259. { class: 'megatron', body: 'megatron', legs: 'megatron', head: 'megatron' },
  260. new ViewText( 'foo' )
  261. );
  262. let result = dispatcher.convert( viewElement, objWithContext );
  263. modelRoot.appendChildren( result );
  264. expect( modelToString( result ) ).to.equal( '<span transformer="megatron">foo</span>' );
  265. } );
  266. it( 'should set different priorities for `toElement` and `toAttribute` conversion', () => {
  267. buildViewConverter().for( dispatcher )
  268. .fromAttribute( 'class' )
  269. .toAttribute( ( viewElement ) => ( { key: 'class', value: viewElement.getAttribute( 'class' ) } ) );
  270. buildViewConverter().for( dispatcher ).fromElement( 'p' ).toElement( 'paragraph' );
  271. let result = dispatcher.convert( new ViewContainerElement( 'p', { class: 'myClass' }, new ViewText( 'foo' ) ), objWithContext );
  272. modelRoot.appendChildren( result );
  273. // Element converter was fired first even though attribute converter was added first.
  274. expect( modelToString( result ) ).to.equal( '<paragraph class="myClass">foo</paragraph>' );
  275. } );
  276. it( 'should overwrite default priorities for converters', () => {
  277. buildViewConverter().for( dispatcher ).fromElement( 'p' ).toElement( 'paragraph' );
  278. buildViewConverter().for( dispatcher )
  279. .fromAttribute( 'class' )
  280. .toAttribute( ( viewElement ) => ( { key: 'class', value: viewElement.getAttribute( 'class' ) } ) );
  281. let result;
  282. result = dispatcher.convert( new ViewContainerElement( 'p', { class: 'myClass' }, new ViewText( 'foo' ) ), objWithContext );
  283. modelRoot.appendChildren( result );
  284. expect( modelToString( result ) ).to.equal( '<paragraph class="myClass">foo</paragraph>' );
  285. buildViewConverter().for( dispatcher )
  286. .from( { name: 'p', class: 'myClass' } ).withPriority( 'high' )
  287. .toElement( 'customP' );
  288. result = dispatcher.convert( new ViewContainerElement( 'p', { class: 'myClass' }, new ViewText( 'foo' ) ), objWithContext );
  289. modelRoot.appendChildren( result );
  290. expect( modelToString( result ) ).to.equal( '<customP>foo</customP>' );
  291. } );
  292. it( 'should overwrite default consumed values', () => {
  293. // Converter (1).
  294. buildViewConverter().for( dispatcher ).fromElement( 'p' ).toElement( 'paragraph' );
  295. // Converter (2).
  296. buildViewConverter().for( dispatcher )
  297. .from( { name: 'p', class: 'decorated' } ).consuming( { class: 'decorated' } )
  298. .toAttribute( 'decorated', true );
  299. // Converter (3).
  300. buildViewConverter().for( dispatcher )
  301. .fromAttribute( 'class', 'small' ).consuming( { class: 'small' } )
  302. .toAttribute( 'size', 'small' );
  303. const viewElement = new ViewContainerElement( 'p', { class: 'decorated small' }, new ViewText( 'foo' ) );
  304. const result = dispatcher.convert( viewElement, objWithContext );
  305. modelRoot.appendChildren( result );
  306. // P element and it's children got converted by the converter (1) and the converter (1) got fired
  307. // because P name was not consumed in converter (2). Converter (3) could consume class="small" because
  308. // only class="decorated" was consumed in converter (2).
  309. expect( modelToString( result ) ).to.equal( '<paragraph decorated="true" size="small">foo</paragraph>' );
  310. } );
  311. it( 'should convert from matcher instance to model', () => {
  312. // Universal class converter, synonymous to .fromAttribute( 'class' ).
  313. buildViewConverter().for( dispatcher )
  314. .from( new ViewMatcher( { class: /.*/ } ) )
  315. .toAttribute( ( viewElement ) => ( { key: 'class', value: viewElement.getAttribute( 'class' ) } ) );
  316. // Universal element converter.
  317. buildViewConverter().for( dispatcher )
  318. .from( new ViewMatcher( { name: /.*/ } ) )
  319. .toElement( ( viewElement ) => new ModelElement( viewElement.name ) );
  320. let viewStructure = new ViewContainerElement( 'div', { class: 'myClass' }, [
  321. new ViewContainerElement( 'abcd', null, new ViewText( 'foo' ) )
  322. ] );
  323. let result = dispatcher.convert( viewStructure, objWithContext );
  324. modelRoot.appendChildren( result );
  325. expect( modelToString( result ) ).to.equal( '<div class="myClass"><abcd>foo</abcd></div>' );
  326. } );
  327. it( 'should filter out structure that is wrong with schema - elements', () => {
  328. buildViewConverter().for( dispatcher ).fromElement( 'div' ).toElement( 'div' );
  329. buildViewConverter().for( dispatcher ).fromElement( 'p' ).toElement( 'paragraph' );
  330. schema.disallow( { name: 'div', inside: '$root' } );
  331. dispatcher.on( 'element', convertToModelFragment(), { priority: 'lowest' } );
  332. let viewElement = new ViewContainerElement( 'div', null,
  333. new ViewContainerElement( 'p', null,
  334. new ViewText( 'foo' )
  335. )
  336. );
  337. let result = dispatcher.convert( viewElement, objWithContext );
  338. expect( modelToString( result ) ).to.equal( '<paragraph>foo</paragraph>' );
  339. } );
  340. it( 'should filter out structure that is wrong with schema - attributes', () => {
  341. buildViewConverter().for( dispatcher ).fromElement( 'p' ).toElement( 'paragraph' );
  342. buildViewConverter().for( dispatcher ).fromElement( 'strong' ).toAttribute( 'bold', true );
  343. schema.disallow( { name: '$text', attributes: 'bold', inside: 'paragraph' } );
  344. dispatcher.on( 'element', convertToModelFragment(), { priority: 'lowest' } );
  345. let viewElement = new ViewContainerElement( 'p', null,
  346. new ViewAttributeElement( 'strong', null,
  347. new ViewText( 'foo' )
  348. )
  349. );
  350. let result = dispatcher.convert( viewElement, objWithContext );
  351. expect( modelToString( result ) ).to.equal( '<paragraph>foo</paragraph>' );
  352. } );
  353. } );