nocobase/packages/plugins/saml/src/client/SAMLList.tsx

import { LoginOutlined } from '@ant-design/icons';
import { css } from '@emotion/css';
import { useAPIClient, useRedirect, useRequest } from '@nocobase/client';
import { Button, Space } from 'antd';
import React, { useEffect, useState } from 'react';

export interface SAMLProvider {
  title: string;
  clientId: string;
  loginUrl: string;
}

export const SAMLList = () => {
  const [windowHandler, setWindowHandler] = useState<Window | undefined>();
  const api = useAPIClient();
  const redirect = useRedirect();

  const { data, loading } = useRequest({
    resource: 'saml',
    action: 'getEnabledProviders',
  });

  /**
   * 打开登录弹出框
   */
  const handleOpen = async (item: SAMLProvider) => {
    const response = await api.request({
      method: 'post',
      url: 'saml:getAuthUrl',
      data: {
        clientId: item.clientId,
      },
    });

    const authUrl = response?.data?.data;
    const { width, height } = screen;

    const win = window.open(
      authUrl,
      '_blank',
      `width=800,height=600,left=${(width - 800) / 2},top=${
        (height - 600) / 2
      },toolbar=no,menubar=no,location=no,status=no`,
    );

    setWindowHandler(win);
  };

  /**
   * 从弹出窗口，发消息回来进行登录
   */
  const handleOIDCLogin = async (event: MessageEvent) => {
    await api.auth.signIn(event.data, 'saml');
    windowHandler.close();
    setWindowHandler(undefined);
    redirect();
  };

  /**
   * 监听弹出窗口的消息
   */
  useEffect(() => {
    if (!windowHandler) return;
    window.addEventListener('message', handleOIDCLogin);
    return () => {
      window.removeEventListener('message', handleOIDCLogin);
    };
  }, [windowHandler]);

  return (
    <Space
      direction="vertical"
      className={css`
        display: flex;
      `}
    >
      {data?.data?.map?.((item) => (
        <Button shape="round" block key={item.clientId} icon={<LoginOutlined />} onClick={() => handleOpen(item)}>
          {item.title}
        </Button>
      ))}
    </Space>
  );
};