feat(core): export useWorkspaceLoader from core (#5898)

Author: pedrobonamin

packages/sanity/src/core/studio/index.ts

@@ -9,4 +9,5 @@ export * from './StudioLayout'
 export * from './StudioProvider'
 export * from './upsell'
 export * from './workspace'
+export * from './workspaceLoader'
 export * from './workspaces'

packages/sanity/src/core/studio/workspaceLoader/ErrorMessage.tsx

@@ -8,13 +8,19 @@ const ListItem = styled(Flex)``
 
 const ErrorMessageRoot = styled(Box).attrs({padding: 4})``
 
+/**
+ * @internal
+ */
 export interface ErrorMessageProps {
   message: string
   stack?: string
   error: unknown
   path: Array<{name: string; type: string}>
 }
 
+/**
+ * @internal
+ */
 export function ErrorMessage({error, message, path, stack}: ErrorMessageProps) {
   useEffect(() => {
     console.error(error)

packages/sanity/src/core/studio/workspaceLoader/WorkspaceLoader.tsx

@@ -3,7 +3,12 @@ import {type ComponentType, type ReactNode, useEffect, useState} from 'react'
 import {combineLatest, of} from 'rxjs'
 import {catchError, map} from 'rxjs/operators'
 
-import {ConfigResolutionError, type Source, type Workspace} from '../../config'
+import {
+  ConfigResolutionError,
+  type Source,
+  type Workspace,
+  type WorkspaceSummary,
+} from '../../config'
 import {useActiveWorkspace} from '../activeWorkspaceMatcher'
 import {SourceProvider} from '../source'
 import {WorkspaceProvider} from '../workspace'
@@ -18,14 +23,13 @@ interface WorkspaceLoaderProps {
   LoadingComponent: ComponentType
 }
 
-function WorkspaceLoader({
-  children,
-  LoadingComponent,
-}: Omit<WorkspaceLoaderProps, 'ConfigErrorsComponent'>) {
+/**
+ * @internal
+ */
+export function useWorkspaceLoader(activeWorkspace: WorkspaceSummary) {
   const [error, handleError] = useState<unknown>(null)
   if (error) throw error
 
-  const {activeWorkspace} = useActiveWorkspace()
   const [workspace, setWorkspace] = useState<Workspace | null>(null)
 
   useEffect(() => {
@@ -69,6 +73,15 @@ function WorkspaceLoader({
     return () => subscription.unsubscribe()
   }, [activeWorkspace])
 
+  return workspace
+}
+
+function WorkspaceLoader({
+  children,
+  LoadingComponent,
+}: Omit<WorkspaceLoaderProps, 'ConfigErrorsComponent'>) {
+  const {activeWorkspace} = useActiveWorkspace()
+  const workspace = useWorkspaceLoader(activeWorkspace)
   if (!workspace) return <LoadingComponent />
 
   // TODO: may need a screen if one of the sources is not logged in. e.g. it
@@ -90,6 +103,9 @@ function WorkspaceLoader({
   )
 }
 
+/**
+ * @internal
+ */
 function WorkspaceLoaderBoundary({ConfigErrorsComponent, ...props}: WorkspaceLoaderProps) {
   const [{error}, setError] = useState<{error: unknown}>({error: null})