Stop empty module display

src/Dodge/Module.hs

@@ -6,18 +6,28 @@ import Dodge.Data
 --import Dodge.Combine.Module
 
 import qualified Data.Map.Strict as M
-import Control.Lens
-import Data.Maybe
+--import Control.Lens
+--import Data.Maybe
 
 --moduleSizes :: Item -> Int
 --moduleSizes = moduleFold (+) (^? modSize) 0
 moduleSizes :: Item -> Int
-moduleSizes = length . _iyModules . _itType
+moduleSizes = sum . fmap moduleSize . _iyModules . _itType
+
+moduleSize :: ItemModuleType -> Int
+moduleSize imt = case imt of
+    EMPTYMODULE -> 0
+    _ -> 1
 
 --moduleStrings :: Item -> [String]
 --moduleStrings = moduleFold (++) (^? modName) []
 moduleStrings :: Item -> [String]
-moduleStrings = map show . M.elems . _iyModules . _itType
+moduleStrings = map moduleName . M.elems . _iyModules . _itType
+
+moduleName :: ItemModuleType -> String
+moduleName imt = case imt of
+    EMPTYMODULE -> ""
+    _ -> '+':show imt
 
 --moduleFold :: (m -> m -> m) -> (ItemModule -> Maybe m) -> m -> Item -> m
 --moduleFold g f e = foldr (g . fromMaybe e . f) e