Merge pull request #6 from moznion/bind_parameter_on_immediately_invoked_function

Bind the argument of `dynamodbRecord()` to the transformed immediately invoked function's parameter

Author: moznion

tests/transformer.test.ts

@@ -280,9 +280,9 @@ describe('dynamodb record transform', () => {
         ['buz', 'qux'],
       ]),
     });
+    expect(record['id']).toEqual({ N: '12345' });
     expect(record['name']).toEqual({ S: 'John Doe' });
     expect(record['tags']).toEqual({ M: { foo: { S: 'bar' }, buz: { S: 'qux' } } });
-    expect(record['id']).toEqual({ N: '12345' });
     expect(Object.keys(record)).toHaveLength(3);
   });
 });

transformer.ts

@@ -246,11 +246,11 @@ function visitNode(node: ts.Node, program: ts.Program): ts.Node | undefined {
     })
     .filter((c): c is ts.ObjectLiteralElementLike => !!c);
 
-  return ts.factory.createImmediatelyInvokedFunctionExpression([
-    ts.factory.createVariableStatement([], [ts.factory.createVariableDeclaration(argVarNameIdent)]),
-    ts.factory.createBinaryExpression(argVarNameIdent, ts.SyntaxKind.EqualsToken, node.arguments[0]),
-    ts.factory.createReturnStatement(ts.factory.createObjectLiteralExpression(objectProps, true)),
-  ] as ts.Statement[]);
+  return ts.factory.createImmediatelyInvokedFunctionExpression(
+    [ts.factory.createReturnStatement(ts.factory.createObjectLiteralExpression(objectProps, true))] as ts.Statement[],
+    ts.factory.createParameterDeclaration([], undefined, argVarNameIdent, undefined, node.typeArguments[0], undefined),
+    node.arguments[0],
+  );
 }
 
 function extractKeyValueTypesFromKeyValuePairMapSyntax(nodes: ts.Node[]): [string, string] | undefined {