move useEntity to separate file

Author: nikgraf

packages/hypergraph-react/src/HypergraphSpaceContext.tsx

@@ -1,8 +1,8 @@
 'use client';
 
-import { Entity, type Id, store } from '@graphprotocol/hypergraph';
+import { Entity, store } from '@graphprotocol/hypergraph';
 import { useSelector } from '@xstate/store/react';
-import * as Schema from 'effect/Schema';
+import type * as Schema from 'effect/Schema';
 import {
   createContext,
   type ReactNode,
@@ -14,8 +14,6 @@ import {
   useSyncExternalStore,
 } from 'react';
 import { useHypergraphApp } from './HypergraphAppContext.js';
-import { useEntityPrivate } from './internal/use-entity-private.js';
-import { useEntityPublic } from './internal/use-entity-public.js';
 import { usePublicSpace } from './internal/use-public-space.js';
 
 // TODO space can be undefined
@@ -182,22 +180,3 @@ export function useQueryLocal<const S extends Entity.AnyNoContext>(type: S, para
 
   return { entities, deletedEntities };
 }
-
-export function useEntity<const S extends Entity.AnyNoContext>(
-  type: S,
-  params: {
-    id: string | Id;
-    space?: string;
-    mode: 'private' | 'public';
-    include?: { [K in keyof Schema.Schema.Type<S>]?: Record<string, Record<string, never>> } | undefined;
-  },
-) {
-  const resultPublic = useEntityPublic(type, { ...params, enabled: params.mode === 'public' });
-  const resultPrivate = useEntityPrivate(type, { ...params, enabled: params.mode === 'private' });
-
-  if (params.mode === 'public') {
-    return resultPublic;
-  }
-
-  return resultPrivate;
-}

packages/hypergraph-react/src/index.ts

@@ -1,28 +1,27 @@
 export { PublishDiff } from './components/publish-diff/publish-diff.js';
 export { createWalletClient } from './create-wallet-client.js';
+export { useSpaces } from './hooks/use-spaces.js';
+export { useExternalAccountInbox } from './hooks/useExternalAccountInbox.js';
+export { useExternalSpaceInbox } from './hooks/useExternalSpaceInbox.js';
+export { useOwnAccountInbox } from './hooks/useOwnAccountInbox.js';
+export { useOwnSpaceInbox } from './hooks/useOwnSpaceInbox.js';
+export { usePublicAccountInboxes } from './hooks/usePublicAccountInboxes.js';
+export { usePublishToPublicSpace } from './hooks/usePublishToSpace.js';
 export {
   HypergraphAppProvider,
   useHypergraphApp,
   useHypergraphAuth,
 } from './HypergraphAppContext.js';
 export {
+  useQueryLocal as _useQueryLocal,
   HypergraphSpaceProvider,
   useCreateEntity,
   useDeleteEntity,
-  useEntity,
   useHardDeleteEntity,
-  useQueryLocal as _useQueryLocal,
   useRemoveRelation,
   useSpace,
   useUpdateEntity,
 } from './HypergraphSpaceContext.js';
-export { useSpaces } from './hooks/use-spaces.js';
-export { useExternalAccountInbox } from './hooks/useExternalAccountInbox.js';
-export { useExternalSpaceInbox } from './hooks/useExternalSpaceInbox.js';
-export { useOwnAccountInbox } from './hooks/useOwnAccountInbox.js';
-export { useOwnSpaceInbox } from './hooks/useOwnSpaceInbox.js';
-export { usePublicAccountInboxes } from './hooks/usePublicAccountInboxes.js';
-export { usePublishToPublicSpace } from './hooks/usePublishToSpace.js';
 export { generateDeleteOps as _generateDeleteOps } from './internal/generate-delete-ops.js';
 export { useCreateEntityPublic as _useCreateEntityPublic } from './internal/use-create-entity-public.js';
 export { useDeleteEntityPublic as _useDeleteEntityPublic } from './internal/use-delete-entity-public.js';
@@ -31,4 +30,5 @@ export { useQueryPublic as _useQueryPublic } from './internal/use-query-public.j
 export { preparePublish } from './prepare-publish.js';
 export { publishOps } from './publish-ops.js';
 export type * from './types.js';
+export { useEntity } from './use-entity.js';
 export { useQuery } from './use-query.js';

packages/hypergraph-react/src/internal/use-entity-private.tsx

@@ -1,5 +1,3 @@
-'use client';
-
 import { Entity, type Id, store } from '@graphprotocol/hypergraph';
 import { useSelector } from '@xstate/store/react';
 import * as Schema from 'effect/Schema';

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

@@ -0,0 +1,25 @@
+'use client';
+
+import type { Entity, Id } from '@graphprotocol/hypergraph';
+import type * as Schema from 'effect/Schema';
+import { useEntityPrivate } from './internal/use-entity-private.js';
+import { useEntityPublic } from './internal/use-entity-public.js';
+
+export function useEntity<const S extends Entity.AnyNoContext>(
+  type: S,
+  params: {
+    id: string | Id;
+    space?: string;
+    mode: 'private' | 'public';
+    include?: { [K in keyof Schema.Schema.Type<S>]?: Record<string, Record<string, never>> } | undefined;
+  },
+) {
+  const resultPublic = useEntityPublic(type, { ...params, enabled: params.mode === 'public' });
+  const resultPrivate = useEntityPrivate(type, { ...params, enabled: params.mode === 'private' });
+
+  if (params.mode === 'public') {
+    return resultPublic;
+  }
+
+  return resultPrivate;
+}

packages/hypergraph-react/test/HypergraphSpaceContext.test.tsx

@@ -10,10 +10,10 @@ import {
   HypergraphSpaceProvider,
   useCreateEntity,
   useDeleteEntity,
-  useEntity,
   useQueryLocal,
   useUpdateEntity,
 } from '../src/HypergraphSpaceContext.js';
+import { useEntity } from '../src/use-entity.js';
 
 afterEach(() => {
   cleanup();