/
array-property-collapser.js
101 lines (82 loc) · 2.64 KB
/
array-property-collapser.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
"use strict";
const Collapser = require("./collapser");
class ArrayPropertyCollapser extends Collapser {
isInitTypeValid(init) {
return init.isArrayExpression();
}
isExpressionTypeValid(expr) {
return expr.isAssignmentExpression();
}
getExpressionChecker(objName, checkReference) {
return expr => {
// checks expr is of form:
// foo[num] = rval
const left = expr.get("left");
if (!left.isMemberExpression()) {
return false;
}
const obj = left.get("object"),
prop = left.get("property");
if (!obj.isIdentifier() || obj.node.name !== objName) {
return false;
}
const checkIndex = num => Number.isInteger(num) && num >= 0;
if (
!(prop.isNumericLiteral() || prop.isStringLiteral()) ||
!checkIndex(Number(prop.node.value))
) {
return false;
}
const right = expr.get("right");
if (checkReference(right)) {
return false;
}
return true;
};
}
extractAssignment(expr) {
return [expr.node.left.property.value, expr.get("right")];
}
addSuccessfully(t, [index, rval], init) {
const elements = init.elements;
for (let i = elements.length; i <= index; i++) {
elements.push(null);
}
if (elements[index] !== null) {
return false;
}
elements[index] = rval.node;
return true;
}
isSizeSmaller({ newInit, oldInit, varDecl, assignments, statements }) {
const anyUndefined = args => args.some(a => a === undefined);
// We make an inexact calculation of how much space we save.
// It's inexact because we don't know how whitespaces will get minimized,
// and other factors.
if (
anyUndefined([
statements[statements.length - 1].node.end,
varDecl.node.end
])
) {
return false;
}
const statementsLength =
statements[statements.length - 1].node.end - varDecl.node.end;
// Approx. formula of the change in `init`'s length =
// (# commas added) + (size of all the new rvals added), where
// # commas added = (difference between the lengths of the old and new arrays)
const numCommaAdded = newInit.elements.length - oldInit.elements.length;
if (
anyUndefined(assignments.map(([, rval]) => rval.node.end)) ||
anyUndefined(assignments.map(([, rval]) => rval.node.start))
) {
return false;
}
const sizeOfRvals = assignments
.map(([, rval]) => rval.node.end - rval.node.start + 1)
.reduce((a, b) => a + b, 0); // add 1 for space in front // sum
return numCommaAdded + sizeOfRvals < statementsLength;
}
}
module.exports = ArrayPropertyCollapser;