diff options
author | cyfraeviolae <cyfraeviolae> | 2024-04-03 03:10:44 -0400 |
---|---|---|
committer | cyfraeviolae <cyfraeviolae> | 2024-04-03 03:10:44 -0400 |
commit | 6d7ba58f880be618ade07f8ea080fe8c4bf8a896 (patch) | |
tree | b1c931051ffcebd2bd9d61d98d6233ffa289bbce /venv/lib/python3.11/site-packages/jsbeautifier/javascript/beautifier.py | |
parent | 4f884c9abc32990b4061a1bb6997b4b37e58ea0b (diff) |
venv
Diffstat (limited to 'venv/lib/python3.11/site-packages/jsbeautifier/javascript/beautifier.py')
-rw-r--r-- | venv/lib/python3.11/site-packages/jsbeautifier/javascript/beautifier.py | 1647 |
1 files changed, 1647 insertions, 0 deletions
diff --git a/venv/lib/python3.11/site-packages/jsbeautifier/javascript/beautifier.py b/venv/lib/python3.11/site-packages/jsbeautifier/javascript/beautifier.py new file mode 100644 index 0000000..fff41a0 --- /dev/null +++ b/venv/lib/python3.11/site-packages/jsbeautifier/javascript/beautifier.py @@ -0,0 +1,1647 @@ +# The MIT License (MIT) +# +# Copyright (c) 2007-2018 Einar Lielmanis, Liam Newman, and contributors. +# +# Permission is hereby granted, free of charge, to any person +# obtaining a copy of this software and associated documentation files +# (the "Software"), to deal in the Software without restriction, +# including without limitation the rights to use, copy, modify, merge, +# publish, distribute, sublicense, and/or sell copies of the Software, +# and to permit persons to whom the Software is furnished to do so, +# subject to the following conditions: +# +# The above copyright notice and this permission notice shall be +# included in all copies or substantial portions of the Software. +# +# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, +# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF +# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND +# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS +# BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN +# ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN +# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE +# SOFTWARE. + +import re +import string +import copy +from ..core.token import Token +from .tokenizer import Tokenizer +from .tokenizer import TOKEN +from .options import BeautifierOptions +from ..core.output import Output + + +def default_options(): + return BeautifierOptions() + + +class BeautifierFlags: + def __init__(self, mode): + self.mode = mode + self.parent = None + self.last_token = Token(TOKEN.START_BLOCK, "") + self.last_word = "" + self.declaration_statement = False + self.declaration_assignment = False + self.multiline_frame = False + self.inline_frame = False + self.if_block = False + self.else_block = False + self.class_start_block = False + self.do_block = False + self.do_while = False + self.import_block = False + self.in_case = False + self.in_case_statement = False + self.case_body = False + self.case_block = False + self.indentation_level = 0 + self.alignment = 0 + self.line_indent_level = 0 + self.start_line_index = 0 + self.ternary_depth = 0 + + def apply_base(self, flags_base, added_newline): + next_indent_level = flags_base.indentation_level + if not added_newline and flags_base.line_indent_level > next_indent_level: + next_indent_level = flags_base.line_indent_level + + self.parent = flags_base + self.last_token = flags_base.last_token + self.last_word = flags_base.last_word + self.indentation_level = next_indent_level + + +OPERATOR_POSITION = { + "before_newline": "before-newline", + "after_newline": "after-newline", + "preserve_newline": "preserve-newline", +} +OPERATOR_POSITION_BEFORE_OR_PRESERVE = [ + OPERATOR_POSITION["before_newline"], + OPERATOR_POSITION["preserve_newline"], +] + + +class MODE: + ( + BlockStatement, + Statement, + ObjectLiteral, + ArrayLiteral, + ForInitializer, + Conditional, + Expression, + ) = range(7) + + +def remove_redundant_indentation(output, frame): + # This implementation is effective but has some issues: + # - can cause line wrap to happen too soon due to indent removal + # after wrap points are calculated + # These issues are minor compared to ugly indentation. + + if ( + frame.multiline_frame + or frame.mode == MODE.ForInitializer + or frame.mode == MODE.Conditional + ): + return + + # remove one indent from each line inside this section + output.remove_indent(frame.start_line_index) + + +def reserved_word(token, word): + return token and token.type == TOKEN.RESERVED and token.text == word + + +def reserved_array(token, words): + return token and token.type == TOKEN.RESERVED and token.text in words + + +_special_word_set = frozenset( + [ + "case", + "return", + "do", + "if", + "throw", + "else", + "await", + "break", + "continue", + "async", + ] +) + + +class Beautifier: + def __init__(self, opts=None): + import jsbeautifier.javascript.acorn as acorn + + self.acorn = acorn + self._options = BeautifierOptions(opts) + + self._blank_state() + + def _blank_state(self, js_source_text=None): + if js_source_text is None: + js_source_text = "" + + # internal flags + self._flags = None + self._previous_flags = None + self._flag_store = [] + self._tokens = None + + if self._options.eol == "auto": + self._options.eol = "\n" + if self.acorn.lineBreak.search(js_source_text or ""): + self._options.eol = self.acorn.lineBreak.search(js_source_text).group() + + baseIndentString = re.search("^[\t ]*", js_source_text).group(0) + self._last_last_text = "" # pre-last token text + + self._output = Output(self._options, baseIndentString) + # If testing the ignore directive, start with output disable set to + # true + self._output.raw = self._options.test_output_raw + + self.set_mode(MODE.BlockStatement) + return js_source_text + + def beautify(self, source_text="", opts=None): + if opts is not None: + self._options = BeautifierOptions(opts) + + source_text = source_text or "" + if self._options.disabled: + return source_text + + source_text = self._blank_state(source_text) + + source_text = self.unpack(source_text, self._options.eval_code) + + self._tokens = Tokenizer(source_text, self._options).tokenize() + + for current_token in self._tokens: + self.handle_token(current_token) + + self._last_last_text = self._flags.last_token.text + self._flags.last_token = current_token + + sweet_code = self._output.get_code(self._options.eol) + + return sweet_code + + def handle_token(self, current_token, preserve_statement_flags=False): + if current_token.type == TOKEN.START_EXPR: + self.handle_start_expr(current_token) + elif current_token.type == TOKEN.END_EXPR: + self.handle_end_expr(current_token) + elif current_token.type == TOKEN.START_BLOCK: + self.handle_start_block(current_token) + elif current_token.type == TOKEN.END_BLOCK: + self.handle_end_block(current_token) + elif current_token.type == TOKEN.WORD: + self.handle_word(current_token) + elif current_token.type == TOKEN.RESERVED: + self.handle_word(current_token) + elif current_token.type == TOKEN.SEMICOLON: + self.handle_semicolon(current_token) + elif current_token.type == TOKEN.STRING: + self.handle_string(current_token) + elif current_token.type == TOKEN.EQUALS: + self.handle_equals(current_token) + elif current_token.type == TOKEN.OPERATOR: + self.handle_operator(current_token) + elif current_token.type == TOKEN.COMMA: + self.handle_comma(current_token) + elif current_token.type == TOKEN.BLOCK_COMMENT: + self.handle_block_comment(current_token, preserve_statement_flags) + elif current_token.type == TOKEN.COMMENT: + self.handle_comment(current_token, preserve_statement_flags) + elif current_token.type == TOKEN.DOT: + self.handle_dot(current_token) + elif current_token.type == TOKEN.EOF: + self.handle_eof(current_token) + elif current_token.type == TOKEN.UNKNOWN: + self.handle_unknown(current_token, preserve_statement_flags) + else: + self.handle_unknown(current_token, preserve_statement_flags) + + def handle_whitespace_and_comments( + self, current_token, preserve_statement_flags=False + ): + newlines = current_token.newlines + keep_whitespace = self._options.keep_array_indentation and self.is_array( + self._flags.mode + ) + + if current_token.comments_before is not None: + for comment_token in current_token.comments_before: + # The cleanest handling of inline comments is to treat them + # as though they aren't there. + # Just continue formatting and the behavior should be logical. + # Also ignore unknown tokens. Again, this should result in better + # behavior. + self.handle_whitespace_and_comments( + comment_token, preserve_statement_flags + ) + self.handle_token(comment_token, preserve_statement_flags) + + if keep_whitespace: + for i in range(newlines): + self.print_newline(i > 0, preserve_statement_flags) + else: # not keep_whitespace + if ( + self._options.max_preserve_newlines != 0 + and newlines > self._options.max_preserve_newlines + ): + newlines = self._options.max_preserve_newlines + + if self._options.preserve_newlines and newlines > 1: + self.print_newline(False, preserve_statement_flags) + for i in range(1, newlines): + self.print_newline(True, preserve_statement_flags) + + def unpack(self, source, evalcode=False): + import jsbeautifier.unpackers as unpackers + + try: + return unpackers.run(source, evalcode) + except unpackers.UnpackingError: + return source + + def is_array(self, mode): + return mode == MODE.ArrayLiteral + + def is_expression(self, mode): + return ( + mode == MODE.Expression + or mode == MODE.ForInitializer + or mode == MODE.Conditional + ) + + _newline_restricted_tokens = frozenset( + ["async", "break", "continue", "return", "throw", "yield"] + ) + + def allow_wrap_or_preserved_newline(self, current_token, force_linewrap=False): + # never wrap the first token of a line. + if self._output.just_added_newline(): + return + + shouldPreserveOrForce = ( + self._options.preserve_newlines and current_token.newlines + ) or force_linewrap + operatorLogicApplies = ( + self._flags.last_token.text in Tokenizer.positionable_operators + or current_token.text in Tokenizer.positionable_operators + ) + + if operatorLogicApplies: + shouldPrintOperatorNewline = ( + self._flags.last_token.text in Tokenizer.positionable_operators + and self._options.operator_position + in OPERATOR_POSITION_BEFORE_OR_PRESERVE + ) or current_token.text in Tokenizer.positionable_operators + shouldPreserveOrForce = shouldPreserveOrForce and shouldPrintOperatorNewline + + if shouldPreserveOrForce: + self.print_newline(preserve_statement_flags=True) + elif self._options.wrap_line_length > 0: + if reserved_array(self._flags.last_token, self._newline_restricted_tokens): + # These tokens should never have a newline inserted between + # them and the following expression. + return + self._output.set_wrap_point() + + def print_newline(self, force_newline=False, preserve_statement_flags=False): + if not preserve_statement_flags: + if ( + self._flags.last_token.text != ";" + and self._flags.last_token.text != "," + and self._flags.last_token.text != "=" + and ( + self._flags.last_token.type != TOKEN.OPERATOR + or self._flags.last_token.text == "--" + or self._flags.last_token.text == "++" + ) + ): + next_token = self._tokens.peek() + while ( + self._flags.mode == MODE.Statement + and not (self._flags.if_block and reserved_word(next_token, "else")) + and not self._flags.do_block + ): + self.restore_mode() + + if self._output.add_new_line(force_newline): + self._flags.multiline_frame = True + + def print_token_line_indentation(self, current_token): + if self._output.just_added_newline(): + line = self._output.current_line + if ( + self._options.keep_array_indentation + and current_token.newlines + and (self.is_array(self._flags.mode) or current_token.text == "[") + ): + line.set_indent(-1) + line.push(current_token.whitespace_before) + self._output.space_before_token = False + elif self._output.set_indent( + self._flags.indentation_level, self._flags.alignment + ): + self._flags.line_indent_level = self._flags.indentation_level + + def print_token(self, current_token, s=None): + if self._output.raw: + self._output.add_raw_token(current_token) + return + + if ( + self._options.comma_first + and current_token.previous + and current_token.previous.type == TOKEN.COMMA + and self._output.just_added_newline() + ): + if self._output.previous_line.last() == ",": + # if the comma was already at the start of the line, + # pull back onto that line and reprint the indentation + popped = self._output.previous_line.pop() + if self._output.previous_line.is_empty(): + self._output.previous_line.push(popped) + self._output.trim(True) + self._output.current_line.pop() + self._output.trim() + + # add the comma in front of the next token + self.print_token_line_indentation(current_token) + self._output.add_token(",") + self._output.space_before_token = True + + if s is None: + s = current_token.text + + self.print_token_line_indentation(current_token) + self._output.non_breaking_space = True + self._output.add_token(s) + if self._output.previous_token_wrapped: + self._flags.multiline_frame = True + + def indent(self): + self._flags.indentation_level += 1 + self._output.set_indent(self._flags.indentation_level, self._flags.alignment) + + def deindent(self): + allow_deindent = self._flags.indentation_level > 0 and ( + (self._flags.parent is None) + or self._flags.indentation_level > self._flags.parent.indentation_level + ) + + if allow_deindent: + self._flags.indentation_level -= 1 + + self._output.set_indent(self._flags.indentation_level, self._flags.alignment) + + def set_mode(self, mode): + if self._flags: + self._flag_store.append(self._flags) + self._previous_flags = self._flags + else: + self._previous_flags = BeautifierFlags(mode) + + self._flags = BeautifierFlags(mode) + self._flags.apply_base(self._previous_flags, self._output.just_added_newline()) + self._flags.start_line_index = self._output.get_line_number() + + self._output.set_indent(self._flags.indentation_level, self._flags.alignment) + + def restore_mode(self): + if len(self._flag_store) > 0: + self._previous_flags = self._flags + self._flags = self._flag_store.pop() + if self._previous_flags.mode == MODE.Statement: + remove_redundant_indentation(self._output, self._previous_flags) + + self._output.set_indent(self._flags.indentation_level, self._flags.alignment) + + def start_of_object_property(self): + return ( + self._flags.parent.mode == MODE.ObjectLiteral + and self._flags.mode == MODE.Statement + and ( + (self._flags.last_token.text == ":" and self._flags.ternary_depth == 0) + or (reserved_array(self._flags.last_token, ["get", "set"])) + ) + ) + + def start_of_statement(self, current_token): + start = False + start = start or ( + reserved_array(self._flags.last_token, ["var", "let", "const"]) + and current_token.type == TOKEN.WORD + ) + start = start or reserved_word(self._flags.last_token, "do") + start = start or ( + not ( + self._flags.parent.mode == MODE.ObjectLiteral + and self._flags.mode == MODE.Statement + ) + and reserved_array(self._flags.last_token, self._newline_restricted_tokens) + and not current_token.newlines + ) + start = start or ( + reserved_word(self._flags.last_token, "else") + and not ( + reserved_word(current_token, "if") + and current_token.comments_before is None + ) + ) + start = start or ( + self._flags.last_token.type == TOKEN.END_EXPR + and ( + self._previous_flags.mode == MODE.ForInitializer + or self._previous_flags.mode == MODE.Conditional + ) + ) + start = start or ( + self._flags.last_token.type == TOKEN.WORD + and self._flags.mode == MODE.BlockStatement + and not self._flags.in_case + and not (current_token.text == "--" or current_token.text == "++") + and self._last_last_text != "function" + and current_token.type != TOKEN.WORD + and current_token.type != TOKEN.RESERVED + ) + start = start or ( + self._flags.mode == MODE.ObjectLiteral + and ( + (self._flags.last_token.text == ":" and self._flags.ternary_depth == 0) + or (reserved_array(self._flags.last_token, ["get", "set"])) + ) + ) + + if start: + self.set_mode(MODE.Statement) + self.indent() + + self.handle_whitespace_and_comments(current_token, True) + + # Issue #276: + # If starting a new statement with [if, for, while, do], push to a new line. + # if (a) if (b) if(c) d(); else e(); else f(); + if not self.start_of_object_property(): + self.allow_wrap_or_preserved_newline( + current_token, + reserved_array(current_token, ["do", "for", "if", "while"]), + ) + return True + else: + return False + + def handle_start_expr(self, current_token): + if self.start_of_statement(current_token): + # The conditional starts the statement if appropriate. + pass + else: + self.handle_whitespace_and_comments(current_token) + + next_mode = MODE.Expression + + if current_token.text == "[": + if ( + self._flags.last_token.type == TOKEN.WORD + or self._flags.last_token.text == ")" + ): + if reserved_array(self._flags.last_token, Tokenizer.line_starters): + self._output.space_before_token = True + self.print_token(current_token) + self.set_mode(next_mode) + self.indent() + if self._options.space_in_paren: + self._output.space_before_token = True + return + + next_mode = MODE.ArrayLiteral + + if self.is_array(self._flags.mode): + if self._flags.last_token.text == "[" or ( + self._flags.last_token.text == "," + and (self._last_last_text == "]" or self._last_last_text == "}") + ): + # ], [ goes to a new line + # }, [ goes to a new line + if not self._options.keep_array_indentation: + self.print_newline() + + if self._flags.last_token.type not in [ + TOKEN.START_EXPR, + TOKEN.END_EXPR, + TOKEN.WORD, + TOKEN.OPERATOR, + TOKEN.DOT, + ]: + self._output.space_before_token = True + + else: + if self._flags.last_token.type == TOKEN.RESERVED: + if self._flags.last_token.text == "for": + self._output.space_before_token = ( + self._options.space_before_conditional + ) + next_mode = MODE.ForInitializer + elif self._flags.last_token.text in ["if", "while", "switch"]: + self._output.space_before_token = ( + self._options.space_before_conditional + ) + next_mode = MODE.Conditional + elif self._flags.last_word in ["await", "async"]: + # Should be a space between await and an IIFE, or async and + # an arrow function + self._output.space_before_token = True + elif ( + self._flags.last_token.text == "import" + and current_token.whitespace_before == "" + ): + self._output.space_before_token = False + elif ( + self._flags.last_token.text in Tokenizer.line_starters + or self._flags.last_token.text == "catch" + ): + self._output.space_before_token = True + + elif self._flags.last_token.type in [TOKEN.EQUALS, TOKEN.OPERATOR]: + # Support of this kind of newline preservation: + # a = (b && + # (c || d)); + if not self.start_of_object_property(): + self.allow_wrap_or_preserved_newline(current_token) + elif self._flags.last_token.type == TOKEN.WORD: + self._output.space_before_token = False + # function name() vs function name () + # function* name() vs function* name () + # async name() vs async name () + # In ES6, you can also define the method properties of an object + # var obj = {a: function() {}} + # It can be abbreviated + # var obj = {a() {}} + # var obj = { a() {}} vs var obj = { a () {}} + # var obj = { * a() {}} vs var obj = { * a () {}} + peek_back_two = self._tokens.peek(-3) + if self._options.space_after_named_function and peek_back_two: + # peek starts at next character so -1 is current token + peek_back_three = self._tokens.peek(-4) + if reserved_array(peek_back_two, ["async", "function"]) or ( + peek_back_two.text == "*" + and reserved_array(peek_back_three, ["async", "function"]) + ): + self._output.space_before_token = True + elif self._flags.mode == MODE.ObjectLiteral: + if (peek_back_two.text == "{" or peek_back_two.text == ",") or ( + peek_back_two.text == "*" + and ( + peek_back_three.text == "{" + or peek_back_three.text == "," + ) + ): + self._output.space_before_token = True + elif self._flags.parent and self._flags.parent.class_start_block: + self._output.space_before_token = True + else: + # Support preserving wrapped arrow function expressions + # a.b('c', + # () => d.e + # ) + self.allow_wrap_or_preserved_newline(current_token) + + # function() vs function (), typeof() vs typeof () + # function*() vs function* (), yield*() vs yield* () + if ( + self._flags.last_token.type == TOKEN.RESERVED + and ( + self._flags.last_word == "function" + or self._flags.last_word == "typeof" + ) + ) or ( + self._flags.last_token.text == "*" + and ( + self._last_last_text in ["function", "yield"] + or ( + self._flags.mode == MODE.ObjectLiteral + and self._last_last_text in ["{", ","] + ) + ) + ): + self._output.space_before_token = ( + self._options.space_after_anon_function + ) + + if ( + self._flags.last_token.text == ";" + or self._flags.last_token.type == TOKEN.START_BLOCK + ): + self.print_newline() + elif ( + self._flags.last_token.type + in [TOKEN.END_EXPR, TOKEN.START_EXPR, TOKEN.END_BLOCK, TOKEN.COMMA] + or self._flags.last_token.text == "." + ): + # do nothing on (( and )( and ][ and ]( and .( + # TODO: Consider whether forcing this is required. Review failing + # tests when removed. + self.allow_wrap_or_preserved_newline(current_token, current_token.newlines) + + self.print_token(current_token) + self.set_mode(next_mode) + + if self._options.space_in_paren: + self._output.space_before_token = True + + # In all cases, if we newline while inside an expression it should be + # indented. + self.indent() + + def handle_end_expr(self, current_token): + # statements inside expressions are not valid syntax, but... + # statements must all be closed when their container closes + while self._flags.mode == MODE.Statement: + self.restore_mode() + + self.handle_whitespace_and_comments(current_token) + + if self._flags.multiline_frame: + self.allow_wrap_or_preserved_newline( + current_token, + current_token.text == "]" + and self.is_array(self._flags.mode) + and not self._options.keep_array_indentation, + ) + + if self._options.space_in_paren: + if ( + self._flags.last_token.type == TOKEN.START_EXPR + and not self._options.space_in_empty_paren + ): + # empty parens are always "()" and "[]", not "( )" or "[ ]" + self._output.space_before_token = False + self._output.trim() + else: + self._output.space_before_token = True + + self.deindent() + self.print_token(current_token) + self.restore_mode() + + remove_redundant_indentation(self._output, self._previous_flags) + + # do {} while () // no statement required after + if self._flags.do_while and self._previous_flags.mode == MODE.Conditional: + self._previous_flags.mode = MODE.Expression + self._flags.do_block = False + self._flags.do_while = False + + def handle_start_block(self, current_token): + self.handle_whitespace_and_comments(current_token) + + # Check if this is a BlockStatement that should be treated as a + # ObjectLiteral + next_token = self._tokens.peek() + second_token = self._tokens.peek(1) + if ( + self._flags.last_word == "switch" + and self._flags.last_token.type == TOKEN.END_EXPR + ): + self.set_mode(MODE.BlockStatement) + self._flags.in_case_statement = True + elif self._flags.case_body: + self.set_mode(MODE.BlockStatement) + elif second_token is not None and ( + ( + second_token.text in [":", ","] + and next_token.type in [TOKEN.STRING, TOKEN.WORD, TOKEN.RESERVED] + ) + or ( + next_token.text in ["get", "set", "..."] + and second_token.type in [TOKEN.WORD, TOKEN.RESERVED] + ) + ): + # We don't support TypeScript,but we didn't break it for a very long time. + # We'll try to keep not breaking it. + if self._last_last_text in [ + "class", + "interface", + ] and second_token.text not in [":", ","]: + self.set_mode(MODE.BlockStatement) + else: + self.set_mode(MODE.ObjectLiteral) + elif ( + self._flags.last_token.type == TOKEN.OPERATOR + and self._flags.last_token.text == "=>" + ): + # arrow function: (param1, paramN) => { statements } + self.set_mode(MODE.BlockStatement) + elif self._flags.last_token.type in [ + TOKEN.EQUALS, + TOKEN.START_EXPR, + TOKEN.COMMA, + TOKEN.OPERATOR, + ] or reserved_array( + self._flags.last_token, ["return", "throw", "import", "default"] + ): + # Detecting shorthand function syntax is difficult by scanning forward, + # so check the surrounding context. + # If the block is being returned, imported, export default, passed as arg, + # assigned with = or assigned in a nested object, treat as an + # ObjectLiteral. + self.set_mode(MODE.ObjectLiteral) + else: + self.set_mode(MODE.BlockStatement) + + if self._flags.last_token: + if reserved_array(self._flags.last_token.previous, ["class", "extends"]): + self._flags.class_start_block = True + + empty_braces = ( + (next_token is not None) + and next_token.comments_before is None + and next_token.text == "}" + ) + empty_anonymous_function = ( + empty_braces + and self._flags.last_word == "function" + and self._flags.last_token.type == TOKEN.END_EXPR + ) + + if ( + self._options.brace_preserve_inline + ): # check for inline, set inline_frame if so + # search forward for newline wanted inside this block + index = 0 + check_token = None + self._flags.inline_frame = True + do_loop = True + while do_loop: + index += 1 + check_token = self._tokens.peek(index - 1) + if check_token.newlines: + self._flags.inline_frame = False + + do_loop = check_token.type != TOKEN.EOF and not ( + check_token.type == TOKEN.END_BLOCK + and check_token.opened == current_token + ) + + if ( + self._options.brace_style == "expand" + or (self._options.brace_style == "none" and current_token.newlines) + ) and not self._flags.inline_frame: + if self._flags.last_token.type != TOKEN.OPERATOR and ( + empty_anonymous_function + or self._flags.last_token.type == TOKEN.EQUALS + or ( + reserved_array(self._flags.last_token, _special_word_set) + and self._flags.last_token.text != "else" + ) + ): + self._output.space_before_token = True + else: + self.print_newline(preserve_statement_flags=True) + else: # collapse || inline_frame + if self.is_array(self._previous_flags.mode) and ( + self._flags.last_token.type == TOKEN.START_EXPR + or self._flags.last_token.type == TOKEN.COMMA + ): + # if we're preserving inline, + # allow newline between comma and next brace. + if self._flags.inline_frame: + self.allow_wrap_or_preserved_newline(current_token) + self._flags.inline_frame = True + self._previous_flags.multiline_frame = ( + self._previous_flags.multiline_frame + or self._flags.multiline_frame + ) + self._flags.multiline_frame = False + elif self._flags.last_token.type == TOKEN.COMMA: + self._output.space_before_token = True + + elif self._flags.last_token.type not in [TOKEN.OPERATOR, TOKEN.START_EXPR]: + if ( + self._flags.last_token.type in [TOKEN.START_BLOCK, TOKEN.SEMICOLON] + and not self._flags.inline_frame + ): + self.print_newline() + else: + self._output.space_before_token = True + + self.print_token(current_token) + self.indent() + + # Except for specific cases, open braces are followed by a new line. + if not empty_braces and not ( + self._options.brace_preserve_inline and self._flags.inline_frame + ): + self.print_newline() + + def handle_end_block(self, current_token): + # statements must all be closed when their container closes + self.handle_whitespace_and_comments(current_token) + + while self._flags.mode == MODE.Statement: + self.restore_mode() + + empty_braces = self._flags.last_token.type == TOKEN.START_BLOCK + + # try inline_frame (only set if opt.braces-preserve-inline) first + if self._flags.inline_frame and not empty_braces: + self._output.space_before_token = True + elif self._options.brace_style == "expand": + if not empty_braces: + self.print_newline() + else: + # skip {} + if not empty_braces: + if ( + self.is_array(self._flags.mode) + and self._options.keep_array_indentation + ): + self._options.keep_array_indentation = False + self.print_newline() + self._options.keep_array_indentation = True + else: + self.print_newline() + + self.restore_mode() + self.print_token(current_token) + + def handle_word(self, current_token): + if current_token.type == TOKEN.RESERVED: + if ( + current_token.text in ["set", "get"] + and self._flags.mode != MODE.ObjectLiteral + ): + current_token.type = TOKEN.WORD + elif current_token.text == "import" and self._tokens.peek().text in [ + "(", + ".", + ]: + current_token.type = TOKEN.WORD + elif current_token.text in ["as", "from"] and not self._flags.import_block: + current_token.type = TOKEN.WORD + elif self._flags.mode == MODE.ObjectLiteral: + next_token = self._tokens.peek() + if next_token.text == ":": + current_token.type = TOKEN.WORD + + if self.start_of_statement(current_token): + # The conditional starts the statement if appropriate. + if ( + reserved_array(self._flags.last_token, ["var", "let", "const"]) + and current_token.type == TOKEN.WORD + ): + self._flags.declaration_statement = True + + elif ( + current_token.newlines + and not self.is_expression(self._flags.mode) + and ( + self._flags.last_token.type != TOKEN.OPERATOR + or ( + self._flags.last_token.text == "--" + or self._flags.last_token.text == "++" + ) + ) + and self._flags.last_token.type != TOKEN.EQUALS + and ( + self._options.preserve_newlines + or not reserved_array( + self._flags.last_token, ["var", "let", "const", "set", "get"] + ) + ) + ): + self.handle_whitespace_and_comments(current_token) + self.print_newline() + else: + self.handle_whitespace_and_comments(current_token) + + if self._flags.do_block and not self._flags.do_while: + if reserved_word(current_token, "while"): + # do {} ## while () + self._output.space_before_token = True + self.print_token(current_token) + self._output.space_before_token = True + self._flags.do_while = True + return + else: + # do {} should always have while as the next word. + # if we don't see the expected while, recover + self.print_newline() + self._flags.do_block = False + + # if may be followed by else, or not + # Bare/inline ifs are tricky + # Need to unwind the modes correctly: if (a) if (b) c(); else d(); else + # e(); + if self._flags.if_block: + if (not self._flags.else_block) and reserved_word(current_token, "else"): + self._flags.else_block = True + else: + while self._flags.mode == MODE.Statement: + self.restore_mode() + + self._flags.if_block = False + + if self._flags.in_case_statement and reserved_array( + current_token, ["case", "default"] + ): + self.print_newline() + if (not self._flags.case_block) and ( + self._flags.case_body or self._options.jslint_happy + ): + self.deindent() + self._flags.case_body = False + self.print_token(current_token) + self._flags.in_case = True + return + + if self._flags.last_token.type in [ + TOKEN.COMMA, + TOKEN.START_EXPR, + TOKEN.EQUALS, + TOKEN.OPERATOR, + ]: + if not self.start_of_object_property() and not ( + # start of object property is different for numeric values with +/- prefix operators + self._flags.last_token.text in ["+", "-"] + and self._last_last_text == ":" + and self._flags.parent.mode == MODE.ObjectLiteral + ): + self.allow_wrap_or_preserved_newline(current_token) + + if reserved_word(current_token, "function"): + if self._flags.last_token.text in ["}", ";"] or ( + self._output.just_added_newline() + and not ( + self._flags.last_token.text in ["(", "[", "{", ":", "=", ","] + or self._flags.last_token.type == TOKEN.OPERATOR + ) + ): + # make sure there is a nice clean space of at least one blank line + # before a new function definition, except in arrays + if ( + not self._output.just_added_blankline() + and current_token.comments_before is None + ): + self.print_newline() + self.print_newline(True) + + if ( + self._flags.last_token.type == TOKEN.RESERVED + or self._flags.last_token.type == TOKEN.WORD + ): + if reserved_array( + self._flags.last_token, ["get", "set", "new", "export"] + ) or reserved_array( + self._flags.last_token, self._newline_restricted_tokens + ): + self._output.space_before_token = True + elif ( + reserved_word(self._flags.last_token, "default") + and self._last_last_text == "export" + ): + self._output.space_before_token = True + elif self._flags.last_token.text == "declare": + # accomodates Typescript declare function formatting + self._output.space_before_token = True + else: + self.print_newline() + elif ( + self._flags.last_token.type == TOKEN.OPERATOR + or self._flags.last_token.text == "=" + ): + # foo = function + self._output.space_before_token = True + elif not self._flags.multiline_frame and ( + self.is_expression(self._flags.mode) or self.is_array(self._flags.mode) + ): + # (function + pass + else: + self.print_newline() + + self.print_token(current_token) + self._flags.last_word = current_token.text + return + + prefix = "NONE" + + if self._flags.last_token.type == TOKEN.END_BLOCK: + if self._previous_flags.inline_frame: + prefix = "SPACE" + elif not reserved_array( + current_token, ["else", "catch", "finally", "from"] + ): + prefix = "NEWLINE" + else: + if self._options.brace_style in ["expand", "end-expand"] or ( + self._options.brace_style == "none" and current_token.newlines + ): + prefix = "NEWLINE" + else: + prefix = "SPACE" + self._output.space_before_token = True + elif ( + self._flags.last_token.type == TOKEN.SEMICOLON + and self._flags.mode == MODE.BlockStatement + ): + # TODO: Should this be for STATEMENT as well? + prefix = "NEWLINE" + elif self._flags.last_token.type == TOKEN.SEMICOLON and self.is_expression( + self._flags.mode + ): + prefix = "SPACE" + elif self._flags.last_token.type == TOKEN.STRING: + prefix = "NEWLINE" + elif ( + self._flags.last_token.type == TOKEN.RESERVED + or self._flags.last_token.type == TOKEN.WORD + or ( + self._flags.last_token.text == "*" + and ( + self._last_last_text in ["function", "yield"] + or ( + self._flags.mode == MODE.ObjectLiteral + and self._last_last_text in ["{", ","] + ) + ) + ) + ): + prefix = "SPACE" + elif self._flags.last_token.type == TOKEN.START_BLOCK: + if self._flags.inline_frame: + prefix = "SPACE" + else: + prefix = "NEWLINE" + elif self._flags.last_token.type == TOKEN.END_EXPR: + self._output.space_before_token = True + prefix = "NEWLINE" + + if ( + reserved_array(current_token, Tokenizer.line_starters) + and self._flags.last_token.text != ")" + ): + if ( + self._flags.inline_frame + or self._flags.last_token.text == "else " + or self._flags.last_token.text == "export" + ): + prefix = "SPACE" + else: + prefix = "NEWLINE" + + if reserved_array(current_token, ["else", "catch", "finally"]): + if ( + ( + not ( + self._flags.last_token.type == TOKEN.END_BLOCK + and self._previous_flags.mode == MODE.BlockStatement + ) + ) + or self._options.brace_style == "expand" + or self._options.brace_style == "end-expand" + or (self._options.brace_style == "none" and current_token.newlines) + ) and not self._flags.inline_frame: + self.print_newline() + else: + self._output.trim(True) + # If we trimmed and there's something other than a close block before us + # put a newline back in. Handles '} // comment' scenario. + if self._output.current_line.last() != "}": + self.print_newline() + + self._output.space_before_token = True + + elif prefix == "NEWLINE": + if reserved_array(self._flags.last_token, _special_word_set): + # no newline between return nnn + self._output.space_before_token = True + elif self._flags.last_token.text == "declare" and reserved_array( + current_token, ["var", "let", "const"] + ): + # accomodates Typescript declare formatting + self._output.space_before_token = True + elif self._flags.last_token.type != TOKEN.END_EXPR: + if ( + self._flags.last_token.type != TOKEN.START_EXPR + or not (reserved_array(current_token, ["var", "let", "const"])) + ) and self._flags.last_token.text != ":": + # no need to force newline on VAR - + # for (var x = 0... + if ( + reserved_word(current_token, "if") + and self._flags.last_token.text == "else" + ): + self._output.space_before_token = True + else: + self.print_newline() + elif ( + reserved_array(current_token, Tokenizer.line_starters) + and self._flags.last_token.text != ")" + ): + self.print_newline() + elif ( + self._flags.multiline_frame + and self.is_array(self._flags.mode) + and self._flags.last_token.text == "," + and self._last_last_text == "}" + ): + self.print_newline() # }, in lists get a newline + elif prefix == "SPACE": + self._output.space_before_token = True + + if current_token.previous and ( + current_token.previous.type == TOKEN.WORD + or current_token.previous.type == TOKEN.RESERVED + ): + self._output.space_before_token = True + + self.print_token(current_token) + self._flags.last_word = current_token.text + + if current_token.type == TOKEN.RESERVED: + if current_token.text == "do": + self._flags.do_block = True + elif current_token.text == "if": + self._flags.if_block = True + elif current_token.text == "import": + self._flags.import_block = True + elif current_token.text == "from" and self._flags.import_block: + self._flags.import_block = False + + def handle_semicolon(self, current_token): + if self.start_of_statement(current_token): + # The conditional starts the statement if appropriate. + # Semicolon can be the start (and end) of a statement + self._output.space_before_token = False + else: + self.handle_whitespace_and_comments(current_token) + + next_token = self._tokens.peek() + while ( + self._flags.mode == MODE.Statement + and not (self._flags.if_block and reserved_word(next_token, "else")) + and not self._flags.do_block + ): + self.restore_mode() + + if self._flags.import_block: + self._flags.import_block = False + + self.print_token(current_token) + + def handle_string(self, current_token): + if ( + current_token.text[0] == "`" + and current_token.newlines == 0 + and current_token.whitespace_before == "" + and ( + self._flags.last_token.type == TOKEN.WORD + or current_token.previous.text == ")" + ) + ): + # This conditional checks backtick strings and makes no changes + pass + elif self.start_of_statement(current_token): + # The conditional starts the statement if appropriate. + # One difference - strings want at least a space before + self._output.space_before_token = True + else: + self.handle_whitespace_and_comments(current_token) + if ( + self._flags.last_token.type in [TOKEN.RESERVED, TOKEN.WORD] + or self._flags.inline_frame + ): + self._output.space_before_token = True + elif self._flags.last_token.type in [ + TOKEN.COMMA, + TOKEN.START_EXPR, + TOKEN.EQUALS, + TOKEN.OPERATOR, + ]: + if not self.start_of_object_property(): + self.allow_wrap_or_preserved_newline(current_token) + elif ( + current_token.text[0] == "`" + and self._flags.last_token.type == TOKEN.END_EXPR + and current_token.previous.text in ["]", ")"] + and current_token.newlines == 0 + ): + self._output.space_before_token = True + else: + self.print_newline() + + self.print_token(current_token) + + def handle_equals(self, current_token): + if self.start_of_statement(current_token): + # The conditional starts the statement if appropriate. + pass + else: + self.handle_whitespace_and_comments(current_token) + + if self._flags.declaration_statement: + # just got an '=' in a var-line, different line breaking rules will + # apply + self._flags.declaration_assignment = True + + self._output.space_before_token = True + self.print_token(current_token) + self._output.space_before_token = True + + def handle_comma(self, current_token): + self.handle_whitespace_and_comments(current_token, True) + + self.print_token(current_token) + self._output.space_before_token = True + + if self._flags.declaration_statement: + if self.is_expression(self._flags.parent.mode): + # do not break on comma, for ( var a = 1, b = 2 + self._flags.declaration_assignment = False + + if self._flags.declaration_assignment: + self._flags.declaration_assignment = False + self.print_newline(preserve_statement_flags=True) + elif self._options.comma_first: + # for comma-first, we want to allow a newline before the comma + # to turn into a newline after the comma, which we will fixup + # later + self.allow_wrap_or_preserved_newline(current_token) + + elif self._flags.mode == MODE.ObjectLiteral or ( + self._flags.mode == MODE.Statement + and self._flags.parent.mode == MODE.ObjectLiteral + ): + if self._flags.mode == MODE.Statement: + self.restore_mode() + + if not self._flags.inline_frame: + self.print_newline() + elif self._options.comma_first: + # EXPR or DO_BLOCK + # for comma-first, we want to allow a newline before the comma + # to turn into a newline after the comma, which we will fixup later + self.allow_wrap_or_preserved_newline(current_token) + + def handle_operator(self, current_token): + isGeneratorAsterisk = current_token.text == "*" and ( + reserved_array(self._flags.last_token, ["function", "yield"]) + or ( + self._flags.last_token.type + in [TOKEN.START_BLOCK, TOKEN.COMMA, TOKEN.END_BLOCK, TOKEN.SEMICOLON] + ) + ) + isUnary = current_token.text in ["+", "-"] and ( + self._flags.last_token.type + in [TOKEN.START_BLOCK, TOKEN.START_EXPR, TOKEN.EQUALS, TOKEN.OPERATOR] + or self._flags.last_token.text in Tokenizer.line_starters + or self._flags.last_token.text == "," + ) + + if self.start_of_statement(current_token): + # The conditional starts the statement if appropriate. + pass + else: + preserve_statement_flags = not isGeneratorAsterisk + self.handle_whitespace_and_comments(current_token, preserve_statement_flags) + + # hack for actionscript's import .*; + if current_token.text == "*" and self._flags.last_token.type == TOKEN.DOT: + self.print_token(current_token) + return + + if current_token.text == "::": + # no spaces around the exotic namespacing syntax operator + self.print_token(current_token) + return + + if current_token.text in ["-", "+"] and self.start_of_object_property(): + # numeric value with +/- symbol in front as a property + self.print_token(current_token) + return + + # Allow line wrapping between operators when operator_position is + # set to before or preserve + if ( + self._flags.last_token.type == TOKEN.OPERATOR + and self._options.operator_position in OPERATOR_POSITION_BEFORE_OR_PRESERVE + ): + self.allow_wrap_or_preserved_newline(current_token) + + if current_token.text == ":" and self._flags.in_case: + self.print_token(current_token) + self._flags.in_case = False + self._flags.case_body = True + if self._tokens.peek().type != TOKEN.START_BLOCK: + self.indent() + self.print_newline() + self._flags.case_block = False + else: + self._output.space_before_token = True + self._flags.case_block = True + + return + + space_before = True + space_after = True + in_ternary = False + + if current_token.text == ":": + if self._flags.ternary_depth == 0: + # Colon is invalid javascript outside of ternary and object, + # but do our best to guess what was meant. + space_before = False + else: + self._flags.ternary_depth -= 1 + in_ternary = True + elif current_token.text == "?": + self._flags.ternary_depth += 1 + + # let's handle the operator_position option prior to any conflicting + # logic + if ( + (not isUnary) + and (not isGeneratorAsterisk) + and self._options.preserve_newlines + and current_token.text in Tokenizer.positionable_operators + ): + isColon = current_token.text == ":" + isTernaryColon = isColon and in_ternary + isOtherColon = isColon and not in_ternary + + if self._options.operator_position == OPERATOR_POSITION["before_newline"]: + # if the current token is : and it's not a ternary statement + # then we set space_before to false + self._output.space_before_token = not isOtherColon + + self.print_token(current_token) + + if (not isColon) or isTernaryColon: + self.allow_wrap_or_preserved_newline(current_token) + + self._output.space_before_token = True + + return + + elif self._options.operator_position == OPERATOR_POSITION["after_newline"]: + # if the current token is anything but colon, or (via deduction) it's a colon and in a ternary statement, + # then print a newline. + self._output.space_before_token = True + + if (not isColon) or isTernaryColon: + if self._tokens.peek().newlines: + self.print_newline(preserve_statement_flags=True) + else: + self.allow_wrap_or_preserved_newline(current_token) + else: + self._output.space_before_token = False + + self.print_token(current_token) + + self._output.space_before_token = True + return + + elif ( + self._options.operator_position == OPERATOR_POSITION["preserve_newline"] + ): + if not isOtherColon: + self.allow_wrap_or_preserved_newline(current_token) + + # if we just added a newline, or the current token is : and it's not a ternary statement, + # then we set space_before to false + self._output.space_before_token = not ( + self._output.just_added_newline() or isOtherColon + ) + + self.print_token(current_token) + + self._output.space_before_token = True + return + + if isGeneratorAsterisk: + self.allow_wrap_or_preserved_newline(current_token) + space_before = False + next_token = self._tokens.peek() + space_after = next_token and next_token.type in [TOKEN.WORD, TOKEN.RESERVED] + elif current_token.text == "...": + self.allow_wrap_or_preserved_newline(current_token) + space_before = self._flags.last_token.type == TOKEN.START_BLOCK + space_after = False + elif current_token.text in ["--", "++", "!", "~"] or isUnary: + if ( + self._flags.last_token.type == TOKEN.COMMA + or self._flags.last_token.type == TOKEN.START_EXPR + ): + self.allow_wrap_or_preserved_newline(current_token) + + space_before = False + space_after = False + + # http://www.ecma-international.org/ecma-262/5.1/#sec-7.9.1 + # if there is a newline between -- or ++ and anything else we + # should preserve it. + if current_token.newlines and ( + current_token.text == "--" + or current_token.text == "++" + or current_token.text == "~" + ): + new_line_needed = ( + reserved_array(self._flags.last_token, _special_word_set) + and current_token.newlines + ) + if new_line_needed and ( + self._previous_flags.if_block or self._previous_flags.else_block + ): + self.restore_mode() + self.print_newline(new_line_needed, True) + + if self._flags.last_token.text == ";" and self.is_expression( + self._flags.mode + ): + # for (;; ++i) + # ^^ + space_before = True + + if self._flags.last_token.type == TOKEN.RESERVED: + space_before = True + elif self._flags.last_token.type == TOKEN.END_EXPR: + space_before = not ( + self._flags.last_token.text == "]" + and current_token.text in ["--", "++"] + ) + elif self._flags.last_token.type == TOKEN.OPERATOR: + # a++ + ++b + # a - -b + space_before = current_token.text in [ + "--", + "-", + "++", + "+", + ] and self._flags.last_token.text in ["--", "-", "++", "+"] + # + and - are not unary when preceeded by -- or ++ operator + # a-- + b + # a * +b + # a - -b + if current_token.text in ["-", "+"] and self._flags.last_token.text in [ + "--", + "++", + ]: + space_after = True + + if ( + ( + self._flags.mode == MODE.BlockStatement + and not self._flags.inline_frame + ) + or self._flags.mode == MODE.Statement + ) and self._flags.last_token.text in ["{", ";"]: + # { foo: --i } + # foo(): --bar + self.print_newline() + + if space_before: + self._output.space_before_token = True + + self.print_token(current_token) + + if space_after: + self._output.space_before_token = True + + def handle_block_comment(self, current_token, preserve_statement_flags): + if self._output.raw: + self._output.add_raw_token(current_token) + if ( + current_token.directives + and current_token.directives.get("preserve") == "end" + ): + # If we're testing the raw output behavior, do not allow a + # directive to turn it off. + self._output.raw = self._options.test_output_raw + return + + if current_token.directives: + self.print_newline(preserve_statement_flags=preserve_statement_flags) + self.print_token(current_token) + if current_token.directives.get("preserve") == "start": + self._output.raw = True + + self.print_newline(preserve_statement_flags=True) + return + + # inline block + if ( + not self.acorn.newline.search(current_token.text) + and not current_token.newlines + ): + self._output.space_before_token = True + self.print_token(current_token) + self._output.space_before_token = True + return + + lines = self.acorn.allLineBreaks.split(current_token.text) + javadoc = False + starless = False + last_indent = current_token.whitespace_before + last_indent_length = len(last_indent) + + # block comment starts with a new line + self.print_newline(preserve_statement_flags=preserve_statement_flags) + + # first line always indented + self.print_token(current_token, lines[0]) + self.print_newline(preserve_statement_flags=preserve_statement_flags) + + if len(lines) > 1: + lines = lines[1:] + javadoc = not any( + l for l in lines if (l.strip() == "" or (l.lstrip())[0] != "*") + ) + starless = all(l.startswith(last_indent) or l.strip() == "" for l in lines) + + if javadoc: + self._flags.alignment = 1 + + for line in lines: + if javadoc: + # javadoc: reformat and re-indent + self.print_token(current_token, line.lstrip()) + elif starless and len(line) > last_indent_length: + # starless: re-indent non-empty content, avoiding trim + self.print_token(current_token, line[last_indent_length:]) + else: + # normal comments output raw + self._output.current_line.set_indent(-1) + self._output.add_token(line) + + # for comments on their own line or more than one line, + # make sure there's a new line after + self.print_newline(preserve_statement_flags=preserve_statement_flags) + + self._flags.alignment = 0 + + def handle_comment(self, current_token, preserve_statement_flags): + if current_token.newlines: + self.print_newline(preserve_statement_flags=preserve_statement_flags) + + if not current_token.newlines: + self._output.trim(True) + + self._output.space_before_token = True + self.print_token(current_token) + self.print_newline(preserve_statement_flags=preserve_statement_flags) + + def handle_dot(self, current_token): + if self.start_of_statement(current_token): + # The conditional starts the statement if appropriate. + pass + else: + self.handle_whitespace_and_comments(current_token, True) + + if re.search("^([0-9])+$", self._flags.last_token.text): + self._output.space_before_token = True + + if reserved_array(self._flags.last_token, _special_word_set): + self._output.space_before_token = False + else: + # allow preserved newlines before dots in general + # force newlines on dots after close paren when break_chained - for + # bar().baz() + self.allow_wrap_or_preserved_newline( + current_token, + self._flags.last_token.text == ")" + and self._options.break_chained_methods, + ) + + # Only unindent chained method dot if this dot starts a new line. + # Otherwise the automatic extra indentation removal + # will handle any over indent + if self._options.unindent_chained_methods and self._output.just_added_newline(): + self.deindent() + + self.print_token(current_token) + + def handle_unknown(self, current_token, preserve_statement_flags): + self.print_token(current_token) + if current_token.text[-1] == "\n": + self.print_newline(preserve_statement_flags=preserve_statement_flags) + + def handle_eof(self, current_token): + # Unwind any open statements + while self._flags.mode == MODE.Statement: + self.restore_mode() + + self.handle_whitespace_and_comments(current_token) |