[DOP-22358] add filter columns (#70)

Co-authored-by: Zabilsya <[email protected]>

Author: Zabilsya

src/entities/transformation/types.ts

@@ -1,8 +1,9 @@
 export enum TransformationType {
-  ROWS_FILTER = 'dataframe_rows_filter',
+  FILTER_ROWS = 'dataframe_rows_filter',
+  FILTER_COLUMNS = 'dataframe_columns_filter',
 }
 
-export enum TransformationRowsFilterType {
+export enum TransformationFilterRowsType {
   IS_NULL = 'is_null',
   IS_NOT_NULL = 'is_not_null',
   EQUAL = 'equal',
@@ -18,30 +19,64 @@ export enum TransformationRowsFilterType {
   REGEXP = 'regexp',
 }
 
-interface TransformationRowsFilterNullable {
+interface TransformationFilterRowsNullable {
   type: Extract<
-    TransformationRowsFilterType,
-    TransformationRowsFilterType.IS_NULL | TransformationRowsFilterType.IS_NOT_NULL
+    TransformationFilterRowsType,
+    TransformationFilterRowsType.IS_NULL | TransformationFilterRowsType.IS_NOT_NULL
   >;
   field: string;
 }
 
-interface TransformationRowsFilterDefault {
+interface TransformationFilterRowsDefault {
   type: Exclude<
-    TransformationRowsFilterType,
-    TransformationRowsFilterType.IS_NULL | TransformationRowsFilterType.IS_NOT_NULL
+    TransformationFilterRowsType,
+    TransformationFilterRowsType.IS_NULL | TransformationFilterRowsType.IS_NOT_NULL
   >;
   field: string;
   value: string;
 }
 
-export interface TransformationRowsFilter {
-  type: TransformationType.ROWS_FILTER;
-  filters: Array<TransformationRowsFilterNullable | TransformationRowsFilterDefault>;
+export interface TransformationFilterRows {
+  type: TransformationType.FILTER_ROWS;
+  filters: Array<TransformationFilterRowsNullable | TransformationFilterRowsDefault>;
 }
 
-export type Transformations = Array<TransformationRowsFilter>;
+export enum TransformationFilterColumnsType {
+  CAST = 'cast',
+  INCLUDE = 'include',
+  RENAME = 'rename',
+}
+
+interface TransformationFilterColumnsInclude {
+  type: TransformationFilterColumnsType.INCLUDE;
+  field: string;
+}
+
+interface TransformationFilterColumnsRename {
+  type: TransformationFilterColumnsType.RENAME;
+  field: string;
+  to: string;
+}
+
+interface TransformationFilterColumnsCast {
+  type: TransformationFilterColumnsType.CAST;
+  field: string;
+  as_type: string;
+}
+
+export interface TransformationFilterColumns {
+  type: TransformationType.FILTER_COLUMNS;
+  filters: Array<
+    TransformationFilterColumnsInclude | TransformationFilterColumnsRename | TransformationFilterColumnsCast
+  >;
+}
+
+export type Transformations = Array<TransformationFilterRows | TransformationFilterColumns>;
 
 export interface TransformationsForm {
-  [TransformationType.ROWS_FILTER]?: TransformationRowsFilter['filters'];
+  [TransformationType.FILTER_ROWS]?: TransformationFilterRows['filters'];
+  [TransformationType.FILTER_COLUMNS]?: TransformationFilterColumns['filters'];
 }
+
+export type TransformationsFormNestedType<T extends keyof TransformationsForm> =
+  Required<TransformationsForm>[T][number]['type'];

src/entities/transformation/ui/FilterColumnsValue/index.tsx

@@ -0,0 +1,26 @@
+import React from 'react';
+import { Form, Input } from 'antd';
+import { TransformationFilterColumnsType } from '@entities/transformation';
+
+import { FilterColumnsValueProps } from './types';
+import classes from './styles.module.less';
+
+export const FilterColumnsValue = ({ name, type }: FilterColumnsValueProps) => {
+  switch (type) {
+    case TransformationFilterColumnsType.INCLUDE:
+    case undefined:
+      return null;
+    case TransformationFilterColumnsType.RENAME:
+      return (
+        <Form.Item className={classes.root} label="Value" name={[name, 'to']} rules={[{ required: true }]}>
+          <Input className="nodrag" size="large" />
+        </Form.Item>
+      );
+    case TransformationFilterColumnsType.CAST:
+      return (
+        <Form.Item className={classes.root} label="Value" name={[name, 'as_type']} rules={[{ required: true }]}>
+          <Input className="nodrag" size="large" />
+        </Form.Item>
+      );
+  }
+};

…nents/FilterRowsValue/styles.module.less …ui/FilterColumnsValue/styles.module.lesssrc/entities/transformation/ui/FilterRows/components/FilterRowsValue/styles.module.less renamed to src/entities/transformation/ui/FilterColumnsValue/styles.module.less src/entities/transformation/ui/FilterRows/components/FilterRowsValue/styles.module.less renamed to src/entities/transformation/ui/FilterColumnsValue/styles.module.less

@@ -0,0 +1,6 @@
+import { TransformationFilterColumnsType } from '@entities/transformation';
+
+export interface FilterColumnsValueProps {
+  name: number;
+  type?: TransformationFilterColumnsType;
+}

src/entities/transformation/ui/FilterColumnsValue/types.ts

@@ -1,15 +1,14 @@
 import React from 'react';
 import { Form, Input } from 'antd';
-
-import { TransformationRowsFilterType } from '../../../../types';
+import { TransformationFilterRowsType } from '@entities/transformation';
 
 import { FilterRowsValueProps } from './types';
 import classes from './styles.module.less';
 
 export const FilterRowsValue = ({ name, type }: FilterRowsValueProps) => {
   switch (type) {
-    case TransformationRowsFilterType.IS_NULL:
-    case TransformationRowsFilterType.IS_NOT_NULL:
+    case TransformationFilterRowsType.IS_NULL:
+    case TransformationFilterRowsType.IS_NOT_NULL:
     case undefined:
       return null;
     default: