mdgriffith/elm-optimize-level-2
1
1
Fork 0
mirror of https://github.com/mdgriffith/elm-optimize-level-2.git synced 2024-11-29 12:46:32 +03:00
Code Issues Packages Projects Releases Wiki Activity

Only perform conversion for object literals which has at least one property name used in an update expression, to save space.

Browse Source
This commit is contained in:
Robin Heggelund Hansen 2021-08-12 15:27:59 +02:00
parent d776377375
commit 7bfcdca3c7
1 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/transforms/recordUpdate.ts
View File

@ -5,8 +5,9 @@ export const recordUpdate = (): ts.TransformerFactory<ts.SourceFile> =>
(context) => (sourceFile) => {
const registry = new RecordRegistry();
const replacedUpdates = ts.visitNode(sourceFile, replaceUpdateStatements(context));
const replacedLiterals = ts.visitNode(replacedUpdates, replaceObjectLiterals(registry, context));
const propSet = new Set<String>();
const replacedUpdates = ts.visitNode(sourceFile, replaceUpdateStatements(propSet, context));
const replacedLiterals = ts.visitNode(replacedUpdates, replaceObjectLiterals(propSet, registry, context));
const recordStatements = createRecordStatements(registry);
replacedLiterals.statements = recordStatements.concat(replacedLiterals.statements);
@ -44,7 +45,7 @@ class RecordRegistry {
}
function replaceUpdateStatements(ctx: ts.TransformationContext) {
function replaceUpdateStatements(propSet: Set<String>, ctx: ts.TransformationContext) {
const visitorHelp = (node: ts.Node): ts.VisitResult<ts.Node> => {
const visitedNode = ts.visitEachChild(node, visitorHelp, ctx);
if (!isUpdateExpression(visitedNode)) {
@ -72,6 +73,11 @@ function replaceUpdateStatements(ctx: ts.TransformationContext) {
])
);
// Add updated properties to propSet
visitedNode.arguments[1].properties.map((it) => it.name.text).forEach((it) => {
propSet.add(it);
});
const propSetters = visitedNode.arguments[1].properties.
map((it) => ts.createExpressionStatement(
ts.createBinary(
@ -103,13 +109,19 @@ function isUpdateExpression(node: ts.Node): boolean {
}
function replaceObjectLiterals(registry: RecordRegistry, ctx: ts.TransformationContext) {
function replaceObjectLiterals(propSet: Set<String>, registry: RecordRegistry, ctx: ts.TransformationContext) {
const visitorHelp = (node: ts.Node): ts.VisitResult<ts.Node> => {
const visitedNode = ts.visitEachChild(node, visitorHelp, ctx);
if (!isRecordLiteral(visitedNode)) {
return visitedNode;
}
// Abort of none of the record properties are used in an update expression
const recordPropNames = visitedNode.properties.map((it) => it.name.text);
if (!recordPropNames.some((it) => propSet.has(it))) {
return visitedNode;
}
const recordClassName = registry.register(visitedNode);
const recordConstruction = ts.createParen(
ts.createNew(