openchamber/packages/ui/src/components/session/sidebar/SidebarHeader.tsx

import React from 'react';
import {
  DropdownMenu,
  DropdownMenuContent,
  DropdownMenuItem,
  DropdownMenuSeparator,
  DropdownMenuTrigger,
} from '@/components/ui/dropdown-menu';
import { Tooltip, TooltipContent, TooltipTrigger } from '@/components/ui/tooltip';
import { cn } from '@/lib/utils';
import { ArrowsMerge } from '@/components/icons/ArrowsMerge';
import { Icon } from "@/components/icon/Icon";
import { useSessionDisplayStore } from '@/stores/useSessionDisplayStore';
import { isVSCodeRuntime } from '@/lib/desktop';
import { useI18n } from '@/lib/i18n';

type Props = {
  hideDirectoryControls: boolean;
  showRecentControls: boolean;
  handleOpenDirectoryDialog: () => void;
  openNewSessionDraft: () => void;
  canOpenMultiRun: boolean;
  openMultiRunLauncher: () => void;
  headerActionIconClass: string;
  headerActionButtonClass: string;
  isSessionSearchOpen: boolean;
  setIsSessionSearchOpen: (open: boolean | ((prev: boolean) => boolean)) => void;
  sessionSearchInputRef: React.RefObject<HTMLInputElement | null>;
  sessionSearchQuery: string;
  setSessionSearchQuery: (value: string) => void;
  hasSessionSearchQuery: boolean;
  searchMatchCount: number;
  collapseAllProjects: () => void;
  expandAllProjects: () => void;
  openScheduledTasksDialog: () => void;
  selectionModeEnabled: boolean;
  onToggleSelectionMode: () => void;
};

export function SidebarHeader(props: Props): React.ReactNode {
  const { t } = useI18n();
  const {
    hideDirectoryControls,
    showRecentControls,
    handleOpenDirectoryDialog,
    openNewSessionDraft,
    canOpenMultiRun,
    openMultiRunLauncher,
    headerActionIconClass,
    headerActionButtonClass,
    isSessionSearchOpen,
    setIsSessionSearchOpen,
    sessionSearchInputRef,
    sessionSearchQuery,
    setSessionSearchQuery,
    hasSessionSearchQuery,
    searchMatchCount,
    collapseAllProjects,
    expandAllProjects,
    openScheduledTasksDialog,
    selectionModeEnabled,
    onToggleSelectionMode,
  } = props;

  const displayMode = useSessionDisplayStore((state) => state.displayMode);
  const showRecentSection = useSessionDisplayStore((state) => state.showRecentSection);
  const showArchivedSessions = useSessionDisplayStore((state) => state.showArchivedSessions);
  const setDisplayMode = useSessionDisplayStore((state) => state.setDisplayMode);
  const toggleRecentSection = useSessionDisplayStore((state) => state.toggleRecentSection);
  const toggleArchivedSessions = useSessionDisplayStore((state) => state.toggleArchivedSessions);
  // VS Code forces the expanded layout, so the mode toggle is meaningless there.
  const showDisplayModeToggle = !isVSCodeRuntime();

  if (hideDirectoryControls) {
    return null;
  }

  return (
    <div className="select-none flex-shrink-0 px-2.5 py-1">
      <div className="flex h-auto min-h-8 flex-col gap-1">
        <div className="flex h-8 items-center justify-between gap-2">
          <div className="flex items-center gap-1.5">
            <Tooltip>
              <TooltipTrigger asChild>
                <button
                  type="button"
                  onClick={handleOpenDirectoryDialog}
                  className={headerActionButtonClass}
                  aria-label={t('sessions.sidebar.header.actions.addProject')}
                >
                  <Icon name="folder-add" className={headerActionIconClass} />
                </button>
              </TooltipTrigger>
              <TooltipContent side="bottom" sideOffset={4}><p>{t('sessions.sidebar.header.actions.addProject')}</p></TooltipContent>
            </Tooltip>

            <Tooltip>
              <TooltipTrigger asChild>
                <button
                  type="button"
                  onClick={openNewSessionDraft}
                  className={headerActionButtonClass}
                  aria-label={t('sessions.sidebar.header.actions.newSession')}
                >
                  <Icon name="chat-new" className={headerActionIconClass} />
                </button>
              </TooltipTrigger>
              <TooltipContent side="bottom" sideOffset={4}><p>{t('sessions.sidebar.header.actions.newSession')}</p></TooltipContent>
            </Tooltip>

            <Tooltip>
              <TooltipTrigger asChild>
                <button
                  type="button"
                  onClick={openMultiRunLauncher}
                  className={headerActionButtonClass}
                  aria-label={t('sessions.sidebar.header.actions.newMultiRun')}
                  disabled={!canOpenMultiRun}
                >
                  <ArrowsMerge className={headerActionIconClass} />
                </button>
              </TooltipTrigger>
              <TooltipContent side="bottom" sideOffset={4}><p>{t('sessions.sidebar.header.actions.newMultiRun')}</p></TooltipContent>
            </Tooltip>

            <Tooltip>
              <TooltipTrigger asChild>
                <button
                  type="button"
                  onClick={openScheduledTasksDialog}
                  className={headerActionButtonClass}
                  aria-label={t('sessions.sidebar.header.actions.scheduledTasks')}
                >
                  <Icon name="calendar-schedule" className={headerActionIconClass} />
                </button>
              </TooltipTrigger>
              <TooltipContent side="bottom" sideOffset={4}><p>{t('sessions.sidebar.header.actions.scheduledTasks')}</p></TooltipContent>
            </Tooltip>
          </div>

          <div className="flex items-center gap-1.5">
            <Tooltip>
              <TooltipTrigger asChild>
                <button
                  type="button"
                  onClick={() => setIsSessionSearchOpen((prev) => !prev)}
                  className={headerActionButtonClass}
                  aria-label={t('sessions.sidebar.header.actions.searchSessions')}
                  aria-expanded={isSessionSearchOpen}
                >
                  <Icon name="search" className={headerActionIconClass} />
                </button>
              </TooltipTrigger>
              <TooltipContent side="bottom" sideOffset={4}><p>{t('sessions.sidebar.header.actions.searchSessions')}</p></TooltipContent>
            </Tooltip>

            <Tooltip>
              <TooltipTrigger asChild>
                <button
                  type="button"
                  onClick={onToggleSelectionMode}
                  className={cn(headerActionButtonClass, selectionModeEnabled && 'bg-interactive-hover text-primary')}
                  aria-label={selectionModeEnabled
                    ? t('sessions.sidebar.header.actions.exitSelection')
                    : t('sessions.sidebar.header.actions.selectSessions')}
                  aria-pressed={selectionModeEnabled}
                >
                  <Icon name="checkbox-multiple" className={headerActionIconClass} />
                </button>
              </TooltipTrigger>
              <TooltipContent side="bottom" sideOffset={4}>
                <p>{selectionModeEnabled
                  ? t('sessions.sidebar.header.actions.exitSelection')
                  : t('sessions.sidebar.header.actions.selectSessions')}</p>
              </TooltipContent>
            </Tooltip>

            <DropdownMenu>
              <Tooltip>
                <TooltipTrigger asChild>
                  <DropdownMenuTrigger asChild>
                    <button
                      type="button"
                      className={headerActionButtonClass}
                      aria-label={t('sessions.sidebar.header.actions.sessionDisplayMode')}
                    >
                      <Icon name="equalizer-2" className={headerActionIconClass} />
                    </button>
                  </DropdownMenuTrigger>
                </TooltipTrigger>
                <TooltipContent side="bottom" sideOffset={4}><p>{t('sessions.sidebar.header.displayMode.label')}</p></TooltipContent>
              </Tooltip>
              <DropdownMenuContent align="end" className="min-w-[160px]">
                {showDisplayModeToggle ? (
                  <>
                    <DropdownMenuItem
                      onClick={() => setDisplayMode('default')}
                      className="flex items-center justify-between"
                    >
                      <span>{t('sessions.sidebar.header.displayMode.default')}</span>
                      {displayMode === 'default' ? <Icon name="check" className="h-4 w-4 text-primary" /> : null}
                    </DropdownMenuItem>
                    <DropdownMenuItem
                      onClick={() => setDisplayMode('minimal')}
                      className="flex items-center justify-between"
                    >
                      <span>{t('sessions.sidebar.header.displayMode.minimal')}</span>
                      {displayMode === 'minimal' ? <Icon name="check" className="h-4 w-4 text-primary" /> : null}
                    </DropdownMenuItem>
                  </>
                ) : null}
                {showRecentControls ? (
                  <>
                    {showDisplayModeToggle ? <DropdownMenuSeparator /> : null}
                    <DropdownMenuItem
                      onClick={toggleRecentSection}
                      className="flex items-center justify-between"
                    >
                      <span>{t('sessions.sidebar.header.displayMode.showRecent')}</span>
                      {showRecentSection ? <Icon name="check" className="h-4 w-4 text-primary" /> : null}
                    </DropdownMenuItem>
                    <DropdownMenuItem
                      onClick={toggleArchivedSessions}
                      className="flex items-center justify-between"
                    >
                      <span>{t('sessions.sidebar.header.displayMode.showArchived')}</span>
                      {showArchivedSessions ? <Icon name="check" className="h-4 w-4 text-primary" /> : null}
                    </DropdownMenuItem>
                  </>
                ) : null}
                <DropdownMenuSeparator />
                <DropdownMenuItem onClick={collapseAllProjects} className="flex items-center gap-2">
                  <Icon name="contract-up-down" className="h-4 w-4" />
                  <span>{t('sessions.sidebar.header.displayMode.collapseAll')}</span>
                </DropdownMenuItem>
                <DropdownMenuItem onClick={expandAllProjects} className="flex items-center gap-2">
                  <Icon name="expand-up-down" className="h-4 w-4" />
                  <span>{t('sessions.sidebar.header.displayMode.expandAll')}</span>
                </DropdownMenuItem>
              </DropdownMenuContent>
            </DropdownMenu>
          </div>
        </div>

        {isSessionSearchOpen ? (
          <div className="pb-1">
            <div className="mb-1 flex items-center justify-between px-0.5 typography-micro text-muted-foreground/80">
              {hasSessionSearchQuery ? (
                <span>{searchMatchCount === 1
                  ? t('sessions.sidebar.header.search.matchCountSingle', { count: searchMatchCount })
                  : t('sessions.sidebar.header.search.matchCountPlural', { count: searchMatchCount })}</span>
              ) : <span />}
              <span>{t('sessions.sidebar.header.search.escapeHint')}</span>
            </div>
            <div className="relative">
              <Icon name="search" className="pointer-events-none absolute left-2 top-1/2 h-4 w-4 -translate-y-1/2 text-muted-foreground" />
              <input
                ref={sessionSearchInputRef}
                value={sessionSearchQuery}
                onChange={(event) => setSessionSearchQuery(event.target.value)}
                placeholder={t('sessions.sidebar.header.search.placeholder')}
                className="h-8 w-full rounded-md border border-border bg-transparent pl-8 pr-8 typography-ui-label text-foreground outline-none focus-visible:ring-2 focus-visible:ring-primary/50"
                onKeyDown={(event) => {
                  if (event.key === 'Escape') {
                    event.stopPropagation();
                    if (hasSessionSearchQuery) {
                      setSessionSearchQuery('');
                    } else {
                      setIsSessionSearchOpen(false);
                    }
                  }
                }}
              />
              {sessionSearchQuery.length > 0 ? (
                <button
                  type="button"
                  onClick={() => setSessionSearchQuery('')}
                  className="absolute right-1 top-1/2 inline-flex h-6 w-6 -translate-y-1/2 items-center justify-center rounded-md text-muted-foreground hover:bg-interactive-hover/60 hover:text-foreground focus-visible:outline-none focus-visible:ring-2 focus-visible:ring-primary/50"
                  aria-label={t('sessions.sidebar.header.search.clear')}
                >
                  <Icon name="close" className="h-3.5 w-3.5" />
                </button>
              ) : null}
            </div>
          </div>
        ) : null}
      </div>
    </div>
  );
}