fix geoshape to support NOINDEX & SORTABLE, clean code

Author: leibale

packages/search/lib/commands/CREATE.spec.ts

@@ -70,6 +70,18 @@ describe('CREATE', () => {
                         ['FT.CREATE', 'index', 'SCHEMA', 'field', 'TEXT', 'WITHSUFFIXTRIE']
                     );
                 });
+
+                it('with INDEXEMPTY', () => {
+                    assert.deepEqual(
+                        transformArguments('index', {
+                            field: {
+                                type: SchemaFieldTypes.TEXT,
+                                INDEXEMPTY: true
+                            }
+                        }),
+                        ['FT.CREATE', 'index', 'SCHEMA', 'field', 'TEXT', 'INDEXEMPTY']
+                    );
+                });
             });
 
             it('NUMERIC', () => {
@@ -148,6 +160,18 @@ describe('CREATE', () => {
                         ['FT.CREATE', 'index', 'SCHEMA', 'field', 'TAG', 'WITHSUFFIXTRIE']
                     );
                 });
+
+                it('with INDEXEMPTY', () => {
+                    assert.deepEqual(
+                        transformArguments('index', {
+                            field: {
+                                type: SchemaFieldTypes.TAG,
+                                INDEXEMPTY: true
+                            }
+                        }),
+                        ['FT.CREATE', 'index', 'SCHEMA', 'field', 'TAG', 'INDEXEMPTY']
+                    );
+                });
             });
 
             describe('VECTOR', () => {
@@ -282,6 +306,18 @@ describe('CREATE', () => {
                         ['FT.CREATE', 'index', 'SCHEMA', 'field', 'TEXT', 'NOINDEX']
                     );
                 });
+
+                it('with INDEXMISSING', () => {
+                    assert.deepEqual(
+                        transformArguments('index', {
+                            field: {
+                                type: SchemaFieldTypes.TEXT,
+                                INDEXMISSING: true
+                            }
+                        }),
+                        ['FT.CREATE', 'index', 'SCHEMA', 'field', 'TEXT', 'INDEXMISSING']
+                    );
+                });
             });
         });
 
@@ -441,45 +477,6 @@ describe('CREATE', () => {
                 );
             });
         });
-
-        describe('Missing Values', () => {
-            it('with INDEXEMPTY', () => {
-                assert.deepEqual(
-                    transformArguments('index', {
-                        field: {
-                            type: SchemaFieldTypes.TEXT,
-                            INDEXEMPTY: true
-                        }
-                    }),
-                    ['FT.CREATE', 'index', 'SCHEMA', 'field', 'TEXT', 'INDEXEMPTY']
-                );
-            });
-
-            it('with INDEXMISSING', () => {
-                assert.deepEqual(
-                    transformArguments('index', {
-                        field: {
-                            type: SchemaFieldTypes.TEXT,
-                            INDEXMISSING: true
-                        }
-                    }),
-                    ['FT.CREATE', 'index', 'SCHEMA', 'field', 'TEXT', 'INDEXMISSING']
-                );
-            });
-
-            it('with INDEXEMPTY and INDEXMISSING', () => {
-                assert.deepEqual(
-                    transformArguments('index', {
-                        field: {
-                            type: SchemaFieldTypes.TEXT,
-                            INDEXEMPTY: true,
-                            INDEXMISSING: true
-                        }
-                    }),
-                    ['FT.CREATE', 'index', 'SCHEMA', 'field', 'TEXT', 'INDEXEMPTY', 'INDEXMISSING']
-                );
-            });
-        });
     });
 
     testUtils.testWithClient('client.ft.create', async client => {

packages/search/lib/commands/index.ts

@@ -188,25 +188,6 @@ export enum SchemaFieldTypes {
     VECTOR = 'VECTOR',
     GEOSHAPE = 'GEOSHAPE'
 }
-
-export interface MissingValues {
-    INDEXEMPTY?: boolean;
-    INDEXMISSING?: boolean;
-}
-
-function pushMissingValues(args: RedisCommandArguments, missingValues?: MissingValues) {
-    if (!missingValues) {
-        return;
-    }
-
-    if (missingValues.INDEXEMPTY) {
-        args.push("INDEXEMPTY");
-    }
-  
-    if (missingValues.INDEXMISSING) {
-        args.push("INDEXMISSING");
-    }
-}
 
 type CreateSchemaField<
     T extends SchemaFieldTypes,
@@ -228,6 +209,20 @@ type CreateSchemaCommonField<
     } & E)
 >;
 
+function pushCommonFieldArguments(args: RedisCommandArguments, fieldOptions: CreateSchemaCommonField<SchemaFieldTypes>) {
+    if (fieldOptions.SORTABLE) {
+        args.push('SORTABLE');
+
+        if (fieldOptions.SORTABLE === 'UNF') {
+            args.push('UNF');
+        }
+    }
+
+    if (fieldOptions.NOINDEX) {
+        args.push('NOINDEX');
+    }
+}
+
 export enum SchemaTextFieldPhonetics {
     DM_EN = 'dm:en',
     DM_FR = 'dm:fr',
@@ -299,7 +294,7 @@ export interface RediSearchSchema {
         CreateSchemaTagField |
         CreateSchemaFlatVectorField |
         CreateSchemaHNSWVectorField |
-        CreateSchemaGeoShapeField
+        CreateSchemaGeoShapeField;
 }
 
 export function pushSchema(args: RedisCommandArguments, schema: RediSearchSchema) {
@@ -335,13 +330,17 @@ export function pushSchema(args: RedisCommandArguments, schema: RediSearchSchema
                     args.push('WITHSUFFIXTRIE');
                 }
 
-                pushMissingValues(args, fieldOptions);
+                pushCommonFieldArguments(args, fieldOptions);
+
+                if (fieldOptions.INDEXEMPTY) {
+                    args.push('INDEXEMPTY');
+                }
 
                 break;
 
             case SchemaFieldTypes.NUMERIC:
             case SchemaFieldTypes.GEO:
-                pushMissingValues(args, fieldOptions);
+                pushCommonFieldArguments(args, fieldOptions);
                 break;
 
             case SchemaFieldTypes.TAG:
@@ -357,7 +356,11 @@ export function pushSchema(args: RedisCommandArguments, schema: RediSearchSchema
                     args.push('WITHSUFFIXTRIE');
                 }
 
-                pushMissingValues(args, fieldOptions);
+                pushCommonFieldArguments(args, fieldOptions);
+
+                if (fieldOptions.INDEXEMPTY) {
+                    args.push('INDEXEMPTY');
+                }
 
                 break;
 
@@ -400,30 +403,20 @@ export function pushSchema(args: RedisCommandArguments, schema: RediSearchSchema
                     }
                 });
 
-                pushMissingValues(args, fieldOptions);
-
-                continue; // vector fields do not contain SORTABLE and NOINDEX options
+                break;
 
             case SchemaFieldTypes.GEOSHAPE:
                 if (fieldOptions.COORD_SYSTEM !== undefined) {
                     args.push('COORD_SYSTEM', fieldOptions.COORD_SYSTEM);
                 }
 
-                pushMissingValues(args, fieldOptions);
-
-                continue; // geo shape fields do not contain SORTABLE and NOINDEX options
-        }
+                pushCommonFieldArguments(args, fieldOptions);
 
-        if (fieldOptions.SORTABLE) {
-            args.push('SORTABLE');
-
-            if (fieldOptions.SORTABLE === 'UNF') {
-                args.push('UNF');
-            }
+                break;
         }
 
-        if (fieldOptions.NOINDEX) {
-            args.push('NOINDEX');
+        if (fieldOptions.INDEXMISSING) {
+            args.push('INDEXMISSING');
         }
     }
 }