{-# LANGUAGE OverloadedStrings #-}

module IRE.Application (
  app
) where

import Control.Monad.Trans (liftIO)
import Network.Wai (Application)
import Network.Wai.Middleware.Static (addBase, hasPrefix, staticPolicy, (>->))
import Network.Wai.Parse (FileInfo(..))
import System.FilePath.Posix ((</>))
import Web.Scotty ( ActionM, ScottyM, file, files, get,
  json, middleware, post, scottyApp )

import IRE.Application.YOLO (findItems)
import IRE.YOLO (Detector)

app :: FilePath -> Detector -> IO Application
app rootDir net = scottyApp $ ire rootDir net

ire :: FilePath -> Detector -> ScottyM ()
ire rootDir net = do
  middleware $ staticPolicy (hasPrefix "static" >-> addBase rootDir)
  get "/" $ file (rootDir </> "webui.html")

  post "/findItems" $ apiFindItems net


apiFindItems :: Detector -> ActionM ()
apiFindItems net = do
  fs <- files
  let fc = head [ fileContent fi | (_, fi) <- fs ]
  resp <- liftIO $ findItems net fc
  json resp