Handle other-modules in exe

Author: Avi-D-coder

src/Hie/Cabal/Parser.hs

@@ -9,6 +9,7 @@ import Data.Char
 import Data.Maybe
 import Data.Text (Text)
 import qualified Data.Text as T
+import System.FilePath.Posix ((</>))
 
 type Name = Text
 
@@ -75,7 +76,7 @@ parseBench = parseSecMain (Comp Bench) "benchmark"
 parseSecMain :: (Name -> Path -> Component) -> Text -> Indent -> Parser [Component]
 parseSecMain c s i = do
   n <- componentHeader i s
-  p <- pathMain (i + 1) ["./"] ""
+  p <- pathMain (i + 1) ["./"] "" []
   pure $ map (c n) p
 
 parseQuoted :: Parser Text
@@ -105,12 +106,18 @@ parseList i = items <|> (emptyOrComLine >> indent i >> items)
           <|> pure []
       pure $ h : t
 
-pathMain :: Indent -> [Text] -> Text -> Parser [Text]
-pathMain i p m =
-  (hsSourceDir i >>= (\p' -> pathMain i p' m))
-    <|> (field i "main-is" (const parseString) >>= pathMain i p)
-    <|> (skipBlockLine i >> pathMain i p m)
-    <|> pure (map (<> "/" <> m) p)
+pathMain :: Indent -> [Text] -> Text -> [Text] -> Parser [Text]
+pathMain i p m o =
+  (hsSourceDir i >>= (\p' -> pathMain i p' m o))
+    <|> (field i "main-is" (const parseString) >>= flip (pathMain i p) o)
+    <|> (field i "other-modules" parseList >>= pathMain i p m)
+    <|> (skipBlockLine i >> pathMain i p m o)
+    <|> pure (map (<//> m) p <> [p' <//> (o' <> ".hs") | p' <- p, o' <- o])
+
+
+(<//>) :: Text -> Text -> Text
+a <//> b = T.pack (T.unpack a </> T.unpack b)
+infixr 5 <//>
 
 parseSec :: Indent -> Text -> (Name -> Path -> Component) -> Parser [Component]
 parseSec i compType compCon = do