Code coverage report for src/node-transformers/node-control-flow-transformers/control-flow-replacers/CallExpressionControlFlowReplacer.ts

# All files / src/node-transformers/node-control-flow-transformers/control-flow-replacers CallExpressionControlFlowReplacer.ts

96.15% Statements 25/26
83.33% Branches 5/6
100% Functions 1/1
95.65% Lines 22/23

    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
    102
    1031x
    1x

    1x

    1x
    1x

    1x

    1x

    432x

    4088x

    4088x
    3682x

    406x
    406x
    406x

    406x

    406x

    406x

    406x

    406x

    406x

    406x

    406x

     import { injectable, inject } from 'inversify';
    import { ServiceIdentifiers } from '../../../container/ServiceIdentifiers';

    import * as ESTree from 'estree';

    import { TCustomNodeFactory } from '../../../types/container/TCustomNodeFactory';
    import { TStatement } from '../../../types/node/TStatement';

    import { ICustomNode } from '../../../interfaces/custom-nodes/ICustomNode';
    import { IOptions } from '../../../interfaces/options/IOptions';
    import { IStorage } from '../../../interfaces/storages/IStorage';

    import { CustomNodes } from '../../../enums/container/CustomNodes';

    import { AbstractControlFlowReplacer } from './AbstractControlFlowReplacer';
    import { Node } from '../../../node/Node';

    @injectable()
    export class CallExpressionControlFlowReplacer extends AbstractControlFlowReplacer {
        /**
         * @type {number}
         */
        private static readonly usingExistingIdentifierChance: number = 0.5;

        /**
         * @param customNodeFactory
         * @param options
         */
        constructor (
            @inject(ServiceIdentifiers.Factory__ICustomNode) customNodeFactory: TCustomNodeFactory,
            @inject(ServiceIdentifiers.IOptions) options: IOptions
        ) {
            super(customNodeFactory, options);
        }

        /**
         * @param callExpressionNode
         * @param parentNode
         * @param controlFlowStorage
         * @returns {ESTree.Node}
         */
        public replace (
            callExpressionNode: ESTree.CallExpression,
            parentNode: ESTree.Node,
            controlFlowStorage: IStorage <ICustomNode>
        ): ESTree.Node {
            const callee: ESTree.Expression = <ESTree.Expression>callExpressionNode.callee;

            if (!Node.isIdentifierNode(callee)) {
                return callExpressionNode;
            }

            const replacerId: string = String(callExpressionNode.arguments.length);
            const callExpressionFunctionCustomNode: ICustomNode = this.customNodeFactory(CustomNodes.CallExpressionFunctionNode);
            const expressionArguments: (ESTree.Expression | ESTree.SpreadElement)[] = callExpressionNode.arguments;

            callExpressionFunctionCustomNode.initialize(expressionArguments);

            const storageKey: string = this.insertCustomNodeToControlFlowStorage(
                callExpressionFunctionCustomNode,
                controlFlowStorage,
                replacerId,
                CallExpressionControlFlowReplacer.usingExistingIdentifierChance
            );

            return this.getControlFlowStorageCallNode(
                controlFlowStorage.getStorageId(),
                storageKey,
                callee,
                expressionArguments
            );
        }

        /**
         * @param controlFlowStorageId
         * @param storageKey
         * @param callee
         * @param expressionArguments
         * @param arguments
         * @returns {ESTree.Node}
         */
        protected getControlFlowStorageCallNode (
            controlFlowStorageId: string,
            storageKey: string,
            callee: ESTree.Expression,
            expressionArguments: (ESTree.Expression | ESTree.SpreadElement)[]
        ): ESTree.Node {
            const controlFlowStorageCallCustomNode: ICustomNode = this.customNodeFactory(
                CustomNodes.CallExpressionControlFlowStorageCallNode
            );

            controlFlowStorageCallCustomNode.initialize(controlFlowStorageId, storageKey, callee, expressionArguments);

            const statementNode: TStatement = controlFlowStorageCallCustomNode.getNode()[0];

            Iif (!statementNode || !Node.isExpressionStatementNode(statementNode)) {
                throw new Error(`\`controlFlowStorageCallCustomNode.getNode()[0]\` should returns array with \`ExpressionStatement\` node`);
            }

            return statementNode.expression;
        }
    }