viewconsumable.js 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601
  1. /**
  2. * @license Copyright (c) 2003-2019, CKSource - Frederico Knabben. All rights reserved.
  3. * For licensing, see LICENSE.md or https://ckeditor.com/legal/ckeditor-oss-license
  4. */
  5. /**
  6. * @module engine/conversion/viewconsumable
  7. */
  8. import { isArray } from 'lodash-es';
  9. import CKEditorError from '@ckeditor/ckeditor5-utils/src/ckeditorerror';
  10. /**
  11. * Class used for handling consumption of view {@link module:engine/view/element~Element elements},
  12. * {@link module:engine/view/text~Text text nodes} and {@link module:engine/view/documentfragment~DocumentFragment document fragments}.
  13. * Element's name and its parts (attributes, classes and styles) can be consumed separately. Consuming an element's name
  14. * does not consume its attributes, classes and styles.
  15. * To add items for consumption use {@link module:engine/conversion/viewconsumable~ViewConsumable#add add method}.
  16. * To test items use {@link module:engine/conversion/viewconsumable~ViewConsumable#test test method}.
  17. * To consume items use {@link module:engine/conversion/viewconsumable~ViewConsumable#consume consume method}.
  18. * To revert already consumed items use {@link module:engine/conversion/viewconsumable~ViewConsumable#revert revert method}.
  19. *
  20. * viewConsumable.add( element, { name: true } ); // Adds element's name as ready to be consumed.
  21. * viewConsumable.add( textNode ); // Adds text node for consumption.
  22. * viewConsumable.add( docFragment ); // Adds document fragment for consumption.
  23. * viewConsumable.test( element, { name: true } ); // Tests if element's name can be consumed.
  24. * viewConsumable.test( textNode ); // Tests if text node can be consumed.
  25. * viewConsumable.test( docFragment ); // Tests if document fragment can be consumed.
  26. * viewConsumable.consume( element, { name: true } ); // Consume element's name.
  27. * viewConsumable.consume( textNode ); // Consume text node.
  28. * viewConsumable.consume( docFragment ); // Consume document fragment.
  29. * viewConsumable.revert( element, { name: true } ); // Revert already consumed element's name.
  30. * viewConsumable.revert( textNode ); // Revert already consumed text node.
  31. * viewConsumable.revert( docFragment ); // Revert already consumed document fragment.
  32. */
  33. export default class ViewConsumable {
  34. /**
  35. * Creates new ViewConsumable.
  36. */
  37. constructor() {
  38. /**
  39. * Map of consumable elements. If {@link module:engine/view/element~Element element} is used as a key,
  40. * {@link module:engine/conversion/viewconsumable~ViewElementConsumables ViewElementConsumables} instance is stored as value.
  41. * For {@link module:engine/view/text~Text text nodes} and
  42. * {@link module:engine/view/documentfragment~DocumentFragment document fragments} boolean value is stored as value.
  43. *
  44. * @protected
  45. * @member {Map.<module:engine/conversion/viewconsumable~ViewElementConsumables|Boolean>}
  46. */
  47. this._consumables = new Map();
  48. }
  49. /**
  50. * Adds {@link module:engine/view/element~Element view element}, {@link module:engine/view/text~Text text node} or
  51. * {@link module:engine/view/documentfragment~DocumentFragment document fragment} as ready to be consumed.
  52. *
  53. * viewConsumable.add( p, { name: true } ); // Adds element's name to consume.
  54. * viewConsumable.add( p, { attributes: 'name' } ); // Adds element's attribute.
  55. * viewConsumable.add( p, { classes: 'foobar' } ); // Adds element's class.
  56. * viewConsumable.add( p, { styles: 'color' } ); // Adds element's style
  57. * viewConsumable.add( p, { attributes: 'name', styles: 'color' } ); // Adds attribute and style.
  58. * viewConsumable.add( p, { classes: [ 'baz', 'bar' ] } ); // Multiple consumables can be provided.
  59. * viewConsumable.add( textNode ); // Adds text node to consume.
  60. * viewConsumable.add( docFragment ); // Adds document fragment to consume.
  61. *
  62. * Throws {@link module:utils/ckeditorerror~CKEditorError CKEditorError} `viewconsumable-invalid-attribute` when `class` or `style`
  63. * attribute is provided - it should be handled separately by providing actual style/class.
  64. *
  65. * viewConsumable.add( p, { attributes: 'style' } ); // This call will throw an exception.
  66. * viewConsumable.add( p, { styles: 'color' } ); // This is properly handled style.
  67. *
  68. * @param {module:engine/view/element~Element|module:engine/view/text~Text|module:engine/view/documentfragment~DocumentFragment} element
  69. * @param {Object} [consumables] Used only if first parameter is {@link module:engine/view/element~Element view element} instance.
  70. * @param {Boolean} consumables.name If set to true element's name will be included.
  71. * @param {String|Array.<String>} consumables.attributes Attribute name or array of attribute names.
  72. * @param {String|Array.<String>} consumables.classes Class name or array of class names.
  73. * @param {String|Array.<String>} consumables.styles Style name or array of style names.
  74. */
  75. add( element, consumables ) {
  76. let elementConsumables;
  77. // For text nodes and document fragments just mark them as consumable.
  78. if ( element.is( 'text' ) || element.is( 'documentFragment' ) ) {
  79. this._consumables.set( element, true );
  80. return;
  81. }
  82. // For elements create new ViewElementConsumables or update already existing one.
  83. if ( !this._consumables.has( element ) ) {
  84. elementConsumables = new ViewElementConsumables();
  85. this._consumables.set( element, elementConsumables );
  86. } else {
  87. elementConsumables = this._consumables.get( element );
  88. }
  89. elementConsumables.add( consumables );
  90. }
  91. /**
  92. * Tests if {@link module:engine/view/element~Element view element}, {@link module:engine/view/text~Text text node} or
  93. * {@link module:engine/view/documentfragment~DocumentFragment document fragment} can be consumed.
  94. * It returns `true` when all items included in method's call can be consumed. Returns `false` when
  95. * first already consumed item is found and `null` when first non-consumable item is found.
  96. *
  97. * viewConsumable.test( p, { name: true } ); // Tests element's name.
  98. * viewConsumable.test( p, { attributes: 'name' } ); // Tests attribute.
  99. * viewConsumable.test( p, { classes: 'foobar' } ); // Tests class.
  100. * viewConsumable.test( p, { styles: 'color' } ); // Tests style.
  101. * viewConsumable.test( p, { attributes: 'name', styles: 'color' } ); // Tests attribute and style.
  102. * viewConsumable.test( p, { classes: [ 'baz', 'bar' ] } ); // Multiple consumables can be tested.
  103. * viewConsumable.test( textNode ); // Tests text node.
  104. * viewConsumable.test( docFragment ); // Tests document fragment.
  105. *
  106. * Testing classes and styles as attribute will test if all added classes/styles can be consumed.
  107. *
  108. * viewConsumable.test( p, { attributes: 'class' } ); // Tests if all added classes can be consumed.
  109. * viewConsumable.test( p, { attributes: 'style' } ); // Tests if all added styles can be consumed.
  110. *
  111. * @param {module:engine/view/element~Element|module:engine/view/text~Text|module:engine/view/documentfragment~DocumentFragment} element
  112. * @param {Object} [consumables] Used only if first parameter is {@link module:engine/view/element~Element view element} instance.
  113. * @param {Boolean} consumables.name If set to true element's name will be included.
  114. * @param {String|Array.<String>} consumables.attributes Attribute name or array of attribute names.
  115. * @param {String|Array.<String>} consumables.classes Class name or array of class names.
  116. * @param {String|Array.<String>} consumables.styles Style name or array of style names.
  117. * @returns {Boolean|null} Returns `true` when all items included in method's call can be consumed. Returns `false`
  118. * when first already consumed item is found and `null` when first non-consumable item is found.
  119. */
  120. test( element, consumables ) {
  121. const elementConsumables = this._consumables.get( element );
  122. if ( elementConsumables === undefined ) {
  123. return null;
  124. }
  125. // For text nodes and document fragments return stored boolean value.
  126. if ( element.is( 'text' ) || element.is( 'documentFragment' ) ) {
  127. return elementConsumables;
  128. }
  129. // For elements test consumables object.
  130. return elementConsumables.test( consumables );
  131. }
  132. /**
  133. * Consumes {@link module:engine/view/element~Element view element}, {@link module:engine/view/text~Text text node} or
  134. * {@link module:engine/view/documentfragment~DocumentFragment document fragment}.
  135. * It returns `true` when all items included in method's call can be consumed, otherwise returns `false`.
  136. *
  137. * viewConsumable.consume( p, { name: true } ); // Consumes element's name.
  138. * viewConsumable.consume( p, { attributes: 'name' } ); // Consumes element's attribute.
  139. * viewConsumable.consume( p, { classes: 'foobar' } ); // Consumes element's class.
  140. * viewConsumable.consume( p, { styles: 'color' } ); // Consumes element's style.
  141. * viewConsumable.consume( p, { attributes: 'name', styles: 'color' } ); // Consumes attribute and style.
  142. * viewConsumable.consume( p, { classes: [ 'baz', 'bar' ] } ); // Multiple consumables can be consumed.
  143. * viewConsumable.consume( textNode ); // Consumes text node.
  144. * viewConsumable.consume( docFragment ); // Consumes document fragment.
  145. *
  146. * Consuming classes and styles as attribute will test if all added classes/styles can be consumed.
  147. *
  148. * viewConsumable.consume( p, { attributes: 'class' } ); // Consume only if all added classes can be consumed.
  149. * viewConsumable.consume( p, { attributes: 'style' } ); // Consume only if all added styles can be consumed.
  150. *
  151. * @param {module:engine/view/element~Element|module:engine/view/text~Text|module:engine/view/documentfragment~DocumentFragment} element
  152. * @param {Object} [consumables] Used only if first parameter is {@link module:engine/view/element~Element view element} instance.
  153. * @param {Boolean} consumables.name If set to true element's name will be included.
  154. * @param {String|Array.<String>} consumables.attributes Attribute name or array of attribute names.
  155. * @param {String|Array.<String>} consumables.classes Class name or array of class names.
  156. * @param {String|Array.<String>} consumables.styles Style name or array of style names.
  157. * @returns {Boolean} Returns `true` when all items included in method's call can be consumed,
  158. * otherwise returns `false`.
  159. */
  160. consume( element, consumables ) {
  161. if ( this.test( element, consumables ) ) {
  162. if ( element.is( 'text' ) || element.is( 'documentFragment' ) ) {
  163. // For text nodes and document fragments set value to false.
  164. this._consumables.set( element, false );
  165. } else {
  166. // For elements - consume consumables object.
  167. this._consumables.get( element ).consume( consumables );
  168. }
  169. return true;
  170. }
  171. return false;
  172. }
  173. /**
  174. * Reverts {@link module:engine/view/element~Element view element}, {@link module:engine/view/text~Text text node} or
  175. * {@link module:engine/view/documentfragment~DocumentFragment document fragment} so they can be consumed once again.
  176. * Method does not revert items that were never previously added for consumption, even if they are included in
  177. * method's call.
  178. *
  179. * viewConsumable.revert( p, { name: true } ); // Reverts element's name.
  180. * viewConsumable.revert( p, { attributes: 'name' } ); // Reverts element's attribute.
  181. * viewConsumable.revert( p, { classes: 'foobar' } ); // Reverts element's class.
  182. * viewConsumable.revert( p, { styles: 'color' } ); // Reverts element's style.
  183. * viewConsumable.revert( p, { attributes: 'name', styles: 'color' } ); // Reverts attribute and style.
  184. * viewConsumable.revert( p, { classes: [ 'baz', 'bar' ] } ); // Multiple names can be reverted.
  185. * viewConsumable.revert( textNode ); // Reverts text node.
  186. * viewConsumable.revert( docFragment ); // Reverts document fragment.
  187. *
  188. * Reverting classes and styles as attribute will revert all classes/styles that were previously added for
  189. * consumption.
  190. *
  191. * viewConsumable.revert( p, { attributes: 'class' } ); // Reverts all classes added for consumption.
  192. * viewConsumable.revert( p, { attributes: 'style' } ); // Reverts all styles added for consumption.
  193. *
  194. * @param {module:engine/view/element~Element|module:engine/view/text~Text|module:engine/view/documentfragment~DocumentFragment} element
  195. * @param {Object} [consumables] Used only if first parameter is {@link module:engine/view/element~Element view element} instance.
  196. * @param {Boolean} consumables.name If set to true element's name will be included.
  197. * @param {String|Array.<String>} consumables.attributes Attribute name or array of attribute names.
  198. * @param {String|Array.<String>} consumables.classes Class name or array of class names.
  199. * @param {String|Array.<String>} consumables.styles Style name or array of style names.
  200. */
  201. revert( element, consumables ) {
  202. const elementConsumables = this._consumables.get( element );
  203. if ( elementConsumables !== undefined ) {
  204. if ( element.is( 'text' ) || element.is( 'documentFragment' ) ) {
  205. // For text nodes and document fragments - set consumable to true.
  206. this._consumables.set( element, true );
  207. } else {
  208. // For elements - revert items from consumables object.
  209. elementConsumables.revert( consumables );
  210. }
  211. }
  212. }
  213. /**
  214. * Creates consumable object from {@link module:engine/view/element~Element view element}. Consumable object will include
  215. * element's name and all its attributes, classes and styles.
  216. *
  217. * @static
  218. * @param {module:engine/view/element~Element} element
  219. * @returns {Object} consumables
  220. */
  221. static consumablesFromElement( element ) {
  222. const consumables = {
  223. name: true,
  224. attributes: [],
  225. classes: [],
  226. styles: []
  227. };
  228. const attributes = element.getAttributeKeys();
  229. for ( const attribute of attributes ) {
  230. // Skip classes and styles - will be added separately.
  231. if ( attribute == 'style' || attribute == 'class' ) {
  232. continue;
  233. }
  234. consumables.attributes.push( attribute );
  235. }
  236. const classes = element.getClassNames();
  237. for ( const className of classes ) {
  238. consumables.classes.push( className );
  239. }
  240. const styles = element.getStyleNames();
  241. for ( const style of styles ) {
  242. consumables.styles.push( style );
  243. }
  244. return consumables;
  245. }
  246. /**
  247. * Creates {@link module:engine/conversion/viewconsumable~ViewConsumable ViewConsumable} instance from
  248. * {@link module:engine/view/node~Node node} or {@link module:engine/view/documentfragment~DocumentFragment document fragment}.
  249. * Instance will contain all elements, child nodes, attributes, styles and classes added for consumption.
  250. *
  251. * @static
  252. * @param {module:engine/view/node~Node|module:engine/view/documentfragment~DocumentFragment} from View node or document fragment
  253. * from which `ViewConsumable` will be created.
  254. * @param {module:engine/conversion/viewconsumable~ViewConsumable} [instance] If provided, given `ViewConsumable` instance will be used
  255. * to add all consumables. It will be returned instead of a new instance.
  256. */
  257. static createFrom( from, instance ) {
  258. if ( !instance ) {
  259. instance = new ViewConsumable();
  260. }
  261. if ( from.is( 'text' ) ) {
  262. instance.add( from );
  263. return instance;
  264. }
  265. // Add `from` itself, if it is an element.
  266. if ( from.is( 'element' ) ) {
  267. instance.add( from, ViewConsumable.consumablesFromElement( from ) );
  268. }
  269. if ( from.is( 'documentFragment' ) ) {
  270. instance.add( from );
  271. }
  272. for ( const child of from.getChildren() ) {
  273. instance = ViewConsumable.createFrom( child, instance );
  274. }
  275. return instance;
  276. }
  277. }
  278. /**
  279. * This is a private helper-class for {@link module:engine/conversion/viewconsumable~ViewConsumable}.
  280. * It represents and manipulates consumable parts of a single {@link module:engine/view/element~Element}.
  281. *
  282. * @private
  283. */
  284. class ViewElementConsumables {
  285. /**
  286. * Creates ViewElementConsumables instance.
  287. */
  288. constructor() {
  289. /**
  290. * Flag indicating if name of the element can be consumed.
  291. *
  292. * @private
  293. * @member {Boolean}
  294. */
  295. this._canConsumeName = null;
  296. /**
  297. * Contains maps of element's consumables: attributes, classes and styles.
  298. *
  299. * @private
  300. * @member {Object}
  301. */
  302. this._consumables = {
  303. attributes: new Map(),
  304. styles: new Map(),
  305. classes: new Map()
  306. };
  307. }
  308. /**
  309. * Adds consumable parts of the {@link module:engine/view/element~Element view element}.
  310. * Element's name itself can be marked to be consumed (when element's name is consumed its attributes, classes and
  311. * styles still could be consumed):
  312. *
  313. * consumables.add( { name: true } );
  314. *
  315. * Attributes classes and styles:
  316. *
  317. * consumables.add( { attributes: 'title', classes: 'foo', styles: 'color' } );
  318. * consumables.add( { attributes: [ 'title', 'name' ], classes: [ 'foo', 'bar' ] );
  319. *
  320. * Throws {@link module:utils/ckeditorerror~CKEditorError CKEditorError} `viewconsumable-invalid-attribute` when `class` or `style`
  321. * attribute is provided - it should be handled separately by providing `style` and `class` in consumables object.
  322. *
  323. * @param {Object} consumables Object describing which parts of the element can be consumed.
  324. * @param {Boolean} consumables.name If set to `true` element's name will be added as consumable.
  325. * @param {String|Array.<String>} consumables.attributes Attribute name or array of attribute names to add as consumable.
  326. * @param {String|Array.<String>} consumables.classes Class name or array of class names to add as consumable.
  327. * @param {String|Array.<String>} consumables.styles Style name or array of style names to add as consumable.
  328. */
  329. add( consumables ) {
  330. if ( consumables.name ) {
  331. this._canConsumeName = true;
  332. }
  333. for ( const type in this._consumables ) {
  334. if ( type in consumables ) {
  335. this._add( type, consumables[ type ] );
  336. }
  337. }
  338. }
  339. /**
  340. * Tests if parts of the {@link module:engine/view/node~Node view node} can be consumed.
  341. *
  342. * Element's name can be tested:
  343. *
  344. * consumables.test( { name: true } );
  345. *
  346. * Attributes classes and styles:
  347. *
  348. * consumables.test( { attributes: 'title', classes: 'foo', styles: 'color' } );
  349. * consumables.test( { attributes: [ 'title', 'name' ], classes: [ 'foo', 'bar' ] );
  350. *
  351. * @param {Object} consumables Object describing which parts of the element should be tested.
  352. * @param {Boolean} consumables.name If set to `true` element's name will be tested.
  353. * @param {String|Array.<String>} consumables.attributes Attribute name or array of attribute names to test.
  354. * @param {String|Array.<String>} consumables.classes Class name or array of class names to test.
  355. * @param {String|Array.<String>} consumables.styles Style name or array of style names to test.
  356. * @returns {Boolean|null} `true` when all tested items can be consumed, `null` when even one of the items
  357. * was never marked for consumption and `false` when even one of the items was already consumed.
  358. */
  359. test( consumables ) {
  360. // Check if name can be consumed.
  361. if ( consumables.name && !this._canConsumeName ) {
  362. return this._canConsumeName;
  363. }
  364. for ( const type in this._consumables ) {
  365. if ( type in consumables ) {
  366. const value = this._test( type, consumables[ type ] );
  367. if ( value !== true ) {
  368. return value;
  369. }
  370. }
  371. }
  372. // Return true only if all can be consumed.
  373. return true;
  374. }
  375. /**
  376. * Consumes parts of {@link module:engine/view/element~Element view element}. This function does not check if consumable item
  377. * is already consumed - it consumes all consumable items provided.
  378. * Element's name can be consumed:
  379. *
  380. * consumables.consume( { name: true } );
  381. *
  382. * Attributes classes and styles:
  383. *
  384. * consumables.consume( { attributes: 'title', classes: 'foo', styles: 'color' } );
  385. * consumables.consume( { attributes: [ 'title', 'name' ], classes: [ 'foo', 'bar' ] );
  386. *
  387. * @param {Object} consumables Object describing which parts of the element should be consumed.
  388. * @param {Boolean} consumables.name If set to `true` element's name will be consumed.
  389. * @param {String|Array.<String>} consumables.attributes Attribute name or array of attribute names to consume.
  390. * @param {String|Array.<String>} consumables.classes Class name or array of class names to consume.
  391. * @param {String|Array.<String>} consumables.styles Style name or array of style names to consume.
  392. */
  393. consume( consumables ) {
  394. if ( consumables.name ) {
  395. this._canConsumeName = false;
  396. }
  397. for ( const type in this._consumables ) {
  398. if ( type in consumables ) {
  399. this._consume( type, consumables[ type ] );
  400. }
  401. }
  402. }
  403. /**
  404. * Revert already consumed parts of {@link module:engine/view/element~Element view Element}, so they can be consumed once again.
  405. * Element's name can be reverted:
  406. *
  407. * consumables.revert( { name: true } );
  408. *
  409. * Attributes classes and styles:
  410. *
  411. * consumables.revert( { attributes: 'title', classes: 'foo', styles: 'color' } );
  412. * consumables.revert( { attributes: [ 'title', 'name' ], classes: [ 'foo', 'bar' ] );
  413. *
  414. * @param {Object} consumables Object describing which parts of the element should be reverted.
  415. * @param {Boolean} consumables.name If set to `true` element's name will be reverted.
  416. * @param {String|Array.<String>} consumables.attributes Attribute name or array of attribute names to revert.
  417. * @param {String|Array.<String>} consumables.classes Class name or array of class names to revert.
  418. * @param {String|Array.<String>} consumables.styles Style name or array of style names to revert.
  419. */
  420. revert( consumables ) {
  421. if ( consumables.name ) {
  422. this._canConsumeName = true;
  423. }
  424. for ( const type in this._consumables ) {
  425. if ( type in consumables ) {
  426. this._revert( type, consumables[ type ] );
  427. }
  428. }
  429. }
  430. /**
  431. * Helper method that adds consumables of a given type: attribute, class or style.
  432. *
  433. * Throws {@link module:utils/ckeditorerror~CKEditorError CKEditorError} `viewconsumable-invalid-attribute` when `class` or `style`
  434. * type is provided - it should be handled separately by providing actual style/class type.
  435. *
  436. * @private
  437. * @param {String} type Type of the consumable item: `attributes`, `classes` or `styles`.
  438. * @param {String|Array.<String>} item Consumable item or array of items.
  439. */
  440. _add( type, item ) {
  441. const items = isArray( item ) ? item : [ item ];
  442. const consumables = this._consumables[ type ];
  443. for ( const name of items ) {
  444. if ( type === 'attributes' && ( name === 'class' || name === 'style' ) ) {
  445. /**
  446. * Class and style attributes should be handled separately in
  447. * {@link module:engine/conversion/viewconsumable~ViewConsumable#add `ViewConsumable#add()`}.
  448. *
  449. * What you have done is trying to use:
  450. *
  451. * consumables.add( { attributes: [ 'class', 'style' ] } );
  452. *
  453. * While each class and style should be registered separately:
  454. *
  455. * consumables.add( { classes: 'some-class', styles: 'font-weight' } );
  456. *
  457. * @error viewconsumable-invalid-attribute
  458. */
  459. throw new CKEditorError( 'viewconsumable-invalid-attribute: Classes and styles should be handled separately.', this );
  460. }
  461. consumables.set( name, true );
  462. }
  463. }
  464. /**
  465. * Helper method that tests consumables of a given type: attribute, class or style.
  466. *
  467. * @private
  468. * @param {String} type Type of the consumable item: `attributes`, `classes` or `styles`.
  469. * @param {String|Array.<String>} item Consumable item or array of items.
  470. * @returns {Boolean|null} Returns `true` if all items can be consumed, `null` when one of the items cannot be
  471. * consumed and `false` when one of the items is already consumed.
  472. */
  473. _test( type, item ) {
  474. const items = isArray( item ) ? item : [ item ];
  475. const consumables = this._consumables[ type ];
  476. for ( const name of items ) {
  477. if ( type === 'attributes' && ( name === 'class' || name === 'style' ) ) {
  478. const consumableName = name == 'class' ? 'classes' : 'styles';
  479. // Check all classes/styles if class/style attribute is tested.
  480. const value = this._test( consumableName, [ ...this._consumables[ consumableName ].keys() ] );
  481. if ( value !== true ) {
  482. return value;
  483. }
  484. } else {
  485. const value = consumables.get( name );
  486. // Return null if attribute is not found.
  487. if ( value === undefined ) {
  488. return null;
  489. }
  490. if ( !value ) {
  491. return false;
  492. }
  493. }
  494. }
  495. return true;
  496. }
  497. /**
  498. * Helper method that consumes items of a given type: attribute, class or style.
  499. *
  500. * @private
  501. * @param {String} type Type of the consumable item: `attributes`, `classes` or `styles`.
  502. * @param {String|Array.<String>} item Consumable item or array of items.
  503. */
  504. _consume( type, item ) {
  505. const items = isArray( item ) ? item : [ item ];
  506. const consumables = this._consumables[ type ];
  507. for ( const name of items ) {
  508. if ( type === 'attributes' && ( name === 'class' || name === 'style' ) ) {
  509. const consumableName = name == 'class' ? 'classes' : 'styles';
  510. // If class or style is provided for consumption - consume them all.
  511. this._consume( consumableName, [ ...this._consumables[ consumableName ].keys() ] );
  512. } else {
  513. consumables.set( name, false );
  514. }
  515. }
  516. }
  517. /**
  518. * Helper method that reverts items of a given type: attribute, class or style.
  519. *
  520. * @private
  521. * @param {String} type Type of the consumable item: `attributes`, `classes` or , `styles`.
  522. * @param {String|Array.<String>} item Consumable item or array of items.
  523. */
  524. _revert( type, item ) {
  525. const items = isArray( item ) ? item : [ item ];
  526. const consumables = this._consumables[ type ];
  527. for ( const name of items ) {
  528. if ( type === 'attributes' && ( name === 'class' || name === 'style' ) ) {
  529. const consumableName = name == 'class' ? 'classes' : 'styles';
  530. // If class or style is provided for reverting - revert them all.
  531. this._revert( consumableName, [ ...this._consumables[ consumableName ].keys() ] );
  532. } else {
  533. const value = consumables.get( name );
  534. if ( value === false ) {
  535. consumables.set( name, true );
  536. }
  537. }
  538. }
  539. }
  540. }