import { useCallback, useMemo, useState, type ChangeEvent } from "react";import { useAtomSet } from "@effect/atom-react";import { Link } from "@tanstack/react-router";import * as Exit from "effect/Exit";import { CheckIcon, CircleIcon, TriangleAlert } from "lucide-react"; import { integrationWriteKeys } from "@executor-js/react/api/reactivity-keys";import {  slugifyNamespace,  useIntegrationIdentity,} from "@executor-js/react/plugins/integration-identity";import { Button } from "@executor-js/react/components/button";import { FieldLabel } from "@executor-js/react/components/field";import { FloatActions } from "@executor-js/react/components/float-actions";import { Input } from "@executor-js/react/components/input";import {  errorMessageFromExit,  FormErrorAlert,  SlugCollisionAlert,  useSlugAlreadyExists,} from "@executor-js/react/lib/integration-add";import { OpenApiSourceDetailsFields } from "@executor-js/plugin-openapi/react"; import { addMicrosoftWorkloads } from "./atoms";import { MicrosoftScopePicker } from "./MicrosoftScopePicker";import {  MICROSOFT_GRAPH_BASE_URL,  MICROSOFT_GRAPH_DEFAULT_PRESET_IDS,  microsoftGraphPresetForId,  microsoftServiceSlug,} from "../sdk/presets";import type { MicrosoftAddWorkloadsInput, MicrosoftAddWorkloadsResult } from "../sdk/plugin";import { MICROSOFT_CUSTOM_WORKLOAD_ID } from "../sdk/plugin"; const MICROSOFT_FAVICON = "https://www.microsoft.com/favicon.ico"; const defaultPresetIds: ReadonlySet<string> = new Set(MICROSOFT_GRAPH_DEFAULT_PRESET_IDS); export type MicrosoftWorkloadIdentityOverride = {  readonly slug: string;  readonly name: string;}; export type MicrosoftCustomWorkloadInput = {  readonly customScopes: readonly string[];  readonly slug: string;  readonly name: string;  readonly description?: string;}; export type AddMicrosoftWorkloadsMutation = (input: {  readonly payload: MicrosoftAddWorkloadsInput;  readonly reactivityKeys: typeof integrationWriteKeys;}) => Promise<Exit.Exit<MicrosoftAddWorkloadsResult, unknown>>; export const microsoftAddWorkloadsPayload = (input: {  readonly presetIds: readonly string[];  readonly identityOverride?: MicrosoftWorkloadIdentityOverride;  readonly custom?: MicrosoftCustomWorkloadInput;  readonly baseUrl?: string;}): MicrosoftAddWorkloadsInput => {  const identityOverride =    input.presetIds.length === 1 && !input.custom ? input.identityOverride : undefined;  const presetWorkloads = input.presetIds.map((presetId: string) => ({    presetId,    ...(identityOverride?.slug.trim() ? { slug: identityOverride.slug.trim() } : {}),    ...(identityOverride?.name.trim() ? { name: identityOverride.name.trim() } : {}),  }));  const custom =    input.custom && input.custom.customScopes.length > 0      ? [          {            custom: {              customScopes: [...input.custom.customScopes],              slug: input.custom.slug,              name: input.custom.name,              ...(input.custom.description?.trim()                ? { description: input.custom.description.trim() }                : {}),            },          },        ]      : [];  const workloads = [...presetWorkloads, ...custom];  const baseUrl = input.baseUrl?.trim() ?? "";  return baseUrl.length > 0 ? { workloads, baseUrl } : { workloads };}; export const submitMicrosoftWorkloadsSelection = (  doAddWorkloads: AddMicrosoftWorkloadsMutation,  input: {    readonly presetIds: readonly string[];    readonly identityOverride?: MicrosoftWorkloadIdentityOverride;    readonly custom?: MicrosoftCustomWorkloadInput;    readonly baseUrl?: string;  },): Promise<Exit.Exit<MicrosoftAddWorkloadsResult, unknown>> =>  doAddWorkloads({    payload: microsoftAddWorkloadsPayload(input),    reactivityKeys: integrationWriteKeys,  }); export type MicrosoftWorkloadResultRow =  | {      readonly status: "added";      readonly presetId: string;      readonly slug: string;      readonly toolCount: number;    }  | {      readonly status: "skipped";      readonly presetId: string;      readonly slug: string;      readonly reason: "already_exists";    }  | {      readonly status: "failed";      readonly presetId: string;      readonly slug: string;      readonly error: string;    }; export const microsoftAddWorkloadsResultRows = (  result: MicrosoftAddWorkloadsResult,): readonly MicrosoftWorkloadResultRow[] => [  ...result.added.map((entry) => ({    status: "added" as const,    presetId: entry.presetId,    slug: String(entry.slug),    toolCount: entry.toolCount,  })),  ...result.skipped.map((entry) => ({    status: "skipped" as const,    presetId: entry.presetId,    slug: String(entry.slug),    reason: entry.reason,  })),  ...result.failed.map((entry) => ({    status: "failed" as const,    presetId: entry.presetId,    slug: String(entry.slug),    error: entry.error,  })),]; export const mergeMicrosoftAddWorkloadsResult = (  previous: MicrosoftAddWorkloadsResult,  next: MicrosoftAddWorkloadsResult,): MicrosoftAddWorkloadsResult => {  const nextPresetIds = new Set(microsoftAddWorkloadsResultRows(next).map((row) => row.presetId));  return {    added: [...previous.added.filter((entry) => !nextPresetIds.has(entry.presetId)), ...next.added],    skipped: [      ...previous.skipped.filter((entry) => !nextPresetIds.has(entry.presetId)),      ...next.skipped,    ],    failed: [      ...previous.failed.filter((entry) => !nextPresetIds.has(entry.presetId)),      ...next.failed,    ],  };}; const microsoftPresetName = (presetId: string): string =>  presetId === MICROSOFT_CUSTOM_WORKLOAD_ID    ? "Custom Graph scopes"    : (microsoftGraphPresetForId(presetId)?.name ?? presetId); export function MicrosoftWorkloadResultPanel(props: {  readonly result: MicrosoftAddWorkloadsResult;  readonly retryingPresetId: string | null;  readonly onRetry: (presetId: string) => void | Promise<void>;}) {  const rows = microsoftAddWorkloadsResultRows(props.result);  if (rows.length === 0) return null;   return (    <section      data-testid="microsoft-add-results"      className="space-y-3 rounded-lg border border-border bg-muted/10 px-3 py-3"    >      <div>        <h2 className="text-sm font-medium text-foreground">Microsoft workloads</h2>        <p className="text-[11px] text-muted-foreground">          Each selected workload is added as its own integration.        </p>      </div>      <ul className="space-y-2">        {rows.map((row: MicrosoftWorkloadResultRow) => {          const presetName = microsoftPresetName(row.presetId);          return (            <li              key={`${row.status}:${row.presetId}`}              data-testid={`add-result-row-${row.presetId}`}              data-state={row.status}              className="flex items-start gap-2 rounded-md border border-border bg-background px-2.5 py-2"            >              <span                className={                  row.status === "added"                    ? "mt-0.5 text-emerald-600"                    : row.status === "skipped"                      ? "mt-0.5 text-muted-foreground"                      : "mt-0.5 text-destructive"                }              >                {row.status === "added" ? (                  <CheckIcon className="size-3.5" />                ) : row.status === "skipped" ? (                  <CircleIcon className="size-3.5" />                ) : (                  <TriangleAlert className="size-3.5" />                )}              </span>              <div className="min-w-0 flex-1 space-y-0.5">                <div className="flex min-w-0 items-center gap-2">                  <span className="truncate text-sm font-medium text-foreground">{presetName}</span>                  <span className="shrink-0 text-[11px] text-muted-foreground">                    {row.status === "added"                      ? "Added"                      : row.status === "skipped"                        ? "Already exists"                        : "Failed"}                  </span>                </div>                {row.status === "added" ? (                  <p className="text-[11px] text-muted-foreground">                    {row.toolCount} tool{row.toolCount === 1 ? "" : "s"} added.                  </p>                ) : row.status === "skipped" ? (                  <p className="text-[11px] text-muted-foreground">                    This integration already exists.                  </p>                ) : (                  <p className="text-[11px] text-destructive">{row.error}</p>                )}              </div>              {row.status === "failed" ? (                <Button                  type="button"                  variant="outline"                  size="xs"                  data-testid={`add-result-retry-${row.presetId}`}                  loading={props.retryingPresetId === row.presetId}                  onClick={() => void props.onRetry(row.presetId)}                >                  Retry                </Button>              ) : (                <Button variant="ghost" size="xs" asChild>                  <Link to="/{-$orgSlug}/integrations/$namespace" params={{ namespace: row.slug }}>                    Open                  </Link>                </Button>              )}            </li>          );        })}      </ul>    </section>  );} const BaseUrlSettings = (props: {  readonly baseUrl: string;  readonly onBaseUrlChange: (value: string) => void;}) => (  <section className="space-y-2 rounded-lg border border-border bg-muted/10 px-3 py-3">    <div className="space-y-1">      <FieldLabel>Microsoft workload settings</FieldLabel>      <p className="text-[11px] text-muted-foreground">        Selected workloads keep their preset names and namespaces.      </p>    </div>    <Input      value={props.baseUrl}      onChange={(event: ChangeEvent<HTMLInputElement>) => props.onBaseUrlChange(event.target.value)}      placeholder={MICROSOFT_GRAPH_BASE_URL}      className="font-mono text-sm"    />  </section>); export default function AddMicrosoftSource(props: {  onComplete: (slug?: string) => void;  onCancel: () => void;  initialNamespace?: string;}) {  const [selectedPresetIds, setSelectedPresetIds] = useState<ReadonlySet<string>>(defaultPresetIds);  const [customScopes, setCustomScopes] = useState<readonly string[]>([]);  const [baseUrl, setBaseUrl] = useState("");  const [descriptionDraft, setDescriptionDraft] = useState<string | null>(null);  const [adding, setAdding] = useState(false);  const [retryingPresetId, setRetryingPresetId] = useState<string | null>(null);  const [addError, setAddError] = useState<string | null>(null);  const [workloadsResult, setWorkloadsResult] = useState<MicrosoftAddWorkloadsResult | null>(null);   const selectedIds = useMemo(() => [...selectedPresetIds], [selectedPresetIds]);  const singleSelectedPreset =    selectedIds.length === 1 ? microsoftGraphPresetForId(selectedIds[0]!) : undefined;  const hasCustomScopes = customScopes.length > 0;   const identity = useIntegrationIdentity({    fallbackName: hasCustomScopes      ? "Custom Microsoft Graph"      : (singleSelectedPreset?.name ?? "Microsoft Graph"),    fallbackNamespace:      props.initialNamespace ??      (hasCustomScopes        ? "microsoft_graph_custom"        : singleSelectedPreset          ? microsoftServiceSlug(singleSelectedPreset.id)          : "microsoft_graph"),  });   const togglePreset = useCallback((presetId: string, checked: boolean) => {    setSelectedPresetIds((current: ReadonlySet<string>) => {      const next = new Set(current);      if (checked) next.add(presetId);      else next.delete(presetId);      return next;    });  }, []);   const addCustomScope = useCallback((scope: string) => {    setCustomScopes((current: readonly string[]) =>      current.includes(scope) ? current : [...current, scope],    );  }, []);   const removeCustomScope = useCallback((scope: string) => {    setCustomScopes((current: readonly string[]) =>      current.filter((entry: string) => entry !== scope),    );  }, []);   const doAddWorkloads = useAtomSet(addMicrosoftWorkloads, { mode: "promiseExit" });   const resolvedSourceId = slugifyNamespace(identity.namespace) || "microsoft_graph_custom";  const resolvedDisplayName =    identity.name.trim() ||    (hasCustomScopes      ? "Custom Microsoft Graph"      : (singleSelectedPreset?.name ?? "Microsoft Graph"));  const resolvedDescription =    descriptionDraft ??    (hasCustomScopes ? "Custom Microsoft Graph scopes." : "Selected Microsoft Graph workloads.");  const customGraphSlugAlreadyExists = useSlugAlreadyExists(    hasCustomScopes ? resolvedSourceId : "",  );  const identityOverride =    selectedIds.length === 1 && !hasCustomScopes      ? { slug: resolvedSourceId, name: resolvedDisplayName }      : undefined;  const customWorkload =    customScopes.length > 0      ? {          customScopes: [...customScopes],          slug: resolvedSourceId,          name: resolvedDisplayName,          description: resolvedDescription,        }      : undefined;  const canAdd =    (selectedIds.length > 0 || customScopes.length > 0) && !customGraphSlugAlreadyExists && !adding;   const handleAdd = async () => {    setAdding(true);    setAddError(null);    setWorkloadsResult(null);    const exit = await submitMicrosoftWorkloadsSelection(doAddWorkloads, {      presetIds: selectedIds,      ...(identityOverride ? { identityOverride } : {}),      ...(customWorkload ? { custom: customWorkload } : {}),      baseUrl,    });    if (Exit.isFailure(exit)) {      setAddError(errorMessageFromExit(exit, "Failed to add Microsoft workloads"));      setAdding(false);      return;    }    setWorkloadsResult(exit.value);    setAdding(false);  };   const handleRetry = async (presetId: string) => {    setRetryingPresetId(presetId);    setAddError(null);    const retryingCustom = presetId === MICROSOFT_CUSTOM_WORKLOAD_ID;    const retryIdentityOverride =      !retryingCustom && identityOverride && selectedIds.length === 1 && selectedIds[0] === presetId        ? identityOverride        : undefined;    const exit = await submitMicrosoftWorkloadsSelection(doAddWorkloads, {      presetIds: retryingCustom ? [] : [presetId],      ...(retryIdentityOverride ? { identityOverride: retryIdentityOverride } : {}),      ...(retryingCustom && customWorkload ? { custom: customWorkload } : {}),      baseUrl,    });    if (Exit.isFailure(exit)) {      setAddError(errorMessageFromExit(exit, "Failed to add Microsoft workload"));      setRetryingPresetId(null);      return;    }    setWorkloadsResult((current) =>      current ? mergeMicrosoftAddWorkloadsResult(current, exit.value) : exit.value,    );    setRetryingPresetId(null);  };   const showIdentityDetails = selectedIds.length === 1 || hasCustomScopes;  const detailTitle = hasCustomScopes    ? "Custom Microsoft Graph scopes"    : (singleSelectedPreset?.name ?? "Microsoft Graph");  const detailSubtitle = hasCustomScopes    ? `${customScopes.length} custom scope${        customScopes.length === 1 ? "" : "s"      } added as its own integration.`    : "This workload is added as its own integration.";   const dismiss = () => {    if (workloadsResult) {      props.onComplete();      return;    }    props.onCancel();  };   return (    <div className="flex flex-1 flex-col gap-6">      <div>        <h1 className="text-xl font-semibold text-foreground">Add Microsoft integration</h1>        <p className="mt-1 text-[13px] text-muted-foreground">          Each selected workload is added as its own integration.        </p>      </div>       <MicrosoftScopePicker        selectedPresetIds={selectedPresetIds}        onToggle={togglePreset}        customScopes={customScopes}        onAddCustomScope={addCustomScope}        onRemoveCustomScope={removeCustomScope}      />       {showIdentityDetails ? (        <OpenApiSourceDetailsFields          title={detailTitle}          subtitle={detailSubtitle}          identity={identity}          {...(hasCustomScopes            ? { description: resolvedDescription, onDescriptionChange: setDescriptionDraft }            : {})}          baseUrl={baseUrl}          onBaseUrlChange={setBaseUrl}          baseUrlLabel="Base URL override (optional)"          baseUrlPlaceholder={MICROSOFT_GRAPH_BASE_URL}          faviconIcon={MICROSOFT_FAVICON}          faviconUrl={baseUrl || MICROSOFT_GRAPH_BASE_URL}        />      ) : (        <BaseUrlSettings baseUrl={baseUrl} onBaseUrlChange={setBaseUrl} />      )}       {customGraphSlugAlreadyExists && !adding && <SlugCollisionAlert slug={resolvedSourceId} />}       {addError && <FormErrorAlert message={addError} />}       {workloadsResult && (        <MicrosoftWorkloadResultPanel          result={workloadsResult}          retryingPresetId={retryingPresetId}          onRetry={handleRetry}        />      )}       <FloatActions>        <Button variant="ghost" onClick={dismiss} disabled={adding || retryingPresetId !== null}>          {workloadsResult ? "Done" : "Cancel"}        </Button>        <Button          data-testid="microsoft-add-submit"          onClick={() => void handleAdd()}          disabled={!canAdd}          loading={adding}        >          Connect Microsoft        </Button>      </FloatActions>    </div>  );}