Skip to content

Commit 221e5fe

Browse files
homeboy-ci[bot]homeboy-ci[bot]
committed
fix: scope cleanup pagination by repo
1 parent 45f4d34 commit 221e5fe

9 files changed

Lines changed: 335 additions & 26 deletions

‎inc/Abilities/WorkspaceAbilities.php‎

Lines changed: 46 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2007,6 +2007,10 @@ private function registerAbilities(): void {
20072007
'type' => 'boolean',
20082008
'description' => 'With apply=true, schedule bounded metadata reconciliation page jobs instead of applying synchronously.',
20092009
),
2010+
'repo' => array(
2011+
'type' => 'string',
2012+
'description' => 'Optional primary repo or worktree handle scope. When supplied, pagination, evidence, and writes are constrained to matching rows.',
2013+
),
20102014
'limit' => array(
20112015
'type' => 'integer',
20122016
'description' => 'Positive page size for bounded dry-run, direct apply, budgeted apply, or job-backed apply.',
@@ -2060,6 +2064,10 @@ private function registerAbilities(): void {
20602064
'type' => 'string',
20612065
'description' => 'Compact time budget for this report page, such as 60s or 10m.',
20622066
),
2067+
'repo' => array(
2068+
'type' => 'string',
2069+
'description' => 'Optional primary repo or worktree handle scope. When supplied, pagination and evidence are constrained to matching rows.',
2070+
),
20632071
),
20642072
),
20652073
'output_schema' => array(
@@ -2103,6 +2111,10 @@ private function registerAbilities(): void {
21032111
'type' => 'string',
21042112
'description' => 'Compact time budget for the underlying active_no_signal report page, such as 60s or 10m.',
21052113
),
2114+
'repo' => array(
2115+
'type' => 'string',
2116+
'description' => 'Optional primary repo or worktree handle scope. When supplied, pagination, evidence, and writes are constrained to matching rows.',
2117+
),
21062118
),
21072119
),
21082120
'output_schema' => array(
@@ -2147,6 +2159,10 @@ private function registerAbilities(): void {
21472159
'type' => 'string',
21482160
'description' => 'Compact time budget for the underlying active_no_signal report page, such as 60s or 10m.',
21492161
),
2162+
'repo' => array(
2163+
'type' => 'string',
2164+
'description' => 'Optional primary repo or worktree handle scope. When supplied, pagination, evidence, and writes are constrained to matching rows.',
2165+
),
21502166
),
21512167
),
21522168
'output_schema' => array(
@@ -2191,6 +2207,10 @@ private function registerAbilities(): void {
21912207
'type' => 'string',
21922208
'description' => 'Compact time budget for the underlying active_no_signal report page, such as 60s or 10m.',
21932209
),
2210+
'repo' => array(
2211+
'type' => 'string',
2212+
'description' => 'Optional primary repo or worktree handle scope. When supplied, pagination, evidence, and writes are constrained to matching rows.',
2213+
),
21942214
),
21952215
),
21962216
'output_schema' => array(
@@ -2235,6 +2255,10 @@ private function registerAbilities(): void {
22352255
'type' => 'string',
22362256
'description' => 'Compact time budget for the underlying active_no_signal report page, such as 60s or 10m.',
22372257
),
2258+
'repo' => array(
2259+
'type' => 'string',
2260+
'description' => 'Optional primary repo or worktree handle scope. When supplied, pagination, evidence, and writes are constrained to matching rows.',
2261+
),
22382262
),
22392263
),
22402264
'output_schema' => array(
@@ -2292,6 +2316,10 @@ private function registerAbilities(): void {
22922316
'type' => 'string',
22932317
'description' => 'Caller source marker forwarded to underlying cleanup abilities.',
22942318
),
2319+
'repo' => array(
2320+
'type' => 'string',
2321+
'description' => 'Optional primary repo or worktree handle scope forwarded to underlying paginated cleanup abilities.',
2322+
),
22952323
),
22962324
),
22972325
'output_schema' => array(
@@ -4262,6 +4290,9 @@ public static function worktreeReconcileMetadata( array $input ): array|\WP_Erro
42624290
if ( isset($input['source']) && '' !== trim( (string) $input['source']) ) {
42634291
$opts['source'] = trim( (string) $input['source']);
42644292
}
4293+
if ( isset($input['repo']) && '' !== trim( (string) $input['repo']) ) {
4294+
$opts['repo'] = trim( (string) $input['repo']);
4295+
}
42654296

42664297
return $workspace->worktree_reconcile_metadata($opts);
42674298
}
@@ -4284,6 +4315,9 @@ public static function worktreeActiveNoSignalReport( array $input ): array|\WP_E
42844315
if ( isset($input['until_budget']) && '' !== trim( (string) $input['until_budget']) ) {
42854316
$opts['until_budget'] = trim( (string) $input['until_budget']);
42864317
}
4318+
if ( isset($input['repo']) && '' !== trim( (string) $input['repo']) ) {
4319+
$opts['repo'] = trim( (string) $input['repo']);
4320+
}
42874321

42884322
return $workspace->worktree_active_no_signal_report($opts);
42894323
}
@@ -4308,6 +4342,9 @@ public static function worktreeActiveNoSignalFinalizedApply( array $input ): arr
43084342
if ( isset($input['until_budget']) && '' !== trim( (string) $input['until_budget']) ) {
43094343
$opts['until_budget'] = trim( (string) $input['until_budget']);
43104344
}
4345+
if ( isset($input['repo']) && '' !== trim( (string) $input['repo']) ) {
4346+
$opts['repo'] = trim( (string) $input['repo']);
4347+
}
43114348

43124349
return $workspace->worktree_active_no_signal_finalized_apply($opts);
43134350
}
@@ -4332,6 +4369,9 @@ public static function worktreeActiveNoSignalEquivalentCleanApply( array $input
43324369
if ( isset($input['until_budget']) && '' !== trim( (string) $input['until_budget']) ) {
43334370
$opts['until_budget'] = trim( (string) $input['until_budget']);
43344371
}
4372+
if ( isset($input['repo']) && '' !== trim( (string) $input['repo']) ) {
4373+
$opts['repo'] = trim( (string) $input['repo']);
4374+
}
43354375

43364376
return $workspace->worktree_active_no_signal_equivalent_clean_apply($opts);
43374377
}
@@ -4356,6 +4396,9 @@ public static function worktreeActiveNoSignalMergedApply( array $input ): array|
43564396
if ( isset($input['until_budget']) && '' !== trim( (string) $input['until_budget']) ) {
43574397
$opts['until_budget'] = trim( (string) $input['until_budget']);
43584398
}
4399+
if ( isset($input['repo']) && '' !== trim( (string) $input['repo']) ) {
4400+
$opts['repo'] = trim( (string) $input['repo']);
4401+
}
43594402

43604403
return $workspace->worktree_active_no_signal_merged_apply($opts);
43614404
}
@@ -4380,6 +4423,9 @@ public static function worktreeActiveNoSignalRemoteCleanApply( array $input ): a
43804423
if ( isset($input['until_budget']) && '' !== trim( (string) $input['until_budget']) ) {
43814424
$opts['until_budget'] = trim( (string) $input['until_budget']);
43824425
}
4426+
if ( isset($input['repo']) && '' !== trim( (string) $input['repo']) ) {
4427+
$opts['repo'] = trim( (string) $input['repo']);
4428+
}
43834429

43844430
return $workspace->worktree_active_no_signal_remote_clean_apply($opts);
43854431
}

‎inc/Cli/Commands/WorkspaceCommand.php‎

Lines changed: 13 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -3796,7 +3796,7 @@ public function worktree( array $args, array $assoc_args ): void {
37963796
}
37973797

37983798
if ( 'active-no-signal-drain' === $operation ) {
3799-
$result = $this->run_worktree_active_no_signal_drain($assoc_args);
3799+
$result = $this->run_worktree_active_no_signal_drain($assoc_args, isset($args[1]) ? (string) $args[1] : '');
38003800
if ( is_wp_error($result) ) {
38013801
$this->render_workspace_error($result);
38023802
return;
@@ -3867,7 +3867,7 @@ public function worktree( array $args, array $assoc_args ): void {
38673867
$input = array();
38683868
$input_builder = (string) ( $operation_config['input_builder'] ?? '' );
38693869
if ( '' !== $input_builder ) {
3870-
$input = $this->{$input_builder}($operation, $assoc_args);
3870+
$input = $this->{$input_builder}($operation, $assoc_args, isset($args[1]) ? (string) $args[1] : '');
38713871
}
38723872

38733873
switch ( $operation ) {
@@ -4024,6 +4024,9 @@ public function worktree( array $args, array $assoc_args ): void {
40244024
}
40254025
break;
40264026
case 'reconcile-metadata':
4027+
if ( ! empty($args[1]) ) {
4028+
$input['repo'] = (string) $args[1];
4029+
}
40274030
$input['dry_run'] = ! empty($assoc_args['dry-run']);
40284031
$input['apply'] = ! empty($assoc_args['apply']);
40294032
$input['via_jobs'] = ! empty($assoc_args['via-jobs']);
@@ -4145,8 +4148,11 @@ public function worktree( array $args, array $assoc_args ): void {
41454148
* @param array<string,mixed> $assoc_args CLI args.
41464149
* @return array<string,mixed>
41474150
*/
4148-
private function build_worktree_active_no_signal_input( string $operation, array $assoc_args ): array {
4151+
private function build_worktree_active_no_signal_input( string $operation, array $assoc_args, string $repo = '' ): array {
41494152
$input = array();
4153+
if ( '' !== trim($repo) ) {
4154+
$input['repo'] = trim($repo);
4155+
}
41504156

41514157
if ( in_array($operation, array( 'active-no-signal-finalized-apply', 'active-no-signal-equivalent-clean-apply', 'active-no-signal-merged-apply', 'active-no-signal-remote-clean-apply' ), true) ) {
41524158
$input['dry_run'] = ! empty($assoc_args['dry-run']);
@@ -4199,7 +4205,7 @@ private function run_worktree_abandoned_orchestration( array $assoc_args ): arra
41994205
* @param array<string,mixed> $assoc_args CLI args.
42004206
* @return array<string,mixed>|\WP_Error
42014207
*/
4202-
private function run_worktree_active_no_signal_drain( array $assoc_args ): array|\WP_Error {
4208+
private function run_worktree_active_no_signal_drain( array $assoc_args, string $repo = '' ): array|\WP_Error {
42034209
if ( ! empty($assoc_args['force']) ) {
42044210
return new \WP_Error('active_no_signal_drain_refuses_force', 'Active/no-signal drain will not force cleanup. Protected blockers remain blocked.', array( 'status' => 400 ));
42054211
}
@@ -4216,6 +4222,9 @@ private function run_worktree_active_no_signal_drain( array $assoc_args ): array
42164222
'apply' => ! empty($assoc_args['apply']),
42174223
'source' => self::CLEANUP_CLI_SOURCE,
42184224
);
4225+
if ( '' !== trim($repo) ) {
4226+
$input['repo'] = trim($repo);
4227+
}
42194228
foreach ( array( 'limit', 'passes', 'offset', 'stage' ) as $key ) {
42204229
if ( array_key_exists($key, $assoc_args) ) {
42214230
$input[ $key ] = $assoc_args[ $key ];

‎inc/Workspace/WorkspaceAbandonedCleanupOrchestrator.php‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -57,6 +57,7 @@ public function run( array $input ): array|\WP_Error {
5757
$stage = str_replace('_', '-', $stage);
5858
$until_budget = isset($input['until_budget']) && '' !== trim( (string) $input['until_budget']) ? trim( (string) $input['until_budget']) : '';
5959
$source = isset($input['source']) && '' !== trim( (string) $input['source']) ? trim( (string) $input['source']) : self::DEFAULT_SOURCE;
60+
$repo_scope = isset($input['repo']) && '' !== trim( (string) $input['repo']) ? trim( (string) $input['repo']) : '';
6061
$deadline = null;
6162
$stage_order = $this->stage_order();
6263

@@ -82,6 +83,9 @@ public function run( array $input ): array|\WP_Error {
8283
'limit' => $limit,
8384
'source' => $source,
8485
);
86+
if ( '' !== $repo_scope ) {
87+
$common_page['repo'] = $repo_scope;
88+
}
8589

8690
if ( $apply ) {
8791
$bounded = $this->run_bounded_apply($abilities['bounded_apply'], $result, true, $force, $limit, $source, 'initial');

‎inc/Workspace/WorkspaceActiveNoSignalCleanup.php‎

Lines changed: 13 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,11 @@ public function worktree_active_no_signal_report( array $opts = array() ): array
2929
? $opts['next_command_operation']
3030
: 'active-no-signal-report';
3131
$next_dry_run = 'active-no-signal-report' !== $next_operation && ! empty($opts['dry_run']);
32+
$scope = $this->normalize_worktree_operation_scope(isset($opts['repo']) ? (string) $opts['repo'] : '');
33+
if ( is_wp_error($scope) ) {
34+
return $scope;
35+
}
36+
$scope_arg = $this->worktree_operation_scope_cli_arg($scope);
3237
if ( $limit <= 0 ) {
3338
return new \WP_Error('invalid_active_no_signal_limit', 'Active/no-signal report --limit must be greater than 0.', array( 'status' => 400 ));
3439
}
@@ -39,7 +44,7 @@ public function worktree_active_no_signal_report( array $opts = array() ): array
3944
}
4045
}
4146

42-
$inventory = $this->worktree_cleanup_inventory_only('', '', false);
47+
$inventory = $this->worktree_cleanup_inventory_only('', '', false, null, 0, null === $scope ? '' : (string) $scope['argument']);
4348
if ( is_wp_error($inventory) ) {
4449
return $inventory;
4550
}
@@ -124,7 +129,7 @@ public function worktree_active_no_signal_report( array $opts = array() ): array
124129
'partial' => null !== $next_offset,
125130
'complete' => null === $next_offset,
126131
'next_offset' => $next_offset,
127-
'next_command' => null === $next_offset ? null : sprintf('studio wp datamachine-code workspace worktree %s%s --limit=%d --offset=%d%s --format=json', $next_operation, $next_dry_run ? ' --dry-run' : '', $limit, $next_offset, null !== $budget_context ? ' --until-budget=' . (string) $budget_context['label'] : ''),
132+
'next_command' => null === $next_offset ? null : sprintf('studio wp datamachine-code workspace worktree %s%s%s --limit=%d --offset=%d%s --format=json', $next_operation, $scope_arg, $next_dry_run ? ' --dry-run' : '', $limit, $next_offset, null !== $budget_context ? ' --until-budget=' . (string) $budget_context['label'] : ''),
128133
);
129134
if ( $budget_stopped ) {
130135
$pagination['partial'] = true;
@@ -140,7 +145,8 @@ public function worktree_active_no_signal_report( array $opts = array() ): array
140145
'summary' => array_merge($summary, array( 'slow_rows' => $this->summarize_slow_worktree_rows($rows) )),
141146
'pagination' => $pagination,
142147
'evidence' => array(
143-
'scope' => 'review-only active/no-signal and lifecycle reconciliation worktree evidence',
148+
'scope' => null === $scope ? 'review-only active/no-signal and lifecycle reconciliation worktree evidence' : 'review-only active/no-signal and lifecycle reconciliation worktree evidence scoped to ' . (string) $scope['argument'],
149+
'repo_scope' => $scope,
144150
'safety' => 'No worktrees or remote branches are deleted. Dirty and unpushed probes are evidence only.',
145151
'budget' => null === $budget_context ? null : $this->summarize_worktree_loop_budget_context($budget_context, $budget_stopped),
146152
'probe_cache' => $probe_cache['stats'],
@@ -484,6 +490,8 @@ private function build_active_no_signal_apply_pagination( array $pagination, str
484490
$budget_label = $matches[1];
485491
}
486492

493+
$scope = $this->normalize_worktree_operation_scope(isset($opts['repo']) ? (string) $opts['repo'] : '');
494+
$scope_arg = is_wp_error($scope) ? '' : $this->worktree_operation_scope_cli_arg($scope);
487495
$budget_arg = '' !== $budget_label ? ' --until-budget=' . $budget_label : '';
488496
$dry_run_arg = $dry_run ? ' --dry-run' : '';
489497
$limit = (int) ( $pagination['limit'] ?? 25 );
@@ -494,8 +502,9 @@ private function build_active_no_signal_apply_pagination( array $pagination, str
494502
}
495503

496504
$pagination['next_command'] = sprintf(
497-
'studio wp datamachine-code workspace worktree %s%s --limit=%d --offset=%d%s --format=json',
505+
'studio wp datamachine-code workspace worktree %s%s%s --limit=%d --offset=%d%s --format=json',
498506
$operation,
507+
$scope_arg,
499508
$dry_run_arg,
500509
$limit,
501510
$next_offset,

‎inc/Workspace/WorkspaceCoreUtilities.php‎

Lines changed: 64 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -230,6 +230,70 @@ public function parse_handle( string $handle ): array {
230230
return WorkspaceHandle::parse($handle)->to_array();
231231
}
232232

233+
/**
234+
* Normalize an optional worktree cleanup/reconciliation scope.
235+
*
236+
* @param string|null $scope Optional primary repo or worktree handle.
237+
* @return array{argument: string, repo: string, handle: string|null}|null|\WP_Error
238+
*/
239+
protected function normalize_worktree_operation_scope( ?string $scope ): array|null|\WP_Error {
240+
$scope = null === $scope ? '' : trim($scope);
241+
if ( '' === $scope ) {
242+
return null;
243+
}
244+
245+
$parsed = $this->parse_handle($scope);
246+
if ( '' === $parsed['repo'] || '' === $parsed['dir_name'] ) {
247+
return new \WP_Error('invalid_worktree_scope', sprintf('Worktree cleanup scope "%s" must be a primary repo or worktree handle.', $scope), array( 'status' => 400 ));
248+
}
249+
250+
return array(
251+
'argument' => $scope,
252+
'repo' => $parsed['repo'],
253+
'handle' => $parsed['is_worktree'] ? $parsed['dir_name'] : null,
254+
);
255+
}
256+
257+
/**
258+
* Whether an inventory/worktree row matches a normalized operation scope.
259+
*
260+
* @param array<string,mixed> $row Inventory or evidence row.
261+
* @param array{argument: string, repo: string, handle: string|null}|null $scope Optional normalized scope.
262+
* @return bool
263+
*/
264+
protected function worktree_row_matches_operation_scope( array $row, ?array $scope ): bool {
265+
if ( null === $scope ) {
266+
return true;
267+
}
268+
269+
$handle = (string) ( $row['handle'] ?? $row['name'] ?? '' );
270+
if ( null !== $scope['handle'] ) {
271+
return $handle === $scope['handle'];
272+
}
273+
274+
$repo = (string) ( $row['repo'] ?? '' );
275+
if ( '' === $repo && '' !== $handle ) {
276+
$repo = $this->parse_handle($handle)['repo'];
277+
}
278+
279+
return $repo === $scope['repo'];
280+
}
281+
282+
/**
283+
* Build the positional CLI scope segment for continuation commands.
284+
*
285+
* @param array{argument: string, repo: string, handle: string|null}|null $scope Optional normalized scope.
286+
* @return string
287+
*/
288+
protected function worktree_operation_scope_cli_arg( ?array $scope ): string {
289+
if ( null === $scope ) {
290+
return '';
291+
}
292+
293+
$argument = (string) $scope['argument'];
294+
return ' ' . ( preg_match('/^[A-Za-z0-9._@-]+$/', $argument) ? $argument : escapeshellarg($argument) );
295+
}
296+
233297
/**
234298
* Require file-operation callers to name a workspace handle explicitly.
235299
*

0 commit comments

Comments
 (0)