module Hercules.Agent.ServiceInfo where

import qualified Hercules.API.Agent.LifeCycle as API.LifeCycle
import qualified Hercules.API.Agent.LifeCycle.ServiceInfo as ServiceInfo
import qualified Hercules.Agent.Client as Client
import Hercules.Error (escalate)
import Network.URI
import Protolude
import qualified Servant.Client

data Env
  = Env
      { agentSocketBaseURL :: URI,
        bulkSocketBaseURL :: URI
      }

newEnv :: Servant.Client.ClientEnv -> IO Env
newEnv clientEnv = do
  serviceInfo <- escalate =<< Servant.Client.runClientM (API.LifeCycle.getServiceInfo Client.lifeCycleClient) clientEnv
  let parse varName text = case parseURI (toS text) of
        Nothing -> panic $ varName <> " invalid: " <> ServiceInfo.agentSocketBaseURL serviceInfo
        Just uri -> uri <$ do
          case uriScheme uri of
            "http:" -> pass
            "https:" -> pass
            x -> panic $ varName <> " has invalid uri scheme" <> toS x
  agentSocketBaseURL_ <- parse "agentSocketBaseURL" $ ServiceInfo.agentSocketBaseURL serviceInfo
  bulkSocketBaseURL_ <- parse "bulkSocketBaseURL" $ ServiceInfo.bulkSocketBaseURL serviceInfo
  pure Env
    { agentSocketBaseURL = agentSocketBaseURL_,
      bulkSocketBaseURL = bulkSocketBaseURL_
    }