javascript-obfuscator/src/custom-nodes/unicode-array-nodes/UnicodeArrayRotateFunctionNode.js

"use strict";
const estraverse = require('estraverse');
const NodeType_1 = require("../../enums/NodeType");
const Node_1 = require('../Node');
const NodeUtils_1 = require("../../NodeUtils");
const Utils_1 = require('../../Utils');
class UnicodeArrayRotateFunctionNode extends Node_1.Node {
    constructor(unicodeArrayRotateFunctionName, unicodeArrayName) {
        super();
        this.unicodeArrayRotateFunctionName = unicodeArrayRotateFunctionName;
        this.unicodeArrayName = unicodeArrayName;
        this.node = this.getNodeStructure();
    }
    appendNode(astTree) {
        estraverse.replace(astTree, {
            leave: (node, parent) => {
                if (NodeUtils_1.NodeUtils.isProgramNode(node)) {
                    NodeUtils_1.NodeUtils.appendNode(node.body, this.getNode());
                    return estraverse.VisitorOption.Break;
                }
                return estraverse.VisitorOption.Skip;
            }
        });
    }
    getNodeIdentifier() {
        return this.unicodeArrayRotateFunctionName;
    }
    getNodeStructure() {
        return {
            'type': NodeType_1.NodeType.FunctionExpression,
            'id': {
                'type': NodeType_1.NodeType.Identifier,
                'name': this.unicodeArrayRotateFunctionName
            },
            'params': [
                {
                    'type': NodeType_1.NodeType.Identifier,
                    'name': 'array'
                },
                {
                    'type': NodeType_1.NodeType.Identifier,
                    'name': 'times'
                },
                {
                    'type': NodeType_1.NodeType.Identifier,
                    'name': 'reverse'
                }
            ],
            'defaults': [],
            'body': {
                'type': NodeType_1.NodeType.BlockStatement,
                'body': [
                    {
                        'type': NodeType_1.NodeType.IfStatement,
                        'test': {
                            'type': NodeType_1.NodeType.BinaryExpression,
                            'operator': '<',
                            'left': {
                                'type': NodeType_1.NodeType.Identifier,
                                'name': 'times'
                            },
                            'right': {
                                'type': NodeType_1.NodeType.Literal,
                                'value': 0,
                                'raw': '0'
                            }
                        },
                        'consequent': {
                            'type': NodeType_1.NodeType.BlockStatement,
                            'body': [
                                {
                                    'type': NodeType_1.NodeType.ReturnStatement,
                                    'argument': null
                                }
                            ]
                        },
                        'alternate': null
                    },
                    {
                        'type': NodeType_1.NodeType.ExpressionStatement,
                        'expression': {
                            'type': NodeType_1.NodeType.AssignmentExpression,
                            'operator': '=',
                            'left': {
                                'type': NodeType_1.NodeType.Identifier,
                                'name': 'reverse'
                            },
                            'right': {
                                'type': NodeType_1.NodeType.LogicalExpression,
                                'operator': '||',
                                'left': {
                                    'type': NodeType_1.NodeType.Identifier,
                                    'name': 'reverse'
                                },
                                'right': {
                                    'type': NodeType_1.NodeType.Literal,
                                    'value': false,
                                    'raw': 'false'
                                }
                            }
                        }
                    },
                    {
                        'type': NodeType_1.NodeType.VariableDeclaration,
                        'declarations': [
                            {
                                'type': NodeType_1.NodeType.VariableDeclarator,
                                'id': {
                                    'type': NodeType_1.NodeType.Identifier,
                                    'name': 'temp'
                                },
                                'init': null
                            }
                        ],
                        'kind': 'var'
                    },
                    {
                        'type': NodeType_1.NodeType.WhileStatement,
                        'test': {
                            'type': NodeType_1.NodeType.UpdateExpression,
                            'operator': '--',
                            'argument': {
                                'type': NodeType_1.NodeType.Identifier,
                                'name': 'times'
                            },
                            'prefix': false
                        },
                        'body': {
                            'type': NodeType_1.NodeType.BlockStatement,
                            'body': [
                                {
                                    'type': NodeType_1.NodeType.IfStatement,
                                    'test': {
                                        'type': NodeType_1.NodeType.UnaryExpression,
                                        'operator': '!',
                                        'argument': {
                                            'type': NodeType_1.NodeType.Identifier,
                                            'name': 'reverse'
                                        },
                                        'prefix': true
                                    },
                                    'consequent': {
                                        'type': NodeType_1.NodeType.BlockStatement,
                                        'body': [
                                            {
                                                'type': NodeType_1.NodeType.ExpressionStatement,
                                                'expression': {
                                                    'type': NodeType_1.NodeType.AssignmentExpression,
                                                    'operator': '=',
                                                    'left': {
                                                        'type': NodeType_1.NodeType.Identifier,
                                                        'name': 'temp'
                                                    },
                                                    'right': {
                                                        'type': NodeType_1.NodeType.CallExpression,
                                                        'callee': {
                                                            'type': NodeType_1.NodeType.MemberExpression,
                                                            'computed': true,
                                                            'object': {
                                                                'type': NodeType_1.NodeType.Identifier,
                                                                'name': 'array'
                                                            },
                                                            'property': {
                                                                'type': NodeType_1.NodeType.Literal,
                                                                'name': 'pop',
                                                                'x-verbatim-property': Utils_1.Utils.stringToUnicode('pop')
                                                            }
                                                        },
                                                        'arguments': []
                                                    }
                                                }
                                            },
                                            {
                                                'type': NodeType_1.NodeType.ExpressionStatement,
                                                'expression': {
                                                    'type': NodeType_1.NodeType.CallExpression,
                                                    'callee': {
                                                        'type': NodeType_1.NodeType.MemberExpression,
                                                        'computed': true,
                                                        'object': {
                                                            'type': NodeType_1.NodeType.Identifier,
                                                            'name': 'array'
                                                        },
                                                        'property': {
                                                            'type': NodeType_1.NodeType.Literal,
                                                            'name': 'unshift',
                                                            'x-verbatim-property': Utils_1.Utils.stringToUnicode('unshift')
                                                        }
                                                    },
                                                    'arguments': [
                                                        {
                                                            'type': NodeType_1.NodeType.Identifier,
                                                            'name': 'temp'
                                                        }
                                                    ]
                                                }
                                            }
                                        ]
                                    },
                                    'alternate': {
                                        'type': NodeType_1.NodeType.BlockStatement,
                                        'body': [
                                            {
                                                'type': NodeType_1.NodeType.ExpressionStatement,
                                                'expression': {
                                                    'type': NodeType_1.NodeType.AssignmentExpression,
                                                    'operator': '=',
                                                    'left': {
                                                        'type': NodeType_1.NodeType.Identifier,
                                                        'name': 'temp'
                                                    },
                                                    'right': {
                                                        'type': NodeType_1.NodeType.CallExpression,
                                                        'callee': {
                                                            'type': NodeType_1.NodeType.MemberExpression,
                                                            'computed': true,
                                                            'object': {
                                                                'type': NodeType_1.NodeType.Identifier,
                                                                'name': 'array'
                                                            },
                                                            'property': {
                                                                'type': NodeType_1.NodeType.Literal,
                                                                'name': 'shift',
                                                                'x-verbatim-property': Utils_1.Utils.stringToUnicode('shift')
                                                            }
                                                        },
                                                        'arguments': []
                                                    }
                                                }
                                            },
                                            {
                                                'type': NodeType_1.NodeType.ExpressionStatement,
                                                'expression': {
                                                    'type': NodeType_1.NodeType.CallExpression,
                                                    'callee': {
                                                        'type': NodeType_1.NodeType.MemberExpression,
                                                        'computed': true,
                                                        'object': {
                                                            'type': NodeType_1.NodeType.Identifier,
                                                            'name': 'array'
                                                        },
                                                        'property': {
                                                            'type': NodeType_1.NodeType.Literal,
                                                            'name': 'push',
                                                            'x-verbatim-property': Utils_1.Utils.stringToUnicode('push')
                                                        }
                                                    },
                                                    'arguments': [
                                                        {
                                                            'type': NodeType_1.NodeType.Identifier,
                                                            'name': 'temp'
                                                        }
                                                    ]
                                                }
                                            }
                                        ]
                                    }
                                }
                            ]
                        }
                    }
                ]
            },
            'generator': false,
            'expression': false
        };
    }
}
exports.UnicodeArrayRotateFunctionNode = UnicodeArrayRotateFunctionNode;