summaryrefslogtreecommitdiff
path: root/packages/cli/src/ui/hooks/useAutoAcceptIndicator.ts
blob: 17cffd404f13c9b6a55ddc4542ccbbac5c07bdbc (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
/**
 * @license
 * Copyright 2025 Google LLC
 * SPDX-License-Identifier: Apache-2.0
 */

import { useState, useEffect } from 'react';
import { useInput } from 'ink';
import type { Config } from '@gemini-code/server';

export interface UseAutoAcceptIndicatorArgs {
  config: Config;
}

export function useAutoAcceptIndicator({
  config,
}: UseAutoAcceptIndicatorArgs): boolean {
  const currentConfigValue = config.getAlwaysSkipModificationConfirmation();
  const [showAutoAcceptIndicator, setShowAutoAcceptIndicator] =
    useState(currentConfigValue);

  useEffect(() => {
    setShowAutoAcceptIndicator(currentConfigValue);
  }, [currentConfigValue]);

  useInput((_input, key) => {
    if (key.tab && key.shift) {
      const alwaysAcceptModificationConfirmations =
        !config.getAlwaysSkipModificationConfirmation();
      config.setAlwaysSkipModificationConfirmation(
        alwaysAcceptModificationConfirmations,
      );
      // Update local state immediately for responsiveness
      setShowAutoAcceptIndicator(alwaysAcceptModificationConfirmations);
    }
  });

  return showAutoAcceptIndicator;
}