{-# LANGUAGE LambdaCase #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE QuasiQuotes #-}
{-# LANGUAGE RecordWildCards #-}
{-# LANGUAGE NoImplicitPrelude #-}

-- | spawns a few processes that serve the que.run website
--
-- : out que-website
module Biz.Que.Site
  ( main,
  )
where

import Alpha
import qualified Control.Concurrent.Async as Async
import qualified Data.ByteString.Char8 as BS
import qualified Data.Ini.Config as Config
import qualified Data.Text as Text
import qualified Data.Text.IO as Text
import Network.HTTP.Req
import qualified Omni.Cli as Cli
import Omni.Test ((@=?))
import qualified Omni.Test as Test
import qualified System.Directory as Directory
import System.FilePath ((</>))
import qualified System.Process as Process

main :: IO ()
main = Cli.main <| Cli.Plan help move test pure

move :: Cli.Arguments -> IO ()
move args = do
  let (src, ns) = case (getArg "src", Text.pack </ getArg "ns") of
        (Just s, Just n) -> (s, n)
        _ -> panic "could not initialize from CLI arguments"
  mKey <- getKey ns
  putText <| "serving " <> Text.pack src <> " at " <> ns
  run mKey ns
    <| Sources
      { index = src </> "Index.md",
        client = src </> "Client.py",
        quescripts = src </> "Quescripts.md",
        style = src </> "Style.css",
        tutorial = src </> "Tutorial.md"
      }
  where
    getArg = Cli.getArg args <. Cli.argument

help :: Cli.Docopt
help =
  [Cli.docopt|
que-website

Usage:
  que-website <src> <ns>
  que-website test
|]

test :: Test.Tree
test = Test.group "Biz.Que.Site" [Test.unit "id" <| 1 @=? (1 :: Integer)]

getKey :: Namespace -> IO (Maybe Key)
getKey ns = do
  home <- Directory.getHomeDirectory
  let file = home </> ".config" </> "que.conf"
  exists <- Directory.doesFileExist file
  unless exists <| panic <| "not found: " <> Text.pack file
  conf <- Text.readFile file
  print (home </> ".config" </> "que.conf")
  auth ns
    |> Config.parseIniFile conf
    |> either errorParsingConf identity
    |> pure

errorParsingConf :: error
errorParsingConf = panic "could not parse ~/.config/que.conf"

data Sources = Sources
  { index :: FilePath,
    quescripts :: FilePath,
    client :: FilePath,
    style :: FilePath,
    tutorial :: FilePath
  }

type Namespace = Text

type Key = Text

auth :: Namespace -> Config.IniParser (Maybe Key)
auth "pub" = pure Nothing
auth ns = Config.sectionMb ns <| Config.field "key"

run :: Maybe Key -> Text -> Sources -> IO ()
run key ns Sources {..} = Async.runConcurrently actions |> void
  where
    actions =
      traverse
        Async.Concurrently
        [ toHtml index +> serve key ns "index" |> forever,
          toHtml quescripts +> serve key ns "quescripts" |> forever,
          BS.readFile client +> serve key ns "client" |> forever,
          toHtml tutorial +> serve key ns "tutorial" |> forever
        ]
    toHtml :: FilePath -> IO ByteString
    toHtml md =
      BS.pack
        </ Process.readProcess
          "pandoc"
          [ "--include-in-header",
            style,
            "-i",
            md,
            "--from",
            "markdown",
            "--to",
            "html"
          ]
          []

serve :: Maybe Key -> Namespace -> Text -> ByteString -> IO ()
serve Nothing "pub" path content =
  runReq defaultHttpConfig <| do
    _ <-
      req
        POST
        (http "que.run" /: "pub" /: path)
        (ReqBodyBs content)
        ignoreResponse
        mempty
    liftIO <| pure ()
serve Nothing p _ _ = panic <| "no auth key provided for ns: " <> p
serve (Just _) ns path content =
  runReq defaultHttpConfig <| do
    let options =
          header "Content-type" "text/html;charset=utf-8"
    -- header "Authorization" (encodeUtf8 key) <> responseTimeout maxBound
    _ <-
      req
        POST
        (http "que.run" /: ns /: path)
        (ReqBodyBs content)
        ignoreResponse
        options
    liftIO <| pure ()