{-# LANGUAGE GADTs #-}
-----------------------------------------------------------------------------
-- |
-- Module      :  Distribution.Simple.Program.Script
-- Copyright   :  Duncan Coutts 2009
--
-- Maintainer  :  cabal-devel@haskell.org
-- Portability :  portable
--
-- This module provides an library interface to the @hc-pkg@ program.
-- Currently only GHC and LHC have hc-pkg programs.

module Distribution.Simple.Program.Script (

    invocationAsSystemScript,
    invocationAsShellScript,
    invocationAsBatchFile,
  ) where

import Prelude ()
import Distribution.Compat.Prelude

import Distribution.Simple.Program.Run
import Distribution.Simple.Utils
import Distribution.System

-- | Generate a system script, either POSIX shell script or Windows batch file
-- as appropriate for the given system.
--
invocationAsSystemScript :: OS -> ProgramInvocation -> String
invocationAsSystemScript :: OS -> ProgramInvocation -> String
invocationAsSystemScript OS
Windows = ProgramInvocation -> String
invocationAsBatchFile
invocationAsSystemScript OS
_       = ProgramInvocation -> String
invocationAsShellScript


-- | Generate a POSIX shell script that invokes a program.
--
invocationAsShellScript :: ProgramInvocation -> String
invocationAsShellScript :: ProgramInvocation -> String
invocationAsShellScript
  ProgramInvocation {
    progInvokePath :: ProgramInvocation -> String
progInvokePath  = String
path,
    progInvokeArgs :: ProgramInvocation -> [String]
progInvokeArgs  = [String]
args,
    progInvokeEnv :: ProgramInvocation -> [(String, Maybe String)]
progInvokeEnv   = [(String, Maybe String)]
envExtra,
    progInvokeCwd :: ProgramInvocation -> Maybe String
progInvokeCwd   = Maybe String
mcwd,
    progInvokeInput :: ProgramInvocation -> Maybe IOData
progInvokeInput = Maybe IOData
minput
  } = [String] -> String
unlines forall a b. (a -> b) -> a -> b
$
          [ String
"#!/bin/sh" ]
       forall a. [a] -> [a] -> [a]
++ forall (t :: * -> *) a b. Foldable t => (a -> [b]) -> t a -> [b]
concatMap (String, Maybe String) -> [String]
setEnv [(String, Maybe String)]
envExtra
       forall a. [a] -> [a] -> [a]
++ [ String
"cd " forall a. [a] -> [a] -> [a]
++ String -> String
quote String
cwd | String
cwd <- forall a. Maybe a -> [a]
maybeToList Maybe String
mcwd ]
       forall a. [a] -> [a] -> [a]
++ [ (case Maybe IOData
minput of
              Maybe IOData
Nothing     -> String
""
              Just IOData
input -> String
"echo " forall a. [a] -> [a] -> [a]
++ String -> String
quote (IOData -> String
iodataToText IOData
input) forall a. [a] -> [a] -> [a]
++ String
" | ")
         forall a. [a] -> [a] -> [a]
++ [String] -> String
unwords (forall a b. (a -> b) -> [a] -> [b]
map String -> String
quote forall a b. (a -> b) -> a -> b
$ String
path forall a. a -> [a] -> [a]
: [String]
args) forall a. [a] -> [a] -> [a]
++ String
" \"$@\""]

  where
    setEnv :: (String, Maybe String) -> [String]
setEnv (String
var, Maybe String
Nothing)  = [String
"unset " forall a. [a] -> [a] -> [a]
++ String
var, String
"export " forall a. [a] -> [a] -> [a]
++ String
var]
    setEnv (String
var, Just String
val) = [String
"export " forall a. [a] -> [a] -> [a]
++ String
var forall a. [a] -> [a] -> [a]
++ String
"=" forall a. [a] -> [a] -> [a]
++ String -> String
quote String
val]

    quote :: String -> String
    quote :: String -> String
quote String
s = String
"'" forall a. [a] -> [a] -> [a]
++ String -> String
escape String
s forall a. [a] -> [a] -> [a]
++ String
"'"

    escape :: String -> String
escape []        = []
    escape (Char
'\'':String
cs) = String
"'\\''" forall a. [a] -> [a] -> [a]
++ String -> String
escape String
cs
    escape (Char
c   :String
cs) = Char
c        forall a. a -> [a] -> [a]
: String -> String
escape String
cs

iodataToText :: IOData -> String
iodataToText :: IOData -> String
iodataToText (IODataText String
str)   = String
str
iodataToText (IODataBinary ByteString
lbs) = ByteString -> String
fromUTF8LBS ByteString
lbs


-- | Generate a Windows batch file that invokes a program.
--
invocationAsBatchFile :: ProgramInvocation -> String
invocationAsBatchFile :: ProgramInvocation -> String
invocationAsBatchFile
  ProgramInvocation {
    progInvokePath :: ProgramInvocation -> String
progInvokePath  = String
path,
    progInvokeArgs :: ProgramInvocation -> [String]
progInvokeArgs  = [String]
args,
    progInvokeEnv :: ProgramInvocation -> [(String, Maybe String)]
progInvokeEnv   = [(String, Maybe String)]
envExtra,
    progInvokeCwd :: ProgramInvocation -> Maybe String
progInvokeCwd   = Maybe String
mcwd,
    progInvokeInput :: ProgramInvocation -> Maybe IOData
progInvokeInput = Maybe IOData
minput
  } = [String] -> String
unlines forall a b. (a -> b) -> a -> b
$
          [ String
"@echo off" ]
       forall a. [a] -> [a] -> [a]
++ forall a b. (a -> b) -> [a] -> [b]
map (String, Maybe String) -> String
setEnv [(String, Maybe String)]
envExtra
       forall a. [a] -> [a] -> [a]
++ [ String
"cd \"" forall a. [a] -> [a] -> [a]
++ String
cwd forall a. [a] -> [a] -> [a]
++ String
"\"" | String
cwd <- forall a. Maybe a -> [a]
maybeToList Maybe String
mcwd ]
       forall a. [a] -> [a] -> [a]
++ case Maybe IOData
minput of
            Maybe IOData
Nothing    ->
                [ String
path forall a. [a] -> [a] -> [a]
++ forall (t :: * -> *) a b. Foldable t => (a -> [b]) -> t a -> [b]
concatMap (Char
' 'forall a. a -> [a] -> [a]
:) [String]
args ]

            Just IOData
input ->
                [ String
"(" ]
             forall a. [a] -> [a] -> [a]
++ [ String
"echo " forall a. [a] -> [a] -> [a]
++ String -> String
escape String
line | String
line <- String -> [String]
lines forall a b. (a -> b) -> a -> b
$ IOData -> String
iodataToText IOData
input ]
             forall a. [a] -> [a] -> [a]
++ [ String
") | "
               forall a. [a] -> [a] -> [a]
++ String
"\"" forall a. [a] -> [a] -> [a]
++ String
path forall a. [a] -> [a] -> [a]
++ String
"\""
               forall a. [a] -> [a] -> [a]
++ forall (t :: * -> *) a b. Foldable t => (a -> [b]) -> t a -> [b]
concatMap (\String
arg -> Char
' 'forall a. a -> [a] -> [a]
:String -> String
quote String
arg) [String]
args ]

  where
    setEnv :: (String, Maybe String) -> String
setEnv (String
var, Maybe String
Nothing)  = String
"set " forall a. [a] -> [a] -> [a]
++ String
var forall a. [a] -> [a] -> [a]
++ String
"="
    setEnv (String
var, Just String
val) = String
"set " forall a. [a] -> [a] -> [a]
++ String
var forall a. [a] -> [a] -> [a]
++ String
"=" forall a. [a] -> [a] -> [a]
++ String -> String
escape String
val

    quote :: String -> String
    quote :: String -> String
quote String
s = String
"\"" forall a. [a] -> [a] -> [a]
++ String -> String
escapeQ String
s forall a. [a] -> [a] -> [a]
++ String
"\""

    escapeQ :: String -> String
escapeQ []       = []
    escapeQ (Char
'"':String
cs) = String
"\"\"\"" forall a. [a] -> [a] -> [a]
++ String -> String
escapeQ String
cs
    escapeQ (Char
c  :String
cs) = Char
c         forall a. a -> [a] -> [a]
: String -> String
escapeQ String
cs

    escape :: String -> String
escape []        = []
    escape (Char
'|':String
cs) = String
"^|" forall a. [a] -> [a] -> [a]
++ String -> String
escape String
cs
    escape (Char
'<':String
cs) = String
"^<" forall a. [a] -> [a] -> [a]
++ String -> String
escape String
cs
    escape (Char
'>':String
cs) = String
"^>" forall a. [a] -> [a] -> [a]
++ String -> String
escape String
cs
    escape (Char
'&':String
cs) = String
"^&" forall a. [a] -> [a] -> [a]
++ String -> String
escape String
cs
    escape (Char
'(':String
cs) = String
"^(" forall a. [a] -> [a] -> [a]
++ String -> String
escape String
cs
    escape (Char
')':String
cs) = String
"^)" forall a. [a] -> [a] -> [a]
++ String -> String
escape String
cs
    escape (Char
'^':String
cs) = String
"^^" forall a. [a] -> [a] -> [a]
++ String -> String
escape String
cs
    escape (Char
c  :String
cs) = Char
c     forall a. a -> [a] -> [a]
: String -> String
escape String
cs