476 lines · 18.1 KB · TypeScript
RawDownload
"use client"; import { CheckIcon } from "lucide-react";import { Radio as RadioPrimitive } from "@base-ui/react/radio";import { useCallback, useMemo, useState } from "react";import {  ProviderInstanceId,  ProviderDriverKind,  type ProviderInstanceConfig,} from "@t3tools/contracts"; import { useSettings, useUpdateSettings } from "../../hooks/useSettings";import { cn } from "../../lib/utils";import { normalizeProviderAccentColor } from "../../providerInstances";import { Button } from "../ui/button";import { ACPRegistryIcon, Gemini, GithubCopilotIcon, PiAgentIcon, type Icon } from "../Icons";import {  Dialog,  DialogDescription,  DialogFooter,  DialogHeader,  DialogPopup,  DialogTitle,} from "../ui/dialog";import { Badge } from "../ui/badge";import { Input } from "../ui/input";import { RadioGroup } from "../ui/radio-group";import { toastManager } from "../ui/toast";import { DRIVER_OPTION_BY_VALUE, DRIVER_OPTIONS } from "./providerDriverMeta";import { ProviderSettingsForm, deriveProviderSettingsFields } from "./ProviderSettingsForm";import { AnimatedHeight } from "../AnimatedHeight"; const PROVIDER_ACCENT_SWATCHES = [  "#2563eb",  "#16a34a",  "#ea580c",  "#dc2626",  "#7c3aed",  "#0891b2",] as const; /** * Normalize a user-provided label into a slug suffix for the instance id. * The full id is formed by prefixing the driver slug — e.g. label "Work" on * driver "codex" becomes `codex_work`. Output is trimmed to 48 chars so the * final composed id stays under the 64-char slug cap enforced by * `ProviderInstanceId` in `@t3tools/contracts`. */function slugifyLabel(value: string): string {  return value    .trim()    .toLowerCase()    .replace(/[^a-z0-9]+/g, "_")    .replace(/^_+|_+$/g, "")    .slice(0, 48);} function deriveInstanceId(driver: ProviderDriverKind, label: string): string {  const slug = slugifyLabel(label);  return slug ? `${driver}_${slug}` : "";} const INSTANCE_ID_PATTERN = /^[a-zA-Z][a-zA-Z0-9_-]*$/;const DEFAULT_DRIVER_KIND = ProviderDriverKind.make("codex");const DEFAULT_DRIVER_OPTION = DRIVER_OPTIONS[0]!;const EMPTY_CONFIG_DRAFT: Record<string, unknown> = {};interface ComingSoonDriverOption {  readonly value: ProviderDriverKind;  readonly label: string;  readonly icon: Icon;} const COMING_SOON_DRIVER_OPTIONS: readonly ComingSoonDriverOption[] = [  {    value: ProviderDriverKind.make("githubCopilot"),    label: "Github Copilot",    icon: GithubCopilotIcon,  },  {    value: ProviderDriverKind.make("gemini"),    label: "Gemini",    icon: Gemini,  },  {    value: ProviderDriverKind.make("acpRegistry"),    label: "ACP Registry",    icon: ACPRegistryIcon,  },  {    value: ProviderDriverKind.make("piAgent"),    label: "Pi Agent",    icon: PiAgentIcon,  },]; /** * Validate an instance id against the same slug rules the server applies in * `ProviderInstanceId` (see `packages/contracts/src/providerInstance.ts`). * Returns a user-facing error string, or `null` if valid. */function validateInstanceId(id: string, existing: ReadonlySet<string>): string | null {  if (id.length === 0) return "Instance ID is required.";  if (id.length > 64) return "Instance ID must be 64 characters or fewer.";  if (!INSTANCE_ID_PATTERN.test(id)) {    return "Instance ID must start with a letter and use only letters, digits, '-', or '_'.";  }  if (existing.has(id)) return `An instance named '${id}' already exists.`;  return null;} interface AddProviderInstanceDialogProps {  open: boolean;  onOpenChange: (open: boolean) => void;} export function AddProviderInstanceDialog({ open, onOpenChange }: AddProviderInstanceDialogProps) {  return (    <Dialog open={open} onOpenChange={onOpenChange}>      {open ? <AddProviderInstanceDialogContent onOpenChange={onOpenChange} /> : null}    </Dialog>  );} function AddProviderInstanceDialogContent({  onOpenChange,}: {  readonly onOpenChange: (open: boolean) => void;}) {  const settings = useSettings();  const { updateSettings } = useUpdateSettings();   const [wizardStep, setWizardStep] = useState(0);  const [driver, setDriver] = useState<ProviderDriverKind>(DEFAULT_DRIVER_KIND);  const [label, setLabel] = useState("");  const [accentColor, setAccentColor] = useState<string>("");  const [manualInstanceId, setManualInstanceId] = useState<string | null>(null);  // Driver-specific config drafts keyed by driver so toggling between drivers  // during the same dialog session does not lose in-progress input.  const [configByDriver, setConfigByDriver] = useState<Record<string, Record<string, unknown>>>({});  // Errors are suppressed until the user has tried to submit once. After that  // they update live so fixing the problem clears the message in place.  const [hasAttemptedSubmit, setHasAttemptedSubmit] = useState(false);   const existingIds = useMemo(    () => new Set(Object.keys(settings.providerInstances ?? {})),    [settings.providerInstances],  );   const driverOption = DRIVER_OPTION_BY_VALUE[driver] ?? DEFAULT_DRIVER_OPTION;  const driverSettingsFields = useMemo(    () => deriveProviderSettingsFields(driverOption),    [driverOption],  );  const instanceId = manualInstanceId ?? deriveInstanceId(driver, label);  const instanceIdError = validateInstanceId(instanceId, existingIds);  const showInstanceIdError = hasAttemptedSubmit && instanceIdError !== null;  const previewLabel = label.trim() || `${driverOption.label} Workspace`;  const wizardSteps = ["Driver", "Identity", "Config"] as const;  const wizardStepSummaries = [driverOption.label, previewLabel, null] as const;   const configDraft = configByDriver[driver] ?? EMPTY_CONFIG_DRAFT;  const setConfigDraft = useCallback(    (config: Record<string, unknown> | undefined) => {      setConfigByDriver((existing) => {        const next = { ...existing };        if (config === undefined || Object.keys(config).length === 0) {          delete next[driver];        } else {          next[driver] = config;        }        return next;      });    },    [driver],  );   const handleSave = useCallback(() => {    setHasAttemptedSubmit(true);    if (instanceIdError !== null) return;     const config = configByDriver[driver] ?? {};    const hasConfig = Object.keys(config).length > 0;    const normalizedAccentColor = normalizeProviderAccentColor(accentColor);     const nextInstance: ProviderInstanceConfig = {      driver,      enabled: true,      ...(label.trim().length > 0 ? { displayName: label.trim() } : {}),      ...(normalizedAccentColor ? { accentColor: normalizedAccentColor } : {}),      ...(hasConfig ? { config } : {}),    };    // `ProviderInstanceId.make` revalidates the slug; we've already checked    // it via `validateInstanceId`, but going through the brand constructor    // keeps the type boundary honest and guards against any future drift in    // the slug rules.    const brandedId = ProviderInstanceId.make(instanceId);    const nextMap = {      ...settings.providerInstances,      [brandedId]: nextInstance,    };    try {      updateSettings({ providerInstances: nextMap });      toastManager.add({        type: "success",        title: "Provider instance added",        description: `${driverOption.label} instance '${instanceId}' was added.`,      });      onOpenChange(false);    } catch (error) {      toastManager.add({        type: "error",        title: "Could not add provider instance",        description: error instanceof Error ? error.message : "Update failed.",      });    }  }, [    driver,    driverOption,    configByDriver,    instanceId,    instanceIdError,    label,    accentColor,    onOpenChange,    settings.providerInstances,    updateSettings,  ]);   return (    <DialogPopup className="max-w-xl overflow-hidden">      <div className="flex min-h-0 flex-col overflow-hidden border-foreground/10 bg-background shadow-2xl">        <DialogHeader className="border-b border-border/70 bg-background">          <DialogTitle>Add provider instance</DialogTitle>          <DialogDescription>            Configure an additional provider instance — for example, a second Codex install pointed            at a different workspace.          </DialogDescription>          <div className="grid grid-cols-3 gap-2">            {wizardSteps.map((step, index) => (              <button                key={step}                type="button"                className={cn(                  "grid min-w-0 grid-cols-[1rem_minmax(0,1fr)] gap-x-2 rounded-lg border px-3 py-2 text-left",                  index === wizardStep                    ? "border-primary bg-primary/10 ring-1 ring-primary/25"                    : index < wizardStep                      ? "border-border bg-background"                      : "border-border bg-muted/40",                )}                onClick={() => setWizardStep(index)}              >                <span                  className={cn(                    "row-span-2 mt-0.5 grid size-4 place-items-center rounded-full border",                    index < wizardStep                      ? "border-primary bg-primary text-primary-foreground"                      : index === wizardStep                        ? "border-primary bg-background"                        : "border-muted-foreground/35 bg-background",                  )}                  aria-hidden                >                  {index < wizardStep ? <CheckIcon className="size-3" /> : null}                </span>                <span className="text-[10px] font-medium uppercase text-muted-foreground">                  Step {index + 1}                </span>                <span className="truncate text-xs font-semibold text-foreground">                  {step}                  {index < wizardStep && wizardStepSummaries[index]                    ? `: ${wizardStepSummaries[index]}`                    : ""}                </span>              </button>            ))}          </div>        </DialogHeader>         <div          data-slot="dialog-panel"          className="space-y-4 border-b border-border/70 bg-muted/20 px-6 py-5"        >          <AnimatedHeight>            <div className={cn("grid gap-2", wizardStep !== 0 && "hidden")}>              <span id="add-instance-driver-label" className="text-xs font-medium text-foreground">                Driver              </span>              <RadioGroup                value={driver}                onValueChange={(value) => setDriver(ProviderDriverKind.make(value))}                aria-labelledby="add-instance-driver-label"                className="grid grid-cols-2 gap-2.5"              >                {DRIVER_OPTIONS.map((option) => {                  const IconComponent = option.icon;                  const isSelected = option.value === driver;                  return (                    <RadioPrimitive.Root                      key={option.value}                      value={option.value}                      className={cn(                        "relative flex cursor-pointer items-center gap-3 rounded-lg border px-3 py-3 text-left outline-none transition-[background-color,border-color,box-shadow]",                        "focus-visible:ring-2 focus-visible:ring-ring focus-visible:ring-offset-1 focus-visible:ring-offset-background",                        isSelected                          ? "border-primary bg-background shadow-sm ring-2 ring-primary/35"                          : "border-border bg-background hover:border-foreground/20 hover:bg-muted/50",                      )}                    >                      <IconComponent className="size-5 shrink-0" aria-hidden />                      <span className="min-w-0 flex-1 truncate text-sm font-medium text-foreground">                        {option.label}                      </span>                      {option.badgeLabel ? (                        <Badge variant="warning" size="sm">                          {option.badgeLabel}                        </Badge>                      ) : null}                    </RadioPrimitive.Root>                  );                })}                {COMING_SOON_DRIVER_OPTIONS.map((option) => {                  const IconComponent = option.icon;                  return (                    <RadioPrimitive.Root                      key={option.value}                      value={option.value}                      disabled                      className={cn(                        "relative flex cursor-not-allowed items-center gap-3 rounded-lg border border-border bg-background px-3 py-3 text-left opacity-55 outline-none",                      )}                    >                      <IconComponent                        className="size-5 shrink-0 text-muted-foreground"                        aria-hidden                      />                      <span className="min-w-0 flex-1 truncate text-sm font-medium text-foreground">                        {option.label}                      </span>                      <Badge variant="warning" size="sm">                        Coming Soon                      </Badge>                    </RadioPrimitive.Root>                  );                })}              </RadioGroup>            </div>             <label className={cn("grid gap-2", wizardStep !== 1 && "hidden")}>              <span className="text-xs font-medium text-foreground">Label</span>              <Input                className="bg-background"                placeholder="e.g. Work"                value={label}                onChange={(event) => setLabel(event.target.value)}              />              <span className="text-[11px] text-muted-foreground">                Shown in the provider list. Optional.              </span>            </label>             <label className={cn("grid gap-2", wizardStep !== 1 && "hidden")}>              <span className="text-xs font-medium text-foreground">Instance ID</span>              <Input                className="bg-background"                placeholder={`${driver}_work`}                value={instanceId}                onChange={(event) => {                  setManualInstanceId(event.target.value);                }}                aria-invalid={showInstanceIdError}              />              {showInstanceIdError ? (                <span className="text-[11px] text-destructive">{instanceIdError}</span>              ) : (                <span className="text-[11px] text-muted-foreground">                  Routing key used by threads and sessions. Letters, digits, '-', or '_'.                </span>              )}            </label>             <div className={cn("grid gap-2", wizardStep !== 1 && "hidden")}>              <span className="text-xs font-medium text-foreground">Accent color</span>              <div className="flex min-w-0 flex-wrap items-center gap-2">                <input                  type="color"                  value={normalizeProviderAccentColor(accentColor) ?? PROVIDER_ACCENT_SWATCHES[0]}                  onChange={(event) => setAccentColor(event.target.value)}                  aria-label="Provider instance accent color"                  className="h-8 w-10 cursor-pointer rounded-xl border border-input bg-background p-0.5"                />                <div className="flex flex-wrap gap-1.5">                  {PROVIDER_ACCENT_SWATCHES.map((swatch) => {                    const selected = accentColor.toLowerCase() === swatch;                    return (                      <button                        key={swatch}                        type="button"                        className={cn(                          "size-6 cursor-pointer rounded-full border transition",                          selected                            ? "scale-110 border-foreground ring-2 ring-ring ring-offset-1 ring-offset-background"                            : "border-black/10 hover:scale-105 dark:border-white/20",                        )}                        style={{ backgroundColor: swatch }}                        onClick={() => setAccentColor(swatch)}                        aria-label={`Use ${swatch} accent`}                      />                    );                  })}                </div>                {accentColor ? (                  <Button                    type="button"                    size="sm"                    variant="ghost"                    className="h-7 px-2 text-xs text-muted-foreground"                    onClick={() => setAccentColor("")}                  >                    Clear                  </Button>                ) : null}              </div>              <span className="text-[11px] text-muted-foreground">                Optional marker shown in the picker.              </span>            </div>             {driverSettingsFields.length > 0 ? (              <div className={cn("grid gap-4", wizardStep !== 2 && "hidden")}>                <ProviderSettingsForm                  definition={driverOption}                  value={configDraft}                  idPrefix={`add-provider-${driver}`}                  variant="dialog"                  onChange={setConfigDraft}                />              </div>            ) : wizardStep === 2 ? (              <div className="grid gap-2">                <p className="text-sm text-muted-foreground">                  This driver has no required configuration. You can add the instance now.                </p>              </div>            ) : null}          </AnimatedHeight>        </div>         <DialogFooter className="border-t bg-background">          <Button            variant="outline"            size="sm"            onClick={() => {              if (wizardStep === 0) {                onOpenChange(false);                return;              }              setWizardStep((step) => Math.max(0, step - 1));            }}          >            {wizardStep === 0 ? "Cancel" : "Back"}          </Button>          {wizardStep < wizardSteps.length - 1 ? (            <Button size="sm" onClick={() => setWizardStep((step) => Math.min(2, step + 1))}>              Next            </Button>          ) : (            <Button size="sm" onClick={handleSave}>              Add instance            </Button>          )}        </DialogFooter>      </div>    </DialogPopup>  );}