equivalent-to.js 8.6 KB

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