From b18bd4eee969681ee532c4898ddaaa0851e6b846 Mon Sep 17 00:00:00 2001
From: Ben Sima <ben@bensima.com>
Date: Tue, 16 Dec 2025 13:24:54 -0500
Subject: Batch web_reader tool, much faster

Added retry with backoff, parallel proccessing, editing pages down to main
content, summarization with haiku. It's so much faster and more reliable
now. Plus improved the logging system and distangled the status UI bar from the
logging module.
---
 Omni/Agent/Status.hs | 157 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 157 insertions(+)
 create mode 100644 Omni/Agent/Status.hs

(limited to 'Omni/Agent/Status.hs')

diff --git a/Omni/Agent/Status.hs b/Omni/Agent/Status.hs
new file mode 100644
index 0000000..ab533c4
--- /dev/null
+++ b/Omni/Agent/Status.hs
@@ -0,0 +1,157 @@
+{-# LANGUAGE OverloadedStrings #-}
+{-# LANGUAGE RecordWildCards #-}
+{-# LANGUAGE NoImplicitPrelude #-}
+
+-- | Status bar UI for the jr worker.
+-- This is NOT a logging module - use Omni.Log for logging.
+module Omni.Agent.Status where
+
+import Alpha
+import Data.IORef (IORef, modifyIORef', newIORef, readIORef, writeIORef)
+import qualified Data.Text as Text
+import qualified Data.Text.IO as TIO
+import Data.Time.Clock (NominalDiffTime, UTCTime, diffUTCTime, getCurrentTime)
+import Data.Time.Format (defaultTimeLocale, parseTimeOrError)
+import qualified Omni.Log as Log
+import qualified System.Console.ANSI as ANSI
+import qualified System.IO as IO
+import System.IO.Unsafe (unsafePerformIO)
+import Text.Printf (printf)
+
+-- | Status of the agent for the UI
+data Status = Status
+  { statusWorker :: Text,
+    statusTask :: Maybe Text,
+    statusThread :: Maybe Text,
+    statusFiles :: Int,
+    statusCredits :: Double,
+    statusStartTime :: UTCTime,
+    statusActivity :: Text
+  }
+  deriving (Show, Eq)
+
+emptyStatus :: Text -> UTCTime -> Status
+emptyStatus workerName startTime =
+  Status
+    { statusWorker = workerName,
+      statusTask = Nothing,
+      statusThread = Nothing,
+      statusFiles = 0,
+      statusCredits = 0.0,
+      statusStartTime = startTime,
+      statusActivity = "Idle"
+    }
+
+-- | Global state for the status bar
+{-# NOINLINE currentStatus #-}
+currentStatus :: IORef Status
+currentStatus = unsafePerformIO (newIORef (emptyStatus "Unknown" defaultStartTime))
+
+defaultStartTime :: UTCTime
+defaultStartTime = parseTimeOrError True defaultTimeLocale "%Y-%m-%d %H:%M:%S %Z" "2000-01-01 00:00:00 UTC"
+
+-- | Initialize the status bar system
+init :: Text -> IO ()
+init workerName = do
+  IO.hSetBuffering IO.stderr IO.LineBuffering
+  startTime <- getCurrentTime
+  writeIORef currentStatus (emptyStatus workerName startTime)
+  -- Reserve 5 lines at bottom
+  IO.hPutStrLn IO.stderr ""
+  IO.hPutStrLn IO.stderr ""
+  IO.hPutStrLn IO.stderr ""
+  IO.hPutStrLn IO.stderr ""
+  IO.hPutStrLn IO.stderr ""
+  ANSI.hCursorUp IO.stderr 5
+
+-- | Update the status
+update :: (Status -> Status) -> IO ()
+update f = do
+  modifyIORef' currentStatus f
+  render
+
+-- | Get the current status
+getStatus :: IO Status
+getStatus = readIORef currentStatus
+
+-- | Set the activity message
+updateActivity :: Text -> IO ()
+updateActivity msg = update (\s -> s {statusActivity = msg})
+
+-- | Log a scrolling message (appears above status bars)
+-- Uses Omni.Log for the actual logging, then re-renders status bar
+log :: Text -> IO ()
+log msg = do
+  -- Clear status bars temporarily
+  ANSI.hClearLine IO.stderr
+  ANSI.hCursorDown IO.stderr 1
+  ANSI.hClearLine IO.stderr
+  ANSI.hCursorDown IO.stderr 1
+  ANSI.hClearLine IO.stderr
+  ANSI.hCursorDown IO.stderr 1
+  ANSI.hClearLine IO.stderr
+  ANSI.hCursorDown IO.stderr 1
+  ANSI.hClearLine IO.stderr
+  ANSI.hCursorUp IO.stderr 4
+
+  -- Use Omni.Log for the actual log message
+  Log.info [msg]
+  Log.br
+
+  -- Re-render status bars at bottom
+  render
+
+-- | Render the five status lines
+render :: IO ()
+render = do
+  Status {..} <- readIORef currentStatus
+  now <- getCurrentTime
+  let taskStr = maybe "None" identity statusTask
+      threadStr = maybe "None" identity statusThread
+      elapsed = diffUTCTime now statusStartTime
+      elapsedStr = formatElapsed elapsed
+
+  -- Line 1: Worker | Thread
+  ANSI.hSetCursorColumn IO.stderr 0
+  ANSI.hClearLine IO.stderr
+  TIO.hPutStr IO.stderr ("[Worker: " <> statusWorker <> "] Thread: " <> threadStr)
+
+  -- Line 2: Task
+  ANSI.hCursorDown IO.stderr 1
+  ANSI.hSetCursorColumn IO.stderr 0
+  ANSI.hClearLine IO.stderr
+  TIO.hPutStr IO.stderr ("Task: " <> taskStr)
+
+  -- Line 3: Files | Credits
+  ANSI.hCursorDown IO.stderr 1
+  ANSI.hSetCursorColumn IO.stderr 0
+  ANSI.hClearLine IO.stderr
+  let creditsStr = Text.pack (printf "%.2f" statusCredits)
+  TIO.hPutStr IO.stderr ("Files: " <> tshow statusFiles <> " | Credits: $" <> creditsStr)
+
+  -- Line 4: Time (elapsed duration)
+  ANSI.hCursorDown IO.stderr 1
+  ANSI.hSetCursorColumn IO.stderr 0
+  ANSI.hClearLine IO.stderr
+  TIO.hPutStr IO.stderr ("Time: " <> elapsedStr)
+
+  -- Line 5: Activity
+  ANSI.hCursorDown IO.stderr 1
+  ANSI.hSetCursorColumn IO.stderr 0
+  ANSI.hClearLine IO.stderr
+  TIO.hPutStr IO.stderr ("> " <> statusActivity)
+
+  -- Return cursor to line 1
+  ANSI.hCursorUp IO.stderr 4
+  IO.hFlush IO.stderr
+
+-- | Format elapsed time as MM:SS or HH:MM:SS
+formatElapsed :: NominalDiffTime -> Text
+formatElapsed elapsed =
+  let totalSecs = floor elapsed :: Int
+      hours = totalSecs `div` 3600
+      mins = (totalSecs `mod` 3600) `div` 60
+      secs = totalSecs `mod` 60
+   in if hours > 0
+        then Text.pack (printf "%02d:%02d:%02d" hours mins secs)
+        else Text.pack (printf "%02d:%02d" mins secs)
-- 
cgit v1.2.3