nocobase/packages/core/client/src/schema-initializer/SchemaInitializerProvider.tsx

import { useFieldSchema } from '@formily/react';
import { isPlainObj } from '@formily/shared';
import get from 'lodash/get';
import React, { createContext, useContext } from 'react';
import { SchemaComponentOptions } from '../schema-component';
import { SchemaInitializer } from './SchemaInitializer';
import * as globals from './buttons';
import * as initializerComponents from './components';
import * as items from './items';

export const SchemaInitializerContext = createContext<any>({});

export interface SchemaInitializerProviderProps {
  components?: any;
  initializers?: Record<string, any>;
}

export const useSchemaInitializer = (name: string, props = {}) => {
  const fieldSchema = useFieldSchema();
  const initializers = useContext(SchemaInitializerContext);
  const render = (component?: any, props?: any) => {
    return component && React.createElement(component, props);
  };

  if (!name) {
    return { exists: false, render: (props?: any) => render(null) };
  }

  const initializer = get(initializers, name || fieldSchema?.['x-initializer']);
  const initializerProps = { ...props, ...fieldSchema?.['x-initializer-props'] };

  if (!initializer) {
    return { exists: false, render: (props?: any) => render(null) };
  }

  if (isPlainObj(initializer)) {
    return {
      exists: true,
      render: (props?: any) => {
        const component = (initializer as any).component || SchemaInitializer.Button;
        return render(component, { ...initializer, ...initializerProps, ...props });
      },
    };
  }

  return {
    exists: true,
    render: (props?: any) => render(initializer, { ...initializerProps, ...props }),
  };
};

export const SchemaInitializerPluginContext = createContext(null);

export const SchemaInitializerProvider: React.FC<SchemaInitializerProviderProps> = (props) => {
  const { initializers, components, children } = props;
  const parentInitializers = useContext(SchemaInitializerContext);

  return (
    <SchemaInitializerContext.Provider value={{ ...globals, ...parentInitializers, ...initializers }}>
      <SchemaComponentOptions components={{ ...items, ...components, ...initializerComponents }}>
        {children}
      </SchemaComponentOptions>
    </SchemaInitializerContext.Provider>
  );
};