upstream-plane/apps/admin/providers/toast.tsx

/**
 * Copyright (c) 2023-present Plane Software, Inc. and contributors
 * SPDX-License-Identifier: AGPL-3.0-only
 * See the LICENSE file for details.
 */

import { createToastManager, ToastProvider } from "@makeplane/propel/components/toast";
import type { ToastData, ToastVariant } from "@makeplane/propel/components/toast";
import { IconButton } from "@makeplane/propel/components/icon-button";
import { Icon } from "@makeplane/propel/components/icon";
import { CloseOutline } from "@makeplane/propel/icons";

// Keep the call-site API the workspace @plane/propel toast exposed, backed by
// @makeplane/propel's toast manager, so consumers only change their import path.
export enum TOAST_TYPE {
  SUCCESS = "success",
  ERROR = "error",
  INFO = "info",
  WARNING = "warning",
}

const TYPE_TO_VARIANT: Record<TOAST_TYPE, ToastVariant> = {
  [TOAST_TYPE.SUCCESS]: "success",
  [TOAST_TYPE.ERROR]: "danger",
  [TOAST_TYPE.INFO]: "info",
  [TOAST_TYPE.WARNING]: "warning",
};

const toastManager = createToastManager();

type SetToastProps = {
  type: TOAST_TYPE;
  title: string;
  message?: string;
};

export const setToast = (props: SetToastProps) =>
  toastManager.add({
    title: props.title,
    description: props.message,
    data: { variant: TYPE_TO_VARIANT[props.type] },
  });

type PromiseToastData<TValue> = {
  title: string;
  message?: (data: TValue) => string;
};

type PromiseToastOptions<TValue> = {
  loading?: string;
  success: PromiseToastData<TValue>;
  error: PromiseToastData<TValue>;
};

export const setPromiseToast = <TValue,>(promise: Promise<TValue>, options: PromiseToastOptions<TValue>): void => {
  void toastManager.promise<TValue, ToastData>(promise, {
    loading: {
      title: options.loading ?? "Loading...",
      data: { variant: "neutral" },
    },
    success: (data) => ({
      title: options.success.title,
      description: options.success.message?.(data),
      data: { variant: "success" },
    }),
    error: (data) => ({
      title: options.error.title,
      description: options.error.message?.(data),
      data: { variant: "danger" },
    }),
  });
};

export function ToastWithTheme() {
  return (
    <ToastProvider
      toastManager={toastManager}
      close={
        <IconButton variant="ghost" size="sm" aria-label="Close notification" icon={<Icon icon={CloseOutline} />} />
      }
    />
  );
}