-- {-# LANGUAGE BlockArguments #-}
module Main where

import qualified Data.ByteString.Lazy as Lbs
import Data.Foldable

import Frontend.Lexer.Tokens
import Frontend.Autogen.Lexer
import Frontend.Autogen.Parser

import qualified Data.Text.Encoding as T
import qualified Data.Text as T
import Control.Monad ( unless )
import System.Environment (getArgs)
import Text.Show.Pretty (pPrint)

import Debug.Trace

main :: IO ()
main = do
  args <- getArgs
  for_ args $ \str -> do
    Main.lex pPrint parseMod =<< Lbs.readFile str

testParse :: String -> IO ()
testParse s = Main.lex print parseMod (Lbs.fromStrict (T.encodeUtf8 (T.pack s)))

testLex :: String -> IO ()
testLex s = Main.lex (const (pure ())) (scan []) (Lbs.fromStrict (T.encodeUtf8 (T.pack s)))

lex :: (a -> IO ()) -> Alex a -> Lbs.ByteString -> IO ()
lex show cont arg = do
  let x = runAlex arg cont
  case x of
    Left e -> print e
    Right x -> show x

scan :: [Token] -> Alex [Token]
scan acc = do
  tok <- alexMonadScan
  sc <- alexGetStartCode
  state <- getUserState
  traceM . unlines $
    [ "----------------"
    , "just lexed: " ++ show tok
    , "sc: "         ++ show sc
    , "sc stack: "   ++ show (startCodes state)
    ]
  case tokenClass tok of
    TokEof -> pure (reverse acc)
    _      -> scan (tok:acc)