mirror of
https://github.com/Avi-D-coder/implicit-hie.git
synced 2024-11-30 00:33:29 +03:00
55 lines
1.6 KiB
Haskell
55 lines
1.6 KiB
Haskell
{-# LANGUAGE MultiWayIf #-}
|
|
|
|
module Main where
|
|
|
|
import Control.Monad
|
|
import Data.Attoparsec.Text
|
|
import qualified Data.Text as T
|
|
import qualified Data.Text.IO as T
|
|
import Hie.Cabal.Parser
|
|
import Hie.Yaml
|
|
import System.Directory
|
|
import System.Directory.Internal
|
|
import System.FilePath.Posix
|
|
|
|
main :: IO ()
|
|
main = do
|
|
pwd <- getCurrentDirectory
|
|
files <- listDirectory pwd
|
|
cfs <- cabalFiles pwd
|
|
let cabal = (cabalHieYaml, "Cabal ")
|
|
stack = (stackHieYaml, "Stack ")
|
|
sOrC =
|
|
if | any (("dist-newstyle" ==) . takeFileName) files -> cabal
|
|
| any ((".stack-work" ==) . takeFileName) files -> stack
|
|
| any (("stack.yaml" ==) . takeFileName) files -> stack
|
|
| otherwise -> cabal
|
|
gen f = do
|
|
f' <- T.readFile f
|
|
case parseOnly parsePackage f' of
|
|
Right r -> do
|
|
let hiePath = fst (splitFileName f) </> "hie.yaml"
|
|
T.writeFile hiePath $ fst sOrC r
|
|
pure ("wrote " <> snd sOrC <> hiePath)
|
|
_ -> pure $ "Could not parse " <> f
|
|
when (null cfs) $ error $
|
|
"No .cabal files found under"
|
|
<> pwd
|
|
<> "\n You may need to run stack build."
|
|
mapM_ (putStrLn <=< gen) cfs
|
|
|
|
cabalFiles :: FilePath -> IO [FilePath]
|
|
cabalFiles f = do
|
|
fs <- listDirectory f
|
|
case filter ((".cabal" ==) . takeExtension) fs of
|
|
h : _ -> pure [f </> h]
|
|
_ ->
|
|
fmap concat . mapM cabalFiles
|
|
=<< filterM
|
|
(fmap (fileTypeIsDirectory . fileTypeFromMetadata) . getFileMetadata)
|
|
( map (f </>) $
|
|
filter
|
|
(`notElem` [".git", "dist", "dist-newstyle", ".stack-work"])
|
|
fs
|
|
)
|