feat: system prompt swapping (#131)

2025-11-27 01:30:11 -08:00
parent 8330534f00
commit 135c19c7d7
10 changed files with 665 additions and 4 deletions
--- a/src/cli/components/SystemPromptSelector.tsx
+++ b/src/cli/components/SystemPromptSelector.tsx
@@ -0,0 +1,111 @@
+// Import useInput from vendored Ink for bracketed paste support
+import { Box, Text, useInput } from "ink";
+import { useMemo, useState } from "react";
+import { SYSTEM_PROMPTS } from "../../agent/promptAssets";
+import { colors } from "./colors";
+
+interface SystemPromptSelectorProps {
+  currentPromptId?: string;
+  onSelect: (promptId: string) => void;
+  onCancel: () => void;
+}
+
+export function SystemPromptSelector({
+  currentPromptId,
+  onSelect,
+  onCancel,
+}: SystemPromptSelectorProps) {
+  const [showAll, setShowAll] = useState(false);
+  const [selectedIndex, setSelectedIndex] = useState(0);
+
+  const featuredPrompts = useMemo(
+    () => SYSTEM_PROMPTS.filter((prompt) => prompt.isFeatured),
+    [],
+  );
+
+  const visiblePrompts = useMemo(() => {
+    if (showAll) return SYSTEM_PROMPTS;
+    if (featuredPrompts.length > 0) return featuredPrompts;
+    return SYSTEM_PROMPTS.slice(0, 3);
+  }, [featuredPrompts, showAll]);
+
+  const hasHiddenPrompts = visiblePrompts.length < SYSTEM_PROMPTS.length;
+  const hasShowAllOption = !showAll && hasHiddenPrompts;
+
+  const totalItems = visiblePrompts.length + (hasShowAllOption ? 1 : 0);
+
+  useInput((_input, key) => {
+    if (key.upArrow) {
+      setSelectedIndex((prev) => Math.max(0, prev - 1));
+    } else if (key.downArrow) {
+      setSelectedIndex((prev) => Math.min(totalItems - 1, prev + 1));
+    } else if (key.return) {
+      if (hasShowAllOption && selectedIndex === visiblePrompts.length) {
+        setShowAll(true);
+        setSelectedIndex(0);
+      } else {
+        const selectedPrompt = visiblePrompts[selectedIndex];
+        if (selectedPrompt) {
+          onSelect(selectedPrompt.id);
+        }
+      }
+    } else if (key.escape) {
+      onCancel();
+    }
+  });
+
+  return (
+    <Box flexDirection="column" gap={1}>
+      <Box>
+        <Text bold color={colors.selector.title}>
+          Select System Prompt (↑↓ to navigate, Enter to select, ESC to cancel)
+        </Text>
+      </Box>
+
+      <Box flexDirection="column">
+        {visiblePrompts.map((prompt, index) => {
+          const isSelected = index === selectedIndex;
+          const isCurrent = prompt.id === currentPromptId;
+
+          return (
+            <Box key={prompt.id} flexDirection="row" gap={1}>
+              <Text
+                color={isSelected ? colors.selector.itemHighlighted : undefined}
+              >
+                {isSelected ? "›" : " "}
+              </Text>
+              <Box flexDirection="row">
+                <Text
+                  bold={isSelected}
+                  color={
+                    isSelected ? colors.selector.itemHighlighted : undefined
+                  }
+                >
+                  {prompt.label}
+                  {isCurrent && (
+                    <Text color={colors.selector.itemCurrent}> (current)</Text>
+                  )}
+                </Text>
+                <Text dimColor> {prompt.description}</Text>
+              </Box>
+            </Box>
+          );
+        })}
+        {hasShowAllOption && (
+          <Box flexDirection="row" gap={1}>
+            <Text
+              color={
+                selectedIndex === visiblePrompts.length
+                  ? colors.selector.itemHighlighted
+                  : undefined
+              }
+            >
+              {selectedIndex === visiblePrompts.length ? "›" : " "}
+            </Text>
+            <Text dimColor>Show all prompts</Text>
+          </Box>
+        )}
+      </Box>
+    </Box>
+  );
+}