equivalent-to.js 8.1 KB

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