import type { DataNode } from 'antd/es/tree';
import type { ReactNode } from 'react';

export interface TreeData<T = string | number> extends Omit<DataNode, 'children'> {
  key: T;
  title: string | number;
  parent?: T;
}

export type TreeNode<T> = Omit<TreeData<T>, 'title'> & {
  title?: ReactNode;
  children?: TreeNode<T>[];
  data: TreeData<T>;
};

export interface OperationType<T> {
  onAdd?: (node: TreeData<T>) => void;
  others?: (node: TreeData<T>) => ReactNode;
}