|
| 1 | +#!/usr/bin/env cabal |
| 2 | +{- cabal: |
| 3 | +build-depends: |
| 4 | + , base >=4.16 |
| 5 | + , bytestring >=0.11 |
| 6 | + , Cabal-syntax ^>=3.10 || ^>=3.12 |
| 7 | +
|
| 8 | +ghc-options: |
| 9 | + -Wall -Wcompat -Wincomplete-uni-patterns |
| 10 | + -Wincomplete-record-updates -Wpartial-fields -Widentities |
| 11 | + -Wredundant-constraints -Wmissing-export-lists |
| 12 | + -Wno-unticked-promoted-constructors -Wunused-packages |
| 13 | +
|
| 14 | +ghc-options: -Werror=missing-deriving-strategies |
| 15 | +
|
| 16 | +ghc-options: -Werror |
| 17 | +
|
| 18 | +ghc-options: -Wmissing-import-lists |
| 19 | +-} |
| 20 | +{-# LANGUAGE LambdaCase #-} |
| 21 | + |
| 22 | +module Main (main) where |
| 23 | + |
| 24 | +import Control.Monad (forM_) |
| 25 | +import qualified Data.ByteString as BS |
| 26 | +import Data.Maybe (mapMaybe) |
| 27 | +import Distribution.PackageDescription.Parsec |
| 28 | + (parseGenericPackageDescriptionMaybe) |
| 29 | +import Distribution.Types.CondTree (ignoreConditions) |
| 30 | +import qualified Distribution.Types.GenericPackageDescription as GenericPackageDescription |
| 31 | +import Distribution.Types.Library (libVisibility) |
| 32 | +import Distribution.Types.LibraryVisibility |
| 33 | + (LibraryVisibility (LibraryVisibilityPrivate, LibraryVisibilityPublic)) |
| 34 | +import Distribution.Types.PackageDescription (package) |
| 35 | +import Distribution.Types.PackageId (pkgName) |
| 36 | +import Distribution.Types.PackageName (unPackageName) |
| 37 | +import Distribution.Types.UnqualComponentName (unUnqualComponentName) |
| 38 | +import System.Environment (getArgs) |
| 39 | +import System.IO (hPutStrLn, stderr) |
| 40 | + |
| 41 | +-- | Given a cabal file, list all public libraries in that cabal file (including |
| 42 | +-- the main library) and print them as valid cabal targets. |
| 43 | +-- |
| 44 | +-- For example: |
| 45 | +-- |
| 46 | +-- @ |
| 47 | +-- ❯ ./scripts/cabal-list-public-library-targets.hs ./blockio/blockio.cabal |
| 48 | +-- Finding sub-libraries with public visibility in './blockio/blockio.cabal' ... |
| 49 | +-- blockio:blockio |
| 50 | +-- blockio:sim |
| 51 | +-- @ |
| 52 | +main :: IO () |
| 53 | +main = do |
| 54 | + cabalFiles <- getArgs |
| 55 | + forM_ cabalFiles $ \cabalFile -> do |
| 56 | + putStrLn ("Finding sub-libraries with public visibility in '" ++ cabalFile ++ "' ...") |
| 57 | + cabalContent <- BS.readFile cabalFile |
| 58 | + case parseGenericPackageDescriptionMaybe cabalContent of |
| 59 | + Nothing -> hPutStrLn stderr $ "error: Could not parse '" <> cabalFile <> "'" |
| 60 | + Just genericPackageDescription -> do |
| 61 | + let packageDescription = GenericPackageDescription.packageDescription genericPackageDescription |
| 62 | + packageName = unPackageName $ pkgName $ package packageDescription |
| 63 | + subLibraries = GenericPackageDescription.condSubLibraries genericPackageDescription |
| 64 | + publicSubLibraries = mapMaybe isPublic subLibraries |
| 65 | + mkCabalTarget s = packageName ++ ":" ++ s |
| 66 | + forM_ (packageName : publicSubLibraries) $ \n -> |
| 67 | + putStrLn $ mkCabalTarget n |
| 68 | + where |
| 69 | + isPublic (componentName, condTree) = |
| 70 | + let vis = libVisibility (fst $ ignoreConditions condTree) in |
| 71 | + case vis of |
| 72 | + LibraryVisibilityPublic -> Just (unUnqualComponentName componentName) |
| 73 | + LibraryVisibilityPrivate -> Nothing |
0 commit comments