{-# LANGUAGE FlexibleContexts #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE ScopedTypeVariables #-}

-- | @futhark run@
module Futhark.CLI.Run (main) where

import Control.Exception
import Control.Monad
import Control.Monad.Except
import Control.Monad.Free.Church
import qualified Data.Map as M
import Data.Maybe
import qualified Data.Text.IO as T
import Futhark.Compiler
import Futhark.Pipeline
import Futhark.Util (toPOSIX)
import Futhark.Util.Options
import Language.Futhark
import qualified Language.Futhark.Interpreter as I
import Language.Futhark.Parser
import qualified Language.Futhark.Semantic as T
import qualified Language.Futhark.TypeChecker as T
import System.Exit
import System.FilePath
import System.IO
import Prelude

-- | Run @futhark run@.
main :: String -> [String] -> IO ()
main :: String -> [String] -> IO ()
main = forall cfg.
cfg
-> [FunOptDescr cfg]
-> String
-> ([String] -> cfg -> Maybe (IO ()))
-> String
-> [String]
-> IO ()
mainWithOptions InterpreterConfig
interpreterConfig [FunOptDescr InterpreterConfig]
options String
"options... <program.fut>" [String] -> InterpreterConfig -> Maybe (IO ())
run
  where
    run :: [String] -> InterpreterConfig -> Maybe (IO ())
run [String
prog] InterpreterConfig
config = forall a. a -> Maybe a
Just forall a b. (a -> b) -> a -> b
$ InterpreterConfig -> String -> IO ()
interpret InterpreterConfig
config String
prog
    run [String]
_ InterpreterConfig
_ = forall a. Maybe a
Nothing

interpret :: InterpreterConfig -> FilePath -> IO ()
interpret :: InterpreterConfig -> String -> IO ()
interpret InterpreterConfig
config String
fp = do
  Either String (Env, Ctx)
pr <- InterpreterConfig -> String -> IO (Either String (Env, Ctx))
newFutharkiState InterpreterConfig
config String
fp
  (Env
tenv, Ctx
ienv) <- case Either String (Env, Ctx)
pr of
    Left String
err -> do
      Handle -> String -> IO ()
hPutStrLn Handle
stderr String
err
      forall a. IO a
exitFailure
    Right (Env, Ctx)
env -> forall (f :: * -> *) a. Applicative f => a -> f a
pure (Env, Ctx)
env

  let entry :: Name
entry = InterpreterConfig -> Name
interpreterEntryPoint InterpreterConfig
config
  Either SyntaxError [Value]
vr <- String -> Text -> Either SyntaxError [Value]
parseValues String
"stdin" forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> IO Text
T.getContents

  [Value]
inps <-
    case Either SyntaxError [Value]
vr of
      Left (SyntaxError Loc
loc String
err) -> do
        Handle -> String -> IO ()
hPutStrLn Handle
stderr forall a b. (a -> b) -> a -> b
$ String
"Input syntax error at " forall a. Semigroup a => a -> a -> a
<> forall a. Located a => a -> String
locStr Loc
loc forall a. Semigroup a => a -> a -> a
<> String
":\n" forall a. Semigroup a => a -> a -> a
<> String
err
        forall a. IO a
exitFailure
      Right [Value]
vs ->
        forall (f :: * -> *) a. Applicative f => a -> f a
pure [Value]
vs

  (QualName VName
fname, TypeBase () ()
ret) <-
    case forall k a. Ord k => k -> Map k a -> Maybe a
M.lookup (Namespace
T.Term, Name
entry) forall a b. (a -> b) -> a -> b
$ Env -> NameMap
T.envNameMap Env
tenv of
      Just QualName VName
fname
        | Just (T.BoundV [TypeParam]
_ StructType
t) <- forall k a. Ord k => k -> Map k a -> Maybe a
M.lookup (forall vn. QualName vn -> vn
qualLeaf QualName VName
fname) forall a b. (a -> b) -> a -> b
$ Env -> Map VName BoundV
T.envVtable Env
tenv ->
            forall (f :: * -> *) a. Applicative f => a -> f a
pure (QualName VName
fname, forall dim as. TypeBase dim as -> TypeBase () ()
toStructural forall a b. (a -> b) -> a -> b
$ forall a b. (a, b) -> b
snd forall a b. (a -> b) -> a -> b
$ forall dim as.
TypeBase dim as -> ([TypeBase dim ()], TypeBase dim ())
unfoldFunType StructType
t)
      Maybe (QualName VName)
_ -> do
        Handle -> String -> IO ()
hPutStrLn Handle
stderr forall a b. (a -> b) -> a -> b
$ String
"Invalid entry point: " forall a. [a] -> [a] -> [a]
++ forall a. Pretty a => a -> String
pretty Name
entry
        forall a. IO a
exitFailure

  case Ctx -> VName -> [Value] -> Either String (F ExtOp Value)
I.interpretFunction Ctx
ienv (forall vn. QualName vn -> vn
qualLeaf QualName VName
fname) [Value]
inps of
    Left String
err -> do
      Handle -> String -> IO ()
hPutStrLn Handle
stderr String
err
      forall a. IO a
exitFailure
    Right F ExtOp Value
run -> do
      Either InterpreterError Value
run' <- forall (m :: * -> *) a.
MonadIO m =>
F ExtOp a -> m (Either InterpreterError a)
runInterpreter' F ExtOp Value
run
      case Either InterpreterError Value
run' of
        Left InterpreterError
err -> do
          forall a. Show a => Handle -> a -> IO ()
hPrint Handle
stderr InterpreterError
err
          forall a. IO a
exitFailure
        Right Value
res ->
          case (Value -> Maybe [Value]
I.fromTuple Value
res, forall dim as. TypeBase dim as -> Maybe [TypeBase dim as]
isTupleRecord TypeBase () ()
ret) of
            (Just [Value]
vs, Just [TypeBase () ()]
ts) -> forall (m :: * -> *) a b c.
Applicative m =>
(a -> b -> m c) -> [a] -> [b] -> m ()
zipWithM_ Value -> TypeBase () () -> IO ()
putValue [Value]
vs [TypeBase () ()]
ts
            (Maybe [Value], Maybe [TypeBase () ()])
_ -> Value -> TypeBase () () -> IO ()
putValue Value
res TypeBase () ()
ret

putValue :: I.Value -> TypeBase () () -> IO ()
putValue :: Value -> TypeBase () () -> IO ()
putValue Value
v TypeBase () ()
t
  | Value -> Bool
I.isEmptyArray Value
v = String -> IO ()
putStrLn forall a b. (a -> b) -> a -> b
$ TypeBase () () -> Value -> String
I.prettyEmptyArray TypeBase () ()
t Value
v
  | Bool
otherwise = String -> IO ()
putStrLn forall a b. (a -> b) -> a -> b
$ forall a. Pretty a => a -> String
pretty Value
v

data InterpreterConfig = InterpreterConfig
  { InterpreterConfig -> Name
interpreterEntryPoint :: Name,
    InterpreterConfig -> Bool
interpreterPrintWarnings :: Bool
  }

interpreterConfig :: InterpreterConfig
interpreterConfig :: InterpreterConfig
interpreterConfig = Name -> Bool -> InterpreterConfig
InterpreterConfig Name
defaultEntryPoint Bool
True

options :: [FunOptDescr InterpreterConfig]
options :: [FunOptDescr InterpreterConfig]
options =
  [ forall a. String -> [String] -> ArgDescr a -> String -> OptDescr a
Option
      String
"e"
      [String
"entry-point"]
      ( forall a. (String -> a) -> String -> ArgDescr a
ReqArg
          ( \String
entry -> forall a b. b -> Either a b
Right forall a b. (a -> b) -> a -> b
$ \InterpreterConfig
config ->
              InterpreterConfig
config {interpreterEntryPoint :: Name
interpreterEntryPoint = String -> Name
nameFromString String
entry}
          )
          String
"NAME"
      )
      String
"The entry point to execute.",
    forall a. String -> [String] -> ArgDescr a -> String -> OptDescr a
Option
      String
"w"
      [String
"no-warnings"]
      (forall a. a -> ArgDescr a
NoArg forall a b. (a -> b) -> a -> b
$ forall a b. b -> Either a b
Right forall a b. (a -> b) -> a -> b
$ \InterpreterConfig
config -> InterpreterConfig
config {interpreterPrintWarnings :: Bool
interpreterPrintWarnings = Bool
False})
      String
"Do not print warnings."
  ]

newFutharkiState ::
  InterpreterConfig ->
  FilePath ->
  IO (Either String (T.Env, I.Ctx))
newFutharkiState :: InterpreterConfig -> String -> IO (Either String (Env, Ctx))
newFutharkiState InterpreterConfig
cfg String
file = forall e (m :: * -> *) a. ExceptT e m a -> m (Either e a)
runExceptT forall a b. (a -> b) -> a -> b
$ do
  (Warnings
ws, Imports
imports, VNameSource
src) <-
    forall err a.
(err -> String) -> Either err a -> ExceptT String IO a
badOnLeft forall a. Show a => a -> String
show
      forall (m :: * -> *) a b. Monad m => (a -> m b) -> m a -> m b
=<< forall (m :: * -> *) a. MonadIO m => IO a -> m a
liftIO
        ( forall e (m :: * -> *) a. ExceptT e m a -> m (Either e a)
runExceptT (forall (m :: * -> *).
(MonadError CompilerError m, MonadIO m) =>
[Name] -> String -> m (Warnings, Imports, VNameSource)
readProgramFile [] String
file)
            forall e a. Exception e => IO a -> (e -> IO a) -> IO a
`catch` \(IOException
err :: IOException) ->
              forall (f :: * -> *) a. Applicative f => a -> f a
pure (forall (m :: * -> *) a. MonadError CompilerError m => String -> m a
externalErrorS (forall a. Show a => a -> String
show IOException
err))
        )
  forall (f :: * -> *). Applicative f => Bool -> f () -> f ()
when (InterpreterConfig -> Bool
interpreterPrintWarnings InterpreterConfig
cfg) forall a b. (a -> b) -> a -> b
$
    forall (m :: * -> *) a. MonadIO m => IO a -> m a
liftIO forall a b. (a -> b) -> a -> b
$
      Handle -> String -> IO ()
hPutStr Handle
stderr forall a b. (a -> b) -> a -> b
$
        forall a. Pretty a => a -> String
pretty Warnings
ws

  let imp :: ImportName
imp = String -> ImportName
T.mkInitialImport String
"."
  Ctx
ienv1 <-
    forall (t :: * -> *) (m :: * -> *) b a.
(Foldable t, Monad m) =>
(b -> a -> m b) -> b -> t a -> m b
foldM (\Ctx
ctx -> forall err a.
(err -> String) -> Either err a -> ExceptT String IO a
badOnLeft forall a. Show a => a -> String
show forall (m :: * -> *) b c a.
Monad m =>
(b -> m c) -> (a -> m b) -> a -> m c
<=< forall (m :: * -> *) a.
MonadIO m =>
F ExtOp a -> m (Either InterpreterError a)
runInterpreter' forall b c a. (b -> c) -> (a -> b) -> a -> c
. Ctx -> (String, Prog) -> F ExtOp Ctx
I.interpretImport Ctx
ctx) Ctx
I.initialCtx forall a b. (a -> b) -> a -> b
$
      forall a b. (a -> b) -> [a] -> [b]
map (forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap FileModule -> Prog
fileProg) Imports
imports
  (Env
tenv1, Dec
d1, VNameSource
src') <-
    forall err a.
(err -> String) -> Either err a -> ExceptT String IO a
badOnLeft forall a. Pretty a => a -> String
pretty forall a b. (a -> b) -> a -> b
$
      forall a b. (a, b) -> b
snd forall a b. (a -> b) -> a -> b
$
        Imports
-> VNameSource
-> Env
-> ImportName
-> UncheckedDec
-> (Warnings, Either TypeError (Env, Dec, VNameSource))
T.checkDec Imports
imports VNameSource
src Env
T.initialEnv ImportName
imp forall a b. (a -> b) -> a -> b
$
          String -> UncheckedDec
mkOpen String
"/prelude/prelude"
  (Env
tenv2, Dec
d2, VNameSource
_) <-
    forall err a.
(err -> String) -> Either err a -> ExceptT String IO a
badOnLeft forall a. Pretty a => a -> String
pretty forall a b. (a -> b) -> a -> b
$
      forall a b. (a, b) -> b
snd forall a b. (a -> b) -> a -> b
$
        Imports
-> VNameSource
-> Env
-> ImportName
-> UncheckedDec
-> (Warnings, Either TypeError (Env, Dec, VNameSource))
T.checkDec Imports
imports VNameSource
src' Env
tenv1 ImportName
imp forall a b. (a -> b) -> a -> b
$
          String -> UncheckedDec
mkOpen forall a b. (a -> b) -> a -> b
$
            String -> String
toPOSIX forall a b. (a -> b) -> a -> b
$
              String -> String
dropExtension String
file
  Ctx
ienv2 <- forall err a.
(err -> String) -> Either err a -> ExceptT String IO a
badOnLeft forall a. Show a => a -> String
show forall (m :: * -> *) a b. Monad m => (a -> m b) -> m a -> m b
=<< forall (m :: * -> *) a.
MonadIO m =>
F ExtOp a -> m (Either InterpreterError a)
runInterpreter' (Ctx -> Dec -> F ExtOp Ctx
I.interpretDec Ctx
ienv1 Dec
d1)
  Ctx
ienv3 <- forall err a.
(err -> String) -> Either err a -> ExceptT String IO a
badOnLeft forall a. Show a => a -> String
show forall (m :: * -> *) a b. Monad m => (a -> m b) -> m a -> m b
=<< forall (m :: * -> *) a.
MonadIO m =>
F ExtOp a -> m (Either InterpreterError a)
runInterpreter' (Ctx -> Dec -> F ExtOp Ctx
I.interpretDec Ctx
ienv2 Dec
d2)
  forall (f :: * -> *) a. Applicative f => a -> f a
pure (Env
tenv2, Ctx
ienv3)
  where
    badOnLeft :: (err -> String) -> Either err a -> ExceptT String IO a
    badOnLeft :: forall err a.
(err -> String) -> Either err a -> ExceptT String IO a
badOnLeft err -> String
_ (Right a
x) = forall (f :: * -> *) a. Applicative f => a -> f a
pure a
x
    badOnLeft err -> String
p (Left err
err) = forall e (m :: * -> *) a. MonadError e m => e -> m a
throwError forall a b. (a -> b) -> a -> b
$ err -> String
p err
err

mkOpen :: FilePath -> UncheckedDec
mkOpen :: String -> UncheckedDec
mkOpen String
f = forall (f :: * -> *) vn. ModExpBase f vn -> SrcLoc -> DecBase f vn
OpenDec (forall (f :: * -> *) vn.
String -> f String -> SrcLoc -> ModExpBase f vn
ModImport String
f forall a. NoInfo a
NoInfo forall a. Monoid a => a
mempty) forall a. Monoid a => a
mempty

runInterpreter' :: MonadIO m => F I.ExtOp a -> m (Either I.InterpreterError a)
runInterpreter' :: forall (m :: * -> *) a.
MonadIO m =>
F ExtOp a -> m (Either InterpreterError a)
runInterpreter' F ExtOp a
m = forall (f :: * -> *) a.
F f a -> forall r. (a -> r) -> (f r -> r) -> r
runF F ExtOp a
m (forall (f :: * -> *) a. Applicative f => a -> f a
pure forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a b. b -> Either a b
Right) forall {f :: * -> *} {b}.
MonadIO f =>
ExtOp (f (Either InterpreterError b))
-> f (Either InterpreterError b)
intOp
  where
    intOp :: ExtOp (f (Either InterpreterError b))
-> f (Either InterpreterError b)
intOp (I.ExtOpError InterpreterError
err) = forall (f :: * -> *) a. Applicative f => a -> f a
pure forall a b. (a -> b) -> a -> b
$ forall a b. a -> Either a b
Left InterpreterError
err
    intOp (I.ExtOpTrace String
w String
v f (Either InterpreterError b)
c) = do
      forall (m :: * -> *) a. MonadIO m => IO a -> m a
liftIO forall a b. (a -> b) -> a -> b
$ Handle -> String -> IO ()
hPutStrLn Handle
stderr forall a b. (a -> b) -> a -> b
$ String
w forall a. [a] -> [a] -> [a]
++ String
": " forall a. [a] -> [a] -> [a]
++ String
v
      f (Either InterpreterError b)
c
    intOp (I.ExtOpBreak Loc
_ BreakReason
_ NonEmpty StackFrame
_ f (Either InterpreterError b)
c) = f (Either InterpreterError b)
c