Add support for getters and setters

praveenvijayan · Feb 9, 2012 · 23fd0b7 · 23fd0b7
1 parent dc122f4
commit 23fd0b7
Show file tree

Hide file tree

Showing 4 changed files with 23 additions and 0 deletions.
diff --git a/beautify.js b/beautify.js
@@ -920,6 +920,10 @@ function js_beautify(js_source_text, options) {
                 } else {
                     prefix = 'NEWLINE';
                 }
+
+                if (token_text === 'function' && (last_text === 'get' || last_text === 'set')) {
+                    prefix = 'SPACE';
+                }
             }
 
             if (flags.if_line && last_type === 'TK_END_EXPR') {

diff --git a/python/jsbeautifier/__init__.py b/python/jsbeautifier/__init__.py
@@ -823,6 +823,9 @@ def handle_word(self, token_text):
             else:
                 prefix = 'NEWLINE'
 
+            if token_text == 'function' and self.last_text in ['get', 'set']:
+                prefix = 'SPACE'
+
         if token_text in ['else', 'catch', 'finally']:
             if self.last_type != 'TK_END_BLOCK' \
                or self.opts.brace_style == 'expand' \

diff --git a/python/jsbeautifier/tests/testjsbeautifier.py b/python/jsbeautifier/tests/testjsbeautifier.py
@@ -409,6 +409,14 @@ def test_beautifier(self):
 
         bt('"foo""bar""baz"', '"foo"\n"bar"\n"baz"')
         bt("'foo''bar''baz'", "'foo'\n'bar'\n'baz'")
+        bt("{\n    get foo() {}\n}")
+        bt("{\n    var a = get\n    foo();\n}")
+        bt("{\n    set foo() {}\n}")
+        bt("{\n    var a = set\n    foo();\n}")
+        bt("var x = {\n    get function()\n}")
+        bt("var x = {\n    set function()\n}")
+        bt("var x = set\n\nfunction() {}")
+
 
     def decodesto(self, input, expectation=None):
         self.assertEqual(

diff --git a/tests/beautify-tests.js b/tests/beautify-tests.js
@@ -460,6 +460,14 @@ function run_beautifier_tests(test_obj)
     bt('"foo""bar""baz"', '"foo"\n"bar"\n"baz"');
     bt("'foo''bar''baz'", "'foo'\n'bar'\n'baz'");
 
+    bt("{\n    get foo() {}\n}");
+    bt("{\n    var a = get\n    foo();\n}");
+    bt("{\n    set foo() {}\n}");
+    bt("{\n    var a = set\n    foo();\n}");
+    bt("var x = {\n    get function()\n}");
+    bt("var x = {\n    set function()\n}");
+    bt("var x = set\n\nfunction() {}");
+
     opts.space_before_conditional = false;
     bt('if(a) b()');