From d4d3426b4e7a039cf855cffff52b7cac2317082b Mon Sep 17 00:00:00 2001
From: Ben Ogle <ogle.ben@gmail.com>
Date: Thu, 18 Jul 2013 11:27:04 -0700
Subject: [PATCH] Add `foldAllAtIndentLevel()`

---
 spec/app/language-mode-spec.coffee | 23 +++++++++++++++++++++++
 src/app/edit-session.coffee        |  3 +++
 src/app/language-mode.coffee       | 12 ++++++++++++
 3 files changed, 38 insertions(+)

diff --git a/spec/app/language-mode-spec.coffee b/spec/app/language-mode-spec.coffee
index e10fd4963..d7ef94221 100644
--- a/spec/app/language-mode-spec.coffee
+++ b/spec/app/language-mode-spec.coffee
@@ -191,6 +191,29 @@ describe "LanguageMode", ->
           fold4 = editSession.lineForScreenRow(3).fold
           expect([fold4.getStartRow(), fold4.getEndRow()]).toEqual [6, 8]
 
+      describe ".foldAllAtIndentLevel()", ->
+        it "folds every foldable range at a given indentLevel", ->
+          editSession.foldAllAtIndentLevel(2)
+
+          fold1 = editSession.lineForScreenRow(6).fold
+          expect([fold1.getStartRow(), fold1.getEndRow()]).toEqual [6, 8]
+          fold1.destroy()
+
+          fold2 = editSession.lineForScreenRow(11).fold
+          expect([fold2.getStartRow(), fold2.getEndRow()]).toEqual [11, 14]
+          fold2.destroy()
+
+        it "does not fold anything but the indentLevel", ->
+          editSession.foldAllAtIndentLevel(0)
+
+          fold1 = editSession.lineForScreenRow(0).fold
+          expect([fold1.getStartRow(), fold1.getEndRow()]).toEqual [0, 19]
+          fold1.destroy()
+
+          fold2 = editSession.lineForScreenRow(5).fold
+          expect(fold2).toBeFalsy()
+
+
 
 
 
diff --git a/src/app/edit-session.coffee b/src/app/edit-session.coffee
index d2ebdf4fe..14647bc8f 100644
--- a/src/app/edit-session.coffee
+++ b/src/app/edit-session.coffee
@@ -531,6 +531,9 @@ class EditSession
   unfoldAll: ->
     @languageMode.unfoldAll()
 
+  foldAllAtIndentLevel: (indentLevel) ->
+    @languageMode.foldAllAtIndentLevel(indentLevel)
+
   # Folds the current row.
   foldCurrentRow: ->
     bufferRow = @bufferPositionForScreenPosition(@getCursorScreenPosition()).row
diff --git a/src/app/language-mode.coffee b/src/app/language-mode.coffee
index b097a67d8..5ded3427b 100644
--- a/src/app/language-mode.coffee
+++ b/src/app/language-mode.coffee
@@ -98,6 +98,18 @@ class LanguageMode
     for row in [@buffer.getLastRow()..0]
       fold.destroy() for fold in @editSession.displayBuffer.foldsStartingAtBufferRow(row)
 
+  # Fold all comment and code blocks at a given indentLevel
+  #
+  # indentLevel - A {Number} indicating indentLevel; 0 based.
+  foldAllAtIndentLevel: (indentLevel) ->
+    for currentRow in [0..@buffer.getLastRow()]
+      [startRow, endRow] = @rowRangeForFoldAtBufferRow(currentRow) ? []
+      continue unless startRow?
+
+      # assumption: startRow will always be the min indent level for the entire range
+      if @editSession.indentationForBufferRow(startRow) == indentLevel
+        @editSession.createFold(startRow, endRow)
+
   # Given a buffer row, creates a fold at it.
   #
   # bufferRow - A {Number} indicating the buffer row