@@ -10,6 +10,7 @@ import (
1010
1111 "github.com/recallnet/mainline/internal/domain"
1212 "github.com/recallnet/mainline/internal/git"
13+ "github.com/recallnet/mainline/internal/policy"
1314 "github.com/recallnet/mainline/internal/state"
1415)
1516
@@ -204,7 +205,7 @@ func waitForIntegratedSubmission(queued queuedSubmission, timeout time.Duration,
204205 ProtectedBranch : queued .Config .Repo .ProtectedBranch ,
205206 SubmissionStatus : queued .Submission .Status ,
206207 }
207- defer populateIntegrationWaitQueueSummary (queued .Store , queued .RepoRecord . ID , & result )
208+ defer populateIntegrationWaitQueueSummary (context . Background (), queued .Store , queued .RepoRecord , queued . Config , & result )
208209 mainEngine := git .NewEngine (queued .Config .Repo .MainWorktree )
209210
210211 ticker := time .NewTicker (pollInterval )
@@ -340,7 +341,7 @@ func waitForSubmissionTarget(queued queuedSubmission, target waitTarget, timeout
340341 ProtectedBranch : queued .Config .Repo .ProtectedBranch ,
341342 SubmissionStatus : queued .Submission .Status ,
342343 }
343- defer populateSubmissionWaitQueueSummary (queued .Store , queued .RepoRecord . ID , & result )
344+ defer populateSubmissionWaitQueueSummary (context . Background (), queued .Store , queued .RepoRecord , queued . Config , & result )
344345 mainEngine := git .NewEngine (queued .Config .Repo .MainWorktree )
345346 ticker := time .NewTicker (pollInterval )
346347 defer ticker .Stop ()
@@ -513,24 +514,24 @@ func waitForSubmissionTarget(queued queuedSubmission, target waitTarget, timeout
513514 }
514515}
515516
516- func populateIntegrationWaitQueueSummary (store state.Store , repoID int64 , result * integrationWaitResult ) {
517+ func populateIntegrationWaitQueueSummary (ctx context. Context , store state.Store , repoRecord state. RepositoryRecord , cfg policy. File , result * integrationWaitResult ) {
517518 if result == nil {
518519 return
519520 }
520- snapshot , err := loadQueueSnapshot ( store , repoID )
521+ snapshot , err := loadRepoStatusSnapshot ( ctx , store , repoRecord , cfg , 0 )
521522 if err != nil {
522523 return
523524 }
524- result .QueueSummary = snapshot .Summary
525+ result .QueueSummary = snapshot .QueueSummary
525526}
526527
527- func populateSubmissionWaitQueueSummary (store state.Store , repoID int64 , result * submissionWaitResult ) {
528+ func populateSubmissionWaitQueueSummary (ctx context. Context , store state.Store , repoRecord state. RepositoryRecord , cfg policy. File , result * submissionWaitResult ) {
528529 if result == nil {
529530 return
530531 }
531- snapshot , err := loadQueueSnapshot ( store , repoID )
532+ snapshot , err := loadRepoStatusSnapshot ( ctx , store , repoRecord , cfg , 0 )
532533 if err != nil {
533534 return
534535 }
535- result .QueueSummary = snapshot .Summary
536+ result .QueueSummary = snapshot .QueueSummary
536537}
0 commit comments