MetadataStorage.js 8.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207
  1. import { TransformationType } from './enums';
  2. /**
  3. * Storage all library metadata.
  4. */
  5. export class MetadataStorage {
  6. constructor() {
  7. // -------------------------------------------------------------------------
  8. // Properties
  9. // -------------------------------------------------------------------------
  10. this._typeMetadatas = new Map();
  11. this._transformMetadatas = new Map();
  12. this._exposeMetadatas = new Map();
  13. this._excludeMetadatas = new Map();
  14. this._ancestorsMap = new Map();
  15. }
  16. // -------------------------------------------------------------------------
  17. // Adder Methods
  18. // -------------------------------------------------------------------------
  19. addTypeMetadata(metadata) {
  20. if (!this._typeMetadatas.has(metadata.target)) {
  21. this._typeMetadatas.set(metadata.target, new Map());
  22. }
  23. this._typeMetadatas.get(metadata.target).set(metadata.propertyName, metadata);
  24. }
  25. addTransformMetadata(metadata) {
  26. if (!this._transformMetadatas.has(metadata.target)) {
  27. this._transformMetadatas.set(metadata.target, new Map());
  28. }
  29. if (!this._transformMetadatas.get(metadata.target).has(metadata.propertyName)) {
  30. this._transformMetadatas.get(metadata.target).set(metadata.propertyName, []);
  31. }
  32. this._transformMetadatas.get(metadata.target).get(metadata.propertyName).push(metadata);
  33. }
  34. addExposeMetadata(metadata) {
  35. if (!this._exposeMetadatas.has(metadata.target)) {
  36. this._exposeMetadatas.set(metadata.target, new Map());
  37. }
  38. this._exposeMetadatas.get(metadata.target).set(metadata.propertyName, metadata);
  39. }
  40. addExcludeMetadata(metadata) {
  41. if (!this._excludeMetadatas.has(metadata.target)) {
  42. this._excludeMetadatas.set(metadata.target, new Map());
  43. }
  44. this._excludeMetadatas.get(metadata.target).set(metadata.propertyName, metadata);
  45. }
  46. // -------------------------------------------------------------------------
  47. // Public Methods
  48. // -------------------------------------------------------------------------
  49. findTransformMetadatas(target, propertyName, transformationType) {
  50. return this.findMetadatas(this._transformMetadatas, target, propertyName).filter(metadata => {
  51. if (!metadata.options)
  52. return true;
  53. if (metadata.options.toClassOnly === true && metadata.options.toPlainOnly === true)
  54. return true;
  55. if (metadata.options.toClassOnly === true) {
  56. return (transformationType === TransformationType.CLASS_TO_CLASS ||
  57. transformationType === TransformationType.PLAIN_TO_CLASS);
  58. }
  59. if (metadata.options.toPlainOnly === true) {
  60. return transformationType === TransformationType.CLASS_TO_PLAIN;
  61. }
  62. return true;
  63. });
  64. }
  65. findExcludeMetadata(target, propertyName) {
  66. return this.findMetadata(this._excludeMetadatas, target, propertyName);
  67. }
  68. findExposeMetadata(target, propertyName) {
  69. return this.findMetadata(this._exposeMetadatas, target, propertyName);
  70. }
  71. findExposeMetadataByCustomName(target, name) {
  72. return this.getExposedMetadatas(target).find(metadata => {
  73. return metadata.options && metadata.options.name === name;
  74. });
  75. }
  76. findTypeMetadata(target, propertyName) {
  77. return this.findMetadata(this._typeMetadatas, target, propertyName);
  78. }
  79. getStrategy(target) {
  80. const excludeMap = this._excludeMetadatas.get(target);
  81. const exclude = excludeMap && excludeMap.get(undefined);
  82. const exposeMap = this._exposeMetadatas.get(target);
  83. const expose = exposeMap && exposeMap.get(undefined);
  84. if ((exclude && expose) || (!exclude && !expose))
  85. return 'none';
  86. return exclude ? 'excludeAll' : 'exposeAll';
  87. }
  88. getExposedMetadatas(target) {
  89. return this.getMetadata(this._exposeMetadatas, target);
  90. }
  91. getExcludedMetadatas(target) {
  92. return this.getMetadata(this._excludeMetadatas, target);
  93. }
  94. getExposedProperties(target, transformationType) {
  95. return this.getExposedMetadatas(target)
  96. .filter(metadata => {
  97. if (!metadata.options)
  98. return true;
  99. if (metadata.options.toClassOnly === true && metadata.options.toPlainOnly === true)
  100. return true;
  101. if (metadata.options.toClassOnly === true) {
  102. return (transformationType === TransformationType.CLASS_TO_CLASS ||
  103. transformationType === TransformationType.PLAIN_TO_CLASS);
  104. }
  105. if (metadata.options.toPlainOnly === true) {
  106. return transformationType === TransformationType.CLASS_TO_PLAIN;
  107. }
  108. return true;
  109. })
  110. .map(metadata => metadata.propertyName);
  111. }
  112. getExcludedProperties(target, transformationType) {
  113. return this.getExcludedMetadatas(target)
  114. .filter(metadata => {
  115. if (!metadata.options)
  116. return true;
  117. if (metadata.options.toClassOnly === true && metadata.options.toPlainOnly === true)
  118. return true;
  119. if (metadata.options.toClassOnly === true) {
  120. return (transformationType === TransformationType.CLASS_TO_CLASS ||
  121. transformationType === TransformationType.PLAIN_TO_CLASS);
  122. }
  123. if (metadata.options.toPlainOnly === true) {
  124. return transformationType === TransformationType.CLASS_TO_PLAIN;
  125. }
  126. return true;
  127. })
  128. .map(metadata => metadata.propertyName);
  129. }
  130. clear() {
  131. this._typeMetadatas.clear();
  132. this._exposeMetadatas.clear();
  133. this._excludeMetadatas.clear();
  134. this._ancestorsMap.clear();
  135. }
  136. // -------------------------------------------------------------------------
  137. // Private Methods
  138. // -------------------------------------------------------------------------
  139. getMetadata(metadatas, target) {
  140. const metadataFromTargetMap = metadatas.get(target);
  141. let metadataFromTarget;
  142. if (metadataFromTargetMap) {
  143. metadataFromTarget = Array.from(metadataFromTargetMap.values()).filter(meta => meta.propertyName !== undefined);
  144. }
  145. const metadataFromAncestors = [];
  146. for (const ancestor of this.getAncestors(target)) {
  147. const ancestorMetadataMap = metadatas.get(ancestor);
  148. if (ancestorMetadataMap) {
  149. const metadataFromAncestor = Array.from(ancestorMetadataMap.values()).filter(meta => meta.propertyName !== undefined);
  150. metadataFromAncestors.push(...metadataFromAncestor);
  151. }
  152. }
  153. return metadataFromAncestors.concat(metadataFromTarget || []);
  154. }
  155. findMetadata(metadatas, target, propertyName) {
  156. const metadataFromTargetMap = metadatas.get(target);
  157. if (metadataFromTargetMap) {
  158. const metadataFromTarget = metadataFromTargetMap.get(propertyName);
  159. if (metadataFromTarget) {
  160. return metadataFromTarget;
  161. }
  162. }
  163. for (const ancestor of this.getAncestors(target)) {
  164. const ancestorMetadataMap = metadatas.get(ancestor);
  165. if (ancestorMetadataMap) {
  166. const ancestorResult = ancestorMetadataMap.get(propertyName);
  167. if (ancestorResult) {
  168. return ancestorResult;
  169. }
  170. }
  171. }
  172. return undefined;
  173. }
  174. findMetadatas(metadatas, target, propertyName) {
  175. const metadataFromTargetMap = metadatas.get(target);
  176. let metadataFromTarget;
  177. if (metadataFromTargetMap) {
  178. metadataFromTarget = metadataFromTargetMap.get(propertyName);
  179. }
  180. const metadataFromAncestorsTarget = [];
  181. for (const ancestor of this.getAncestors(target)) {
  182. const ancestorMetadataMap = metadatas.get(ancestor);
  183. if (ancestorMetadataMap) {
  184. if (ancestorMetadataMap.has(propertyName)) {
  185. metadataFromAncestorsTarget.push(...ancestorMetadataMap.get(propertyName));
  186. }
  187. }
  188. }
  189. return metadataFromAncestorsTarget
  190. .slice()
  191. .reverse()
  192. .concat((metadataFromTarget || []).slice().reverse());
  193. }
  194. getAncestors(target) {
  195. if (!target)
  196. return [];
  197. if (!this._ancestorsMap.has(target)) {
  198. const ancestors = [];
  199. for (let baseClass = Object.getPrototypeOf(target.prototype.constructor); typeof baseClass.prototype !== 'undefined'; baseClass = Object.getPrototypeOf(baseClass.prototype.constructor)) {
  200. ancestors.push(baseClass);
  201. }
  202. this._ancestorsMap.set(target, ancestors);
  203. }
  204. return this._ancestorsMap.get(target);
  205. }
  206. }
  207. //# sourceMappingURL=MetadataStorage.js.map