equivalent-to.js 8.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303
  1. import {
  2. AST_Array,
  3. AST_Atom,
  4. AST_Await,
  5. AST_BigInt,
  6. AST_Binary,
  7. AST_Block,
  8. AST_Call,
  9. AST_Catch,
  10. AST_Chain,
  11. AST_Class,
  12. AST_ClassProperty,
  13. AST_ClassPrivateProperty,
  14. AST_ConciseMethod,
  15. AST_Conditional,
  16. AST_Debugger,
  17. AST_Definitions,
  18. AST_Destructuring,
  19. AST_Directive,
  20. AST_Do,
  21. AST_Dot,
  22. AST_DotHash,
  23. AST_EmptyStatement,
  24. AST_Expansion,
  25. AST_Export,
  26. AST_Finally,
  27. AST_For,
  28. AST_ForIn,
  29. AST_ForOf,
  30. AST_If,
  31. AST_Import,
  32. AST_ImportMeta,
  33. AST_Jump,
  34. AST_LabeledStatement,
  35. AST_Lambda,
  36. AST_LoopControl,
  37. AST_NameMapping,
  38. AST_NewTarget,
  39. AST_Node,
  40. AST_Number,
  41. AST_Object,
  42. AST_ObjectGetter,
  43. AST_ObjectKeyVal,
  44. AST_ObjectProperty,
  45. AST_ObjectSetter,
  46. AST_PrefixedTemplateString,
  47. AST_PrivateIn,
  48. AST_PrivateMethod,
  49. AST_PropAccess,
  50. AST_RegExp,
  51. AST_Sequence,
  52. AST_SimpleStatement,
  53. AST_String,
  54. AST_Super,
  55. AST_Switch,
  56. AST_SwitchBranch,
  57. AST_Symbol,
  58. AST_TemplateSegment,
  59. AST_TemplateString,
  60. AST_This,
  61. AST_Toplevel,
  62. AST_Try,
  63. AST_Unary,
  64. AST_VarDef,
  65. AST_While,
  66. AST_With,
  67. AST_Yield
  68. } from "./ast.js";
  69. const shallow_cmp = (node1, node2) => {
  70. return (
  71. node1 === null && node2 === null
  72. || node1.TYPE === node2.TYPE && node1.shallow_cmp(node2)
  73. );
  74. };
  75. export const equivalent_to = (tree1, tree2) => {
  76. if (!shallow_cmp(tree1, tree2)) return false;
  77. const walk_1_state = [tree1];
  78. const walk_2_state = [tree2];
  79. const walk_1_push = walk_1_state.push.bind(walk_1_state);
  80. const walk_2_push = walk_2_state.push.bind(walk_2_state);
  81. while (walk_1_state.length && walk_2_state.length) {
  82. const node_1 = walk_1_state.pop();
  83. const node_2 = walk_2_state.pop();
  84. if (!shallow_cmp(node_1, node_2)) return false;
  85. node_1._children_backwards(walk_1_push);
  86. node_2._children_backwards(walk_2_push);
  87. if (walk_1_state.length !== walk_2_state.length) {
  88. // Different number of children
  89. return false;
  90. }
  91. }
  92. return walk_1_state.length == 0 && walk_2_state.length == 0;
  93. };
  94. const pass_through = () => true;
  95. AST_Node.prototype.shallow_cmp = function () {
  96. throw new Error("did not find a shallow_cmp function for " + this.constructor.name);
  97. };
  98. AST_Debugger.prototype.shallow_cmp = pass_through;
  99. AST_Directive.prototype.shallow_cmp = function(other) {
  100. return this.value === other.value;
  101. };
  102. AST_SimpleStatement.prototype.shallow_cmp = pass_through;
  103. AST_Block.prototype.shallow_cmp = pass_through;
  104. AST_EmptyStatement.prototype.shallow_cmp = pass_through;
  105. AST_LabeledStatement.prototype.shallow_cmp = function(other) {
  106. return this.label.name === other.label.name;
  107. };
  108. AST_Do.prototype.shallow_cmp = pass_through;
  109. AST_While.prototype.shallow_cmp = pass_through;
  110. AST_For.prototype.shallow_cmp = function(other) {
  111. return (this.init == null ? other.init == null : this.init === other.init) && (this.condition == null ? other.condition == null : this.condition === other.condition) && (this.step == null ? other.step == null : this.step === other.step);
  112. };
  113. AST_ForIn.prototype.shallow_cmp = pass_through;
  114. AST_ForOf.prototype.shallow_cmp = pass_through;
  115. AST_With.prototype.shallow_cmp = pass_through;
  116. AST_Toplevel.prototype.shallow_cmp = pass_through;
  117. AST_Expansion.prototype.shallow_cmp = pass_through;
  118. AST_Lambda.prototype.shallow_cmp = function(other) {
  119. return this.is_generator === other.is_generator && this.async === other.async;
  120. };
  121. AST_Destructuring.prototype.shallow_cmp = function(other) {
  122. return this.is_array === other.is_array;
  123. };
  124. AST_PrefixedTemplateString.prototype.shallow_cmp = pass_through;
  125. AST_TemplateString.prototype.shallow_cmp = pass_through;
  126. AST_TemplateSegment.prototype.shallow_cmp = function(other) {
  127. return this.value === other.value;
  128. };
  129. AST_Jump.prototype.shallow_cmp = pass_through;
  130. AST_LoopControl.prototype.shallow_cmp = pass_through;
  131. AST_Await.prototype.shallow_cmp = pass_through;
  132. AST_Yield.prototype.shallow_cmp = function(other) {
  133. return this.is_star === other.is_star;
  134. };
  135. AST_If.prototype.shallow_cmp = function(other) {
  136. return this.alternative == null ? other.alternative == null : this.alternative === other.alternative;
  137. };
  138. AST_Switch.prototype.shallow_cmp = pass_through;
  139. AST_SwitchBranch.prototype.shallow_cmp = pass_through;
  140. AST_Try.prototype.shallow_cmp = function(other) {
  141. return (this.body === other.body) && (this.bcatch == null ? other.bcatch == null : this.bcatch === other.bcatch) && (this.bfinally == null ? other.bfinally == null : this.bfinally === other.bfinally);
  142. };
  143. AST_Catch.prototype.shallow_cmp = function(other) {
  144. return this.argname == null ? other.argname == null : this.argname === other.argname;
  145. };
  146. AST_Finally.prototype.shallow_cmp = pass_through;
  147. AST_Definitions.prototype.shallow_cmp = pass_through;
  148. AST_VarDef.prototype.shallow_cmp = function(other) {
  149. return this.value == null ? other.value == null : this.value === other.value;
  150. };
  151. AST_NameMapping.prototype.shallow_cmp = pass_through;
  152. AST_Import.prototype.shallow_cmp = function(other) {
  153. return (this.imported_name == null ? other.imported_name == null : this.imported_name === other.imported_name) && (this.imported_names == null ? other.imported_names == null : this.imported_names === other.imported_names) && (this.attributes == null ? other.attributes == null : this.attributes === other.attributes);
  154. };
  155. AST_ImportMeta.prototype.shallow_cmp = pass_through;
  156. AST_Export.prototype.shallow_cmp = function(other) {
  157. return (this.exported_definition == null ? other.exported_definition == null : this.exported_definition === other.exported_definition) && (this.exported_value == null ? other.exported_value == null : this.exported_value === other.exported_value) && (this.exported_names == null ? other.exported_names == null : this.exported_names === other.exported_names) && (this.attributes == null ? other.attributes == null : this.attributes === other.attributes) && this.module_name === other.module_name && this.is_default === other.is_default;
  158. };
  159. AST_Call.prototype.shallow_cmp = pass_through;
  160. AST_Sequence.prototype.shallow_cmp = pass_through;
  161. AST_PropAccess.prototype.shallow_cmp = pass_through;
  162. AST_Chain.prototype.shallow_cmp = pass_through;
  163. AST_Dot.prototype.shallow_cmp = function(other) {
  164. return this.property === other.property;
  165. };
  166. AST_DotHash.prototype.shallow_cmp = function(other) {
  167. return this.property === other.property;
  168. };
  169. AST_Unary.prototype.shallow_cmp = function(other) {
  170. return this.operator === other.operator;
  171. };
  172. AST_Binary.prototype.shallow_cmp = function(other) {
  173. return this.operator === other.operator;
  174. };
  175. AST_PrivateIn.prototype.shallow_cmp = pass_through;
  176. AST_Conditional.prototype.shallow_cmp = pass_through;
  177. AST_Array.prototype.shallow_cmp = pass_through;
  178. AST_Object.prototype.shallow_cmp = pass_through;
  179. AST_ObjectProperty.prototype.shallow_cmp = pass_through;
  180. AST_ObjectKeyVal.prototype.shallow_cmp = function(other) {
  181. return this.key === other.key && this.quote === other.quote;
  182. };
  183. AST_ObjectSetter.prototype.shallow_cmp = function(other) {
  184. return this.static === other.static;
  185. };
  186. AST_ObjectGetter.prototype.shallow_cmp = function(other) {
  187. return this.static === other.static;
  188. };
  189. AST_ConciseMethod.prototype.shallow_cmp = function(other) {
  190. return this.static === other.static;
  191. };
  192. AST_PrivateMethod.prototype.shallow_cmp = function(other) {
  193. return this.static === other.static;
  194. };
  195. AST_Class.prototype.shallow_cmp = function(other) {
  196. return (this.name == null ? other.name == null : this.name === other.name) && (this.extends == null ? other.extends == null : this.extends === other.extends);
  197. };
  198. AST_ClassProperty.prototype.shallow_cmp = function(other) {
  199. return this.static === other.static
  200. && (typeof this.key === "string"
  201. ? this.key === other.key
  202. : true /* AST_Node handled elsewhere */);
  203. };
  204. AST_ClassPrivateProperty.prototype.shallow_cmp = function(other) {
  205. return this.static === other.static;
  206. };
  207. AST_Symbol.prototype.shallow_cmp = function(other) {
  208. return this.name === other.name;
  209. };
  210. AST_NewTarget.prototype.shallow_cmp = pass_through;
  211. AST_This.prototype.shallow_cmp = pass_through;
  212. AST_Super.prototype.shallow_cmp = pass_through;
  213. AST_String.prototype.shallow_cmp = function(other) {
  214. return this.value === other.value;
  215. };
  216. AST_Number.prototype.shallow_cmp = function(other) {
  217. return this.value === other.value;
  218. };
  219. AST_BigInt.prototype.shallow_cmp = function(other) {
  220. return this.value === other.value;
  221. };
  222. AST_RegExp.prototype.shallow_cmp = function (other) {
  223. return (
  224. this.value.flags === other.value.flags
  225. && this.value.source === other.value.source
  226. );
  227. };
  228. AST_Atom.prototype.shallow_cmp = pass_through;