{-
Module:   Module
Description :  Description
Maintainer  : matt
Portability : Linux
  http://www.serpentine.com/criterion/tutorial.html

-}
{-# LANGUAGE DataKinds #-}
{-# LANGUAGE FlexibleContexts #-}
{-# LANGUAGE QuasiQuotes #-}
{-# LANGUAGE TemplateHaskell #-}
{-# LANGUAGE TypeApplications #-}
import Criterion
import Criterion.Main
import MptcpAnalyzer.Cache
import MptcpAnalyzer.Loader (loadPcapIntoFrame)
import MptcpAnalyzer.Map
import MptcpAnalyzer.Pcap (buildFrameFromStreamId)
import Tshark.Main (defaultTsharkPrefs)

import Polysemy (Final, Members, Sem, runFinal)
import qualified Polysemy as P
import qualified Polysemy.Embed as P
import qualified Polysemy.IO as P
import qualified Polysemy.Internal as P
import qualified Polysemy.State as P

import Control.Monad.IO.Class (liftIO)
import Data.Either (fromRight)
import MptcpAnalyzer.Stream
import Polysemy.Log (Log)
import qualified Polysemy.Log as Log
import Polysemy.Log.Colog (interpretLogStdout)

-- tableTypes "LCols" "data/left1.csv"
-- tableTypes "RCols" "data/right1.csv"
-- tableTypes "SmCols" "data/left_summary.csv"

-- lfi :: IO (Frame LCols)
-- lfi = inCoreAoS (readTable "data/left1.csv")

-- rfi :: IO (Frame RCols)
-- rfi = inCoreAoS (readTable "data/right1.csv")

-- smfi :: IO (Frame SmCols)
-- smfi = inCoreAoS (readTable "data/left_summary.csv")
cacheConfig :: CacheConfig
cacheConfig = CacheConfig {
  cacheFolder = "/tmp"
  , cacheEnabled = True
}

-- logs/cache
main :: IO ()
main = defaultMain [
  bench "loadPcap" $ nfIO (runTests)
  -- TODO bench mapping
    -- bench "inner1a" $ nf (\x -> mapTcpConnection aframe x >> pure () ) (fromRight (error "could not load frame 1")  frame2)
  ]


runTests :: IO ()
runTests = do
  _ <- P.runM
    $ interpretLogStdout
    $ runCache cacheConfig
      inputLoop
  putStrLn "finished"


inputLoop :: (Members '[P.Embed IO, Log , Cache] r) => Sem r ()
inputLoop = do
  frame1 <- loadPcapIntoFrame defaultTsharkPrefs "examples/client_2_cleaned.pcapng"
  frame2 <- loadPcapIntoFrame defaultTsharkPrefs "examples/server_2_cleaned.pcapng"

  case buildFrameFromStreamId (fromRight undefined frame1) (StreamId 0) of
    Left err -> error err
    Right aframe -> P.embed $ defaultMain [
      bench "inner1a" $ nf (\x -> mapTcpConnection aframe x >> pure () ) (fromRight (error "could not load frame 1")  frame2)
      ]