From 95f5dffb70a06d84a6c05b4df2e17b29bd93942a Mon Sep 17 00:00:00 2001
From: Sebastian Wild <wildsebastian@users.noreply.github.com>
Date: Sat, 17 Nov 2018 13:48:58 +0100
Subject: Implement behaviour for cabal new-* (#61)

Add list of components to Unit data type to handle v2 based builds
per cabal unit.
---
 src/CabalHelper/Compiletime/Program/CabalInstall.hs | 7 ++++++-
 src/CabalHelper/Compiletime/Program/Stack.hs        | 3 ++-
 2 files changed, 8 insertions(+), 2 deletions(-)

(limited to 'src/CabalHelper/Compiletime/Program')

diff --git a/src/CabalHelper/Compiletime/Program/CabalInstall.hs b/src/CabalHelper/Compiletime/Program/CabalInstall.hs
index a4df188..637d577 100644
--- a/src/CabalHelper/Compiletime/Program/CabalInstall.hs
+++ b/src/CabalHelper/Compiletime/Program/CabalInstall.hs
@@ -1,3 +1,5 @@
+{-# LANGUAGE DataKinds #-}
+
 -- cabal-helper: Simple interface to Cabal's configuration state
 -- Copyright (C) 2018  Daniel Gröber <cabal-helper@dxld.at>
 --
@@ -181,7 +183,7 @@ cabalWithGHCProgOpts = concat
       else []
   ]
 
-planUnits :: CP.PlanJson -> IO [Unit]
+planUnits :: CP.PlanJson -> IO [Unit 'V2]
 planUnits plan = do
     units <- fmap catMaybes $ mapM takeunit $ Map.elems $ CP.pjUnits plan
     case lefts units of
@@ -195,13 +197,16 @@ planUnits plan = do
       { uType=CP.UnitTypeLocal
       , uDistDir=Just distdirv1
       , uPkgSrc=Just (CP.LocalUnpackedPackage pkgdir)
+      , uComps=comps
       } = do
         cabal_file <- Cabal.findCabalFile pkgdir
+        let uiV2Components = map (Text.unpack . CP.dispCompName) $ Map.keys comps
         return $ Just $ Right $ Unit
           { uUnitId     = UnitId $ Text.unpack (coerce (CP.uId u))
           , uPackageDir = pkgdir
           , uCabalFile  = CabalFile cabal_file
           , uDistDir    = DistDirLib distdirv1
+          , uImpl       = UnitImplV2 {..}
           }
     takeunit u@CP.Unit {uType=CP.UnitTypeLocal} =
       return $ Just $ Left u
diff --git a/src/CabalHelper/Compiletime/Program/Stack.hs b/src/CabalHelper/Compiletime/Program/Stack.hs
index 4f3680f..f4ada8f 100644
--- a/src/CabalHelper/Compiletime/Program/Stack.hs
+++ b/src/CabalHelper/Compiletime/Program/Stack.hs
@@ -36,7 +36,7 @@ import Prelude
 import CabalHelper.Compiletime.Types
 import CabalHelper.Compiletime.Types.RelativePath
 
-getUnit :: QueryEnvI c 'Stack -> CabalFile -> IO Unit
+getUnit :: QueryEnvI c 'Stack -> CabalFile -> IO (Unit 'Stack)
 getUnit qe cabal_file@(CabalFile cabal_file_path) = do
   let pkgdir = takeDirectory cabal_file_path
   let pkg_name = dropExtension $ takeFileName cabal_file_path
@@ -47,6 +47,7 @@ getUnit qe cabal_file@(CabalFile cabal_file_path) = do
     , uPackageDir = pkgdir
     , uCabalFile  = cabal_file
     , uDistDir    = DistDirLib distdirv1
+    , uImpl       = UnitImplStack
     }
 
 -- TODO: patch ghc/ghc-pkg program paths like in ghc-mod when using stack so
-- 
cgit v1.2.3