tachybase_todo/packages/client/src/schema-initializer/Initializers/utils.ts

import { ISchema, Schema, useFieldSchema } from '@formily/react';
import { uid } from '@formily/shared';
import { SchemaInitializerItemOptions } from '../';
import { useCollection } from '../../collection-manager';
import { useDesignable } from '../../schema-component';

export const gridRowColWrap = (schema: ISchema) => {
  return {
    type: 'void',
    'x-component': 'Grid.Row',
    properties: {
      [uid()]: {
        type: 'void',
        'x-component': 'Grid.Col',
        properties: {
          [schema.name || uid()]: schema,
        },
      },
    },
  };
};

export const removeTableColumn = (schema, cb) => {
  cb(schema, {
    removeParentsIfNoChildren: true,
    // breakRemoveOn: (s) => {
    //   return s['x-component'].startsWith('Table.');
    // },
  });
};

export const removeGridFormItem = (schema, cb) => {
  cb(schema, {
    removeParentsIfNoChildren: true,
    breakRemoveOn: {
      'x-component': 'Grid',
    },
  });
};

const findTableColumn = (schema: Schema, key: string, action: string, deepth: number = 0) => {
  return schema.reduceProperties((buf, s) => {
    if (s[key] === action) {
      return s;
    }
    const c = s.reduceProperties((buf, s) => {
      if (s[key] === action) {
        return s;
      }
      return buf;
    });
    if (c) {
      return c;
    }
    return buf;
  });
};

export const useTableColumnInitializerFields = () => {
  const { name, fields } = useCollection();
  return (
    fields
      // .filter((field) => field?.uiSchema?.title)
      .map((field) => {
        return {
          type: 'item',
          title: field?.uiSchema?.title || field.name,
          schema: {
            name: field.name,
            'x-collection-field': `${name}.${field.name}`,
            'x-component': 'CollectionField',
          },
          component: 'CollectionFieldInitializer',
          find: findTableColumn,
          remove: removeTableColumn,
        } as SchemaInitializerItemOptions;
      })
  );
};

export const useFormItemInitializerFields = () => {
  const { name, fields } = useCollection();
  return fields?.map((field) => {
    if (field.interface === 'subTable') {
      return {
        type: 'item',
        title: field?.uiSchema?.title || field.name,
        component: 'SubTableFieldInitializer',
        remove: removeGridFormItem,
        field,
        schema: {
          type: 'void',
          name: field.name,
          'x-designer': 'FormItem.Designer',
          'x-component': 'div',
          'x-decorator': 'FormItem',
          'x-collection-field': `${name}.${field.name}`,
        },
      } as SchemaInitializerItemOptions;
    }
    return {
      type: 'item',
      title: field?.uiSchema?.title || field.name,
      component: 'CollectionFieldInitializer',
      remove: removeGridFormItem,
      schema: {
        name: field.name,
        'x-designer': 'FormItem.Designer',
        'x-component': 'CollectionField',
        'x-decorator': 'FormItem',
        'x-collection-field': `${name}.${field.name}`,
      },
    } as SchemaInitializerItemOptions;
  });
};

const findSchema = (schema: Schema, key: string, action: string) => {
  return schema.reduceProperties((buf, s) => {
    if (s[key] === action) {
      return s;
    }
    const c = findSchema(s, key, action);
    if (c) {
      return c;
    }
    return buf;
  });
};

const removeSchema = (schema, cb) => {
  return cb(schema);
};

export const useCurrentSchema = (action: string, key: string, find = findSchema, rm = removeSchema) => {
  const fieldSchema = useFieldSchema();
  const { remove } = useDesignable();
  const schema = find(fieldSchema, key, action);
  return {
    schema,
    exists: !!schema,
    remove() {
      schema && rm(schema, remove);
    },
  };
};