probo/apps/console/src/components/obligations/LinkedObligationsDialog.tsx

// Copyright (c) 2025-2026 Probo Inc <hello@probo.com>.
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
// SOFTWARE.

import {
  getObligationStatusLabel,
  getObligationStatusVariant,
} from "@probo/helpers";
import {
  Badge,
  Button,
  Dialog,
  DialogContent,
  DialogFooter,
  IconMagnifyingGlass,
  IconPlusLarge,
  IconTrashCan,
  InfiniteScrollTrigger,
  Input,
  Spinner,
} from "@probo/ui";
import { type ReactNode, Suspense, useMemo, useState } from "react";
import { useTranslation } from "react-i18next";
import { useLazyLoadQuery, usePaginationFragment } from "react-relay";
import { graphql } from "relay-runtime";

import type {
  LinkedObligationsDialogFragment$data,
  LinkedObligationsDialogFragment$key,
} from "#/__generated__/core/LinkedObligationsDialogFragment.graphql";
import type { LinkedObligationsDialogQuery } from "#/__generated__/core/LinkedObligationsDialogQuery.graphql";
import type { LinkedObligationsDialogQuery_fragment } from "#/__generated__/core/LinkedObligationsDialogQuery_fragment.graphql";
import { useOrganizationId } from "#/hooks/useOrganizationId";
import type { NodeOf } from "#/types";

const obligationsQuery = graphql`
  query LinkedObligationsDialogQuery($organizationId: ID!) {
    organization: node(id: $organizationId) {
      id
      ... on Organization {
        ...LinkedObligationsDialogFragment
      }
    }
  }
`;

const obligationsFragment = graphql`
  fragment LinkedObligationsDialogFragment on Organization
  @refetchable(queryName: "LinkedObligationsDialogQuery_fragment")
  @argumentDefinitions(
    first: { type: "Int", defaultValue: 20 }
    order: { type: "ObligationOrder", defaultValue: null }
    after: { type: "CursorKey", defaultValue: null }
    before: { type: "CursorKey", defaultValue: null }
    last: { type: "Int", defaultValue: null }
  ) {
    obligations(
      first: $first
      after: $after
      last: $last
      before: $before
      orderBy: $order
    ) @connection(key: "LinkedObligationsDialogQuery_obligations") {
      edges {
        node {
          id
          area
          source
          status
          owner {
            fullName
          }
        }
      }
    }
  }
`;

type Props = {
  children: ReactNode;
  connectionId: string;
  disabled?: boolean;
  linkedObligations?: { id: string }[];
  onLink: (obligationId: string) => void;
  onUnlink: (obligationId: string) => void;
};

export function LinkedObligationDialog({ children, ...props }: Props) {
  const { t } = useTranslation();

  return (
    <Dialog trigger={children} title={t("linkedObligationsDialog.title")}>
      <DialogContent>
        <Suspense fallback={<Spinner centered />}>
          <LinkedObligationsDialogContent {...props} />
        </Suspense>
      </DialogContent>
      <DialogFooter exitLabel={t("linkedObligationsDialog.actions.close")} />
    </Dialog>
  );
}

function LinkedObligationsDialogContent(props: Omit<Props, "children">) {
  const organizationId = useOrganizationId();
  const query = useLazyLoadQuery<LinkedObligationsDialogQuery>(
    obligationsQuery,
    {
      organizationId,
    },
    { fetchPolicy: "network-only" },
  );
  const { data, loadNext, hasNext, isLoadingNext } = usePaginationFragment<
    LinkedObligationsDialogQuery_fragment,
    LinkedObligationsDialogFragment$key
  >(
    obligationsFragment,
    query.organization as LinkedObligationsDialogFragment$key,
  );
  const { t } = useTranslation();
  const [search, setSearch] = useState("");
  const obligations = useMemo(
    () => data.obligations?.edges?.map(edge => edge.node) ?? [],
    [data.obligations],
  );
  const linkedIds = useMemo(() => {
    return new Set(props.linkedObligations?.map(o => o.id) ?? []);
  }, [props.linkedObligations]);

  const filteredObligations = useMemo(() => {
    return obligations.filter(
      obligation =>
        obligation.area?.toLowerCase().includes(search.toLowerCase())
        || obligation.source?.toLowerCase().includes(search.toLowerCase())
        || obligation.owner?.fullName
          ?.toLowerCase()
          .includes(search.toLowerCase()),
    );
  }, [obligations, search]);

  return (
    <>
      <div className="flex items-center gap-2 sticky top-0 relative py-4 bg-linear-to-b from-50% from-level-2 to-level-2/0 px-6">
        <Input
          icon={IconMagnifyingGlass}
          placeholder={t("linkedObligationsDialog.searchPlaceholder")}
          onValueChange={setSearch}
        />
      </div>
      <div className="divide-y divide-border-low">
        {filteredObligations.map(obligation => (
          <ObligationRow
            key={obligation.id}
            obligation={obligation}
            linkedObligations={linkedIds}
            onLink={props.onLink}
            onUnlink={props.onUnlink}
            disabled={props.disabled}
          />
        ))}
        {hasNext && (
          <InfiniteScrollTrigger
            loading={isLoadingNext}
            onView={() => loadNext(20)}
          />
        )}
      </div>
    </>
  );
}

type Obligation = NodeOf<LinkedObligationsDialogFragment$data["obligations"]>;

function ObligationRow(props: {
  obligation: Obligation;
  linkedObligations: Set<string>;
  onLink: (obligationId: string) => void;
  onUnlink: (obligationId: string) => void;
  disabled?: boolean;
}) {
  const { t } = useTranslation();
  const isLinked = props.linkedObligations.has(props.obligation.id);

  const onToggle = () => {
    if (isLinked) {
      props.onUnlink(props.obligation.id);
    } else {
      props.onLink(props.obligation.id);
    }
  };

  return (
    <div className="flex items-center justify-between p-4 hover:bg-level-1">
      <div className="flex-1 min-w-0">
        <div className="flex items-center gap-3">
          <div className="flex-1 min-w-0">
            <div className="text-sm font-medium text-txt-primary truncate">
              {props.obligation.area || t("linkedObligationsDialog.noArea")}
              {props.obligation.source || t("linkedObligationsDialog.noSource")}
            </div>
            <div className="text-xs text-txt-secondary">
              {props.obligation.owner?.fullName
                || t("linkedObligationsDialog.unassigned")}
            </div>
          </div>
          <Badge variant={getObligationStatusVariant(props.obligation.status)}>
            {getObligationStatusLabel(props.obligation.status)}
          </Badge>
        </div>
      </div>
      <Button
        variant={isLinked ? "secondary" : "primary"}
        icon={isLinked ? IconTrashCan : IconPlusLarge}
        onClick={onToggle}
        disabled={props.disabled}
        className="ml-6"
      >
        {isLinked
          ? t("linkedObligationsDialog.actions.unlink")
          : t("linkedObligationsDialog.actions.link")}
      </Button>
    </div>
  );
}