module Language where

import qualified Ast as A
import Compiler
import Lexer (scan)
import Parser (parse)
import Test.HUnit
import Text.Parsec (runParser)
import Text.Parsec.Pos (newPos)

assertAst :: String -> String -> [A.Expr] -> Assertion
assertAst tcase input expected = do
  r <- return $ runParser (scan parse) () "test" input
  case r of
    Left e -> assertFailure $ show e
    Right ast -> assertEqual tcase ast expected

testCase = TestCase $ do
  assertAst "parse module" "module main" [A.Module "main" $ newPos "test" 1 1]
  assertAst
    "parse a function"
    "module main\n\
    \def fn() { }"
    [ A.Module "main" $ newPos "test" 1 1,
      A.Func "fn" [] Nothing [] False False $ newPos "test" 2 1
    ]
  assertAst
    "parse a function with parameters"
    "module main\n\
    \def fn(a: u8) { }"
    [ A.Module "main" $ newPos "test" 1 1,
      A.Func "fn" [("a", A.Type "u8", newPos "test" 2 8)] Nothing [] False False $ newPos "test" 2 1
    ]
  assertAst
    "parse a function with return value"
    "module main\n\
    \def fn(): u8 { }"
    [ A.Module "main" $ newPos "test" 1 1,
      A.Func "fn" [] (Just $ A.Type "u8") [] False False $ newPos "test" 2 1
    ]
  assertAst
    "parse a function call"
    "module main\n\
    \def fn() { }\n\
    \fn();"
    [ A.Module "main" $ newPos "test" 1 1,
      A.Func "fn" [] Nothing [] False False $ newPos "test" 2 1,
      A.Call (A.Var "fn" $ newPos "test" 3 1) [] $ newPos "test" 3 1
    ]
  assertAst
    "parse a function call with arguments"
    "module main\n\
    \def fn(a: u8) { }\n\
    \fn(10);"
    [ A.Module "main" $ newPos "test" 1 1,
      A.Func "fn" [("a", A.Type "u8", newPos "test" 2 8)] Nothing [] False False $ newPos "test" 2 1,
      A.Call (A.Var "fn" $ newPos "test" 3 1) [A.Num 10 $ newPos "test" 3 4] $ newPos "test" 3 1
    ]

language = [testCase]