fix: add create cursor util

packages/twenty-front/src/modules/apollo/optimistic-effect/utils/triggerCreateRecordsOptimisticEffect.ts

@@ -1,6 +1,5 @@
 import { ApolloCache, StoreObject } from '@apollo/client';
 import { isNonEmptyString } from '@sniptt/guards';
-import { Buffer } from 'buffer';
 
 import { triggerUpdateRelationsOptimisticEffect } from '@/apollo/optimistic-effect/utils/triggerUpdateRelationsOptimisticEffect';
 import { ObjectMetadataItem } from '@/object-metadata/types/ObjectMetadataItem';
@@ -11,6 +10,7 @@ import { RecordGqlNode } from '@/object-record/graphql/types/RecordGqlNode';
 import { isRecordMatchingFilter } from '@/object-record/record-filter/utils/isRecordMatchingFilter';
 
 import { CachedObjectRecordQueryVariables } from '@/apollo/types/CachedObjectRecordQueryVariables';
+import { createCursor } from '@/apollo/utils/createCursor';
 import { isDefined } from '~/utils/isDefined';
 import { parseApolloStoreFieldName } from '~/utils/parseApolloStoreFieldName';
 
@@ -128,13 +128,7 @@ export const triggerCreateRecordsOptimisticEffect = ({
               );
 
               if (recordToCreateReference && !recordAlreadyInCache) {
-                const cursor = Buffer.from(
-                  JSON.stringify({
-                    position: recordToCreate.position,
-                    id: recordToCreate.id,
-                  }),
-                  'utf-8',
-                ).toString('base64');
+                const cursor = createCursor(recordToCreate);
 
                 const edge = {
                   __typename: getEdgeTypename(objectMetadataItem.nameSingular),

packages/twenty-front/src/modules/apollo/utils/__tests__/createCursor.test.ts

@@ -0,0 +1,55 @@
+import { createCursor } from '@/apollo/utils/createCursor';
+import { ObjectRecord } from '@/object-record/types/ObjectRecord';
+
+describe('createCursor', () => {
+  it('should create a cursor with id only', () => {
+    const record: ObjectRecord = { __typename: 'ObjectRecord', id: '123' };
+    const cursor = createCursor(record);
+    const decoded = JSON.parse(Buffer.from(cursor, 'base64').toString('utf-8'));
+
+    expect(decoded).toEqual({ id: '123' });
+  });
+
+  it('should create a cursor with id and position', () => {
+    const record: ObjectRecord = {
+      __typename: 'ObjectRecord',
+      id: '123',
+      position: 1,
+    };
+    const cursor = createCursor(record);
+    const decoded = JSON.parse(Buffer.from(cursor, 'base64').toString('utf-8'));
+
+    expect(decoded).toEqual({ id: '123', position: 1 });
+  });
+
+  it('should create a cursor with id and position as 0', () => {
+    const record: ObjectRecord = {
+      __typename: 'ObjectRecord',
+      id: '123',
+      position: 0,
+    };
+    const cursor = createCursor(record);
+    const decoded = JSON.parse(Buffer.from(cursor, 'base64').toString('utf-8'));
+
+    expect(decoded).toEqual({ id: '123', position: 0 });
+  });
+
+  it('should create a cursor with id and ignore extra fields', () => {
+    const record: ObjectRecord = {
+      __typename: 'ObjectRecord',
+      id: '123',
+      position: 1,
+      extra: 'extra',
+    };
+    const cursor = createCursor(record);
+    const decoded = JSON.parse(Buffer.from(cursor, 'base64').toString('utf-8'));
+
+    expect(decoded).toEqual({ id: '123', position: 1 });
+  });
+
+  it('should throw an error if record does not have an id', () => {
+    const record = { position: 1 } as any;
+
+    expect(() => createCursor(record)).toThrow();
+  });
+});

packages/twenty-front/src/modules/apollo/utils/createCursor.ts

@@ -0,0 +1,21 @@
+import { ObjectRecord } from '@/object-record/types/ObjectRecord';
+import { isDefined } from '~/utils/isDefined';
+
+export const createCursor = (record: ObjectRecord) => {
+  if (!('id' in record) || !isDefined(record.id)) {
+    throw new Error('Record does not have an id');
+  }
+
+  const payload: {
+    id: string;
+    position?: number;
+  } = {
+    id: record.id,
+  };
+
+  if ('position' in record) {
+    payload.position = record.position;
+  }
+
+  return Buffer.from(JSON.stringify(payload), 'utf-8').toString('base64');
+};