Main.idr

module Main

import Parser
import Lang
import Bounded
import RawLang
import CheckLang
import Interp
import System
import Debug.Trace

testProg : List RInstr
testProg = [RFl (RLABEL "Start"),
                  RIOi RREADNUM,
                  RStk (RPUSH 10),
                  RFl (RCALL "addup"),
                  RIOi ROUTPUTNUM,
                  RFl (RJUMP "Start"),
                  RFl (RLABEL "addup"),
                  RAr RADD,
                  RFl RRETURN]

{-
dumpI : Instr x y l -> String
dumpI (Stk (PUSH n)) = "PUSH " ++ show n
dumpI (Stk DUP) = "DUP"
dumpI (Stk (COPY n)) = "COPY " ++ show n
dumpI (Stk SWAP) = "SWAP"
dumpI (Stk DISCARD) = "DISCARD"
dumpI (Stk (SLIDE n)) = "SLIDE " ++ show n
dumpI = ?dumpI_rhs

dumpI (Ar ADD) = "ADD"
dumpI (Ar SUB) = "SUB"
dumpI (Ar MUL) = "MUL"
dumpI (Ar DIV) = "DIV"
dumpI (Ar MOD) = "MOD"

dumpI (Hp STORE) = "STORE"
dumpI (Hp RETRIEVE) = "RETRIEVE"

dumpI (Fl (LABEL x)) = "LABEL " ++ show x
dumpI (Fl (CALL x)) = "CALL " ++ show x
dumpI (Fl (JUMP x)) = "JUMP " ++ show x
dumpI (Fl (JZ x)) = "JZ " ++ show x
dumpI (Fl (JNEG x)) = "JNEG " ++ show x
dumpI (Fl RETURN) = "RETURN"
dumpI (Fl END) = "END"

dumpI (IOi OUTPUT) = "OUTPUT"
dumpI (IOi OUTPUTNUM) = "OUTPUTNUM"
dumpI (IOi READCHAR) = "READCHAR"
dumpI (IOi READNUM) = "READNUM"
-}

dumpIStk : StackInst x y l -> String
dumpIStk (PUSH n) = "PUSH " ++ show n
dumpIStk DUP = "DUP"
dumpIStk (COPY n) = "COPY " ++ show n
dumpIStk SWAP = "SWAP"
dumpIStk DISCARD = "DISCARD"
dumpIStk (SLIDE n) = "SLIDE " ++ show n

dumpIArith : ArithInst x y l -> String
dumpIArith ADD = "ADD"
dumpIArith SUB = "SUB"
dumpIArith MUL = "MUL"
dumpIArith DIV = "DIV"
dumpIArith MOD = "MOD"

dumpIHeap : HeapInst x y l -> String
dumpIHeap STORE = "STORE"
dumpIHeap RETRIEVE = "RETRIEVE"

dumpIFlow : FlowInst x y l -> String
dumpIFlow (LABEL x) = "LABEL " ++ show x
dumpIFlow (CALL x) = "CALL " ++ show x
dumpIFlow (JUMP x) = "JUMP " ++ show x
dumpIFlow (JZ x) = "JZ " ++ show x
dumpIFlow (JNEG x) = "JNEG " ++ show x
dumpIFlow RETURN = "RETURN"
dumpIFlow END = "END"

dumpIIO : IOInst x y l -> String
dumpIIO OUTPUT = "OUTPUT"
dumpIIO OUTPUTNUM = "OUTPUTNUM"
dumpIIO READCHAR = "READCHAR"
dumpIIO READNUM = "READNUM"

dumpI : Instr x y l -> String
dumpI (Stk i) = dumpIStk i
dumpI (Ar i) = dumpIArith i
dumpI (Hp i) = dumpIHeap i
dumpI (Fl i) = dumpIFlow i
dumpI (IOi i) = dumpIIO i
dumpI (Check x i) = "CHECK " ++ show x ++ " : " ++ dumpI i

dump : Prog x y l -> String
dump [] = ""
dump (x :: xs) = dumpI x ++ "\n" ++ dump xs

tspan : List Char -> (List Char, List Char)
tspan []      = ([], [])
tspan (x::xs) =
  if isDigit x then
    let (ys, zs) = trace (show xs) (tspan xs) in
      (x::ys, zs)
  else
    ([], x::xs)

main : IO ()
main = do xs <- getArgs
          case xs of
               (_ :: prog :: _) =>
--                   print (tspan (unpack ("1234" ++ prog)))
--                    print $ parseNum (unpack ("      \t\t          \t\t  \t\t \n fooasdklfjsahdlkfjashdflkjashfsldkjfhalksdjfh" ++ prog))
                   do Right src <- readFile prog
                          | Left err => putStrLn ("Error reading " ++ prog ++
                                                   "(" ++ show err ++ ")")
                      let raw = parse src
                      case check raw of
                           Just (_ ** m) => do -- putStrLn (dump (program m))
                                               loop m
                                               -- gcInfo
                           Nothing => putStrLn "FAIL"
               _ => putStrLn "Usage: wspace <file>"