add first filter (#314)

Author: nikgraf

apps/events/src/components/playground.tsx

@@ -17,6 +17,7 @@ export const Playground = () => {
         jobOffers: {},
       },
     },
+    first: 2,
   });
   const [isDeleting, setIsDeleting] = useState(false);
   const [isCreating, setIsCreating] = useState(false);

packages/hypergraph-react/src/internal/types.ts

@@ -5,4 +5,5 @@ export type QueryPublicParams<S extends Entity.AnyNoContext> = {
   enabled: boolean;
   // TODO: for multi-level nesting it should only allow the allowed properties instead of Record<string, Record<string, never>>
   include?: { [K in keyof Schema.Schema.Type<S>]?: Record<string, Record<string, never>> } | undefined;
+  first?: number | undefined;
 };

packages/hypergraph-react/src/internal/use-query-public.tsx

@@ -10,10 +10,11 @@ import { GEO_API_TESTNET_ENDPOINT } from './constants.js';
 import type { QueryPublicParams } from './types.js';
 
 const entitiesQueryDocumentLevel0 = gql`
-query entities($spaceId: UUID!, $typeIds: [UUID!]!) {
+query entities($spaceId: UUID!, $typeIds: [UUID!]!, $first: Int) {
   entities(
     filter: {
       relations: {some: {typeId: {is: "8f151ba4-de20-4e3c-9cb4-99ddf96f48f1"}, toEntityId: {in: $typeIds}}}, spaceIds: {in: [$spaceId]}},
+    first: $first
   ) {
     id
     name
@@ -26,8 +27,10 @@ query entities($spaceId: UUID!, $typeIds: [UUID!]!) {
 `;
 
 const entitiesQueryDocumentLevel1 = gql`
-query entities($spaceId: UUID!, $typeIds: [UUID!]!, $relationTypeIdsLevel1: [UUID!]!) {
-  entities(filter: {
+query entities($spaceId: UUID!, $typeIds: [UUID!]!, $relationTypeIdsLevel1: [UUID!]!, $first: Int) {
+  entities(
+    first: $first
+    filter: {
     relations: {some: {typeId: {is: "8f151ba4-de20-4e3c-9cb4-99ddf96f48f1"}, toEntityId: {in: $typeIds}}}, spaceIds: {in: [$spaceId]}}) {
     id
     name
@@ -53,8 +56,10 @@ query entities($spaceId: UUID!, $typeIds: [UUID!]!, $relationTypeIdsLevel1: [UUI
 `;
 
 const entitiesQueryDocumentLevel2 = gql`
-query entities($spaceId: UUID!, $typeIds: [UUID!]!, $relationTypeIdsLevel1: [UUID!]!, $relationTypeIdsLevel2: [UUID!]!) {
-  entities(filter: {
+query entities($spaceId: UUID!, $typeIds: [UUID!]!, $relationTypeIdsLevel1: [UUID!]!, $relationTypeIdsLevel2: [UUID!]!, $first: Int) {
+  entities(
+    first: $first
+    filter: {
     relations: {some: {typeId: {is: "8f151ba4-de20-4e3c-9cb4-99ddf96f48f1"}, toEntityId: {in: $typeIds}}}, spaceIds: {in: [$spaceId]}}) {
     id
     name
@@ -290,7 +295,7 @@ export const parseResult = <S extends Entity.AnyNoContext>(
 };
 
 export const useQueryPublic = <S extends Entity.AnyNoContext>(type: S, params?: QueryPublicParams<S>) => {
-  const { enabled = true, include } = params ?? {};
+  const { enabled = true, include, first = 100 } = params ?? {};
   const { space } = useHypergraphSpaceInternal();
   const mapping = useSelector(store, (state) => state.context.mapping);
 
@@ -327,6 +332,7 @@ export const useQueryPublic = <S extends Entity.AnyNoContext>(type: S, params?:
       mappingEntry?.typeIds,
       relationTypeIdsLevel1,
       relationTypeIdsLevel2,
+      // TODO should `first` be in here?
     ],
     queryFn: async () => {
       let queryDocument = entitiesQueryDocumentLevel0;
@@ -342,6 +348,7 @@ export const useQueryPublic = <S extends Entity.AnyNoContext>(type: S, params?:
         typeIds: mappingEntry?.typeIds || [],
         relationTypeIdsLevel1,
         relationTypeIdsLevel2,
+        first,
       });
       return result;
     },

packages/hypergraph-react/src/use-query.tsx

@@ -10,6 +10,7 @@ type QueryParams<S extends Entity.AnyNoContext> = {
   // TODO: for multi-level nesting it should only allow the allowed properties instead of Record<string, Record<string, never>>
   include?: { [K in keyof Schema.Schema.Type<S>]?: Record<string, Record<string, never>> } | undefined;
   space?: string;
+  first?: number | undefined;
 };
 
 // @ts-expect-error TODO: remove this function
@@ -144,8 +145,8 @@ const getDiff = <S extends Entity.AnyNoContext>(
 const preparePublishDummy = () => undefined;
 
 export function useQuery<const S extends Entity.AnyNoContext>(type: S, params: QueryParams<S>) {
-  const { mode, filter, include, space } = params;
-  const publicResult = useQueryPublic(type, { enabled: mode === 'public', include });
+  const { mode, filter, include, space, first } = params;
+  const publicResult = useQueryPublic(type, { enabled: mode === 'public', include, first });
   const localResult = useQueryLocal(type, { enabled: mode === 'private', filter, include, space });
   // const mapping = useSelector(store, (state) => state.context.mapping);
   // const generateUpdateOps = useGenerateUpdateOps(type, mode === 'merged');