Switch to using a reaction for in-progress tracking

This commit is contained in:
Brian McKenna 2026-02-26 06:56:36 +11:00
parent af967210c6
commit 93becd5020
4 changed files with 150 additions and 49 deletions

View File

@ -106,11 +106,17 @@ parseClaudeJsonOutput raw =
case fromJSON resultVal of case fromJSON resultVal of
Success v -> Right v Success v -> Right v
Error e -> Left $ "Failed to parse structured_output: " <> e Error e -> Left $ "Failed to parse structured_output: " <> e
Nothing -> Left "Missing 'structured_output' field in Claude output" Nothing ->
-- Check if Claude reported an error (e.g. max turns reached)
case parseMaybe extractSubtype val of
Just subtype -> Left $ "Claude failed with: " <> T.unpack subtype
Nothing -> Left "Missing 'structured_output' field in Claude output"
Left err -> Left $ "Failed to parse Claude JSON: " <> err Left err -> Left $ "Failed to parse Claude JSON: " <> err
where where
extractField :: Value -> Parser Value extractField :: Value -> Parser Value
extractField = withObject "envelope" (.: "structured_output") extractField = withObject "envelope" (.: "structured_output")
extractSubtype :: Value -> Parser Text
extractSubtype = withObject "envelope" (.: "subtype")
-- Parse Claude's JSON envelope for freeform text: { "result": "..." } -- Parse Claude's JSON envelope for freeform text: { "result": "..." }
parseClaudeTextOutput :: String -> Either String Text parseClaudeTextOutput :: String -> Either String Text
@ -120,11 +126,16 @@ parseClaudeTextOutput raw =
case parseMaybe extractField val of case parseMaybe extractField val of
Just (String resultText) -> Right resultText Just (String resultText) -> Right resultText
Just _ -> Left "Expected string 'result' in Claude output" Just _ -> Left "Expected string 'result' in Claude output"
Nothing -> Left "Missing 'result' field in Claude output" Nothing ->
case parseMaybe extractSubtype val of
Just subtype -> Left $ "Claude failed with: " <> T.unpack subtype
Nothing -> Left "Missing 'result' field in Claude output"
Left err -> Left $ "Failed to parse Claude JSON: " <> err Left err -> Left $ "Failed to parse Claude JSON: " <> err
where where
extractField :: Value -> Parser Value extractField :: Value -> Parser Value
extractField = withObject "envelope" (.: "result") extractField = withObject "envelope" (.: "result")
extractSubtype :: Value -> Parser Text
extractSubtype = withObject "envelope" (.: "subtype")
strToLBS :: String -> BL.ByteString strToLBS :: String -> BL.ByteString
strToLBS = BL.fromStrict . TE.encodeUtf8 . T.pack strToLBS = BL.fromStrict . TE.encodeUtf8 . T.pack

View File

@ -34,6 +34,7 @@ data IssueCommentEvent = IssueCommentEvent
{ iceAction :: Text { iceAction :: Text
, iceIsPull :: Bool , iceIsPull :: Bool
, iceNumber :: Int , iceNumber :: Int
, iceCommentId :: Int
, iceCommentBody :: Text , iceCommentBody :: Text
, iceSender :: Text , iceSender :: Text
} }
@ -42,6 +43,7 @@ data PRReviewEvent = PRReviewEvent
{ prreAction :: Text { prreAction :: Text
, prreNumber :: Int , prreNumber :: Int
, prreTitle :: Text , prreTitle :: Text
, prreReviewId :: Int
, prreReviewBody :: Text , prreReviewBody :: Text
, prreSender :: Text , prreSender :: Text
} }
@ -50,6 +52,7 @@ data PRReviewCommentEvent = PRReviewCommentEvent
{ prrceAction :: Text { prrceAction :: Text
, prrceNumber :: Int , prrceNumber :: Int
, prrceTitle :: Text , prrceTitle :: Text
, prrceCommentId :: Int
, prrceCommentBody :: Text , prrceCommentBody :: Text
, prrceSender :: Text , prrceSender :: Text
} }
@ -86,13 +89,15 @@ instance FromJSON IssueCommentEvent where
pure $ case pr of pure $ case pr of
Just (Object _) -> True Just (Object _) -> True
_ -> False _ -> False
comment <- o .: "comment" comment <- o .: "comment"
body <- comment .: "body" commentId <- comment .: "id"
sender <- parseSender o body <- comment .: "body"
sender <- parseSender o
pure IssueCommentEvent pure IssueCommentEvent
{ iceAction = action { iceAction = action
, iceIsPull = isPull , iceIsPull = isPull
, iceNumber = number , iceNumber = number
, iceCommentId = commentId
, iceCommentBody = body , iceCommentBody = body
, iceSender = sender , iceSender = sender
} }
@ -103,13 +108,15 @@ instance FromJSON PRReviewEvent where
pr <- o .: "pull_request" pr <- o .: "pull_request"
number <- pr .: "number" number <- pr .: "number"
title <- pr .: "title" title <- pr .: "title"
review <- o .: "review" review <- o .: "review"
body <- review .:? "body" .!= "" reviewId <- review .: "id"
sender <- parseSender o body <- review .:? "body" .!= ""
sender <- parseSender o
pure PRReviewEvent pure PRReviewEvent
{ prreAction = action { prreAction = action
, prreNumber = number , prreNumber = number
, prreTitle = title , prreTitle = title
, prreReviewId = reviewId
, prreReviewBody = body , prreReviewBody = body
, prreSender = sender , prreSender = sender
} }
@ -120,13 +127,15 @@ instance FromJSON PRReviewCommentEvent where
pr <- o .: "pull_request" pr <- o .: "pull_request"
number <- pr .: "number" number <- pr .: "number"
title <- pr .: "title" title <- pr .: "title"
comment <- o .: "comment" comment <- o .: "comment"
body <- comment .: "body" commentId <- comment .: "id"
sender <- parseSender o body <- comment .: "body"
sender <- parseSender o
pure PRReviewCommentEvent pure PRReviewCommentEvent
{ prrceAction = action { prrceAction = action
, prrceNumber = number , prrceNumber = number
, prrceTitle = title , prrceTitle = title
, prrceCommentId = commentId
, prrceCommentBody = body , prrceCommentBody = body
, prrceSender = sender , prrceSender = sender
} }

View File

@ -8,7 +8,12 @@ module Bot.Gitea
, getPrReviews , getPrReviews
, createComment , createComment
, updateComment , updateComment
, deleteComment
, postReview , postReview
, addCommentReaction
, removeCommentReaction
, addIssueReaction
, removeIssueReaction
, PrInfo (..) , PrInfo (..)
, GiteaComment (..) , GiteaComment (..)
, GiteaReview (..) , GiteaReview (..)
@ -137,6 +142,44 @@ apiPostRaw gc path body = do
resp <- httpLbs req (gcManager gc) resp <- httpLbs req (gcManager gc)
pure (statusCode $ responseStatus resp, responseBody resp) pure (statusCode $ responseStatus resp, responseBody resp)
apiDeleteWithBody :: GiteaClient -> Text -> Value -> IO ()
apiDeleteWithBody gc path body = do
let url = T.unpack (gcBaseUrl gc <> path)
initReq <- parseRequest url
let req = initReq
{ method = "DELETE"
, requestHeaders =
[ ("Authorization", "token " <> TE.encodeUtf8 (gcToken gc))
, ("Accept", "application/json")
, ("Content-Type", "application/json")
]
, requestBody = RequestBodyLBS (encode body)
}
resp <- httpLbs req (gcManager gc)
let sc = statusCode (responseStatus resp)
if sc >= 200 && sc < 300
then pure ()
else fail $ "API DELETE " <> url <> " failed with status " <> show sc
<> ": " <> BS.unpack (BL.toStrict $ responseBody resp)
apiDelete :: GiteaClient -> Text -> IO ()
apiDelete gc path = do
let url = T.unpack (gcBaseUrl gc <> path)
initReq <- parseRequest url
let req = initReq
{ method = "DELETE"
, requestHeaders =
[ ("Authorization", "token " <> TE.encodeUtf8 (gcToken gc))
, ("Accept", "application/json")
]
}
resp <- httpLbs req (gcManager gc)
let sc = statusCode (responseStatus resp)
if sc >= 200 && sc < 300
then pure ()
else fail $ "API DELETE " <> url <> " failed with status " <> show sc
<> ": " <> BS.unpack (BL.toStrict $ responseBody resp)
-- API operations -- API operations
getCurrentUser :: GiteaClient -> IO Text getCurrentUser :: GiteaClient -> IO Text
@ -260,3 +303,29 @@ postReview gc prNum body comments = do
then pure () then pure ()
else fail $ "postReview failed with status " <> show sc else fail $ "postReview failed with status " <> show sc
<> ": " <> BS.unpack (BL.toStrict respBody) <> ": " <> BS.unpack (BL.toStrict respBody)
deleteComment :: GiteaClient -> Int -> IO ()
deleteComment gc commentId =
apiDelete gc (repoPath gc <> "/issues/comments/" <> T.pack (show commentId))
addCommentReaction :: GiteaClient -> Int -> Text -> IO ()
addCommentReaction gc commentId reaction = do
let payload = object ["content" .= reaction]
_ <- apiPostRaw gc (repoPath gc <> "/issues/comments/" <> T.pack (show commentId) <> "/reactions") payload
pure ()
removeCommentReaction :: GiteaClient -> Int -> Text -> IO ()
removeCommentReaction gc commentId reaction = do
let payload = object ["content" .= reaction]
apiDeleteWithBody gc (repoPath gc <> "/issues/comments/" <> T.pack (show commentId) <> "/reactions") payload
addIssueReaction :: GiteaClient -> Int -> Text -> IO ()
addIssueReaction gc issueNum reaction = do
let payload = object ["content" .= reaction]
_ <- apiPostRaw gc (repoPath gc <> "/issues/" <> T.pack (show issueNum) <> "/reactions") payload
pure ()
removeIssueReaction :: GiteaClient -> Int -> Text -> IO ()
removeIssueReaction gc issueNum reaction = do
let payload = object ["content" .= reaction]
apiDeleteWithBody gc (repoPath gc <> "/issues/" <> T.pack (show issueNum) <> "/reactions") payload

View File

@ -2,7 +2,7 @@ module Bot.Review
( handleEvent ( handleEvent
) where ) where
import Control.Exception (SomeException, catch) import Control.Exception (SomeException, catch, finally)
import Data.Text (Text) import Data.Text (Text)
import Data.Text qualified as T import Data.Text qualified as T
import System.IO (hPutStrLn, stderr) import System.IO (hPutStrLn, stderr)
@ -37,14 +37,29 @@ handleEvent cfg event = do
dispatch :: Config -> GiteaClient -> Event -> Int -> IO () dispatch :: Config -> GiteaClient -> Event -> Int -> IO ()
dispatch cfg client event prNum = do dispatch cfg client event prNum = do
-- Create tracking comment -- Add eyes reaction to acknowledge
commentId <- createComment client prNum "\x23F3 Thinking..." addReaction event
putStrLn $ "Created tracking comment: " <> show commentId putStrLn "Added eyes reaction"
-- Ensure tracking comment is updated even on error
let jobUrl = cfgGiteaUrl cfg <> "/" <> cfgRepoOwner cfg <> "/" <> cfgRepoName cfg let jobUrl = cfgGiteaUrl cfg <> "/" <> cfgRepoOwner cfg <> "/" <> cfgRepoName cfg
<> "/actions/runs/" <> cfgRunId cfg <> "/actions/runs/" <> cfgRunId cfg
let run = do
ctx <- gatherContext client prNum
case event of
EvPullRequest _ ->
handleReview cfg client event prNum jobUrl ctx
EvIssueComment ice ->
handleReply cfg client (iceCommentBody ice) prNum jobUrl ctx
EvPRReview prre ->
handleReply cfg client (prreReviewBody prre) prNum jobUrl ctx
EvPRReviewComment prrce ->
handleReply cfg client (prrceCommentBody prrce) prNum jobUrl ctx
let onError :: SomeException -> IO () let onError :: SomeException -> IO ()
onError ex = do onError ex = do
let errMsg = T.pack (show ex) let errMsg = T.pack (show ex)
@ -54,27 +69,27 @@ dispatch cfg client event prNum = do
, "" , ""
, T.take 2000 errMsg , T.take 2000 errMsg
] ]
updateComment client commentId body (do _ <- createComment client prNum body; pure ())
`catch` \(_ :: SomeException) -> pure () `catch` \(_ :: SomeException) -> pure ()
let run = do let removeReaction_ =
-- Gather PR context removeReaction event
ctx <- gatherContext client prNum `catch` \(_ :: SomeException) -> pure ()
case event of (run `catch` onError) `finally` removeReaction_
EvPullRequest _ -> do
handleReview cfg client event prNum commentId jobUrl ctx
EvIssueComment _ -> do where
handleReview cfg client event prNum commentId jobUrl ctx reaction = "eyes"
addReaction = \case
EvPRReview prre -> do EvPullRequest _ -> addIssueReaction client prNum reaction
handleReply cfg client (prreReviewBody prre) prNum commentId jobUrl ctx EvIssueComment ice -> addCommentReaction client (iceCommentId ice) reaction
EvPRReview prre -> addCommentReaction client (prreReviewId prre) reaction
EvPRReviewComment prrce -> do EvPRReviewComment prrce -> addCommentReaction client (prrceCommentId prrce) reaction
handleReply cfg client (prrceCommentBody prrce) prNum commentId jobUrl ctx removeReaction = \case
EvPullRequest _ -> removeIssueReaction client prNum reaction
run `catch` onError EvIssueComment ice -> removeCommentReaction client (iceCommentId ice) reaction
EvPRReview prre -> removeCommentReaction client (prreReviewId prre) reaction
EvPRReviewComment prrce -> removeCommentReaction client (prrceCommentId prrce) reaction
gatherContext :: GiteaClient -> Int -> IO PRContext gatherContext :: GiteaClient -> Int -> IO PRContext
gatherContext client prNum = do gatherContext client prNum = do
@ -89,8 +104,8 @@ gatherContext client prNum = do
, ctxReviews = reviews , ctxReviews = reviews
} }
handleReview :: Config -> GiteaClient -> Event -> Int -> Int -> Text -> PRContext -> IO () handleReview :: Config -> GiteaClient -> Event -> Int -> Text -> PRContext -> IO ()
handleReview cfg client event prNum commentId jobUrl ctx = do handleReview cfg client event prNum jobUrl ctx = do
let prompt = buildReviewPrompt cfg event ctx let prompt = buildReviewPrompt cfg event ctx
result <- invokeClaudeReview prompt result <- invokeClaudeReview prompt
case result of case result of
@ -101,28 +116,23 @@ handleReview cfg client event prNum commentId jobUrl ctx = do
, "" , ""
, T.pack err , T.pack err
] ]
updateComment client commentId body _ <- createComment client prNum body
pure ()
Right review -> do Right review -> do
-- Update tracking comment with summary
let sender = eventSender event let sender = eventSender event
body = T.unlines summary = T.unlines
[ "**Claude** reviewed @" <> sender <> "'s PR \x2014 [View job run](" <> jobUrl <> ")" [ "**Claude** reviewed @" <> sender <> "'s PR \x2014 [View job run](" <> jobUrl <> ")"
, "" , ""
, "---" , "---"
, "" , ""
, roSummary review , roSummary review
] ]
updateComment client commentId body inlineComments = map toInlineComment (roComments review)
postReviewWithFallback client prNum summary inlineComments
-- Post review with inline comments if any handleReply :: Config -> GiteaClient -> Text -> Int -> Text -> PRContext -> IO ()
let inlineComments = map toInlineComment (roComments review) handleReply cfg client triggerBody prNum jobUrl ctx = do
if null inlineComments
then pure ()
else postReviewWithFallback client prNum (roSummary review) inlineComments
handleReply :: Config -> GiteaClient -> Text -> Int -> Int -> Text -> PRContext -> IO ()
handleReply cfg client triggerBody _prNum commentId jobUrl ctx = do
let prompt = buildReplyPrompt cfg triggerBody ctx let prompt = buildReplyPrompt cfg triggerBody ctx
result <- invokeClaudeReply prompt result <- invokeClaudeReply prompt
case result of case result of
@ -133,7 +143,8 @@ handleReply cfg client triggerBody _prNum commentId jobUrl ctx = do
, "" , ""
, T.pack err , T.pack err
] ]
updateComment client commentId body _ <- createComment client prNum body
pure ()
Right replyText -> do Right replyText -> do
let body = T.unlines let body = T.unlines
@ -143,7 +154,8 @@ handleReply cfg client triggerBody _prNum commentId jobUrl ctx = do
, "" , ""
, replyText , replyText
] ]
updateComment client commentId body _ <- createComment client prNum body
pure ()
toInlineComment :: ClaudeComment -> InlineComment toInlineComment :: ClaudeComment -> InlineComment
toInlineComment cc = InlineComment toInlineComment cc = InlineComment