cinny/src/app/components/CloseBehaviorPrompt.tsx

import React, { useCallback, useEffect, useState } from 'react';
import FocusTrap from 'focus-trap-react';
import {
  Box,
  Button,
  Checkbox,
  Dialog,
  Header,
  Overlay,
  OverlayBackdrop,
  OverlayCenter,
  Text,
  config,
} from 'folds';
import { invokeTauri, isTauri, tauriInvoke, useTauriEvent } from '../hooks/useTauri';
import { useModalStyle } from '../hooks/useModalStyle';

/**
 * cinny-desktop #5: the first time the window is closed, ask whether Lotus Chat
 * should keep running in the tray (calls, messages, notifications) or quit —
 * and, in the same moment, whether it should start when you sign in. Asked
 * once; changeable in Settings → General. The native side sends
 * `lotus-close-requested` only while the choice is still "ask", and never
 * during a call (a call always goes to the tray).
 */
export function CloseBehaviorPrompt() {
  const [open, setOpen] = useState(false);
  const [launchOnLogin, setLaunchOnLogin] = useState(false);
  const modalStyle = useModalStyle(420);

  // Tells the native side we're listening (see get_close_behavior).
  useEffect(() => {
    if (!isTauri()) return;
    tauriInvoke()?.('get_close_behavior').catch(() => undefined);
  }, []);

  useTauriEvent('lotus-close-requested', () => setOpen(true));

  const choose = useCallback(
    (value: 'tray' | 'quit') => {
      if (launchOnLogin) invokeTauri('plugin:autostart|enable');
      setOpen(false);
      invokeTauri('resolve_close_request', { value });
    },
    [launchOnLogin],
  );

  if (!open) return null;
  return (
    <Overlay open backdrop={<OverlayBackdrop />}>
      <OverlayCenter>
        <FocusTrap
          focusTrapOptions={{
            initialFocus: '#close-behavior-keep',
            // Dismissing without choosing keeps the window open; the next close
            // asks again.
            onDeactivate: () => setOpen(false),
            clickOutsideDeactivates: true,
          }}
        >
          <Dialog
            variant="Surface"
            role="dialog"
            aria-modal="true"
            aria-labelledby="close-behavior-title"
            aria-describedby="close-behavior-body"
            style={modalStyle}
          >
            <Header
              style={{
                padding: `0 ${config.space.S400}`,
                borderBottomWidth: config.borderWidth.B300,
              }}
              variant="Surface"
              size="500"
            >
              <Text as="h2" size="H4" id="close-behavior-title">
                Keep Lotus Chat running?
              </Text>
            </Header>
            <Box style={{ padding: config.space.S400 }} direction="Column" gap="400">
              <Text priority="400" id="close-behavior-body">
                Lotus Chat can keep running in the system tray when you close this window, so
                messages, notifications and calls still reach you. Open it again from the tray icon.
              </Text>
              <Box as="label" alignItems="Center" gap="200" style={{ cursor: 'pointer' }}>
                <Checkbox
                  size="300"
                  variant="Primary"
                  checked={launchOnLogin}
                  onClick={(e: React.MouseEvent<HTMLInputElement>) =>
                    setLaunchOnLogin(e.currentTarget.checked)
                  }
                />
                <Text size="T300">Start Lotus Chat when I sign in to my computer</Text>
              </Box>
              <Box direction="Column" gap="200">
                <Button id="close-behavior-keep" variant="Primary" onClick={() => choose('tray')}>
                  <Text size="B400">Keep running in the tray</Text>
                </Button>
                <Button variant="Secondary" fill="Soft" onClick={() => choose('quit')}>
                  <Text size="B400">Quit when I close the window</Text>
                </Button>
              </Box>
              <Text size="T200" priority="300">
                You can change this in Settings → General.
              </Text>
            </Box>
          </Dialog>
        </FocusTrap>
      </OverlayCenter>
    </Overlay>
  );
}