/**
 * This file is part of the NocoBase (R) project.
 * Copyright (c) 2020-2024 NocoBase Co., Ltd.
 * Authors: NocoBase Team.
 *
 * This project is dual-licensed under AGPL-3.0 and NocoBase Commercial License.
 * For more information, please refer to: https://www.nocobase.com/agreement.
 */

import { App, ConfigProvider, theme } from 'antd';
import React, { createContext, useContext, useEffect } from 'react';
import { FlowEngineContext } from './flowContext';
import { FlowContextProvider } from './FlowContextProvider';
import { FlowEngine } from './flowEngine';
import { useDialog, useDrawer, usePage, usePopover } from './views';
import { FlowViewer } from './views/FlowView';

interface FlowEngineProviderProps {
  engine: FlowEngine;
  children: React.ReactNode;
}

const FlowEngineReactContext = createContext<FlowEngine | null>(null);

export const FlowEngineProvider: React.FC<FlowEngineProviderProps> = React.memo((props) => {
  const { engine, children } = props;
  if (!engine) {
    throw new Error('FlowEngineProvider must be supplied with an engine.');
  }
  return (
    <FlowEngineReactContext.Provider value={engine}>
      <FlowContextProvider context={engine.context}>{children}</FlowContextProvider>
    </FlowEngineReactContext.Provider>
  );
});

FlowEngineProvider.displayName = 'FlowEngineProvider';

export const FlowEngineGlobalsContextProvider: React.FC<{ children: React.ReactNode }> = ({ children }) => {
  const { modal, message, notification } = App.useApp();
  const [drawer, contextHolder] = useDrawer();
  const [embed, pageContextHolder] = usePage();
  const [popover, popoverContextHolder] = usePopover();
  const [dialog, dialogContextHolder] = useDialog();
  const engine = useFlowEngine();
  const config = useContext(ConfigProvider.ConfigContext);
  const { token } = theme.useToken();
  const isDarkTheme = React.useMemo(() => {
    const algorithm = config?.theme?.algorithm;
    if (Array.isArray(algorithm)) {
      return algorithm.includes(theme.darkAlgorithm);
    }
    return algorithm === theme.darkAlgorithm;
  }, [config]);

  // 这些全局能力需要在 children 首次渲染前就可读，不能等到 effect 后再挂到上下文。
  engine.context.defineProperty('viewer', {
    cache: false,
    get: (ctx) => new FlowViewer(ctx, { drawer, embed, popover, dialog }),
  });
  for (const item of Object.entries({
    antdConfig: config,
    modal,
    message,
    notification,
  })) {
    const [key, value] = item;
    if (value) {
      engine.context.defineProperty(key, { value });
    }
  }
  // 将 themeToken 定义为 observable, 使组件能够响应主题的变更。
  engine.context.defineProperty('themeToken', {
    get: () => token,
    observable: true,
    cache: true,
  });
  // 统一把暗色模式暴露到 Flow 上下文，避免 flow 侧继续依赖 global-theme。
  engine.context.defineProperty('isDarkTheme', {
    get: () => isDarkTheme,
    observable: true,
    cache: true,
    info: {
      description: 'Whether current theme algorithm is dark mode.',
      detail: 'boolean',
    },
  });

  useEffect(() => {
    engine.reactView.refresh();
  }, [engine, drawer, modal, message, notification, config, popover, token, dialog, embed, isDarkTheme]);

  return (
    <ConfigProvider {...config} locale={engine.context.locales?.antd} popupMatchSelectWidth={false}>
      {children}
      {contextHolder as any}
      {popoverContextHolder as any}
      {pageContextHolder as any}
      {dialogContextHolder as any}
      {/* The modal context is provided by App.useApp() */}
    </ConfigProvider>
  );
};
// 不 throw Error 怎么处理？
export const useFlowEngine = ({ throwError = true } = {}): FlowEngine => {
  const context = useContext(FlowEngineReactContext);
  if (!context && throwError) {
    // This error should ideally not be hit if FlowEngineProvider is used correctly at the root
    // and always supplied with an engine.
    console.warn(
      'useFlowEngine must be used within a FlowEngineProvider, and FlowEngineProvider must be supplied with an engine.',
    );
    return;
  }
  return context;
};

export const useFlowEngineContext = (): FlowEngineContext => {
  const engine = useFlowEngine();
  return engine.context as FlowEngineContext;
};