schema.js 9.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373
  1. /**
  2. * @license Copyright (c) 2003-2017, CKSource - Frederico Knabben. All rights reserved.
  3. * For licensing, see LICENSE.md.
  4. */
  5. import CKEditorError from '@ckeditor/ckeditor5-utils/src/ckeditorerror';
  6. import ObservableMixin from '@ckeditor/ckeditor5-utils/src/observablemixin';
  7. import mix from '@ckeditor/ckeditor5-utils/src/mix';
  8. /**
  9. * @module engine/model/schema
  10. */
  11. /**
  12. * @mixes module:utils/emittermixin~ObservableMixin
  13. */
  14. export default class Schema {
  15. constructor() {
  16. this._sourceRules = {};
  17. // TODO events
  18. this.decorate( 'checkChild' );
  19. this.decorate( 'checkAttribute' );
  20. this.on( 'checkAttribute', ( evt, args ) => {
  21. args[ 0 ] = normalizeContext( args[ 0 ] );
  22. }, { priority: 'highest' } );
  23. this.on( 'checkChild', ( evt, args ) => {
  24. args[ 0 ] = normalizeContext( args[ 0 ] );
  25. }, { priority: 'highest' } );
  26. }
  27. register( itemName, rules ) {
  28. if ( this._sourceRules[ itemName ] ) {
  29. // TODO docs
  30. throw new CKEditorError( 'schema-cannot-register-item-twice: A single item cannot be registered twice in the schema.' );
  31. }
  32. this._sourceRules[ itemName ] = [
  33. Object.assign( {}, rules )
  34. ];
  35. this._clearCache();
  36. }
  37. extend( itemName, rules ) {
  38. // TODO it should not throw if we want to allow e.g. adding attrs before element is registered
  39. // (which may be done by another feature).
  40. if ( !this._sourceRules[ itemName ] ) {
  41. // TODO docs
  42. throw new CKEditorError( 'schema-cannot-extend-missing-item: Cannot extend an item which was not registered yet.' );
  43. }
  44. this._sourceRules[ itemName ].push( Object.assign( {}, rules ) );
  45. this._clearCache();
  46. }
  47. getRules() {
  48. if ( !this._compiledRules ) {
  49. this._compile();
  50. }
  51. return this._compiledRules;
  52. }
  53. /**
  54. * @param {module:engine/model/node~Node|String} item
  55. */
  56. getRule( item ) {
  57. let itemName;
  58. if ( typeof item == 'string' ) {
  59. itemName = item;
  60. } else if ( item.is && item.is( 'text' ) ) {
  61. itemName = '$text';
  62. }
  63. // Element or context item.
  64. else {
  65. itemName = item.name;
  66. }
  67. return this.getRules()[ itemName ];
  68. }
  69. isRegistered( itemName ) {
  70. return !!this.getRule( itemName );
  71. }
  72. isBlock( itemName ) {
  73. const rule = this.getRule( itemName );
  74. return !!( rule && rule.isBlock );
  75. }
  76. isLimit( itemName ) {
  77. const rule = this.getRule( itemName );
  78. return !!( rule && rule.isLimit );
  79. }
  80. isObject( itemName ) {
  81. const rule = this.getRule( itemName );
  82. return !!( rule && rule.isObject );
  83. }
  84. /**
  85. * @param {module:engine/model/element~Element|module:engine/model/position~Position|Array.<String>} context
  86. * @param {module:engine/model/node~Node|String}
  87. */
  88. checkChild( context, child ) {
  89. const rule = this.getRule( child );
  90. if ( !rule ) {
  91. return false;
  92. }
  93. return this._checkContextMatch( rule, context );
  94. }
  95. /**
  96. * @param {module:engine/model/node~Node} context
  97. * @param {String}
  98. */
  99. checkAttribute( context, attributeName ) {
  100. const rule = this.getRule( context[ context.length - 1 ] );
  101. if ( !rule ) {
  102. return false;
  103. }
  104. return rule.allowAttributes.includes( attributeName );
  105. }
  106. /**
  107. * Returns the lowest {@link module:engine/model/schema~Schema#isLimit limit element} containing the entire
  108. * selection or the root otherwise.
  109. *
  110. * @param {module:engine/model/selection~Selection} selection Selection which returns the common ancestor.
  111. * @returns {module:engine/model/element~Element}
  112. */
  113. getLimitElement( selection ) {
  114. // Find the common ancestor for all selection's ranges.
  115. let element = Array.from( selection.getRanges() )
  116. .reduce( ( node, range ) => {
  117. if ( !node ) {
  118. return range.getCommonAncestor();
  119. }
  120. return node.getCommonAncestor( range.getCommonAncestor() );
  121. }, null );
  122. while ( !this.isLimit( element.name ) ) {
  123. if ( element.parent ) {
  124. element = element.parent;
  125. } else {
  126. break;
  127. }
  128. }
  129. return element;
  130. }
  131. removeDisallowedAttributes() {
  132. // TODO
  133. }
  134. _clearCache() {
  135. this._compiledRules = null;
  136. }
  137. _compile() {
  138. const compiledRules = {};
  139. const sourceRules = this._sourceRules;
  140. const itemNames = Object.keys( sourceRules );
  141. for ( const itemName of itemNames ) {
  142. compiledRules[ itemName ] = compileBaseItemRule( sourceRules[ itemName ], itemName );
  143. }
  144. for ( const itemName of itemNames ) {
  145. compileAllowContentOf( compiledRules, itemName );
  146. }
  147. for ( const itemName of itemNames ) {
  148. compileAllowWhere( compiledRules, itemName );
  149. }
  150. for ( const itemName of itemNames ) {
  151. compileAllowAttributesOf( compiledRules, itemName );
  152. }
  153. for ( const itemName of itemNames ) {
  154. cleanUpAllowIn( compiledRules, itemName );
  155. cleanUpAllowAttributes( compiledRules, itemName );
  156. }
  157. this._compiledRules = compiledRules;
  158. }
  159. _checkContextMatch( rule, context, contextItemIndex = context.length - 1 ) {
  160. const contextItem = context[ contextItemIndex ];
  161. if ( rule.allowIn.includes( contextItem.name ) ) {
  162. if ( contextItemIndex == 0 ) {
  163. return true;
  164. } else {
  165. const parentRule = this.getRule( contextItem );
  166. return this._checkContextMatch( parentRule, context, contextItemIndex - 1 );
  167. }
  168. } else {
  169. return false;
  170. }
  171. }
  172. }
  173. mix( Schema, ObservableMixin );
  174. function compileBaseItemRule( sourceItemRules, itemName ) {
  175. const itemRule = {
  176. name: itemName,
  177. allowIn: [],
  178. allowContentOf: [],
  179. allowWhere: [],
  180. allowAttributes: [],
  181. allowAttributesOf: []
  182. };
  183. copyTypes( sourceItemRules, itemRule );
  184. copyProperty( sourceItemRules, itemRule, 'allowIn' );
  185. copyProperty( sourceItemRules, itemRule, 'allowContentOf' );
  186. copyProperty( sourceItemRules, itemRule, 'allowWhere' );
  187. copyProperty( sourceItemRules, itemRule, 'allowAttributes' );
  188. copyProperty( sourceItemRules, itemRule, 'allowAttributesOf' );
  189. makeInheritAllWork( sourceItemRules, itemRule );
  190. return itemRule;
  191. }
  192. function compileAllowContentOf( compiledRules, itemName ) {
  193. for ( const allowContentOfItemName of compiledRules[ itemName ].allowContentOf ) {
  194. // The allowContentOf property may point to an unregistered element.
  195. if ( compiledRules[ allowContentOfItemName ] ) {
  196. const allowedChildren = getAllowedChildren( compiledRules, allowContentOfItemName );
  197. allowedChildren.forEach( allowedItem => {
  198. allowedItem.allowIn.push( itemName );
  199. } );
  200. }
  201. }
  202. delete compiledRules[ itemName ].allowContentOf;
  203. }
  204. function compileAllowWhere( compiledRules, itemName ) {
  205. for ( const allowWhereItemName of compiledRules[ itemName ].allowWhere ) {
  206. const inheritFrom = compiledRules[ allowWhereItemName ];
  207. // The allowWhere property may point to an unregistered element.
  208. if ( inheritFrom ) {
  209. const allowedIn = inheritFrom.allowIn;
  210. compiledRules[ itemName ].allowIn.push( ...allowedIn );
  211. }
  212. }
  213. delete compiledRules[ itemName ].allowWhere;
  214. }
  215. function compileAllowAttributesOf( compiledRules, itemName ) {
  216. for ( const allowAttributeOfItem of compiledRules[ itemName ].allowAttributesOf ) {
  217. const inheritFrom = compiledRules[ allowAttributeOfItem ];
  218. if ( inheritFrom ) {
  219. const inheritAttributes = inheritFrom.allowAttributes;
  220. compiledRules[ itemName ].allowAttributes.push( ...inheritAttributes );
  221. }
  222. }
  223. delete compiledRules[ itemName ].allowAttributesOf;
  224. }
  225. // Remove items which weren't registered (because it may break some checks or we'd need to complicate them).
  226. // Make sure allowIn doesn't contain repeated values.
  227. function cleanUpAllowIn( compiledRules, itemName ) {
  228. const itemRule = compiledRules[ itemName ];
  229. const existingItems = itemRule.allowIn.filter( itemToCheck => compiledRules[ itemToCheck ] );
  230. itemRule.allowIn = Array.from( new Set( existingItems ) );
  231. }
  232. function cleanUpAllowAttributes( compiledRules, itemName ) {
  233. const itemRule = compiledRules[ itemName ];
  234. itemRule.allowAttributes = Array.from( new Set( itemRule.allowAttributes ) );
  235. }
  236. function copyTypes( sourceItemRules, itemRule ) {
  237. for ( const sourceItemRule of sourceItemRules ) {
  238. const typeNames = Object.keys( sourceItemRule ).filter( name => name.startsWith( 'is' ) );
  239. for ( const name of typeNames ) {
  240. itemRule[ name ] = sourceItemRule[ name ];
  241. }
  242. }
  243. }
  244. function copyProperty( sourceItemRules, itemRule, propertyName ) {
  245. for ( const sourceItemRule of sourceItemRules ) {
  246. if ( typeof sourceItemRule[ propertyName ] == 'string' ) {
  247. itemRule[ propertyName ].push( sourceItemRule[ propertyName ] );
  248. } else if ( Array.isArray( sourceItemRule[ propertyName ] ) ) {
  249. itemRule[ propertyName ].push( ...sourceItemRule[ propertyName ] );
  250. }
  251. }
  252. }
  253. function makeInheritAllWork( sourceItemRules, itemRule ) {
  254. for ( const sourceItemRule of sourceItemRules ) {
  255. const inheritFrom = sourceItemRule.inheritAllFrom;
  256. if ( inheritFrom ) {
  257. itemRule.allowContentOf.push( inheritFrom );
  258. itemRule.allowWhere.push( inheritFrom );
  259. itemRule.allowAttributesOf.push( inheritFrom );
  260. }
  261. }
  262. }
  263. function getAllowedChildren( compiledRules, itemName ) {
  264. const itemRule = compiledRules[ itemName ];
  265. return getValues( compiledRules ).filter( rule => rule.allowIn.includes( itemRule.name ) );
  266. }
  267. function normalizeContext( ctx ) {
  268. if ( Array.isArray( ctx ) ) {
  269. return ctx.map( mapContextItem );
  270. }
  271. // Item or position (PS. It's ok that Position#getAncestors() doesn't accept params).
  272. else {
  273. return ctx.getAncestors( { includeSelf: true } ).map( mapContextItem );
  274. }
  275. }
  276. function mapContextItem( ctxItem ) {
  277. if ( typeof ctxItem == 'string' ) {
  278. return {
  279. name: ctxItem,
  280. * getAttributes() {}
  281. };
  282. } else {
  283. return {
  284. name: ctxItem.is( 'text' ) ? '$text' : ctxItem.name,
  285. * getAttributes() {
  286. yield* ctxItem.getAttributes();
  287. }
  288. };
  289. }
  290. }
  291. function getValues( obj ) {
  292. return Object.keys( obj ).map( key => obj[ key ] );
  293. }