Add stack-repos and cabal-name executables (input-output-hk#72)

These will be used to generate default cache argument `stackProject` in
haskell.nix.

Author: hamishmack

nix-tools/cabal-name/Main.hs

@@ -0,0 +1,7 @@
+module Main where
+
+import CabalName (doCabalName)
+import CabalName.CLI (parseCabalNameArgs)
+
+main :: IO ()
+main = parseCabalNameArgs >>= doCabalName

nix-tools/cabal.project

@@ -1,4 +1,4 @@
-packages: . 
+packages: .
 
 -- Needs https://github.com/input-output-hk/iohk-nix/commit/6a8c29117eff36ce975e02e01efc8b25d93fcb90#diff-6fb0c6517b547a8baf082d5d2d604842
 -- to work with the data-dir issues when building components.

nix-tools/lib/CabalName.hs

@@ -0,0 +1,13 @@
+module CabalName
+  ( doCabalName
+  ) where
+
+import Stack2nix.Project (findCabalFiles)
+import Cabal2Nix (cabalFilePkgName)
+
+import CabalName.CLI (Args(..))
+
+doCabalName :: Args -> IO ()
+doCabalName args =
+  findCabalFiles (argHpackUse args) (argPackageDir args)
+    >>= mapM_ (putStr . cabalFilePkgName) . take 1

nix-tools/lib/CabalName/CLI.hs

@@ -0,0 +1,29 @@
+module CabalName.CLI
+  ( Args(..)
+  , HpackUse(..)
+  , parseCabalNameArgs
+  ) where
+
+import Options.Applicative hiding (option)
+import Data.Semigroup ((<>))
+import Stack2nix.CLI (HpackUse(..))
+
+--------------------------------------------------------------------------------
+-- CLI Arguments
+data Args = Args
+  { argPackageDir :: FilePath
+  , argHpackUse   :: HpackUse
+  } deriving Show
+
+-- Argument Parser
+args :: Parser Args
+args = Args
+  <$> argument str ( metavar "DIR" <> help "Directory containing the package source" )
+  <*> flag UsePackageYamlFirst IgnorePackageYaml (long "ignore-package-yaml" <> help "disable hpack run and use only cabal disregarding package.yaml existence")
+
+parseCabalNameArgs :: IO Args
+parseCabalNameArgs = execParser opts
+  where opts = info (args <**> helper)
+          ( fullDesc
+         <> progDesc "Find the name of the packeage in the specified directory"
+         <> header "cabal-name - extract the name of a package" )

nix-tools/stack2nix/Stack2nix.hs renamed to nix-tools/lib/Stack2nix.hs

@@ -152,7 +152,7 @@ packages2nix args pkgs =
                     writeDoc nixFile =<<
                       prettyNix <$> cabal2nix True (argDetailLevel args) src cabalFile
                     return $ fromString pkg $= mkPath False nix
-       (DVCS (Git url rev) subdirs) ->
+       (DVCS (Git url rev) _ subdirs) ->
          fmap concat . forM subdirs $ \subdir ->
          do cacheHits <- liftIO $ cacheHits (argCacheFile args) url rev subdir
             case cacheHits of

nix-tools/stack2nix/Stack2nix/CLI.hs renamed to nix-tools/lib/Stack2nix/CLI.hs

@@ -112,7 +112,7 @@ instance Text CabalRev where
 data Dependency
   = PkgIndex PackageIdentifier (Maybe (Either Sha256 CabalRev)) -- ^ overridden package in the stackage index
   | LocalPath String -- ^ Some local package (potentially overriding a package in the index as well)
-  | DVCS Location [FilePath] -- ^ One or more packages fetched from git or similar.
+  | DVCS Location (Maybe Sha256) [FilePath] -- ^ One or more packages fetched from git or similar.
   -- TODO: Support archives.
   -- | Archive ...
   deriving (Show)
@@ -213,6 +213,7 @@ instance FromJSON Dependency where
             return . LocalPath . dropTrailingSlash . T.unpack
           parseDVCS = withObject "DVCS" $ \o -> DVCS
             <$> (o .: "location" <|> parseJSON p)
+            <*> o .:? "nix-sha256" .!= Nothing
             <*> o .:? "subdirs" .!= ["."]
 
           -- drop trailing slashes. Nix doesn't like them much;

nix-tools/stack2nix/Stack2nix/Cache.hs renamed to nix-tools/lib/Stack2nix/Cache.hs

@@ -0,0 +1,46 @@
+{-# LANGUAGE LambdaCase #-}
+{-# LANGUAGE NamedFieldPuns #-}
+{-# LANGUAGE OverloadedStrings #-}
+{-# LANGUAGE DeriveGeneric #-}
+
+module StackRepos
+  ( doStackRepos
+  , stack2nix
+  ) where
+
+import Data.Aeson (ToJSON(..))
+import Data.Aeson.Encode.Pretty (encodePretty)
+import qualified Data.ByteString.Lazy as LBS (writeFile)
+import Data.Yaml (decodeFileEither)
+
+import GHC.Generics (Generic)
+
+import Stack2nix.Stack (Stack(..), Dependency(..), Location(..))
+import Stack2nix.External.Resolve
+
+import StackRepos.CLI (Args(..))
+
+data SourceRepos = SourceRepos
+  { url     :: String
+  , rev     :: String
+  , sha256  :: Maybe String
+  , subdirs :: [FilePath]
+  } deriving (Show, Eq, Ord, Generic)
+
+instance ToJSON SourceRepos
+
+doStackRepos :: Args -> IO ()
+doStackRepos args = do
+  evalue <- decodeFileEither (argStackYaml args)
+  case evalue of
+    Left e -> error (show e)
+    Right value -> stack2nix args
+                    =<< resolveSnapshot (argStackYaml args) value
+
+stack2nix :: Args -> Stack -> IO ()
+stack2nix args (Stack _ _ pkgs _ _) =
+  LBS.writeFile "repos.json" $ encodePretty (
+    pkgs >>= (\case
+       (DVCS (Git url rev) sha256 subdirs) ->
+         [SourceRepos { url, rev, sha256, subdirs }]
+       _ -> []))

nix-tools/stack2nix/Stack2nix/External/Resolve.hs renamed to nix-tools/lib/Stack2nix/External/Resolve.hs

@@ -0,0 +1,27 @@
+module StackRepos.CLI
+  ( Args(..)
+  , HpackUse(..)
+  , parseStackReposArgs
+  ) where
+
+import Options.Applicative hiding (option)
+import Data.Semigroup ((<>))
+import Stack2nix.CLI (HpackUse(..))
+
+--------------------------------------------------------------------------------
+-- CLI Arguments
+newtype Args = Args
+  { argStackYaml :: FilePath
+  } deriving Show
+
+-- Argument Parser
+args :: Parser Args
+args = Args
+  <$> strOption ( long "stack-yaml" <> value "stack.yaml" <> showDefault <> metavar "FILE" <> help "Override project stack.yaml" )
+
+parseStackReposArgs :: IO Args
+parseStackReposArgs = execParser opts
+  where opts = info (args <**> helper)
+          ( fullDesc
+         <> progDesc "Collect information about remote source packages used by Stack"
+         <> header "stack-repos - extract the details of remote repos" )

nix-tools/stack2nix/Stack2nix/Project.hs renamed to nix-tools/lib/Stack2nix/Project.hs

@@ -17,23 +17,42 @@ library
   exposed-modules:     Cabal2Nix
                      , Cabal2Nix.Util
                      , Cabal2Nix.Plan
+                     , CabalName
+                     , CabalName.CLI
                      , Distribution.Nixpkgs.Fetch
+                     , StackRepos
+                     , StackRepos.CLI
+                     , Stack2nix
+                     , Stack2nix.Cache
+                     , Stack2nix.CLI
+                     , Stack2nix.External.Resolve
+                     , Stack2nix.Project
+                     , Stack2nix.Stack
   build-depends:       base >=4 && <4.13
-                     , hnix
-                     , aeson
-                     , unordered-containers
-                     , process
-                     , deepseq
-                     , transformers
-                     , data-fix
                      , Cabal >= 2.4
-                     , text
-                     , filepath
-                     , directory
+                     , aeson
+                     , aeson-pretty
+                     , base16-bytestring
                      , bytestring
                      , cryptohash-sha256
-                     , base16-bytestring
+                     , data-fix
+                     , deepseq
+                     , directory
+                     , extra
+                     , filepath
+                     , hnix
                      , hpack
+                     , http-client
+                     , http-client-tls
+                     , http-types
+                     , optparse-applicative
+                     , prettyprinter
+                     , process
+                     , text
+                     , transformers
+                     , unordered-containers
+                     , yaml
+
   hs-source-dirs:      lib
   default-language:    Haskell2010
 
@@ -139,34 +158,8 @@ executable lts-to-nix
 executable stack-to-nix
   ghc-options:         -Wall
   main-is:             Main.hs
-  other-modules:       Stack2nix
-                     , Stack2nix.Cache
-                     , Stack2nix.CLI
-                     , Stack2nix.External.Resolve
-                     , Stack2nix.Project
-                     , Stack2nix.Stack
   build-depends:       base >=4 && <4.13
                      , nix-tools
-                     , transformers
-                     , hnix
-                     , yaml
-                     , aeson
-                     , microlens
-                     , microlens-aeson
-                     , text
-                     , Cabal
-                     , vector
-                     , prettyprinter
-                     , directory
-                     , filepath
-                     , extra
-                     , hpack
-                     , bytestring
-                     , optparse-applicative
-                     , http-client-tls
-                     , http-client
-                     , http-types
-                     , unordered-containers
   hs-source-dirs:      stack2nix
   default-language:    Haskell2010
 
@@ -182,3 +175,19 @@ executable truncate-index
                      , time
   hs-source-dirs:      truncate-index
   default-language:    Haskell2010
+
+executable stack-repos
+  ghc-options:         -Wall
+  main-is:             Main.hs
+  build-depends:       base >=4 && <4.13
+                     , nix-tools
+  hs-source-dirs:      stack-repos
+  default-language:    Haskell2010
+
+executable cabal-name
+  ghc-options:         -Wall
+  main-is:             Main.hs
+  build-depends:       base >=4 && <4.13
+                     , nix-tools
+  hs-source-dirs:      cabal-name
+  default-language:    Haskell2010

nix-tools/stack2nix/Stack2nix/Stack.hs renamed to nix-tools/lib/Stack2nix/Stack.hs

@@ -0,0 +1,7 @@
+module Main where
+
+import StackRepos (doStackRepos)
+import StackRepos.CLI (parseStackReposArgs)
+
+main :: IO ()
+main = parseStackReposArgs >>= doStackRepos