Push to journalctl

src/HomeAssistant/Runtime/Bus.hs

@@ -34,6 +34,9 @@ import Data.UUID (toText)
 import AFRP (Request(..), Event(..))
 import Control.Monad.IO.Class (MonadIO, liftIO)
 import HomeAssistant.Runtime.Flags (Flags)
+import Katip.Scribes.Journal (mkJournalScribe)
+import System.Environment (lookupEnv)
+import Data.Maybe (isJust)
 
 -- | Shared runtime state: inbound is a broadcast channel (controllers
 -- read from 'dupTChan' copies), outbound queues service calls for the
@@ -50,8 +53,7 @@ data Bus = Bus
 
 withBus :: Severity -> AppMetrics -> Flags -> (Bus -> IO a) -> IO a
 withBus severity metrics flags callback = do
-  handleScribe <- mkHandleScribe ColorIfTerminal stdout (permitItem severity) V2
-  let makeLogEnv = registerScribe "stdout" handleScribe defaultScribeSettings =<< initLogEnv "hass-controller" "production"
+  let makeLogEnv = registerJournalScribe =<< registerHandleScribe =<< initLogEnv "hass-controller" "production"
   -- closeScribes will stop accepting new logs, flush existing ones and clean up resources
   bracket makeLogEnv closeScribes $ \le -> do
     bus <- Bus
@@ -63,6 +65,18 @@ withBus severity metrics flags callback = do
           <*> pure metrics
           <*> pure flags
     callback bus
+  where
+    registerJournalScribe :: LogEnv -> IO LogEnv
+    registerJournalScribe le = do
+      journalScribe <- mkJournalScribe (permitItem severity) V2
+      registerScribe "journalctl" journalScribe defaultScribeSettings le
+    registerHandleScribe le = do
+      systemdUnit <- isJust <$> lookupEnv "INVOCATION_ID"
+      if systemdUnit
+         then pure le -- ignore stdout when running in systemd
+         else do
+           handleScribe <- mkHandleScribe ColorIfTerminal stdout (permitItem severity) V2
+           registerScribe "stdout" handleScribe defaultScribeSettings le
 
 recordInbound :: Bus -> IO ()
 recordInbound bus = inc (amTriggersIn (busMetrics bus))