Allow configuring the base indentation level

Author: vinistock

lib/prettier_print.rb

@@ -360,6 +360,11 @@ def self.for(output)
   # a forced line, or the maximum width.
   MODE_FLAT = 2
 
+  # The default indentation for printing is zero, assuming that the code starts
+  # at the top level. That can be changed if desired to start from a different
+  # indentation level.
+  DEFAULT_INDENTATION = 0
+
   # This is a convenience method which is same as follows:
   #
   #   begin
@@ -373,11 +378,12 @@ def self.format(
     output = "".dup,
     maxwidth = 80,
     newline = DEFAULT_NEWLINE,
-    genspace = DEFAULT_GENSPACE
+    genspace = DEFAULT_GENSPACE,
+    indentation = DEFAULT_INDENTATION
   )
     q = new(output, maxwidth, newline, &genspace)
     yield q
-    q.flush
+    q.flush(indentation)
     output
   end
 
@@ -481,7 +487,7 @@ def current_group
 
   # Flushes all of the generated print tree onto the output buffer, then clears
   # the generated tree from memory.
-  def flush
+  def flush(base_indentation = DEFAULT_INDENTATION)
     # First, get the root group, since we placed one at the top to begin with.
     doc = groups.first
 
@@ -491,7 +497,7 @@ def flush
 
     # This is our command stack. A command consists of a triplet of an
     # indentation level, the mode (break or flat), and a doc node.
-    commands = [[0, MODE_BREAK, doc]]
+    commands = [[base_indentation, MODE_BREAK, doc]]
 
     # This is a small optimization boolean. It keeps track of whether or not
     # when we hit a group node we should check if it fits on the same line.

test/prettyprint_test.rb

@@ -517,4 +517,69 @@ def test_27
 
 end
 
+class DifferentIndentationLevelExample < Test::Unit::TestCase # :nodoc:
+  def format(indentation)
+    PrettierPrint.format(
+      ''.dup,
+      2,
+      PrettierPrint::DEFAULT_NEWLINE,
+      PrettierPrint::DEFAULT_GENSPACE,
+      indentation
+    ) {|q|
+      q.group {
+        q.breakable_empty
+        q.text("[")
+
+        q.indent {
+          q.breakable_empty
+          q.text("1")
+        }
+
+        q.breakable_empty
+        q.text("]")
+      }
+    }.delete_prefix("\n")
+  end
+
+  def test_default
+    expected = <<'End'.chomp
+[
+  1
+]
+End
+
+    assert_equal(expected, format(0))
+  end
+
+  def test_level_2
+    expected = <<'End'.chomp
+  [
+    1
+  ]
+End
+
+    assert_equal(expected, format(2))
+  end
+
+  def test_level_4
+    expected = <<'End'.chomp
+    [
+      1
+    ]
+End
+
+    assert_equal(expected, format(4))
+  end
+
+  def test_level_6
+    expected = <<'End'.chomp
+      [
+        1
+      ]
+End
+
+    assert_equal(expected, format(6))
+  end
+end
+
 end