kowainik · chshersh · Jul 17, 2018 · Jul 17, 2018 · Jul 17, 2018 · Jul 17, 2018
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -11,6 +11,8 @@
   UseTravis-specific env variable `TRAVIS_BUILD_DIR` in created travis file.
 * [#97](https://github.com/kowainik/summoner/issues/97):
   Add cabal to created travis file.
+* [#96](https://github.com/kowainik/summoner/issues/96):
+  Add option to choose `cabal`, `stack` or both.
 
 1.0.4
 =====

diff --git a/README.md b/README.md
@@ -64,6 +64,10 @@ Here is the list of the options that could be configured for your needs:
 
 ###### Global keys
 
+* `cabal` – `true` if you want to build you project with `cabal`,
+             `false` if you don't. If not specified it would be asked during each run of the `summoner`.
+* `stack` – `true` if you want to build your project with `stack`,
+             `false` if you don't. If not specified it would be asked during each run of the `summoner`.
 * `owner` – `GitHub` login.
 * `fullName` – full name.
 * `email` – e-mail address.
@@ -114,13 +118,16 @@ The other way to specify some particular `.toml` file is `summon PROJECTNAME --f
 See the basic usage syntax below (you can check it out with `summon --help` command):
 
 ```
-summon PROJECT_NAME [with [OPTIONS]] [without [OPTIONS]]
+summon PROJECT_NAME [--cabal] [--stack]
+       [with [OPTIONS]] [without [OPTIONS]]
        [-f|--file FILENAME]  [--prelude-package PACKAGE_NAME]
        [--prelude-module MODULE_NAME]
 
 Available global options:
   -h, --help               Show this help text
   -v, --version            Show summoner's version
+  --cabal                  Cabal support for the project
+  --stack                  Stack support for the project
   -f, --file FILENAME      Path to the toml file with configurations. If not
                            specified '~/.summoner.toml' will be used if present
   --prelude-package PACKAGE_NAME

diff --git a/src/Summoner/CLI.hs b/src/Summoner/CLI.hs
@@ -187,9 +187,21 @@ preludeModP = strOption
    <> metavar "MODULE_NAME"
    <> help "Name for the module of the custom prelude to use in the project"
 
+cabalP :: Parser Decision
+cabalP = flag Idk Yes
+       $ long "cabal"
+      <> help "Cabal support for the project"
+
+stackP :: Parser Decision
+stackP = flag Idk Yes
+       $ long "stack"
+      <> help "Stack support for the project"
+
 optsP :: Parser InitOpts
 optsP = do
     projectName <- strArgument (metavar "PROJECT_NAME")
+    cabal   <- cabalP
+    stack   <- stackP
     with    <- optional withP
     without <- optional withoutP
     file    <- optional fileP
@@ -199,6 +211,8 @@ optsP = do
     pure $ InitOpts projectName file
         $ (maybeToMonoid $ with <> without)
             { cPrelude = Last $ Prelude <$> preludePack <*> preludeMod
+            , cCabal = cabal
+            , cStack = stack
             }
 
 versionP :: Parser (a -> a)

diff --git a/src/Summoner/Config.hs b/src/Summoner/Config.hs
@@ -49,6 +49,8 @@ data ConfigP (p :: Phase) = Config
     , cEmail      :: p :- Text
     , cLicense    :: p :- License
     , cGhcVer     :: p :- [GhcVer]
+    , cCabal      :: Decision
+    , cStack      :: Decision
     , cGitHub     :: Decision
     , cTravis     :: Decision
     , cAppVey     :: Decision
@@ -91,6 +93,8 @@ defaultConfig = Config
     , cEmail    = Last (Just "xrom.xkov@gmail.com")
     , cLicense  = Last (Just $ License "MIT")
     , cGhcVer   = Last (Just [])
+    , cCabal    = Idk
+    , cStack    = Idk
     , cGitHub   = Idk
     , cTravis   = Idk
     , cAppVey   = Idk
@@ -112,6 +116,8 @@ configT = Config
     <*> lastT Toml.text "email"       .= cEmail
     <*> lastT license   "license"     .= cLicense
     <*> lastT ghcVerArr "ghcVersions" .= cGhcVer
+    <*> decision        "cabal"       .= cCabal
+    <*> decision        "stack"       .= cStack
     <*> decision        "github"      .= cGitHub
     <*> decision        "travis"      .= cTravis
     <*> decision        "appveyor"    .= cAppVey
@@ -169,7 +175,9 @@ finalise Config{..} = Config
     <*> fin  "fullName"   cFullName
     <*> fin  "email"      cEmail
     <*> fin  "license"    cLicense
-    <*> fin  "ghcersions" cGhcVer
+    <*> fin  "ghcVersions" cGhcVer
+    <*> pure cCabal
+    <*> pure cStack
     <*> pure cGitHub
     <*> pure cTravis
     <*> pure cAppVey

diff --git a/src/Summoner/Project.hs b/src/Summoner/Project.hs
@@ -14,15 +14,16 @@ import NeatInterpolation (text)
 import System.Info (os)
 import System.Process (readProcess)
 
-import Summoner.Ansi (infoMessage, successMessage)
+import Summoner.Ansi (errorMessage, infoMessage, successMessage)
 import Summoner.Config (Config, ConfigP (..))
 import Summoner.Default (currentYear, defaultGHC)
 import Summoner.License (License (..), customizeLicense, githubLicenseQueryNames, licenseNames)
 import Summoner.Process ()
 import Summoner.ProjectData (CustomPrelude (..), Decision (..), ProjectData (..), parseGhcVer,
                              showGhcVer, supportedGhcVers)
 import Summoner.Question (checkUniqueName, choose, chooseYesNo, chooseYesNoBool, falseMessage,
-                          query, queryDef, queryManyRepeatOnFail, trueMessage)
+                          query, queryDef, queryManyRepeatOnFail, targetMessageWithText,
+                          trueMessage)
 import Summoner.Template (createStackTemplate)
 import Summoner.Text (intercalateMap, packageToModule)
 import Summoner.Tree (showTree, traverseTree)
@@ -39,6 +40,9 @@ decisionToBool decision target = case decision of
 generateProject :: Text -> Config -> IO ()
 generateProject projectName Config{..} = do
     repo        <- checkUniqueName projectName
+    -- decide cabal stack or both
+    (cabal, stack) <- getCabalStack (cCabal, cStack)
+
     owner       <- queryDef "Repository owner: " cOwner
     description <- query "Short project description: "
     nm          <- queryDef "Author: " cFullName
@@ -67,7 +71,7 @@ generateProject projectName Config{..} = do
     -- Library/Executable/Tests/Benchmarks flags
     github <- decisionToBool cGitHub "GitHub integration"
     travis <- ifGithub github "Travis CI integration" cTravis
-    appVey <- ifGithub github "AppVeyor CI integration" cAppVey
+    appVey <- ifGithub (stack && github) "AppVeyor CI integration" cAppVey
     privat <- ifGithub github "private repository" cPrivate
     script <- decisionToBool cScript "build script"
     isLib  <- decisionToBool cLib "library target"
@@ -165,3 +169,23 @@ generateProject projectName Config{..} = do
             chooseYesNo "custom prelude" yesDo noDo
         Last prelude@(Just (Prelude p _)) ->
             prelude <$ successMessage ("Custom prelude " <> p <> " will be used in the project")
+
+    -- get what build tool to use in the project
+    -- If user chose only one during CLI, we assume to use only that one.
+    getCabalStack :: (Decision, Decision) -> IO (Bool, Bool)
+    getCabalStack = \case
+        (Idk, Idk) -> decisionToBool cCabal "cabal" >>= \c ->
+            if c then decisionToBool cStack "stack" >>= \s -> pure (c, s)
+            else stackMsg True >> pure (False, True)
+        (Nop, Nop) -> errorMessage "Neither cabal nor stack was chosen" >> exitFailure
+        (Yes, Yes) -> output (True, True)
+        (Yes, _)   -> output (True, False)
+        (_, Yes)   -> output (False, True)
+        (Nop, Idk) -> output (False, True)
+        (Idk, Nop) -> output (True, False)
+      where
+        output :: (Bool, Bool) -> IO (Bool, Bool)
+        output x@(c, s) = cabalMsg c >> stackMsg s >> pure x
+
+        cabalMsg c = targetMessageWithText c "Cabal" "used in this project"
+        stackMsg c = targetMessageWithText c "Stack" "used in this project"
diff --git a/src/Summoner/ProjectData.hs b/src/Summoner/ProjectData.hs
@@ -46,6 +46,8 @@ data ProjectData = ProjectData
     , base           :: Text -- ^ Base library to use
     , prelude        :: Maybe CustomPrelude  -- ^ custom prelude to be used
     , extensions     :: [Text] -- ^ default extensions
+    , cabal          :: Bool
+    , stack          :: Bool
     } deriving (Show)
 
 -- | Used for detecting the user decision during CLI input.

diff --git a/src/Summoner/Question.hs b/src/Summoner/Question.hs
@@ -13,6 +13,8 @@ module Summoner.Question
        , queryManyRepeatOnFail
        , checkUniqueName
 
+       , targetMessageWithText
+
        , trueMessage
        , falseMessage
        ) where
@@ -71,10 +73,13 @@ chooseYesNoBool :: Text -> IO Bool
 chooseYesNoBool target = chooseYesNo target (pure True) (pure False)
 
 targetMessage :: Bool -> Text -> IO Bool
-targetMessage result target = do
+targetMessage result target = targetMessageWithText result target "added to the project"
+
+targetMessageWithText :: Bool -> Text -> Text -> IO Bool
+targetMessageWithText result target text = do
     let (color, actionResult) = if result
-          then (Green, " will be added to the project")
-          else (Cyan,  " won't be added to the project")
+          then (Green, " will be " <> text)
+          else (Cyan,  " won't be" <> text)
 
     beautyPrint [italic, bold, setColor color] $ "  " <> headToUpper target
     beautyPrint [setColor color] actionResult