Data.Tree.Extra: add setSubForest

3 files changed, 9 insertions, 5 deletions

diff --git a/much.cabal b/much.cabal
index 9aae2a2..fb33c6d 100644
--- a/much.cabal
+++ b/much.cabal
@@ -66,6 +66,7 @@ library
                  , Codec.MIME.QuotedPrintable
                  , Codec.MIME.Type
                  , Data.Aeson.Extends
+                 , Data.Tree.Extra
   build-depends:   base
                  , aeson
                  , attoparsec
diff --git a/src/Data/Tree/Extra.hs b/src/Data/Tree/Extra.hs
new file mode 100644
index 0000000..256af08
--- /dev/null
+++ b/src/Data/Tree/Extra.hs
@@ -0,0 +1,7 @@
+module Data.Tree.Extra where
+
+import Data.Tree (Forest, Tree(subForest))
+
+
+setSubForest :: Forest a -> Tree a -> Tree a
+setSubForest sf t = t { subForest = sf }
diff --git a/src/Much/Action.hs b/src/Much/Action.hs
index 4b1d4b1..d76e503 100644
--- a/src/Much/Action.hs
+++ b/src/Much/Action.hs
@@ -5,13 +5,13 @@ module Much.Action where
 
 import Blessings.String.WCWidth
 import Data.Maybe
+import Data.Tree.Extra (setSubForest)
 import Scanner
 import Much.State
 import Much.TagUtils
 import Much.TreeSearch
 import Much.TreeView
 import Much.TreeZipperUtils
-import qualified Data.Tree as Tree
 import qualified Data.Tree.Zipper as Z
 import qualified Notmuch
 import qualified Notmuch.Message as Notmuch
@@ -216,7 +216,3 @@ topOverrun State{..} =
 botOverrun :: State -> Int
 botOverrun State{..} =
     max 0 (linearPos cursor - yoffset - (screenHeight - length headBuffer - 1))
-
-
-setSubForest :: Tree.Forest a -> Tree.Tree a -> Tree.Tree a
-setSubForest sf t = t { Tree.subForest = sf }