@@ -16,6 +16,7 @@ import { useLayoutSetIdFromUrl } from 'src/features/form/layoutSets/useCurrentLa
1616import { useInstanceDataQuery , useLaxInstanceId } from 'src/features/instance/InstanceContext' ;
1717import { useProcessQuery } from 'src/features/instance/useProcessQuery' ;
1818import { makeLikertChildId } from 'src/layout/Likert/Generator/makeLikertChildId' ;
19+ import { fetchLayoutsForInstance } from 'src/queries/queries' ;
1920import type { QueryDefinition } from 'src/core/queries/usePrefetchQuery' ;
2021import type { CompExternal , ILayoutCollection , ILayouts } from 'src/layout/layout' ;
2122import type { IExpandedWidthLayouts , IHiddenLayoutsExternal } from 'src/types' ;
@@ -39,10 +40,14 @@ export function useLayoutQueryDef(
3940 return {
4041 queryKey : [ 'formLayouts' , layoutSetId , enabled ] ,
4142 queryFn : layoutSetId
42- ? ( ) =>
43- fetchLayouts ( layoutSetId , features . addInstanceIdentifierToLayoutRequests ? instanceId : undefined ) . then (
44- ( layouts ) => processLayouts ( layouts , layoutSetId , defaultDataModelType ) ,
45- )
43+ ? async ( ) => {
44+ const shouldUseInstanceEndpoint = features . addInstanceIdentifierToLayoutRequests && instanceId ;
45+ const layouts = shouldUseInstanceEndpoint
46+ ? await fetchLayoutsForInstance ( layoutSetId , instanceId )
47+ : await fetchLayouts ( layoutSetId ) ;
48+
49+ return processLayouts ( layouts , layoutSetId , defaultDataModelType ) ;
50+ }
4651 : skipToken ,
4752 enabled : enabled && ! ! layoutSetId ,
4853 } ;
0 commit comments