module Kempe.File ( tcFile
                  , warnFile
                  , dumpMono
                  , dumpTyped
                  , dumpCDecl
                  , cDeclFile
                  , irFile
                  , x86File
                  , armFile
                  , dumpX86
                  , dumpArm
                  , compile
                  , armCompile
                  , dumpIR
                  ) where

-- common b/w test suite and exec, repl utils
import           Control.Applicative       ((<|>))
import           Control.Composition       ((.*))
import           Control.Exception         (Exception, throwIO)
import           Control.Monad             ((<=<))
import           Data.Bifunctor            (bimap)
import           Data.Functor              (void)
import           Data.Semigroup            ((<>))
import           Data.Tuple.Ext            (fst3)
import           Data.Typeable             (Typeable)
import           Kempe.AST
import qualified Kempe.Asm.Arm.Type        as Arm
import qualified Kempe.Asm.X86.Type        as X86
import           Kempe.CGen
import           Kempe.Check.Lint
import           Kempe.Check.Pattern
import           Kempe.Check.TopLevel
import           Kempe.Error
import           Kempe.IR
import           Kempe.Lexer
import           Kempe.Module
import           Kempe.Pipeline
import           Kempe.Proc.As             as As
import qualified Kempe.Proc.Nasm           as Nasm
import           Kempe.Shuttle
import           Kempe.TyAssign
import           Language.C.AST
import           Prettyprinter             (Doc, hardline)
import           Prettyprinter.Render.Text (putDoc)

tcFile :: FilePath -> IO (Either (Error ()) ())
tcFile fp = do
    (maxU, m) <- parseProcess fp
    pure $ do
        void $ runTypeM maxU (checkModule m)
        mErr $ checkModuleExhaustive (void <$> m)

warnFile :: FilePath -> IO (Maybe (Warning AlexPosn))
warnFile fp = do
    (_, m) <- parseProcess fp
    pure (topLevelCheck m <|> lint m)

yeetIO :: Exception e => Either e a -> IO a
yeetIO = either throwIO pure

cDeclFile :: FilePath -> IO [CFunc]
cDeclFile fp = do
    (i, m) <- parseProcess fp
    (mTyped, _) <- yeetIO $ runTypeM i (assignModule m)
    pure $ cGen mTyped

dumpCDecl :: FilePath -> IO ()
dumpCDecl = putDoc . prettyHeaders <=< cDeclFile

dumpTyped :: FilePath -> IO ()
dumpTyped fp = do
    (i, m) <- parseProcess fp
    (mTyped, _) <- yeetIO $ runTypeM i (assignModule m)
    putDoc $ prettyTypedModule mTyped

dumpMono :: FilePath -> IO ()
dumpMono fp = do
    (i, m) <- parseProcess fp
    (mMono, _) <- yeetIO $ monomorphize i m
    putDoc $ prettyTypedModule (fmap (bimap fromMonoConsAnn fromMono) mMono)
    where fromMono (is, os) = StackType is os
          fromMonoConsAnn (ConsAnn _ _ ty) = fromMono ty

dumpIR :: Typeable a => Int -> Declarations a c b -> Doc ann
dumpIR = prettyIR . fst3 .* irGen

dumpX86 :: Typeable a => Int -> Declarations a c b -> Doc ann
dumpX86 = X86.prettyAsm .* x86Alloc

dumpArm :: Typeable a => Int -> Declarations a c b -> Doc ann
dumpArm = Arm.prettyAsm .* armAlloc

irFile :: FilePath -> IO ()
irFile fp = do
    res <- parseProcess fp
    putDoc $ uncurry dumpIR res <> hardline

x86File :: FilePath -> IO ()
x86File fp = do
    res <- parseProcess fp
    putDoc $ uncurry dumpX86 res <> hardline

armFile :: FilePath -> IO ()
armFile fp = do
    res <- parseProcess fp
    putDoc $ uncurry dumpArm res -- don't need hardline b/c arm pp adds it already

compile :: FilePath
        -> FilePath
        -> Bool -- ^ Debug symbols?
        -> IO ()
compile fp o dbg = do
    res <- parseProcess fp
    Nasm.writeO (uncurry dumpX86 res) o dbg

armCompile :: FilePath
           -> FilePath
           -> Bool -- ^ Debug symbols?
           -> IO ()
armCompile fp o dbg = do
    res <- parseProcess fp
    As.writeO (uncurry dumpArm res) o dbg